#include <ranges>

#include "UVGrid.h"
#include "readdata.h"

#define sizeError2 "The sizes of the hydrodynamic data files are different"
#define sizeError "The sizes of the hydrodynamicU or -V files does not correspond with the sizes of the grid file"

using namespace std;

UVGrid::UVGrid() {
    auto us = readHydrodynamicU();
    auto vs = readHydrodynamicV();
    if (us.size() != vs.size()) {
        throw domain_error(sizeError2);
    }

    tie(times, lats, lons) = readGrid();

    timeSize = times.size();
    latSize = lats.size();
    lonSize = lons.size();

    size_t gridSize = timeSize * latSize * lonSize;
    if (gridSize != us.size()) {
        throw domain_error(sizeError);
    }

    uvData.reserve(gridSize);

    for (auto vel: views::zip(us, vs)) {
        uvData.push_back(Vel(vel));
    }
}

const Vel &UVGrid::operator[](size_t timeIndex, size_t latIndex, size_t lonIndex) const {
    if(timeIndex < 0 or timeIndex >= timeSize
        or latIndex < 0 or latIndex >= latSize
        or lonIndex < 0 or lonIndex >= lonSize) {
        throw std::out_of_range("Index out of bounds");
    }
    size_t index = timeIndex * (latSize * lonSize) + latIndex * lonSize + lonIndex;
    return uvData[index];
}

double UVGrid::lonStep() const {
    return lons[1] - lons[0];
}

double UVGrid::latStep() const {
    return lats[1] - lats[0];
}

int UVGrid::timeStep() const {
    return times[1] - times[0];
}

void UVGrid::streamSlice(ostream &os, size_t t) {
    for (int x = 0; x < latSize; x++) {
        for (int y = 0; y < lonSize; y++) {
            auto vel = (*this)[t,x,y];
            os << vel << " ";
        }
      os << endl;
    }
}