#define MICHLIB_NOSOURCE
#include "GRIDVEL.h"

MString GRIDVELData::Info() const
{
 if(!isOk()) return "";

 // clang-format off
 return MString() +
  " Region: (" + u.Xb() + " : " + u.Xe() + ") x (" + u.Yb() + " : " + u.Ye() + ")\n" +
  " Grid: " + u.Nx() + "x" + u.Ny() + "\n" +
  " Supported variables: u, v, U2";
 // clang-format on
}

MString GRIDVELData::Open(const CLArgs& args)
{
 MString uname = args.contains("u") ? args.at("u") : "";
 MString vname = args.contains("v") ? args.at("v") : "";
 if(!uname.Exist()) return "File with u component not specified";
 if(!vname.Exist()) return "File with v component not specified";

 unit = args.contains("unit") ? args.at("unit") : "cm/s";
 fill = args.contains("fill") ? args.at("fill").ToReal() : 1e10;

 u.Open(uname);
 if(!u.Opened()) return "Can't open file: " + uname;
 v.Open(vname);
 if(!v.Opened()) return "Can't open file: " + vname;
 if(u != v) return "Files " + uname + " and " + vname + " have different grids";

 return "";
}

bool GRIDVELData::Read(const MString& vname, std::map<MString, GRIDVELData::Data>& cache, size_t i) const
{
 if(cache.contains(vname)) return true;
 if(!isOk()) return false;

 // Only rectangular grids are supported
 real xs = (u.Xe() - u.Xb()) / (u.Nx() - 1);
 real ys = (u.Ye() - u.Yb()) / (u.Ny() - 1);

 Data out(u.Nx(), u.Ny(), u.Xb(), u.Yb(), xs, ys, MString(unit));

 // U and U2 from u and v
 if(vname == "U" || vname == "U2")
 {
  bool square = vname == "U2";
  if(!(Read("u", cache, i) && Read("v", cache, i))) return false;
  cache[vname]      = cache.at("u").CopyGrid();
  auto&       U     = cache.at(vname);
  const auto& udata = cache.at("u");
  const auto& vdata = cache.at("v");
  if(udata.Unit().Exist()) U.SetUnit(square ? ("(" + udata.Unit() + ")2") : udata.Unit());
  for(size_t ind = 0; ind < U.N(); ind++)
  {
   if(udata.IsFill(ind) || vdata.IsFill(ind))
    U.V(ind) = U.Fillval();
   else
    U.V(ind) = square ? (udata(ind) * udata(ind) + vdata(ind) * vdata(ind)) : michlib::Hypot(udata(ind), vdata(ind));
  }
  return true;
 }

 if(vname == "u" || vname == "v")
 {
  bool isu = vname == "u";
  for(IType ix = 0; ix < u.Nx(); ix++)
   for(IType iy = 0; iy < u.Ny(); iy++)
   {
    if(isu ? (u(ix, iy) == NAN || u(ix, iy) >= 1e10) : (v(ix, iy) == NAN || v(ix, iy) >= 1e10))
     out.V(ix, iy) = out.Fillval();
    else
     out.V(ix, iy) = isu ? u(ix, iy) : v(ix, iy);
   }

  if(out)
  {
   cache[vname] = std::move(out);
   return true;
  }
 }

 return false;
}