ply_export_header.cc

Go to the documentation of this file.

/* SPDX-FileCopyrightText: 2023 Blender Authors
 *
 * SPDX-License-Identifier: GPL-2.0-or-later */
 
#include "BKE_blender_version.h"
 
#include "IO_ply.hh"
#include "ply_data.hh"
#include "ply_export_header.hh"
#include "ply_file_buffer.hh"
 
namespace blender::io::ply {
 
void write_header(FileBuffer &buffer,
                  const PlyData &ply_data,
                  const PLYExportParams &export_params)
{
  buffer.write_string("ply");
 
  StringRef format = export_params.ascii_format ? "ascii" : "binary_little_endian";
  buffer.write_string("format " + format + " 1.0");
 
  StringRef version = BKE_blender_version_string();
  buffer.write_string("comment Created in Blender version " + version);
 
  buffer.write_header_element("vertex", int32_t(ply_data.vertices.size()));
  buffer.write_header_scalar_property("float", "x");
  buffer.write_header_scalar_property("float", "y");
  buffer.write_header_scalar_property("float", "z");
 
  if (!ply_data.vertex_normals.is_empty()) {
    buffer.write_header_scalar_property("float", "nx");
    buffer.write_header_scalar_property("float", "ny");
    buffer.write_header_scalar_property("float", "nz");
  }
 
  if (!ply_data.vertex_colors.is_empty()) {
    buffer.write_header_scalar_property("uchar", "red");
    buffer.write_header_scalar_property("uchar", "green");
    buffer.write_header_scalar_property("uchar", "blue");
    buffer.write_header_scalar_property("uchar", "alpha");
  }
 
  if (!ply_data.uv_coordinates.is_empty()) {
    buffer.write_header_scalar_property("float", "s");
    buffer.write_header_scalar_property("float", "t");
  }
 
  for (const PlyCustomAttribute &attr : ply_data.vertex_custom_attr) {
    buffer.write_header_scalar_property("float", attr.name);
  }
 
  if (!ply_data.face_sizes.is_empty()) {
    buffer.write_header_element("face", int(ply_data.face_sizes.size()));
    buffer.write_header_list_property("uchar", "uint", "vertex_indices");
  }
 
  if (!ply_data.edges.is_empty()) {
    buffer.write_header_element("edge", int(ply_data.edges.size()));
    buffer.write_header_scalar_property("int", "vertex1");
    buffer.write_header_scalar_property("int", "vertex2");
  }
 
  buffer.write_string("end_header");
  buffer.write_to_file();
}
 
}  // namespace blender::io::ply