forked from Kitware/VTK
-
Notifications
You must be signed in to change notification settings - Fork 0
/
vtkOBJImporterInternals.h
160 lines (131 loc) · 4.5 KB
/
vtkOBJImporterInternals.h
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
/*=========================================================================
Program: Visualization Toolkit
Module: vtkOBJImporterInternals.h
Copyright (c) Ken Martin, Will Schroeder, Bill Lorensen
All rights reserved.
See Copyright.txt or http://www.kitware.com/Copyright.htm for details.
This software is distributed WITHOUT ANY WARRANTY; without even
the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
PURPOSE. See the above copyright notice for more information.
=========================================================================*/
#ifndef vtkOBJImporterInternals_h
#define vtkOBJImporterInternals_h
#include "vtkIOImportModule.h" // For export macro
#include "vtkPolyDataAlgorithm.h"
#include <map> // for std::map
#include <string> // for std::string
#include <vector> // for std::vector
VTK_ABI_NAMESPACE_BEGIN
class vtkActor;
struct VTKIOIMPORT_EXPORT vtkOBJImportedMaterial
{
std::string name;
std::string texture_filename;
double amb[3];
double diff[3];
double spec[3];
double map_Kd_scale[3];
double map_Kd_offset[3];
int illum;
double reflect;
double refract;
double trans;
double specularPower;
double glossy;
double refract_index;
const char* GetClassName() { return "vtkOBJImportedMaterial"; }
vtkOBJImportedMaterial();
};
VTKIOIMPORT_EXPORT
void obj_set_material_defaults(vtkOBJImportedMaterial* mtl);
struct vtkOBJImportedPolyDataWithMaterial;
class VTKIOIMPORT_EXPORT vtkOBJPolyDataProcessor : public vtkPolyDataAlgorithm
{
public:
static vtkOBJPolyDataProcessor* New();
vtkTypeMacro(vtkOBJPolyDataProcessor, vtkPolyDataAlgorithm);
void PrintSelf(ostream& os, vtkIndent indent) override;
// Description:
// Specify file name of Wavefront .obj file.
void SetFileName(const char* arg)
{
if (arg == nullptr)
{
return;
}
if (this->FileName == arg)
{
return;
}
FileName = std::string(arg);
}
void SetMTLfileName(const char* arg)
{
if (arg == nullptr)
{
return;
}
if (this->MTLFileName == arg)
{
return;
}
MTLFileName = std::string(arg);
this->DefaultMTLFileName = false;
}
void SetTexturePath(const char* arg)
{
TexturePath = std::string(arg);
if (TexturePath.empty())
return;
#if defined(_WIN32)
const char sep = '\\';
#else
const char sep = '/';
#endif
if (TexturePath.at(TexturePath.size() - 1) != sep)
TexturePath += sep;
}
const std::string& GetTexturePath() const { return TexturePath; }
const std::string& GetFileName() const { return FileName; }
const std::string& GetMTLFileName() const { return MTLFileName; }
vtkSetMacro(VertexScale, double);
vtkGetMacro(VertexScale, double);
vtkGetMacro(SuccessParsingFiles, int);
virtual vtkPolyData* GetOutput(int idx);
int GetNumberOfOutputs();
vtkOBJImportedMaterial* GetMaterial(int k);
std::string GetTextureFilename(int idx); // return string by index
double VertexScale; // scale vertices by this during import
std::vector<vtkOBJImportedMaterial*> parsedMTLs;
std::map<std::string, vtkOBJImportedMaterial*> mtlName_to_mtlData;
// our internal parsing/storage
std::vector<vtkOBJImportedPolyDataWithMaterial*> poly_list;
// what gets returned to client code via GetOutput()
std::vector<vtkSmartPointer<vtkPolyData>> outVector_of_vtkPolyData;
std::vector<vtkSmartPointer<vtkActor>> actor_list;
/////////////////////
std::vector<vtkOBJImportedMaterial*> ParseOBJandMTL(std::string filename, int& result_code);
void ReadVertices(bool gotFirstUseMaterialTag, char* pLine, float xyz, int lineNr,
const double v_scale, bool everything_ok, vtkPoints* points, const bool use_scale);
protected:
vtkOBJPolyDataProcessor();
~vtkOBJPolyDataProcessor() override;
int RequestData(vtkInformation*, vtkInformationVector**, vtkInformationVector*) override
/*override*/;
vtkSetMacro(SuccessParsingFiles, int);
std::string FileName; // filename (.obj) being read
std::string MTLFileName; // associated .mtl to *.obj, typically it is *.obj.mtl
bool DefaultMTLFileName; // tells whether default of *.obj.mtl to be used
std::string TexturePath;
int SuccessParsingFiles;
private:
vtkOBJPolyDataProcessor(const vtkOBJPolyDataProcessor&) = delete;
void operator=(const vtkOBJPolyDataProcessor&) = delete;
};
class vtkRenderWindow;
class vtkRenderer;
VTKIOIMPORT_EXPORT
void bindTexturedPolydataToRenderWindow(
vtkRenderWindow* renderWindow, vtkRenderer* renderer, vtkOBJPolyDataProcessor* reader);
VTK_ABI_NAMESPACE_END
#endif