objToEggConverter.h

Go to the documentation of this file.

/**
 * PANDA 3D SOFTWARE
 * Copyright (c) Carnegie Mellon University.  All rights reserved.
 *
 * All use of this software is subject to the terms of the revised BSD
 * license.  You should have received a copy of this license along
 * with this source code in a file named "LICENSE."
 *
 * @file objToEggConverter.h
 * @author drose
 * @date 2010-12-07
 */
 
#ifndef OBJTOEGGCONVERTER_H
#define OBJTOEGGCONVERTER_H
 
#include "pandatoolbase.h"
 
#include "somethingToEggConverter.h"
#include "eggVertexPool.h"
#include "eggGroup.h"
#include "geomVertexData.h"
#include "geomVertexWriter.h"
#include "geomPrimitive.h"
#include "geomNode.h"
#include "pandaNode.h"
#include "pvector.h"
#include "epvector.h"
 
/**
 * Convert an Obj file to egg data.
 */
class ObjToEggConverter : public SomethingToEggConverter {
public:
  ObjToEggConverter();
  ObjToEggConverter(const ObjToEggConverter &copy);
  ~ObjToEggConverter();
 
  virtual SomethingToEggConverter *make_copy();
 
  virtual std::string get_name() const;
  virtual std::string get_extension() const;
  virtual bool supports_compressed() const;
  virtual bool supports_convert_to_node(const LoaderOptions &options) const;
 
  virtual bool convert_file(const Filename &filename);
  virtual PT(PandaNode) convert_to_node(const LoaderOptions &options, const Filename &filename);
 
protected:
  bool process(const Filename &filename);
  bool process_line(const std::string &line);
  bool process_ref_plane_res(const std::string &line);
 
  bool process_v(vector_string &words);
  bool process_vt(vector_string &words);
  bool process_xvt(vector_string &words);
  bool process_xvc(vector_string &words);
  bool process_vn(vector_string &words);
  bool process_f(vector_string &words);
  bool process_g(vector_string &words);
 
  EggVertex *get_face_vertex(const std::string &face_reference);
  void generate_egg_points();
 
  bool process_node(const Filename &filename);
  bool process_line_node(const std::string &line);
 
  bool process_f_node(vector_string &words);
  bool process_g_node(vector_string &words);
 
  void generate_points();
  int add_synth_normal(const LVecBase3d &normal);
 
  // Read from the obj file.
  int _line_number;
  typedef epvector<LVecBase4d> Vec4Table;
  typedef epvector<LVecBase3d> Vec3Table;
  typedef epvector<LVecBase2d> Vec2Table;
  typedef pmap<LVecBase3d, int> UniqueVec3Table;
 
  Vec4Table _v_table;
  Vec3Table _vn_table, _rgb_table;
  Vec3Table _vt_table;
  Vec2Table _xvt_table;
  Vec3Table _synth_vn_table;
  UniqueVec3Table _unique_synth_vn_table;
  LVecBase2d _ref_plane_res;
  bool _v4_given, _vt3_given;
  bool _f_given;
 
  pset<std::string> _ignored_tags;
 
  // Structures filled when creating an egg file.
  PT(EggVertexPool) _vpool;
  PT(EggGroup) _root_group;
  EggGroup *_current_group;
 
  // Structures filled when creating a PandaNode directly.
  PT(PandaNode) _root_node;
 
  class VertexEntry {
  public:
    VertexEntry();
    VertexEntry(const ObjToEggConverter *converter, const std::string &obj_vertex);
 
    INLINE bool operator < (const VertexEntry &other) const;
    INLINE bool operator == (const VertexEntry &other) const;
    INLINE bool matches_except_normal(const VertexEntry &other) const;
 
    // The 1-based vertex, texcoord, and normal index numbers appearing in the
    // obj file for this vertex.  0 if the index number is not given.
    int _vi, _vti, _vni;
 
    // The 1-based index number to the synthesized normal, if needed.
    int _synth_vni;
  };
  typedef pmap<VertexEntry, int> UniqueVertexEntries;
  typedef pvector<VertexEntry> VertexEntries;
 
  class VertexData {
  public:
    VertexData(PandaNode *parent, const std::string &name);
 
    int add_vertex(const ObjToEggConverter *converter, const VertexEntry &entry);
    void add_triangle(const ObjToEggConverter *converter, const VertexEntry &v0,
                      const VertexEntry &v1, const VertexEntry &v2,
                      int synth_vni);
    void close_geom(const ObjToEggConverter *converter);
 
    PT(PandaNode) _parent;
    std::string _name;
    PT(GeomNode) _geom_node;
 
    PT(GeomPrimitive) _prim;
    VertexEntries _entries;
    UniqueVertexEntries _unique_entries;
 
    bool _v4_given, _vt3_given;
    bool _vt_given, _rgb_given, _vn_given;
  };
 
  VertexData *_current_vertex_data;
 
  friend class VertexData;
};
 
#include "objToEggConverter.I"
 
#endif