ply_export_data.cc

Go to the documentation of this file.

/* SPDX-FileCopyrightText: 2023 Blender Authors
 *
 * SPDX-License-Identifier: GPL-2.0-or-later */
 
#include "ply_export_data.hh"
#include "ply_data.hh"
#include "ply_file_buffer.hh"
 
#include "BLI_math_vector.hh"
 
namespace blender::io::ply {
 
void write_vertices(FileBuffer &buffer, const PlyData &ply_data)
{
  for (int i = 0; i < ply_data.vertices.size(); i++) {
    buffer.write_vertex(ply_data.vertices[i].x, ply_data.vertices[i].y, ply_data.vertices[i].z);
 
    if (!ply_data.vertex_normals.is_empty()) {
      buffer.write_vertex_normal(ply_data.vertex_normals[i].x,
                                 ply_data.vertex_normals[i].y,
                                 ply_data.vertex_normals[i].z);
    }
 
    if (!ply_data.vertex_colors.is_empty()) {
      /* PLY colors currently are exported as bytes, make sure inputs are clamped. */
      float4 color = math::clamp(ply_data.vertex_colors[i], 0.0f, 1.0f) * 255.0f;
      buffer.write_vertex_color(uchar(color.x), uchar(color.y), uchar(color.z), uchar(color.w));
    }
 
    if (!ply_data.uv_coordinates.is_empty()) {
      buffer.write_UV(ply_data.uv_coordinates[i].x, ply_data.uv_coordinates[i].y);
    }
 
    for (const PlyCustomAttribute &attr : ply_data.vertex_custom_attr) {
      buffer.write_data(attr.data[i]);
    }
 
    buffer.write_vertex_end();
  }
  buffer.write_to_file();
}
 
void write_faces(FileBuffer &buffer, const PlyData &ply_data)
{
  const uint32_t *indices = ply_data.face_vertices.data();
  for (uint32_t face_size : ply_data.face_sizes) {
    buffer.write_face(char(face_size), Span<uint32_t>(indices, face_size));
    indices += face_size;
  }
  buffer.write_to_file();
}
void write_edges(FileBuffer &buffer, const PlyData &ply_data)
{
  for (const std::pair<int, int> &edge : ply_data.edges) {
    buffer.write_edge(edge.first, edge.second);
  }
  buffer.write_to_file();
}
}  // namespace blender::io::ply