#ifndef __READLATTPAR__
#define __READLATTPAR__

// a slight modification on ReadLattice class (if not inheritance)
// to enable parallel reading/writing of "Gauge Connection Format" lattice data

#include <cstdlib>	// exit()
#include <cstdio>
#include <fstream>
#include <iostream>
#include <map>
#include <string>

#include <util/gjp.h>
#include <util/vector.h>
#include <util/lattice.h>
#include <util/data_types.h>
#include <util/verbose.h>
#include <util/error.h>
#include <alg/do_arg.h>
#include <util/qioarg.h>
#include <util/fpconv.h>
#include <util/iostyle.h>
#include <util/latheader.h>

CPS_START_NAMESPACE

class  ReadLatticeParallel : public QioControl
{
  // which determines parallel reading or serial reading

 private:
  const char *cname;
  LatticeHeader hd;
  static bool UseParIO;
  const int default_concur=16;

 public:
  // ctor for 2-step loading
 ReadLatticeParallel()
   : QioControl(), cname("ReadLatticeParallel")
    { 
      //CK set architecture-dependent default IO style rather than hardcoding it.
      setDefault();
    }

  // ctor invoking loading behavior
 ReadLatticeParallel(Lattice & lat, const char * filename, const Float chkprec = 0.01): QioControl(), cname("ReadLatticeParallel")
    {
    //CK set architecture-dependent default IO style rather than hardcoding it.
    QioArg rd_arg(filename,chkprec);
	rd_arg.ConcurIONumber=default_concur;
    read(lat,rd_arg);
  }

  // ctor invoking loading behavior
 ReadLatticeParallel(Lattice & lat, const QioArg & rd_arg) 
   : QioControl(), cname("ReadLatticeParallel")
    {
      //CK set architecture-dependent default IO style rather than hardcoding it.
      setDefault();

      read(lat,rd_arg);
    }
  
  virtual ~ReadLatticeParallel() {}

  void read(Lattice &lat, const char *filename, const Float chkprec = 0.01){
    QioArg rd_arg(filename,chkprec);
	rd_arg.ConcurIONumber=default_concur;
    read(lat,rd_arg);
  }

  void read(Lattice & lat, const QioArg & rd_arg);

  std::string getEnsembleId(){ return hd.ensemble_id; } // CAREFULL: only on MasterNode!!
  std::string getEnsembleLabel(){ return hd.ensemble_label; } // CAREFULL: only on MasterNode!!
  int getSequenceNumber();// same on all nodes!

 private:
  bool CheckPlaqLinktrace(Lattice & lat, const QioArg & rd_arg,
			  const Float plaq_inheader, const Float linktrace_inheader);

 public:
  inline static void setParallel() { UseParIO = 1; }
  inline static void setSerial() { UseParIO = 0; }
  inline static void setDefault() { UseParIO = 1; }
  inline int parIO() const { return UseParIO; }
};


class ReadLatticeSerial : public ReadLatticeParallel {
 private:
  const char * cname;

 public:
  // ctor for 2-step loading
  ReadLatticeSerial() : ReadLatticeParallel(), cname("ReadLatticeSerial") {
    setSerial();
  }

  // ctor invoking loading behavior
  ReadLatticeSerial(Lattice & lat, const char * filename, const Float chkprec = 0.01) 
    : ReadLatticeParallel(), cname("ReadLatticeSerial") {
    setSerial();
    read(lat, filename, chkprec);
  }

  // ctor invoking loading behavior
  ReadLatticeSerial(Lattice & lat, const QioArg & rd_arg) 
    : ReadLatticeParallel(), cname("ReadLatticeSerial") {
    setSerial();
    read(lat, rd_arg);
  }
  
  virtual ~ReadLatticeSerial() {}
};


CPS_END_NAMESPACE
#endif