mirror of
https://github.com/qgis/QGIS.git
synced 2025-02-27 00:33:48 -05:00
110 lines
3.4 KiB
C++
110 lines
3.4 KiB
C++
/*
|
|
MDAL - Mesh Data Abstraction Library (MIT License)
|
|
Copyright (C) 2018 Peter Petrik (zilolv at gmail dot com)
|
|
*/
|
|
|
|
#ifndef MDAL_CF_HPP
|
|
#define MDAL_CF_HPP
|
|
|
|
#include <string>
|
|
#include <vector>
|
|
#include <map>
|
|
#include <stddef.h>
|
|
#include <set>
|
|
|
|
#include "mdal_data_model.hpp"
|
|
#include "mdal.h"
|
|
#include "mdal_utils.hpp"
|
|
#include "mdal_netcdf.hpp"
|
|
#include "mdal_driver.hpp"
|
|
|
|
namespace MDAL
|
|
{
|
|
class CFDimensions
|
|
{
|
|
public:
|
|
enum Type
|
|
{
|
|
UnknownType = 0, //!< Unknown
|
|
Vertex1D, //!< Vertex in 1D mesh
|
|
Vertex2D, //!< Vertex in 2D mesh
|
|
Line1D, //!< Line joining 1D vertices
|
|
Face2DEdge, //!< Edge of 2D Face
|
|
Face2D, //!< 2D (Polygon) Face
|
|
Time, //!< Time steps
|
|
MaxVerticesInFace //!< Maximum number of vertices in a face
|
|
};
|
|
|
|
//! Returns type of dimensions based on NetCDF array id
|
|
Type type( int ncid ) const;
|
|
//! Returns number of items for the type
|
|
size_t size( Type type ) const;
|
|
//! Sets a dimension
|
|
void setDimension( Type type, size_t count, int ncid = -1 );
|
|
//! Returns whether the type is one that case be used for datasets definition
|
|
bool isDatasetType( Type type ) const;
|
|
|
|
private:
|
|
std::map<Type, size_t> mCount;
|
|
std::map<int, Type> mNcId;
|
|
};
|
|
|
|
struct CFDatasetGroupInfo
|
|
{
|
|
std::string name; //!< Dataset group name
|
|
CFDimensions::Type outputType;
|
|
bool is_vector;
|
|
size_t nTimesteps;
|
|
int ncid_x; //!< NetCDF variable id
|
|
int ncid_y; //!< NetCDF variable id
|
|
size_t arr_size;
|
|
};
|
|
typedef std::map<std::string, CFDatasetGroupInfo> cfdataset_info_map; // name -> DatasetInfo
|
|
|
|
//! NetCDF Climate and Forecast (CF) Metadata Conventions
|
|
//! http://cfconventions.org
|
|
class DriverCF: public Driver
|
|
{
|
|
public:
|
|
DriverCF( const std::string &name,
|
|
const std::string &longName,
|
|
const std::string &filters );
|
|
virtual ~DriverCF() override = default;
|
|
bool canRead( const std::string &uri ) override;
|
|
std::unique_ptr< Mesh > load( const std::string &fileName, MDAL_Status *status ) override;
|
|
|
|
protected:
|
|
virtual CFDimensions populateDimensions( const NetCDFFile &ncFile ) = 0;
|
|
virtual void populateFacesAndVertices( Vertices &vertices, Faces &faces ) = 0;
|
|
virtual void addBedElevation( MDAL::Mesh *mesh ) = 0;
|
|
virtual std::string getCoordinateSystemVariableName() = 0;
|
|
virtual std::set<std::string> ignoreNetCDFVariables() = 0;
|
|
virtual std::string nameSuffix( CFDimensions::Type type ) = 0;
|
|
virtual void parseNetCDFVariableMetadata( int varid, const std::string &variableName,
|
|
std::string &name, bool *is_vector, bool *is_x ) = 0;
|
|
|
|
void setProjection( MDAL::Mesh *m );
|
|
cfdataset_info_map parseDatasetGroupInfo();
|
|
void parseTime( std::vector<double> × );
|
|
std::shared_ptr<MDAL::Dataset> createFace2DDataset(
|
|
std::shared_ptr<MDAL::DatasetGroup> group,
|
|
size_t ts,
|
|
const MDAL::CFDatasetGroupInfo &dsi,
|
|
const std::vector<double> &vals_x,
|
|
const std::vector<double> &vals_y,
|
|
double fill_val_x, double fill_val_y );
|
|
|
|
void addDatasetGroups( Mesh *mesh,
|
|
const std::vector<double> ×,
|
|
const cfdataset_info_map &dsinfo_map );
|
|
|
|
|
|
std::string mFileName;
|
|
NetCDFFile mNcFile;
|
|
CFDimensions mDimensions;
|
|
};
|
|
|
|
} // namespace MDAL
|
|
|
|
#endif // MDAL_CF_HPP
|