108 lines
3.5 KiB
C++
108 lines
3.5 KiB
C++
// Copyright 2016 The Draco Authors.
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
//
|
|
#ifndef DRACO_IO_OBJ_ENCODER_H_
|
|
#define DRACO_IO_OBJ_ENCODER_H_
|
|
|
|
#include <unordered_map>
|
|
|
|
#include "draco/core/encoder_buffer.h"
|
|
#include "draco/mesh/corner_table.h"
|
|
#include "draco/mesh/mesh.h"
|
|
|
|
namespace draco {
|
|
|
|
// Class for encoding input draco::Mesh or draco::PointCloud into the Wavefront
|
|
// OBJ format.
|
|
class ObjEncoder {
|
|
public:
|
|
ObjEncoder();
|
|
|
|
// Encodes the mesh or a point cloud and saves it into a file.
|
|
// Returns false when either the encoding failed or when the file couldn't be
|
|
// opened.
|
|
bool EncodeToFile(const PointCloud &pc, const std::string &file_name);
|
|
bool EncodeToFile(const Mesh &mesh, const std::string &file_name);
|
|
|
|
// Encodes the mesh or the point cloud into a buffer.
|
|
bool EncodeToBuffer(const PointCloud &pc, EncoderBuffer *out_buffer);
|
|
bool EncodeToBuffer(const Mesh &mesh, EncoderBuffer *out_buffer);
|
|
|
|
protected:
|
|
bool EncodeInternal();
|
|
EncoderBuffer *buffer() const { return out_buffer_; }
|
|
bool ExitAndCleanup(bool return_value);
|
|
|
|
private:
|
|
typedef AttributeValueIndex PositionIndex;
|
|
typedef std::map<PositionIndex, PointIndex> PolygonEdges;
|
|
bool GetAddedEdges();
|
|
bool GetSubObjects();
|
|
bool EncodeMaterialFileName();
|
|
bool EncodePositions();
|
|
bool EncodeTextureCoordinates();
|
|
bool EncodeNormals();
|
|
bool EncodeFaces();
|
|
bool EncodePolygonalFaces();
|
|
bool EncodeFaceAttributes(FaceIndex face_id);
|
|
bool EncodeSubObject(FaceIndex face_id);
|
|
bool EncodeMaterial(FaceIndex face_id);
|
|
bool EncodeFaceCorner(FaceIndex face_id, int local_corner_id);
|
|
bool EncodeFaceCorner(PointIndex vert_index);
|
|
|
|
void EncodeFloat(float val);
|
|
void EncodeFloatList(float *vals, int num_vals);
|
|
void EncodeInt(int32_t val);
|
|
bool IsNewEdge(const CornerTable &ct, CornerIndex ci) const;
|
|
void FindOriginalFaceEdges(FaceIndex face_index,
|
|
const CornerTable &corner_table,
|
|
std::vector<bool> *triangle_visited,
|
|
PolygonEdges *polygon_edges);
|
|
|
|
// Various attributes used by the encoder. If an attribute is not used, it is
|
|
// set to nullptr.
|
|
const PointAttribute *pos_att_;
|
|
const PointAttribute *tex_coord_att_;
|
|
const PointAttribute *normal_att_;
|
|
const PointAttribute *material_att_;
|
|
const PointAttribute *sub_obj_att_;
|
|
|
|
// Stores per-corner triangulation information for polygon reconstruction.
|
|
const PointAttribute *added_edges_att_;
|
|
|
|
// Buffer used for encoding float/int numbers.
|
|
char num_buffer_[20];
|
|
|
|
EncoderBuffer *out_buffer_;
|
|
|
|
const PointCloud *in_point_cloud_;
|
|
const Mesh *in_mesh_;
|
|
|
|
// Store sub object name for each value.
|
|
std::unordered_map<int, std::string> sub_obj_id_to_name_;
|
|
// Current sub object id of faces.
|
|
int current_sub_obj_id_;
|
|
|
|
// Store material name for each value in material attribute.
|
|
std::unordered_map<int, std::string> material_id_to_name_;
|
|
// Current material id of faces.
|
|
int current_material_id_;
|
|
|
|
std::string file_name_;
|
|
};
|
|
|
|
} // namespace draco
|
|
|
|
#endif // DRACO_IO_OBJ_ENCODER_H_
|