QGIS/external/mdal/frmts/mdal_flo2d.hpp

/*
 MDAL - Mesh Data Abstraction Library (MIT License)
 Copyright (C) 2018 Peter Petrik (zilolv at gmail dot com)
*/

#ifndef MDAL_FLO2D_HPP
#define MDAL_FLO2D_HPP

#include <string>

#include "mdal_data_model.hpp"
#include "mdal_memory_data_model.hpp"
#include "mdal.h"
#include "mdal_driver.hpp"

class HdfGroup;
class HdfFile;

namespace MDAL
{
  class DriverFlo2D: public Driver
  {
    public:
      DriverFlo2D();
      ~DriverFlo2D( ) override = default;
      DriverFlo2D *create() override;

      bool canReadMesh( const std::string &uri ) override;
      bool canReadDatasets( const std::string &uri ) override;

      std::unique_ptr< Mesh > load( const std::string &resultsFile, const std::string &meshName = "" ) override;
      void load( const std::string &uri, Mesh *mesh ) override;
      bool persist( DatasetGroup *group ) override;

    private:
      struct CellCenter
      {
        size_t id;
        double x;
        double y;
        std::vector<int> conn; // north, east, south, west cell center index, -1 boundary Vertex
      };

      std::unique_ptr< MDAL::MemoryMesh > mMesh;
      std::string mDatFileName;

      void createMesh( const std::vector<CellCenter> &cells, double half_cell_size );
      void parseOUTDatasets( const std::string &datFileName, const std::vector<double> &elevations );
      bool parseHDF5Datasets( MDAL::MemoryMesh *mesh, const std::string &timedepFileName );
      void parseVELFPVELOCFile( const std::string &datFileName );
      void parseDEPTHFile( const std::string &datFileName, const std::vector<double> &elevations );
      void parseTIMDEPFile( const std::string &datFileName, const std::vector<double> &elevations );
      void parseFPLAINFile( std::vector<double> &elevations, const std::string &datFileName, std::vector<CellCenter> &cells );
      void parseCADPTSFile( const std::string &datFileName, std::vector<CellCenter> &cells );
      void addStaticDataset( std::vector<double> &vals, const std::string &groupName, const std::string &datFileName );
      static MDAL::Vertex createVertex( size_t position, double half_cell_size, const CellCenter &cell );
      static double calcCellSize( const std::vector<CellCenter> &cells );

      // Write API
      bool addToHDF5File( DatasetGroup *group );
      bool saveNewHDF5File( DatasetGroup *group );
      bool appendGroup( HdfFile &file, DatasetGroup *dsGroup, HdfGroup &groupTNOR );

  };

} // namespace MDAL
#endif //MDAL_FLO2D_HPP