[load_save] obj can save edges

Author: alemuntoni

vclib/core/include/vclib/load_save/obj/save.h

@@ -31,6 +31,7 @@
 #include <vclib/load_save/settings.h>
 #include <vclib/misc/logger.h>
 #include <vclib/space/complex/mesh_info.h>
+#include <vclib/space/core/texture.h>
 
 #include <map>
 
@@ -79,6 +80,21 @@ ObjMaterial objMaterialFromFace(
     return mat;
 }
 
+template<EdgeConcept EdgeType, MeshConcept MeshType>
+ObjMaterial objMaterialFromEdge(const EdgeType& e, const MeshInfo& fi)
+{
+    ObjMaterial mat;
+    if constexpr (HasPerEdgeColor<MeshType>) {
+        if (fi.hasEdgeColors()) {
+            mat.hasColor = true;
+            mat.Kd.x()   = e.color().redF();
+            mat.Kd.y()   = e.color().greenF();
+            mat.Kd.z()   = e.color().blueF();
+        }
+    }
+    return mat;
+}
+
 template<
     ElementConcept ElementType,
     MeshConcept    MeshType,
@@ -97,6 +113,7 @@ void writeElementObjMaterial(
     ObjMaterial    mat;
     constexpr bool EL_IS_VERTEX = ElementType::ELEMENT_ID == ElemId::VERTEX;
     constexpr bool EL_IS_FACE   = ElementType::ELEMENT_ID == ElemId::FACE;
+    constexpr bool EL_IS_EDGE   = ElementType::ELEMENT_ID == ElemId::EDGE;
 
     if constexpr (EL_IS_VERTEX) {
         mat = objMaterialFromVertex<typename MeshType::VertexType, MeshType>(
@@ -105,6 +122,9 @@ void writeElementObjMaterial(
     if constexpr (EL_IS_FACE) {
         mat = objMaterialFromFace(e, m, fi);
     }
+    if constexpr (EL_IS_EDGE) {
+        mat = objMaterialFromEdge<typename MeshType::EdgeType, MeshType>(e, fi);
+    }
     if (!mat.isEmpty()) {
         static const std::string MATERIAL_PREFIX = "MATERIAL_";
         std::string              mname; // name of the material of the vertex
@@ -200,6 +220,9 @@ void saveObj(
 
     // vertices
     using VertexType = MeshType::VertexType;
+
+    fp << std::endl << "# Vertices" << std::endl;
+
     for (const VertexType& v : m.vertices()) {
         if (useMtl) { // mtl management
             detail::writeElementObjMaterial<VertexType, MeshType>(
@@ -243,6 +266,8 @@ void saveObj(
         using VertexType = MeshType::VertexType;
         using FaceType   = MeshType::FaceType;
 
+        fp << std::endl << "# Faces" << std::endl;
+
         // indices of vertices that do not consider deleted vertices
         std::vector<uint> vIndices = m.vertexCompactIndices();
 
@@ -295,6 +320,34 @@ void saveObj(
             fp << std::endl;
         }
     }
+
+    if constexpr (HasEdges<MeshType>) {
+        using VertexType = MeshType::VertexType;
+        using EdgeType = MeshType::EdgeType;
+
+        fp << std::endl << "# Edges" << std::endl;
+
+        // indices of vertices that do not consider deleted vertices
+        std::vector<uint> vIndices = m.vertexCompactIndices();
+
+        for (const EdgeType& e : m.edges()) {
+            if (useMtl) { // mtl management
+                detail::writeElementObjMaterial(
+                    e,
+                    m,
+                    meshInfo,
+                    lastMaterial,
+                    materialMap,
+                    fp,
+                    *mtlfp,
+                    settings,
+                    log);
+            }
+            fp << "l ";
+            fp << vIndices[m.index(e.vertex(0))] + 1 << " ";
+            fp << vIndices[m.index(e.vertex(1))] + 1 << std::endl;
+        }
+    }
 }
 
 } // namespace detail

vclib/core/include/vclib/load_save/ply/detail/edge.h

@@ -68,6 +68,8 @@ void writePlyEdges(
                 io::writeProperty(file, 0, p.type, format);
             }
         }
+        if (!format.isBinary)
+            file << std::endl;
     }
 }