/********************************************************************************

* *

* This file is part of IfcOpenShell. *

* *

* IfcOpenShell is free software: you can redistribute it and/or modify *

* it under the terms of the Lesser GNU General Public License as published by *

* the Free Software Foundation, either version 3.0 of the License, or *

* (at your option) any later version. *

* *

* IfcOpenShell is distributed in the hope that it will be useful, *

* but WITHOUT ANY WARRANTY; without even the implied warranty of *

* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the *

* Lesser GNU General Public License for more details. *

* *

* You should have received a copy of the Lesser GNU General Public License *

* along with this program. If not, see <http://www.gnu.org/licenses/>. *

* *

********************************************************************************/

#ifndef HDFSERIALIZER_H

#define HDFSERIALIZER_H

#ifdef WITH_HDF5

#include <set>

#include <string>

#include <fstream>

#include "H5Cpp.h"

#include "../serializers/serializers_api.h"

#include "../ifcgeom_schema_agnostic/GeometrySerializer.h"

#define USE_BINARY

class SERIALIZERS_API HdfSerializer : public GeometrySerializer {

private:

const std::string hdf_filename;

unsigned int vcount_total;

H5::H5File file;

SerializerSettings settings_;

static const H5std_string DATASET_NAME_POSITIONS;

static const H5std_string DATASET_NAME_UVCOORDS;

static const H5std_string DATASET_NAME_NORMALS;

static const H5std_string DATASET_NAME_INDICES;

static const H5std_string DATASET_NAME_EDGES;

static const H5std_string DATASET_NAME_MATERIAL_IDS;

static const H5std_string DATASET_NAME_ITEM_IDS;

static const H5std_string DATASET_NAME_MATERIALS;

static const H5std_string DATASET_NAME_OCCT;

static const H5std_string DATASET_NAME_PLACEMENT;

static const H5std_string GROUP_NAME_MESH;

struct surface_style_serialization {

const char* name;

const char* original_name;

// 0 if unset

unsigned int id;

// nan if unset

double diffuse[3];

double specular[3];

double transparency;

double specularity;

};

struct brep_element {

int id;

double matrix[4][4];

#ifdef USE_BINARY

hvl_t shape_serialization;

#else

const char* shape_serialization;

#endif

surface_style_serialization surface_style;

};

H5::CompType compound;

H5::CompType style_compound;

H5::StrType str_type;

H5::ArrayType double4x4, double3;

H5::DataType shape_type;

private:

std::map<std::string, boost::shared_ptr<IfcGeom::Representation::BRep>> brep_cache_;

std::map<std::string, boost::shared_ptr<IfcGeom::Representation::Triangulation>> triangulation_cache_;

std::map<std::string, std::string> group_cache_;

H5::Group createRepresentationGroup(const H5::Group& element_group, const std::string& gid);

void read_surface_style(surface_style_serialization& sss, std::shared_ptr<IfcGeom::SurfaceStyle>& style_ptr);

void write_style(surface_style_serialization& data, const IfcGeom::SurfaceStyle& s);

public:

HdfSerializer(const std::string& hdf_filename, const SerializerSettings& settings, bool read_only=false);

virtual ~HdfSerializer() {}

bool ready();

void writeHeader();

H5::Group write(const IfcGeom::Element* o);

void write(const IfcGeom::BRepElement* o);

void write(const IfcGeom::TriangulationElement* o);

void remove(const std::string& guid);

IfcGeom::Element* read(IfcParse::IfcFile& f, const std::string& guid, const std::string&, read_type rt = READ_BREP);

void finalize() {}

bool isTesselated() const { return false; }

void setUnitNameAndMagnitude(const std::string& /*name*/, float /*magnitude*/) {}

void setFile(IfcParse::IfcFile*) {}

};

#endif

#endif