PLYWriter.hh 7.41 KB
Newer Older
Jan Möbius's avatar
Jan Möbius committed
1
/* ========================================================================= *
Jan Möbius's avatar
Jan Möbius committed
2
3
 *                                                                           *
 *                               OpenMesh                                    *
Jan Möbius's avatar
Jan Möbius committed
4
 *           Copyright (c) 2001-2015, RWTH-Aachen University                 *
Jan Möbius's avatar
Typo    
Jan Möbius committed
5
 *           Department of Computer Graphics and Multimedia                  *
Jan Möbius's avatar
Jan Möbius committed
6
7
 *                          All rights reserved.                             *
 *                            www.openmesh.org                               *
Jan Möbius's avatar
Jan Möbius committed
8
 *                                                                           *
9
 *---------------------------------------------------------------------------*
Jan Möbius's avatar
Jan Möbius committed
10
11
 * This file is part of OpenMesh.                                            *
 *---------------------------------------------------------------------------*
Jan Möbius's avatar
Jan Möbius committed
12
 *                                                                           *
Jan Möbius's avatar
Jan Möbius committed
13
14
15
 * Redistribution and use in source and binary forms, with or without        *
 * modification, are permitted provided that the following conditions        *
 * are met:                                                                  *
Jan Möbius's avatar
Jan Möbius committed
16
 *                                                                           *
Jan Möbius's avatar
Jan Möbius committed
17
18
 * 1. Redistributions of source code must retain the above copyright notice, *
 *    this list of conditions and the following disclaimer.                  *
Jan Möbius's avatar
Jan Möbius committed
19
 *                                                                           *
Jan Möbius's avatar
Jan Möbius committed
20
21
22
 * 2. Redistributions in binary form must reproduce the above copyright      *
 *    notice, this list of conditions and the following disclaimer in the    *
 *    documentation and/or other materials provided with the distribution.   *
Jan Möbius's avatar
Jan Möbius committed
23
 *                                                                           *
Jan Möbius's avatar
Jan Möbius committed
24
25
26
 * 3. Neither the name of the copyright holder nor the names of its          *
 *    contributors may be used to endorse or promote products derived from   *
 *    this software without specific prior written permission.               *
27
 *                                                                           *
Jan Möbius's avatar
Jan Möbius committed
28
29
30
31
32
33
34
35
36
37
38
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS       *
 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED *
 * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A           *
 * PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER *
 * OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,  *
 * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,       *
 * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR        *
 * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF    *
 * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING      *
 * NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS        *
 * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.              *
Jan Möbius's avatar
Jan Möbius committed
39
40
 *                                                                           *
 * ========================================================================= */
41

42

Jan Möbius's avatar
Jan Möbius committed
43
44
45
46
47
48
49
50
51
52


//=============================================================================
//
//  Implements a writer module for PLY files
//
//=============================================================================


#ifndef __PLYWRITER_HH__
Jan Möbius's avatar
Jan Möbius committed
53
#define __PLYWRITER_HH__
Jan Möbius's avatar
Jan Möbius committed
54
55
56
57
58


//=== INCLUDES ================================================================

#include <string>
Matthias Möller's avatar
Matthias Möller committed
59
#include <ostream>
60
#include <vector>
Jan Möbius's avatar
Jan Möbius committed
61
62
63
64
65

#include <OpenMesh/Core/System/config.h>
#include <OpenMesh/Core/Utils/SingletonT.hh>
#include <OpenMesh/Core/IO/exporter/BaseExporter.hh>
#include <OpenMesh/Core/IO/writer/BaseWriter.hh>
66
#include <OpenMesh/Core/Utils/GenProg.hh>
Jan Möbius's avatar
Jan Möbius committed
67
68
69
70
71
72
73
74
75
76
77
78


//== NAMESPACES ===============================================================


namespace OpenMesh {
namespace IO {


//=== IMPLEMENTATION ==========================================================


Jan Möbius's avatar
Jan Möbius committed
79
80
/**
    Implementation of the PLY format writer. This class is singleton'ed by
Jan Möbius's avatar
Jan Möbius committed
81
82
83
84
    SingletonT to PLYWriter.

    currently supported options:
    - VertexColors
85
    - FaceColors
Jan Möbius's avatar
Jan Möbius committed
86
87
88
    - Binary
    - Binary -> MSB
*/
89
class OPENMESHDLLEXPORT _PLYWriter_ : public BaseWriter
Jan Möbius's avatar
Jan Möbius committed
90
91
92
93
94
{
public:

  _PLYWriter_();

95
96
97
  /// Destructor
  virtual ~_PLYWriter_() {};

Jan Möbius's avatar
Jan Möbius committed
98
99
  std::string get_description() const override { return "PLY polygon file format"; }
  std::string get_extensions()  const override  { return "ply"; }
Jan Möbius's avatar
Jan Möbius committed
100

Jan Möbius's avatar
Jan Möbius committed
101
  bool write(const std::string&, BaseExporter&, Options, std::streamsize _precision = 6) const override;
102

Jan Möbius's avatar
Jan Möbius committed
103
  bool write(std::ostream&, BaseExporter&, Options, std::streamsize _precision = 6) const override;
Jan Möbius's avatar
Jan Möbius committed
104

Jan Möbius's avatar
Jan Möbius committed
105
  size_t binary_size(BaseExporter& _be, Options _opt) const override;
Jan Möbius's avatar
Jan Möbius committed
106
107

  enum ValueType {
108
    Unsupported = 0,
Jan Möbius's avatar
Jan Möbius committed
109
    ValueTypeFLOAT32, ValueTypeFLOAT,
110
111
112
113
    ValueTypeINT32, ValueTypeINT , ValueTypeUINT,
    ValueTypeUCHAR, ValueTypeCHAR, ValueTypeUINT8,
    ValueTypeUSHORT, ValueTypeSHORT,
    ValueTypeDOUBLE
Jan Möbius's avatar
Jan Möbius committed
114
115
116
117
118
  };

private:
  mutable Options options_;

119
120
121
122
123
124
125
126
127
128
129
  struct CustomProperty
  {
    ValueType type;
    const BaseProperty*  property;
    CustomProperty(const BaseProperty* const _p):type(Unsupported),property(_p){}
  };

  const char* nameOfType_[12];

  /// write custom persistant properties into the header for the current element, returns all properties, which were written sorted
  std::vector<CustomProperty> writeCustomTypeHeader(std::ostream& _out, BaseKernel::const_prop_iterator _begin, BaseKernel::const_prop_iterator _end) const;
130
131
132
133
134
135
136
137
138
139
140
141
  template<bool binary>
  void write_customProp(std::ostream& _our, const CustomProperty& _prop, size_t _index) const;
  template<typename T>
  void writeProxy(ValueType _type, std::ostream& _out, T _value, OpenMesh::GenProg::TrueType /*_binary*/) const
  {
    writeValue(_type, _out, _value);
  }
  template<typename T>
  void writeProxy(ValueType _type, std::ostream& _out, T _value, OpenMesh::GenProg::FalseType /*_binary*/) const
  {
    _out << " " << _value;
  }
142

Jan Möbius's avatar
Jan Möbius committed
143
protected:
144
145
146
147
  void writeValue(ValueType _type, std::ostream& _out, signed char value) const;
  void writeValue(ValueType _type, std::ostream& _out, unsigned char value) const;
  void writeValue(ValueType _type, std::ostream& _out, short value) const;
  void writeValue(ValueType _type, std::ostream& _out, unsigned short value) const;
148
149
150
  void writeValue(ValueType _type, std::ostream& _out, int value) const;
  void writeValue(ValueType _type, std::ostream& _out, unsigned int value) const;
  void writeValue(ValueType _type, std::ostream& _out, float value) const;
151
  void writeValue(ValueType _type, std::ostream& _out, double value) const;
Jan Möbius's avatar
Jan Möbius committed
152

153
154
  bool write_ascii(std::ostream& _out, BaseExporter&, Options) const;
  bool write_binary(std::ostream& _out, BaseExporter&, Options) const;
155
156
  /// write header into the stream _out. Returns custom properties (vertex and face) which are written into the header
  void write_header(std::ostream& _out, BaseExporter& _be, Options& _opt, std::vector<CustomProperty>& _ovProps, std::vector<CustomProperty>& _ofProps) const;
Jan Möbius's avatar
Jan Möbius committed
157
158
159
160
161
162
163
164
};


//== TYPE DEFINITION ==========================================================


/// Declare the single entity of the PLY writer.
extern _PLYWriter_  __PLYWriterInstance;
Mike Kremer's avatar
Mike Kremer committed
165
OPENMESHDLLEXPORT _PLYWriter_& PLYWriter();
Jan Möbius's avatar
Jan Möbius committed
166
167
168
169
170
171
172
173


//=============================================================================
} // namespace IO
} // namespace OpenMesh
//=============================================================================
#endif
//=============================================================================