odm/actions/actiontsc.h

#pragma once
#include "BFileW.h"
#include "actiondep.h"
#include "ncfilew.h"
#include "ncfuncs.h"
#include <memory>

using michlib::BFileW;

ADD_ACTION(TSC, tsc, ReadPSupported<Source> || ReadSupported<Source>);

template<class D> MString ActionTSC::DoAction(const CLArgs& args, D& ds)
{
 auto [reg, regerr] = GetRegion<D>(args);
 if(regerr.Exist()) return regerr;

 auto resop = ds.Open(args);
 if(resop.Exist()) return "Can't open source: " + resop;

 michlib_internal::ParameterListEx pars;
 pars.UsePrefix("");
 pars.SetParameter("source", args.at("source"));

 auto [tindexes, err] = GetTIndexes(ds, args, pars);
 if(err.Exist()) return err;

 MString varstring;
 if(args.contains("var"))
  varstring = args.at("var");
 else
 {
  if(args.contains("vars"))
   varstring = args.at("vars");
  else
  {
   if constexpr(HasDefVars<D>)
    varstring = ds.DefaultVars();
   else
    return "Variables not specified";
  }
 }

 auto                 vlist = michlib::Split_on_words(varstring, " ,", false);
 std::vector<MString> vnames{std::make_move_iterator(std::begin(vlist)), std::make_move_iterator(std::end(vlist))};
 {
  std::set<MString> used;
  for(const auto& vname: vnames)
  {
   if(used.contains(vname)) return "Duplicate variable " + vname + " in list " + varstring;
   if(!ds.CheckVar(vname)) return "Variable " + vname + " not exists in this dataset";
   used.insert(vname);
  }
 }

 pars.SetParameter("variables", varstring);

 std::unique_ptr<const BaseParameters> sourcepars;
 if constexpr(ParametersSupported<D>)
 {
  if constexpr(ParametersRequiredRegion<D>)
  {
   auto [p, err] = ds.Parameters(pars, args, reg);
   if(err.Exist()) return err;
   sourcepars.reset(p);
  }
  else
  {
   auto [p, err] = ds.Parameters(pars, args);
   if(err.Exist()) return err;
   sourcepars.reset(p);
  }
 }
 auto p = sourcepars.get();

 auto data = Read(ds, vnames, p, tindexes);
 if(data.size() != vnames.size()) return "Can't read data";

 for(size_t i = 0; i < vnames.size(); i++)
 {
  if(!data[i].Unit().Exist()) michlib::errmessage("Unknown measurement unit for variable " + vnames[i] + "!");
  if(!data[i].StandartName().Exist()) data[i].SetStandartName(NCFuncs::Name2StName(vnames[i]));
  if(!data[i].LongName().Exist()) data[i].SetLongName(NCFuncs::Name2LongName(vnames[i]));
 }

 MString name   = args.contains("out") ? args.at("out") : "out.bin";
 MString outfmt = args.contains("format") ? args.at("format") : (GetExt(name) == "nc" ? "nc" : "bin");

 if(outfmt == "bin")
 {
  BFileW fw;

  size_t ic = 0;
  fw.Create(name, 2 + data.size());
  fw.SetColumnName(++ic, "Longitude");
  fw.SetColumnName(++ic, "Latitude");
  for(size_t i = 0; i < data.size(); i++) fw.SetColumnName(++ic, vnames[i] + ", " + (data[i].Unit().Exist() ? data[i].Unit() : "unknown"));
  fw.SetParameters(pars);
  for(size_t r = 0; r < data[0].N(); r++)
  {
   fw.Write(data[0].Lon(r));
   fw.Write(data[0].Lat(r));
   for(size_t i = 0; i < data.size(); i++) fw.Write(data[i].IsFill(r) ? NAN : data[i](r));
  }
  fw.Finalize();
  fw.Close();
  return "";
 }

 if(outfmt == "nc" || outfmt == "netcdf")
 {
  int     compress = 3;
  NCFileW fw;
  MString err;

  if(args.contains("compress")) compress = args.at("compress").ToInt();

  if(!err.Exist()) err = fw.Create(data[0], name, args.at("_cmdline"), compress);
  for(size_t i = 0; i < data.size(); i++)
   if(!err.Exist()) err = fw.AddVariable(vnames[i], data[i].StandartName(), data[i].LongName(), data[i].Unit(), data[i].Comment());
  if(!err.Exist()) err = fw.WriteGrid(data[0]);
  for(size_t i = 0; i < data.size(); i++)
   if(!err.Exist()) err = fw.WriteVariable(data[i], vnames[i]);
  if(err.Exist()) return err;
  fw.Close();

  return "";
 }

 return "Unknown format: " + outfmt;
};