//
// Program to create a file containing an set of beam position numbers
// after reading them from a database.
// The access is by specific run and version (simple keyed access).
//
// To run the program:
// checkout the Alignment package
// gmake Alignment.tbin
//
// bin/$BFARCH/SvxBeamReadDB DATABASE_IDENTIFIER RUN VERSION beam_filename
//
// The file is in the format define by the Db container operator <<
//
// 11/00 Ray Culbertson, from an example by Mark Lancaster
//
// -----------------------------------------------------------------------
//

#include <cstdlib>
#include <fstream>
#include <string>
#include <vector>

#include "DBObjects/TimeStampDefs.hh"
#include "CalibDB/RunListDefs.hh"
#include "Alignment/SvxBeam.hh"
#include "Alignment/CotBeam.hh"
#include "AlignmentDB/SvxBeamPosition.Defs.hh"
#include "CalibDB/RunListDefs.hh"


using namespace std;

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

  if (argc < 4)
    {
      std::cerr << " Wrong # of arguments : format =   " << argv[0] ;
      std::cerr << " mode first_run last_run" << std::endl;
      return -1;
    }

  int    mode    = atoi(argv[1]);
  int    firstRun = atoi(argv[2]);
  int    lastRun  = atoi(argv[3]);

  std::string dbid("onotl_prd");


  // find the valid entries
  int* goodrunc;
  int ngoodc=0;
  int* goodruns;
  int ngoods=0;
  
  RunList_mgr runIom(dbid,"RunList");
  if( !runIom.isValid() ) {std::cerr<< "Error opening RunList database" << std::endl;}
  RunListKey rlk("CotBeamPosition","UNDEFINED",
		 Run::undefined,Version::undefined,
		 PrimType::undefined, "UNDEFINED");
  RunList_var row;
  if(runIom.get(rlk,row)!=Result::success) {
    std::cerr << "Failure of runIom.get() to retrieve db contents" << std::endl;  }
  
  goodrunc = new int[row->size()];

  for(RunList::iterator iter=row->begin(); iter!=row->end(); iter++) {
    if(iter->run()>=firstRun && iter->run()<=lastRun) {
      goodrunc[ngoodc++] = iter->run();
    }
  }

  rlk.setTable("SvxBeamPosition");
  if(runIom.get(rlk,row)!=Result::success) {
    std::cerr << "Failure of runIom.get() to retrieve db contents" << std::endl;  }
  
  goodruns = new int[row->size()];
  for(RunList::iterator iter=row->begin(); iter!=row->end(); iter++) {
    if(iter->run()>=firstRun && iter->run()<=lastRun) {
      bool matchc=false;
      for(int i=0; i<ngoodc; i++) {
	if(goodrunc[i] == iter->run()) matchc=true;
      }
      if(matchc) {
	goodruns[ngoods++] = iter->run();
      } else {
	std::cout << " found a run with svx but not cot: "<< iter->run() << std::endl;
      }
    }
  }

  std::cout << "Found "<< ngoodc << " cot entries" << std::endl;
  std::cout << "Found "<< ngoods << " svx entries" << std::endl;


  SvxBeam svxBeam("default");
  CotBeam cotBeam("default");

  SvxBeamPositionContainer scont;
  SvxBeamPosition_mgr siom(dbid,"SvxBeamPosition");
  CotBeamPositionContainer ccont;
  CotBeamPosition_mgr ciom(dbid,"CotBeamPosition");

  if(mode==0) {


    // 100030
    scont.addParent(RunListKey(CID(99668)));
    scont.addParent(RunListKey(CID(99670)));
    scont.addParent(RunListKey(CID(99672)));
    scont.addParent(RunListKey(CID(99674)));
    scont.addParent(RunListKey(CID(99676)));
    
    for(int run=firstRun; run<=lastRun; run++) {
      bool ok = false;
      for(int i=0; i<ngoods; i++) if(goodruns[i]==run) ok = true;
      if(ok) {
	
	cotBeam.setHistoryCode(17);
	svxBeam.setHistoryCode(17);
	cotBeam.loadRun(run);
	svxBeam.loadRun(run);
	if(svxBeam.isValid() && cotBeam.isValid()) {

	  std::cout << "Run " << run << std::endl;
	  
	  SvxBeamPosition svx = svxBeam.getRecord();
	  CotBeamPosition cot = cotBeam.getRecord();
	  // set new history code
	  svx._Flag0 = (18 | (svx._Flag0 & 0xFFFF0000));
	  cot._Flag0 = (18 | (cot._Flag0 & 0xFFFF0000));
	  if(run>=152595 && run<=154012) {
	    // fix wrong alignment
	    svx._BeamX += -0.00519;
	    svx._BeamY +=  0.01261;
	    svx._SlopeX += -0.0000041;
	    svx._SlopeY += 0.0000398;
	  } else {
	    // fix bias
	    svx._BeamX += -0.0004;
	    svx._BeamY +=  0.0025;
	    svx._SlopeX += -0.000015;
	    svx._SlopeY +=  0.000030;
	  }
	  
	  // put it back in db
	  RunListKey qk(run,-1) ;
	  qk.setDataStatus("GOOD");
	  qk.newVersion();
	  qk.setAlgorithm("4.8.4a");

	  scont.clear();
	  ccont.clear();
	  
	  scont.push_back(svx);

	  ccont.push_back(cot);
	  
	  if(siom.put(qk,scont)!=Result::success) {
	    std::cerr << " ERROR: Put failed for SvxBeamPosition Run = " << 
	      run << std::endl;
	  }
	  
	  if(ciom.put(qk,ccont)!=Result::success) {
	    std::cerr << " ERROR: Put failed for CotBeamPosition Run = " << 
	      run << std::endl;
	  }
	  
	  
	} else {
	  std::cout << "Could not find history 17 " <<  run
		    << "   cot: "<< cotBeam.isValid() 
		    << " svx: "<< svxBeam.isValid() 
		    << std::endl;
	} // end both beamlines retrieved
      } // end ok
    }  // end run loop
  } // endif mode=0

  if(mode==1) {

    for(int run=firstRun; run<=lastRun; run++) {
      bool ok = false;
      for(int i=0; i<ngoodc; i++) if(goodrunc[i]==run) ok = true;
      if(ok) {
	
	cotBeam.setHistoryCode(17);
	cotBeam.loadRun(run);

	if(cotBeam.isValid()) {

	  
	  CotBeamPosition cot = cotBeam.getRecord();

	  std::cout << "Run " << run << "    " 
		    << cot._BeamX << " " << cot._BeamY << std::endl;

	  // set new history code
	  cot._Flag0 = (19 | (cot._Flag0 & 0xFFFF0000));

	  // put it back in db
	  RunListKey qk(run,-1) ;
	  qk.setDataStatus("GOOD");
	  qk.newVersion();
	  qk.setAlgorithm("4.8.4a_08_copy");

	  ccont.clear();
	  
	  ccont.push_back(cot);
	  
	  if(ciom.put(qk,ccont)!=Result::success) {
	    std::cerr << " ERROR: Put failed for CotBeamPosition Run = " << 
	      run << std::endl;
	  }
	  
	  
	} else {
	  std::cout << "Could not find history 17 " <<  run
		    << "   cot: "<< cotBeam.isValid() 
		    << std::endl;
	} // beamlines retrieved
      } // end ok
    }  // end run loop
  } // endif mode=1


  if(mode==2) {


    // 100034 21 TEST (after COT alignment)
    scont.addParent(RunListKey(CID(215832)));
    scont.addParent(RunListKey(CID(215834)));
    scont.addParent(RunListKey(CID(215836)));
    scont.addParent(RunListKey(CID(215838)));
    scont.addParent(RunListKey(CID(215840)));


    for(int run=firstRun; run<=lastRun; run++) {
      bool ok = false;
      for(int i=0; i<ngoodc; i++) if(goodrunc[i]==run) ok = true;
      if(ok) {
	
	//typedef DataContainer<SvxBeamPosition> SvxBeamPositionContainer;

	//  **********  cot beam
	ccont.clear();
	  
	cotBeam.setHistoryCode(19);
	cotBeam.loadRun(run);

	if(cotBeam.isValid()) {
	  
	  CotBeamPositionContainer oldCont = cotBeam.getContainer();
	  for(int ic=0; ic<oldCont.size(); ic++) {
	    CotBeamPosition cot = oldCont[ic];

	    std::cout << "Cot Run " << run << "    " 
		      << cot._BeamX << " " << cot._BeamY << std::endl;

	    // set new history code
	    cot._Flag0 = (20 | (cot._Flag0 & 0xFFFF0000));

	    // adjust values
	    
	    cot._BeamX += -0.0019;
	    cot._BeamY +=  0.0002;
	    cot._SlopeX +=  0.000026;
	    cot._SlopeY += -0.000002;

	    //also adjust for VxPrim fixes
	    cot._SlopeX += -0.000100;
	    cot._SlopeY += -0.000050;

	    ccont.push_back(cot);
	  
	  } // loop over container entries

	  // put it back in db
	  RunListKey qk(run,-1) ;
	  qk.setDataStatus("GOOD");
	  qk.newVersion();
	  qk.setAlgorithm("19_patch_to_20");
	  
	  if(ciom.put(qk,ccont)!=Result::success) {
	    std::cerr << " ERROR: Put failed for CotBeamPosition Run = " << 
	      run << std::endl;
	  }
	  
	} else {
	  std::cout << "Could not find history 19 " <<  run
		    << "   cot: "<< cotBeam.isValid() 
		    << std::endl;
	} // beamlines retrieved


	//  **********  svx beam


	scont.clear();
	  
	svxBeam.setHistoryCode(19);
	svxBeam.loadRun(run);

	if(svxBeam.isValid()) {
	  
	  SvxBeamPositionContainer SoldCont = svxBeam.getContainer();
	  for(int ic=0; ic<SoldCont.size(); ic++) {
	    SvxBeamPosition svx = SoldCont[ic];

	    std::cout << "Svx Run " << run << "    " 
		      << svx._BeamX << " " << svx._BeamY << std::endl;

	    // set new history code
	    svx._Flag0 = (20 | (svx._Flag0 & 0xFFFF0000));

	    // adjust values
	    
	    if(run<149717) {
	      // period 1 
	      svx._BeamX  += -0.0023;
	      svx._BeamY  += -0.0016;
	      svx._SlopeX += 1.4e-05;
	      svx._SlopeY += 0;
	    } else if(run<158636) {
	      // period 2 
	      svx._BeamX  += -0.0019;
	      svx._BeamY  += -0.0048;
	      svx._SlopeX += 4.4e-05;
	      svx._SlopeY += 1.7e-05;
	    } else {
	      // period 3 
	      svx._BeamX  += -0.0029;
	      svx._BeamY  += -0.0055;
	      svx._SlopeX += 2.5e-05;
	      svx._SlopeY += -1.7e-05;
	    }


	    scont.push_back(svx);
	  
	  } // loop over container entries

	  // put it back in db
	  RunListKey qk(run,-1) ;
	  qk.setDataStatus("GOOD");
	  qk.newVersion();
	  qk.setAlgorithm("19_patch_to_20");

	  if(siom.put(qk,scont)!=Result::success) {
	    std::cerr << " ERROR: Put failed for SvxBeamPosition Run = " << 
	      run << std::endl;
	  }
	  
	} else {
	  std::cout << "Could not find history 19 " <<  run
		    << "   svx: "<< svxBeam.isValid() 
		    << std::endl;
	} // beamlines retrieved



      } // end ok
    }  // end run loop
  } // endif mode=2


   return 0;
}