0030691: Data Exchange - implement import of mesh data from files in glTF format
[occt.git] / src / RWGltf / RWGltf_TriangulationReader.hxx
1 // Author: Kirill Gavrilov
2 // Copyright (c) 2019 OPEN CASCADE SAS
3 //
4 // This file is part of Open CASCADE Technology software library.
5 //
6 // This library is free software; you can redistribute it and/or modify it under
7 // the terms of the GNU Lesser General Public License version 2.1 as published
8 // by the Free Software Foundation, with special exception defined in the file
9 // OCCT_LGPL_EXCEPTION.txt. Consult the file LICENSE_LGPL_21.txt included in OCCT
10 // distribution for complete text of the license and disclaimer of any warranty.
11 //
12 // Alternatively, this file may be used under the terms of Open CASCADE
13 // commercial license or contractual agreement.
14
15 #ifndef _RWGltf_TriangulationReader_HeaderFile
16 #define _RWGltf_TriangulationReader_HeaderFile
17
18 #include <RWGltf_PrimitiveArrayReader.hxx>
19
20 //! RWGltf_PrimitiveArrayReader implementation creating Poly_Triangulation.
21 class RWGltf_TriangulationReader : public RWGltf_PrimitiveArrayReader
22 {
23   DEFINE_STANDARD_RTTIEXT(RWGltf_TriangulationReader, RWGltf_PrimitiveArrayReader)
24 public:
25
26   //! Empty constructor.
27   Standard_EXPORT RWGltf_TriangulationReader();
28
29 protected:
30
31   //! Create Poly_Triangulation from collected data
32   Standard_EXPORT virtual Handle(Poly_Triangulation) result() Standard_OVERRIDE;
33
34   //! Reset cache before loading primitive array.
35   Standard_EXPORT virtual void reset() Standard_OVERRIDE;
36
37   //! Fill triangulation data and ignore non-triangulation primitives.
38   //! @param theStream   input stream to read from
39   //! @param theName     entity name for logging errors
40   //! @param theAccessor buffer accessor
41   //! @param theType     array type
42   //! @param theMode     primitive mode
43   //! @return FALSE on error
44   Standard_EXPORT virtual bool readBuffer (std::istream& theStream,
45                                            const TCollection_AsciiString& theName,
46                                            const RWGltf_GltfAccessor& theAccessor,
47                                            RWGltf_GltfArrayType theType,
48                                            RWGltf_GltfPrimitiveMode theMode) Standard_OVERRIDE;
49
50 protected: //! @name interface for filling triangulation data
51
52   //! Resize array of position nodes to specified size.
53   virtual bool setNbPositionNodes (Standard_Integer theNbNodes)
54   {
55     if (theNbNodes <= 0)
56     {
57       return false;
58     }
59     myTriangulation->ChangeNodes().Resize (1, theNbNodes, false);
60     return true;
61   }
62
63   //! Set node position.
64   //! @param theIndex node index starting from 1
65   //! @param thePnt   node position
66   virtual void setNodePosition (Standard_Integer theIndex,
67                                 const gp_Pnt& thePnt)
68   {
69     myTriangulation->ChangeNode (theIndex) = thePnt;
70   }
71
72   //! Resize array of UV nodes to specified size.
73   virtual bool setNbUVNodes (Standard_Integer theNbNodes)
74   {
75     if (theNbNodes <= 0
76      || myTriangulation->NbNodes() != theNbNodes)
77     {
78       return false;
79     }
80     myTriangulation->ChangeUVNodes().Resize (1, theNbNodes, false);
81     return true;
82   }
83
84   //! Set node UV texture coordinates.
85   //! @param theIndex node index starting from 1
86   //! @param theUV    node UV coordinates
87   virtual void setNodeUV (Standard_Integer theIndex,
88                           const gp_Pnt2d& theUV)
89   {
90     myTriangulation->ChangeUVNode (theIndex) = theUV;
91   }
92
93   //! Resize array of nodes normals to specified size.
94   virtual bool setNbNormalNodes (Standard_Integer theNbNodes)
95   {
96     if (theNbNodes <= 0
97      || myTriangulation->NbNodes() != theNbNodes)
98     {
99       return false;
100     }
101     myTriangulation->SetNormals (new TShort_HArray1OfShortReal (1, theNbNodes * 3));
102     return true;
103   }
104
105   //! Set node normal.
106   //! @param theIndex  node index starting from 1
107   //! @param theNormal node normal
108   virtual void setNodeNormal (Standard_Integer theIndex,
109                               const gp_Dir& theNormal)
110   {
111     myTriangulation->SetNormal (theIndex, theNormal);
112   }
113
114   //! Resize array of triangles to specified size.
115   virtual bool setNbTriangles (Standard_Integer theNbTris)
116   {
117     if (theNbTris >= 1)
118     {
119       myTriangulation->ChangeTriangles().Resize (1, theNbTris, false);
120       return true;
121     }
122     return false;
123   }
124
125   //! Add triangle element.
126   //! @param theIndex    triangle index starting from 1
127   //! @param theTriangle triangle nodes starting from 1
128   //! @return FALSE if node indexes are out of range
129   virtual bool setTriangle (Standard_Integer theIndex,
130                             const Poly_Triangle& theTriangle)
131   {
132     if (theTriangle.Value (1) < myTriangulation->Nodes().Lower() || theTriangle.Value (1) > myTriangulation->Nodes().Upper()
133      || theTriangle.Value (2) < myTriangulation->Nodes().Lower() || theTriangle.Value (2) > myTriangulation->Nodes().Upper()
134      || theTriangle.Value (3) < myTriangulation->Nodes().Lower() || theTriangle.Value (3) > myTriangulation->Nodes().Upper())
135     {
136       return false;
137     }
138     myTriangulation->ChangeTriangle (theIndex) = theTriangle;
139     return true;
140   }
141
142 protected:
143
144   Handle(Poly_Triangulation) myTriangulation;
145
146 };
147
148 #endif // _RWGltf_TriangulationReader_HeaderFile