6 changed files with 166 additions and 39 deletions
@ -0,0 +1,148 @@
|
||||
#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 |
||||
Loading…
Reference in new issue