1 /*
2 Open Asset Import Library (assimp)
3 ----------------------------------------------------------------------
4 
5 Copyright (c) 2006-2021, assimp team
6 
7 All rights reserved.
8 
9 Redistribution and use of this software in source and binary forms,
10 with or without modification, are permitted provided that the
11 following conditions are met:
12 
13 * Redistributions of source code must retain the above
14   copyright notice, this list of conditions and the
15   following disclaimer.
16 
17 * Redistributions in binary form must reproduce the above
18   copyright notice, this list of conditions and the
19   following disclaimer in the documentation and/or other
20   materials provided with the distribution.
21 
22 * Neither the name of the assimp team, nor the names of its
23   contributors may be used to endorse or promote products
24   derived from this software without specific prior
25   written permission of the assimp team.
26 
27 THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
28 "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
29 LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
30 A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
31 OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
32 SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
33 LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
34 DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
35 THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
36 (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
37 OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
38 
39 ----------------------------------------------------------------------
40 */
41 
42 /** @file STLLoader.h
43  *  Declaration of the STL importer class.
44  */
45 #ifndef AI_STLLOADER_H_INCLUDED
46 #define AI_STLLOADER_H_INCLUDED
47 
48 #include <assimp/BaseImporter.h>
49 #include <assimp/types.h>
50 
51 // Forward declarations
52 struct aiNode;
53 
54 namespace Assimp {
55 
56 // ---------------------------------------------------------------------------
57 /**
58  * @brief   Importer class for the sterolithography STL file format.
59  */
60 class STLImporter : public BaseImporter {
61 public:
62     /**
63      * @brief STLImporter, the class default constructor.
64      */
65     STLImporter();
66 
67     /**
68      * @brief   The class destructor.
69      */
70     ~STLImporter() override;
71 
72     /**
73      * @brief   Returns whether the class can handle the format of the given file.
74      *  See BaseImporter::CanRead() for details.
75      */
76     bool CanRead( const std::string& pFile, IOSystem* pIOHandler, bool checkSig) const override;
77 
78 protected:
79 
80     /**
81      * @brief   Return importer meta information.
82      *  See #BaseImporter::GetInfo for the details
83      */
84     const aiImporterDesc* GetInfo () const override;
85 
86     /**
87      * @brief   Imports the given file into the given scene structure.
88     * See BaseImporter::InternReadFile() for details
89     */
90     void InternReadFile( const std::string& pFile, aiScene* pScene,
91         IOSystem* pIOHandler) override;
92 
93     /**
94      * @brief   Loads a binary .stl file
95      * @return true if the default vertex color must be used as material color
96      */
97     bool LoadBinaryFile();
98 
99     /**
100      * @brief   Loads a ASCII text .stl file
101      */
102     void LoadASCIIFile( aiNode *root );
103 
104     void pushMeshesToNode( std::vector<unsigned int> &meshIndices, aiNode *node );
105 
106 protected:
107 
108     /** Buffer to hold the loaded file */
109     const char* mBuffer;
110 
111     /** Size of the file, in bytes */
112     unsigned int mFileSize;
113 
114     /** Output scene */
115     aiScene* mScene;
116 
117     /** Default vertex color */
118     aiColor4D mClrColorDefault;
119 };
120 
121 } // end of namespace Assimp
122 
123 #endif // AI_3DSIMPORTER_H_IN
124