/*
 * Test_diskvector.cpp, part of Hadrons (https://github.com/aportelli/Hadrons)
 *
 * Copyright (C) 2015 - 2023
 *
 * Author: Antonin Portelli <antonin.portelli@me.com>
 * Author: Ryan Abbott <rabbott@mit.edu>
 *
 * Hadrons is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 2 of the License, or
 * (at your option) any later version.
 *
 * Hadrons is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with Hadrons.  If not, see <http://www.gnu.org/licenses/>.
 *
 * See the full license in the file "LICENSE" in the top level distribution 
 * directory.
 */

/*  END LEGAL */
#define DV_DEBUG
#include <Hadrons/DiskVector.hpp>

using namespace Grid;
using namespace HADRONS_NAMESPACE;

GRID_SERIALIZABLE_ENUM(Enum, undef, red, 1, blue, 2, green, 3);

class Object: Serializable {
public:
  GRID_SERIALIZABLE_CLASS_MEMBERS(Object,
                                  Enum, e,
                                  SpinColourMatrix, scm);
};

#ifdef HAVE_HDF5
typedef Hdf5Reader TestReader;
typedef Hdf5Writer TestWriter;
#else
typedef BinaryReader TestReader;
typedef BinaryWriter TestWriter;
#endif

int main(int argc, char *argv[])
{
    Grid_init(&argc, &argv);

    GridSerialRNG rng;
    Object        obj, v2w, v2r, v13w, v13r;

    SerializableDiskVector<Object, TestReader, TestWriter> v("diskvector_test", 1000, 4);

    obj.e = Enum::red;
    random(rng, obj.scm);
    v[32] = obj;
    random(rng, obj.scm);
    v[2] = obj;
    v2w  = obj;
    random(rng, obj.scm);
    v[6] = obj;
    random(rng, obj.scm);
    v[7] = obj;
    random(rng, obj.scm);
    v[8] = obj;
    random(rng, obj.scm);
    v[9] = obj;
    random(rng, obj.scm);
    v[10] = obj;
    random(rng, obj.scm);
    v[11] = obj;
    random(rng, obj.scm);
    v[12] = obj;
    random(rng, obj.scm);
    v[13] = obj;
    v13w  = obj;
    random(rng, obj.scm);
    v[14] = obj;
    random(rng, obj.scm);
    v[15] = obj;

    v2r = v[2];
    LOG(Message) << "v[2] correct? " 
                 << ((v2r == v2w) ? "yes" : "no" ) << std::endl;
    v13r = v[13];
    LOG(Message) << "v[13] correct? " 
                 << ((v13r == v13w) ? "yes" : "no" ) << std::endl;
    LOG(Message) << "hit ratio " << v.hitRatio() << std::endl;

    EigenDiskVector<ComplexD>         w("eigendiskvector_test", 1000, 4);
    EigenDiskVector<ComplexD>::Matrix m,n;

    w[2] = EigenDiskVectorMat<ComplexD>::Random(2000, 2000);
    m    = w[2];
    w[2] = m;
    w[3] = EigenDiskVectorMat<ComplexD>::Random(2000, 2000);
    w[4] = EigenDiskVectorMat<ComplexD>::Random(2000, 2000);
    w[5] = EigenDiskVectorMat<ComplexD>::Random(2000, 2000);
    w[6] = EigenDiskVectorMat<ComplexD>::Random(2000, 2000);
    w[7] = EigenDiskVectorMat<ComplexD>::Random(2000, 2000);
    n    = w[2];
    LOG(Message) << "w[2] correct? " 
                 << ((m == n) ? "yes" : "no" ) << std::endl;
    LOG(Message) << "hit ratio " << w.hitRatio() << std::endl;

    Grid_finalize();
    
    return EXIT_SUCCESS;
}