/*
 * (C) Copyright 1996- ECMWF.
 *
 * This software is licensed under the terms of the Apache Licence Version 2.0
 * which can be obtained at http://www.apache.org/licenses/LICENSE-2.0.
 *
 * In applying this licence, ECMWF does not waive the privileges and immunities
 * granted to it by virtue of its status as an intergovernmental organisation nor
 * does it submit to any jurisdiction.
 */


#include "mir/netcdf/OutputDataset.h"

#include <netcdf.h>

#include <ostream>

#include "mir/netcdf/Attribute.h"
#include "mir/netcdf/Dimension.h"
#include "mir/netcdf/DummyInputVariable.h"
#include "mir/netcdf/DummyMatrix.h"
#include "mir/netcdf/Exceptions.h"
#include "mir/netcdf/MergePlan.h"
#include "mir/netcdf/Variable.h"
#include "mir/util/Log.h"


namespace mir::netcdf {


OutputDataset::OutputDataset(const std::string& path, NCFileCache& cache, int format) :
    Dataset(path), format_(format != 0 ? format : NC_FORMAT_NETCDF4_CLASSIC), cache_(cache) {}


OutputDataset::~OutputDataset() = default;


void OutputDataset::print(std::ostream& out) const {
    out << "OutputDataset[path=" << path_ << "]";
}


void OutputDataset::merge(Dataset& other) {

    if (dimensions_.empty() && attributes_.empty() && variables_.empty()) {
        // First time, just adopt
        for (const auto& j : other.dimensions()) {
            (j.second)->clone(*this);
        }

        for (const auto& j : other.attributes()) {
            (j.second)->clone(*this);
        }

        for (const auto& j : other.variables()) {
            (j.second)->clone(*this);
        }
        return;
    }

    bool more = true;
    while (more) {
        more = false;
        for (const auto& j : other.variables()) {
            bool found = false;
            for (auto& k : variables_) {
                if ((k.second)->sameAs(*(j.second))) {
                    found = true;
                    break;
                }
            }
            if (!found) {
                // Variable not in output file
                Log::info() << "MISSING in output " << *(j.second) << std::endl;
                Variable* v = j.second;
                v->clone(*this)->setMatrix(new DummyMatrix(*v));
                more = true;
                break;
            }
        }
    }

    more = true;
    while (more) {
        more = false;
        for (auto& k : variables_) {
            bool found = false;
            for (const auto& j : other.variables()) {
                if ((k.second)->sameAs(*(j.second))) {
                    found = true;
                    break;
                }
            }
            if (!found) {
                // Variable not in input file
                Log::info() << "MISSING in input " << *(k.second) << std::endl;
                other.add(new DummyInputVariable(other, *(k.second)));
                more = true;
                break;
            }
        }
    }

    MergePlan plan(*this);

    mergeAttributes(other);

    for (const auto& j : other.variables()) {
        bool found = false;
        for (auto& k : variables_) {
            if ((k.second)->sameAs(*(j.second))) {
                (k.second)->merge(*(j.second), plan);
                found = true;
            }
        }
        ASSERT(found);
    }

    plan.execute();
}


void OutputDataset::save() const {
    int flags = 0;
    switch (format_) {
        case NC_FORMAT_CLASSIC:
            flags = 0;
            break;

        case NC_FORMAT_64BIT:
            flags = NC_64BIT_OFFSET;
            break;

        case NC_FORMAT_NETCDF4:
            flags = NC_NETCDF4;
            break;

        case NC_FORMAT_NETCDF4_CLASSIC:
            flags = NC_NETCDF4 | NC_CLASSIC_MODEL;
            break;
    }

    int nc;

    NC_CALL(nc_create(path_.c_str(), flags | NC_WRITE, &nc), path_);
    NC_CALL(nc_set_fill(nc, NC_NOFILL, NULL), path_);


    // Log::info() << "Save dimensions" << std::endl;
    for (const auto& j : dimensions_) {
        if ((j.second)->inUse()) {
            // Log::info() << "Define " << *(j.second) << std::endl;
            (j.second)->create(nc);
        }
    }

    // Log::info() << "Save attributes" << std::endl;
    for (const auto& j : attributes_) {
        // Log::info() << "Define " << *(j.second) << std::endl;
        (j.second)->create(nc);
    }

    // Log::info() << "Save variables" << std::endl;

    for (const auto& j : variables_) {
        // Log::info() << "Define " << *(j.second) << std::endl;
        (j.second)->create(nc);
    }

    NC_CALL(nc_enddef(nc), path_);

    for (const auto& j : variables_) {
        Log::info() << "Save " << *(j.second) << std::endl;
        (j.second)->save(nc);
    }

    NC_CALL(nc_close(nc), path_);
}


}  // namespace mir::netcdf
