ply_importer.h

Go to the documentation of this file.

#ifndef LICORNEA_PLY_IMPORTER_H_
#define LICORNEA_PLY_IMPORTER_H_

#include <fstream>
#include <string>
#include <cassert>
#include <vector>
#include <array>
#include <cstdlib>
#include <cstdint>
#include <limits>
#include <utility>
#include <memory>
#include <cctype>
#include <stdexcept>

#include "common.h"
#include "io.h"
#include "point.h"
    
namespace tlz {
    
class ply_importer_error : public std::runtime_error {
public:
    using std::runtime_error::runtime_error;
};



class ply_importer {
private:
    const static std::size_t maximal_ascii_element_line_length_ = 256;

    enum property_type {
        none, int8, uint8, int16, uint16, int32, uint32, float32, float64, list
    };

    struct property {
        property_type type = none;
        std::ptrdiff_t offset;
        std::ptrdiff_t index;
        
        explicit operator bool() const { return (type != none); }
    };
        
    std::ifstream file_;
    line_delimitor line_delimitor_;
    std::ifstream::pos_type vertex_data_start_; 
    std::size_t number_of_vertices_; 
    enum { binary_big_endian, binary_little_endian, ascii } format_; 
    std::size_t vertex_length_; 
    std::size_t number_of_properties_; 
    bool has_rgb_; 
    bool has_normal_; 
    bool has_weight_; 
    property x_, y_, z_, r_, g_, b_, nx_, ny_, nz_, w_; 
    
    std::ptrdiff_t current_element_; 


    void skip_lines_(std::size_t n) {
        while(n--) skip_line(file_, line_delimitor_);
    }
    
    void read_line_(std::string& ln) {
        read_line(file_, ln, line_delimitor_);
    }
    
    bool is_host_endian_binary_() const {
        if(host_is_little_endian) return (format_ == binary_little_endian);
        else return (format_ == binary_big_endian);
    }
    
    property* identify_property_(const std::string& nm);
    static property_type identify_property_type_(const std::string& nm);
    static std::size_t property_type_size_(property_type t);
    void read_header_();
    
    template<typename Point> void read_ascii_(Point* buffer, std::size_t n);
    template<typename Point> void read_binary_(Point* buffer, std::size_t n);
    
    void read_ascii_point_(point_xyz& out_point, const char* props[]) const;
    void read_ascii_point_(point_full& out_point, const char* props[]) const;
    void read_binary_point_(point_xyz& out_point, byte* data) const;
    void read_binary_point_(point_full& out_point, byte* data) const;
    template<typename T> T read_binary_property_(const property& prop, byte* data) const;

public:
    explicit ply_importer(const std::string& filename, line_delimitor ld = line_delimitor::unknown);

    std::size_t size() const;
    
    void rewind();
    std::ptrdiff_t tell() const;
    void read(point_xyz*, std::size_t sz);
    void read(point_full*, std::size_t sz);

    bool is_binary() const { return format_ != ascii; }
    bool is_ascii() const { return format_ == ascii; }
};

}

#endif