You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
149 lines
6.7 KiB
149 lines
6.7 KiB
4 months ago
|
#pragma once
|
||
|
#include "ncsimple.h"
|
||
|
#include "zarr.h"
|
||
|
#include <variant>
|
||
|
|
||
|
using NCZarrBase = std::variant<Zarr, NCSimple>;
|
||
|
|
||
|
#define DEFFUNC(NAME) \
|
||
|
auto NAME() const \
|
||
|
{ \
|
||
|
return V([](const auto& arg) { return arg.NAME(); }); \
|
||
|
}
|
||
|
#define DEFFUNC1(NAME) \
|
||
|
auto NAME(const MString& v) const \
|
||
|
{ \
|
||
|
return V([&v = std::as_const(v)](const auto& arg) { return arg.NAME(v); }); \
|
||
|
}
|
||
|
#define DEFFUNC2(NAME) \
|
||
|
auto NAME(const MString& v1, const MString& v2) const \
|
||
|
{ \
|
||
|
return V([&v1 = std::as_const(v1), &v2 = std::as_const(v2)](const auto& arg) { return arg.NAME(v1, v2); }); \
|
||
|
}
|
||
|
|
||
|
class NCZarr: private NCZarrBase, private DimReqDef
|
||
|
{
|
||
|
template<class Visitor> auto V(Visitor&& visitor) const { return std::visit(std::forward<Visitor>(visitor), *static_cast<const NCZarrBase*>(this)); }
|
||
|
|
||
|
public:
|
||
|
explicit operator bool() const
|
||
|
{
|
||
|
return V([](const auto& arg) { return static_cast<bool>(arg); });
|
||
|
}
|
||
|
|
||
|
DEFFUNC(NDim)
|
||
|
DEFFUNC1(NDim)
|
||
|
|
||
|
DEFFUNC(NAtt)
|
||
|
DEFFUNC1(NAtt)
|
||
|
|
||
|
DEFFUNC(AttNames)
|
||
|
DEFFUNC1(AttNames)
|
||
|
|
||
|
DEFFUNC(VarNames)
|
||
|
DEFFUNC1(VarT)
|
||
|
DEFFUNC1(VarFill)
|
||
|
|
||
|
DEFFUNC(DimNames)
|
||
|
DEFFUNC1(DimNames)
|
||
|
|
||
|
DEFFUNC1(DimSize)
|
||
|
|
||
|
DEFFUNC2(AttT)
|
||
|
DEFFUNC2(AttInt)
|
||
|
DEFFUNC2(AttUInt)
|
||
|
DEFFUNC2(AttReal)
|
||
|
DEFFUNC2(AttString)
|
||
|
DEFFUNC2(AttBool)
|
||
|
|
||
|
DEFFUNC1(AttT)
|
||
|
DEFFUNC1(AttInt)
|
||
|
DEFFUNC1(AttUInt)
|
||
|
DEFFUNC1(AttReal)
|
||
|
DEFFUNC1(AttString)
|
||
|
DEFFUNC1(AttBool)
|
||
|
|
||
|
DEFFUNC1(HasDim)
|
||
|
DEFFUNC1(HasVar)
|
||
|
DEFFUNC1(HasAtt)
|
||
|
DEFFUNC2(HasAtt)
|
||
|
|
||
|
DEFFUNC(Vars)
|
||
|
DEFFUNC(Dims)
|
||
|
|
||
|
template<class Data, class Transform> Error Read(const MString& vname, Data& data, Transform transform, const char* request) const
|
||
|
{
|
||
|
return V([&vname = std::as_const(vname), &data = data, &transform = transform, request = request](const auto& arg) { return arg.Read(vname, data, transform, request); });
|
||
|
}
|
||
|
|
||
|
template<class Data, class Transform> Error Read(const MString& vname, Data& data, Transform transform, DimReq&& req1) const
|
||
|
{
|
||
|
return V([&vname = std::as_const(vname), &data = data, &transform = transform, &req1 = req1](const auto& arg) { return arg.Read(vname, data, transform, std::move(req1)); });
|
||
|
}
|
||
|
|
||
|
template<class Data, class Transform> Error Read(const MString& vname, Data& data, Transform transform, DimReq&& req1, DimReq&& req2) const
|
||
|
{
|
||
|
return V([&vname = std::as_const(vname), &data = data, &transform = transform, &req1 = req1, &req2 = req2](const auto& arg)
|
||
|
{ return arg.Read(vname, data, transform, std::move(req1), std::move(req2)); });
|
||
|
}
|
||
|
|
||
|
template<class Data, class Transform>
|
||
|
Error Read(const MString& vname, Data& data, Transform transform, DimReq&& req1, DimReq&& req2, DimReq&& req3) const
|
||
|
{
|
||
|
return V([&vname = std::as_const(vname), &data = data, &transform = transform, &req1 = req1, &req2 = req2, &req3 = req3](const auto& arg)
|
||
|
{ return arg.Read(vname, data, transform, std::move(req1), std::move(req2), std::move(req3)); });
|
||
|
}
|
||
|
|
||
|
template<class Data, class Transform>
|
||
|
Error Read(const MString& vname, Data& data, Transform transform, DimReq&& req1, DimReq&& req2, DimReq&& req3, DimReq&& req4) const
|
||
|
{
|
||
|
return V([&vname = std::as_const(vname), &data = data, &transform = transform, &req1 = req1, &req2 = req2, &req3 = req3, &req4 = req4](const auto& arg)
|
||
|
{ return arg.Read(vname, data, transform, std::move(req1), std::move(req2), std::move(req3), std::move(req4)); });
|
||
|
}
|
||
|
|
||
|
template<class Data, class Transform> Error Read(const MString& vname, Data& data, Transform transform) const
|
||
|
{
|
||
|
return V([&vname = std::as_const(vname), &data = data, &transform = transform](const auto& arg) { return arg.Read(vname, data, transform); });
|
||
|
}
|
||
|
|
||
|
template<class Data, class Transform> Error Read(const MString& vname, Data& data, Transform transform, const std::vector<DimReqDef::DimReq>& reqs) const
|
||
|
{
|
||
|
return V([&vname = std::as_const(vname), &data = data, &transform = transform, &reqs = std::as_const(reqs)](const auto& arg) { return arg.Read(vname, data, transform, reqs); });
|
||
|
}
|
||
|
|
||
|
template<class Data, class Transform> Error Read(const MString& vname, Data& data, Transform transform, const MString& request) const
|
||
|
{
|
||
|
return V([&vname = std::as_const(vname), &data = data, &transform = transform, &request = std::as_const(request)](const auto& arg)
|
||
|
{ return arg.Read(vname, data, transform, request); });
|
||
|
}
|
||
|
|
||
|
template<class Type> Error Read(const MString& vname, std::vector<Type>& out) const
|
||
|
{
|
||
|
return V([&vname = std::as_const(vname), &out = out](const auto& arg) { return arg.Read(vname, out); });
|
||
|
}
|
||
|
|
||
|
Error OpenNC(const MString& filename)
|
||
|
{
|
||
|
auto pv = static_cast<NCZarrBase*>(this);
|
||
|
|
||
|
*pv = NCZarrBase(NCSimple());
|
||
|
auto p = std::get_if<NCSimple>(pv);
|
||
|
if(p == nullptr) return Error("NCZarr::OpenNC", "Impossible error!");
|
||
|
return p->Open(filename);
|
||
|
}
|
||
|
|
||
|
Error OpenZarr(const MString& product, const MString& dataset, bool time = true)
|
||
|
{
|
||
|
auto pv = static_cast<NCZarrBase*>(this);
|
||
|
|
||
|
*pv = NCZarrBase(Zarr());
|
||
|
auto p = std::get_if<Zarr>(pv);
|
||
|
if(p == nullptr) return Error("NCZarr::OpenNC", "Impossible error!");
|
||
|
return p->Open(product, dataset, time);
|
||
|
}
|
||
|
};
|
||
|
|
||
|
#undef DEFFUNC
|
||
|
#undef DEFFUNC1
|
||
|
#undef DEFFUNC2
|