#include <complex>
#include <cstring>

#include "latlib/o815/o815.h"

#include "sim.hpp"

o815 *O815;
sim *Sim;

//const complex<double> _i_ = complex<double>(0.0,1.0);

#include "obs_phi2.hpp"
#include "obs_plaq.hpp"
#include "obs_corrphiphi.hpp"
#include "obs_corrchichi.hpp"
#include "obs_corrphichi.hpp"
#include "obs_corrphi.hpp"
#include "obs_corrchi.hpp"
#include "obs_corriso10.hpp"
#include "obs_corriso00.hpp"
#include "obs_corrtetra.hpp"
#include "obs_diagcorr.hpp"

o815::comoption specOps[] = {
  { "kappaone", required_argument, NULL, 'r', "set inverse mass kappa_1", "min:max:inc" },
  { "kappatwo", required_argument, NULL, 's', "set inverse mass kappa_2", "min:max:inc" },
  { "beta", required_argument, NULL, 'b', "set inverse gauge coupling beta", "min:max:inc" },
  { "lambdaone", required_argument, NULL, 'p', "set quartic coupling lambda_1", "min:max:inc" },
  { "lambdatwo", required_argument, NULL, 'q', "set quartic coupling lambda_2", "min:max:inc" },
  { "kappa", required_argument, NULL, 'k', "set kappa_1 = kappa_2 = kappa", "min:max:inc" },
  { "lambda", required_argument, NULL, 'l', "set lambda_1 = lambda_2 = lambda", "min:max:inc" },
  { "", 0, NULL, 0, "", "" }
};

void parseSpecOps() 
{
  for (int isopt = 0; isopt < O815->parsedSpecOps.size(); isopt++)
    switch(O815->parsedSpecOps[isopt].first) {
    case 'r':
      O815->paraQ->addRange("kappaone", O815->parsedSpecOps[isopt].second);
      break;
    case 's':
      O815->paraQ->addRange("kappatwo", O815->parsedSpecOps[isopt].second);
      break;
    case 'b':
      O815->paraQ->addRange("beta", O815->parsedSpecOps[isopt].second);
      break;
    case 'p':
      O815->paraQ->addRange("lambdaone", O815->parsedSpecOps[isopt].second);
      break;
    case 'q':
      O815->paraQ->addRange("lambdatwo", O815->parsedSpecOps[isopt].second);
      break;
    case 'l':
      O815->paraQ->addRange("lambdaone", O815->parsedSpecOps[isopt].second);
      O815->paraQ->linkParas("lambdatwo", "lambdaone");
      break;
    case 'k':
      O815->paraQ->addRange("kappaone", O815->parsedSpecOps[isopt].second);
      O815->paraQ->linkParas("kappatwo", "kappaone");
      break;
    }
}

void helpHeader() 
{ 
  cout << "Usage: ./u1casc-ordinary [OPTIONS] [obs1] ... [obsN]" << endl << endl;
}

void parseLonelyArgs()
{
  for (vector<char*>::iterator lonit = O815->lonelyArgs.begin(); lonit != O815->lonelyArgs.end(); ++lonit) {
    if ( strcmp(*lonit, "phi2") == 0 ) {
      *O815->out->log << "MASTER: registered observable: phi2" << endl << flush;
      O815->observables.push_back(new obs_phi2(O815));
    }
    else if ( strcmp(*lonit, "plaq") == 0 ) {
      *O815->out->log << "MASTER: registered observable: plaq" << endl << flush;
      O815->observables.push_back(new obs_plaq(O815));
    }
    else if ( strcmp(*lonit, "corrphiphi") == 0 ) {
      *O815->out->log << "MASTER: registered observable: corrphiphi" << endl << flush;
      O815->observables.push_back(new obs_corrphiphi(O815));
    }
    else if ( strcmp(*lonit, "corrchichi") == 0 ) {
      *O815->out->log << "MASTER: registered observable: corrchichi" << endl << flush;
      O815->observables.push_back(new obs_corrchichi(O815));
    }
    else if ( strcmp(*lonit, "corrphichi") == 0 ) {
      *O815->out->log << "MASTER: registered observable: corrphichi" << endl << flush;
      O815->observables.push_back(new obs_corrphichi(O815));
    }
    else if ( strcmp(*lonit, "corrphi") == 0 ) {
      *O815->out->log << "MASTER: registered observable: corrphi" << endl << flush;
      O815->observables.push_back(new obs_corrphi(O815));
    }
    else if ( strcmp(*lonit, "corrchi") == 0 ) {
      *O815->out->log << "MASTER: registered observable: corrchi" << endl << flush;
      O815->observables.push_back(new obs_corrchi(O815));
    }
    else if ( strcmp(*lonit, "corriso10") == 0 ) {
      *O815->out->log << "MASTER: registered observable: corriso10" << endl << flush;
      O815->observables.push_back(new obs_corriso10(O815));
    }
    else if ( strcmp(*lonit, "corriso00") == 0 ) {
      *O815->out->log << "MASTER: registered observable: corriso00" << endl << flush;
      O815->observables.push_back(new obs_corriso00(O815));
    }
    else if ( strcmp(*lonit, "corrtetra") == 0 ) {
      *O815->out->log << "MASTER: registered observable: corrtetra" << endl << flush;
      O815->observables.push_back(new obs_corrtetra(O815));
    }
    else if ( strcmp(*lonit, "diagcorr") == 0 ) {
      *O815->out->log << "MASTER: registered observable: diagcorr" << endl << flush;
      O815->observables.push_back(new obs_diagcorr(O815));
    }
  }
}

int main (int argc, char *argv[])
{
  O815 = new o815(argc, argv, "u1casc-ordinary", specOps, &helpHeader);

  O815->addPara("beta", 1);
  O815->addPara("lambdaone", 1);
  O815->addPara("lambdatwo", 1);
  O815->addPara("kappaone", 1);  
  O815->addPara("kappatwo", 1);

  parseSpecOps();
  O815->postParaInit();
  O815->Sim = new sim(O815);
  parseLonelyArgs();
  O815->mainLoop();

  delete O815;
  return 0;
}