MeshImporter.h

Go to the documentation of this file.

/* SPDX-FileCopyrightText: 2023 Blender Authors
 *
 * SPDX-License-Identifier: GPL-2.0-or-later */

 
#pragma once
 
#include <map>
#include <vector>
 
#include "COLLADAFWIndexList.h"
#include "COLLADAFWInstanceGeometry.h"
#include "COLLADAFWMaterialBinding.h"
#include "COLLADAFWMesh.h"
#include "COLLADAFWMeshVertexData.h"
#include "COLLADAFWNode.h"
#include "COLLADAFWPolygons.h"
#include "COLLADAFWTypes.h"
#include "COLLADAFWUniqueId.h"
 
#include "collada_utils.h"
 
#include "BLI_math_vector_types.hh"
 
#include "DNA_material_types.h"
#include "DNA_mesh_types.h"
#include "DNA_object_types.h"
#include "DNA_scene_types.h"
 
class ArmatureImporter;
struct MLoopCol;
 
/* only for ArmatureImporter to "see" MeshImporter::get_object_by_geom_uid */
class MeshImporterBase {
 public:
  virtual Object *get_object_by_geom_uid(const COLLADAFW::UniqueId &geom_uid) = 0;
  virtual Mesh *get_mesh_by_geom_uid(const COLLADAFW::UniqueId &mesh_uid) = 0;
  virtual std::string *get_geometry_name(const std::string &mesh_name) = 0;
};
 
class UVDataWrapper {
  COLLADAFW::MeshVertexData *mVData;
 
 public:
  UVDataWrapper(COLLADAFW::MeshVertexData &vdata);
 
#ifdef COLLADA_DEBUG
  void print();
#endif
 
  void getUV(int uv_index, float *uv);
};
 
class VCOLDataWrapper {
  COLLADAFW::MeshVertexData *mVData;
 
 public:
  VCOLDataWrapper(COLLADAFW::MeshVertexData &vdata);
  void get_vcol(int v_index, MLoopCol *mloopcol);
};
 
class MeshImporter : public MeshImporterBase {
 private:
  UnitConverter *unitconverter;
  bool use_custom_normals;
 
  Main *m_bmain;
  Scene *scene;
  ViewLayer *view_layer;
 
  ArmatureImporter *armature_importer;
 
  std::map<std::string, std::string> mesh_geom_map;       /* needed for correct shape key naming */
  std::map<COLLADAFW::UniqueId, Mesh *> uid_mesh_map;     /* geometry unique id-to-mesh map */
  std::map<COLLADAFW::UniqueId, Object *> uid_object_map; /* geom UID-to-object */
  std::vector<Object *> imported_objects;                 /* list of imported objects */
 
  /* this structure is used to assign material indices to faces
   * it holds a portion of Mesh faces and corresponds to a DAE primitive list
   * (`<triangles>`, `<polylist>`, etc.) */
  struct Primitive {
    int face_index;
    int *material_indices;
    uint faces_num;
  };
  using MaterialIdPrimitiveArrayMap = std::map<COLLADAFW::MaterialId, std::vector<Primitive>>;
  /* crazy name! */
  std::map<COLLADAFW::UniqueId, MaterialIdPrimitiveArrayMap> geom_uid_mat_mapping_map;
  /* < materials that have already been mapped to a geometry.
   * A pair/of geom UID and mat UID, one geometry can have several materials. */
  std::multimap<COLLADAFW::UniqueId, COLLADAFW::UniqueId> materials_mapped_to_geom;
 
  bool set_poly_indices(int *face_verts, int loop_index, const uint *indices, int loop_count);
 
  void set_face_uv(blender::float2 *mloopuv,
                   UVDataWrapper &uvs,
                   int start_index,
                   COLLADAFW::IndexList &index_list,
                   int count);
 
  void set_vcol(MLoopCol *mloopcol,
                VCOLDataWrapper &vob,
                int loop_index,
                COLLADAFW::IndexList &index_list,
                int count);
 
#ifdef COLLADA_DEBUG
  void print_index_list(COLLADAFW::IndexList &index_list);
#endif

  bool is_nice_mesh(COLLADAFW::Mesh *mesh);
 
  void read_vertices(COLLADAFW::Mesh *mesh, Mesh *blender_mesh);

  bool primitive_has_useable_normals(COLLADAFW::MeshPrimitive *mp);
  bool primitive_has_faces(COLLADAFW::MeshPrimitive *mp);

  static void mesh_add_edges(Mesh *mesh, int len);
 
  uint get_loose_edge_count(COLLADAFW::Mesh *mesh);

  void allocate_poly_data(COLLADAFW::Mesh *collada_mesh, Mesh *mesh);
 
  /* TODO: import uv set names */
  void read_polys(COLLADAFW::Mesh *mesh,
                  Mesh *blender_mesh,
                  blender::Vector<blender::float3> &loop_normals);
  void read_lines(COLLADAFW::Mesh *mesh, Mesh *blender_mesh);
  uint get_vertex_count(COLLADAFW::Polygons *mp, int index);
 
  void get_vector(float v[3], COLLADAFW::MeshVertexData &arr, int i, int stride);
 
  bool is_flat_face(uint *nind, COLLADAFW::MeshVertexData &nor, int count);

  std::vector<Object *> get_all_users_of(Mesh *reference_mesh);
 
 public:
  MeshImporter(UnitConverter *unitconv,
               bool use_custom_normals,
               ArmatureImporter *arm,
               Main *bmain,
               Scene *sce,
               ViewLayer *view_layer);
 
  Object *get_object_by_geom_uid(const COLLADAFW::UniqueId &geom_uid) override;
 
  Mesh *get_mesh_by_geom_uid(const COLLADAFW::UniqueId &geom_uid) override;

  void optimize_material_assignements();

  void assign_material_to_geom(COLLADAFW::MaterialBinding cmaterial,
                               std::map<COLLADAFW::UniqueId, Material *> &uid_material_map,
                               Object *ob,
                               const COLLADAFW::UniqueId *geom_uid,
                               short mat_index);
 
  Object *create_mesh_object(COLLADAFW::Node *node,
                             COLLADAFW::InstanceGeometry *geom,
                             bool isController,
                             std::map<COLLADAFW::UniqueId, Material *> &uid_material_map);

  bool write_geometry(const COLLADAFW::Geometry *geom);
  std::string *get_geometry_name(const std::string &mesh_name) override;
};