//-----------------------------------------------------------
// File and Version Information:
// $Id$
//
// Description:
//      Implementation of TpcLaserGrid
//
//
// Environment:
//      Software developed for the GEM-TPC detector
//
// Author List:
//      Felix Boehmer      TUM            (original author)
//
//-----------------------------------------------------------


#include "TpcLaserGrid.h" 
#include "TpcDigiPar.h"
#include "TpcDigiMapper.h"
#include "RKTrackRep.h"

#include <iostream>
#include <string>
#include <fstream>
#include <sstream>

#include "TVector3.h"


TpcLaserGrid* TpcLaserGrid::fInstance = NULL;

std::string trim(std::string &str, const std::string &trimChars = " ")
{
  std::string result = str.erase(str.find_last_not_of(trimChars) + 1);
  return result.erase(0, result.find_first_not_of(trimChars));
}

std::string trimComment(std::string &str, const std::string &trimChars = " ")
{
  int pos = str.find_first_of(trimChars);
  if(pos>=0) 
    str.erase(pos);
  return str;
}

TpcLaserGrid::TpcLaserGrid() 
  : fInit(false)
{;}
  

TpcLaserGrid::~TpcLaserGrid() {
  std::map<unsigned int, TpcLaser*>::iterator it;
  for(it=fLaserMap.begin(); it!=fLaserMap.end(); it++) 
    delete it->second;
  
}

TpcLaserGrid*
TpcLaserGrid::Instance() {
  if(fInstance == NULL) 
    fInstance = new TpcLaserGrid();
  return fInstance;
}

void
TpcLaserGrid::Destruct() {
  if(fInstance != NULL)
    delete fInstance;
  fInstance = NULL;
  //fInit = false;
}

void 
TpcLaserGrid::ReadFile(const TString& filename) {
  if(!fInit) {
    std::cerr<<"ERROR: TpcLaserGrid::ReadFile(): Geometry was not initialized! Call Init() first!"
	     <<"\nAbrorting ..."<<std::endl;
    return;
  }
  fLaserMap.clear();
  fTrackMap.clear();

  double startX, startY, startZ, dirX, dirY, dirZ, iondens, width, time;
  char dirSwitch;
  unsigned int id;
  std::string dir("d");
  std::string end("e");
      
  std::ifstream infile(filename, std::fstream::in);
  if (!infile.good()) 
  {
    std::cerr<<"ERROR: TpcLaserGrid::ReadFile(): File "<<filename.Data()<<" not ok, aborting ..."<<std::endl;
    return;
  }   
  
  std::string iline;
  while (infile.good())
  {
    iline.clear();
    getline(infile, iline);
    std::string copy = iline;
    
    //strip left, right and comments:
    trim(iline," ");
    trimComment(iline, "#");
        
    //ignore blank lines
    if(iline.size()==0) {
      //std::cout<<"Ignoring blank line ... "<<std::endl;
      continue;
    }
     
    //the standard read-in
    std::stringstream ss;
    ss<<iline.c_str();
    ss>>dirSwitch>>id>>startX>>startY>>startZ>>dirX>>dirY>>dirZ>>iondens>>width>>time;
    
    std::string dirStr(1,dirSwitch); 
    //decide which sort of vector we have
    bool isEnd = true;
    bool ok = false;
    if(dirStr==dir) {
      ok = true;
      isEnd = false;
    }
    if(dirStr==end) 
      ok = true;
    if(!ok) {
      std::cout<<"ERROR: TpcLaserGrid::ReadFile(): Invalid direction/end specifier in line: \n"
	       <<"        "<<copy<<std::endl;
      return;
    }
    
    if(fLaserMap.count(id)) {
      std::cout<<"ERROR: TpcLaserGrid::ReadFile(): Id "<<id<<" already taken! Complete line: \n"
	       <<"        "<<copy<<std::endl;
      return;
    }
    TVector3 startV(startX,startY,startZ);
    TVector3 endV(dirX,dirY,dirZ);
    
    fLaserMap[id] = new TpcLaser(id, fRMin, fRMax, fZMin, fZMax,
    				 startV, endV,
    				 iondens,width,time,isEnd);
    
    
    TVector3 lDir = (fLaserMap[id])->getDir();
    lDir.SetMag(1.);
    TVector3 lStart = (fLaserMap[id])->getStart();
    double mom = 100.; //100 GeV/c, but should not matter for what we need here
    
    fTrackMap[id] = new RKTrackRep(lStart, mom*lDir,
				   lStart*0.001, mom*lDir*0.001,211);  //211: pi+
    //TAKE CARE THAT YOU USE ZERO MAGNETIC FIELD IN THE MC SIM!
  }
  std::cout<<"TpcLaserGrid::ReadFile(): Created "<<fLaserMap.size()<<" tracks"<<std::endl;
  infile.close();
}


void
TpcLaserGrid::Init(TpcDigiPar* par) {
  if(fInit) 
    std::cout<<"WARNING: TpcLaserGrid::Init(): ReInitialization of geometry information!"
	     <<" Be careful what you do ..."<<std::endl;
  fZMin = par->windowMin();
  fZMax = par->windowMax();
  fRMin = par->getRMin();
  fRMax = par->getRMax();
  fInit = true;
  
}


void
TpcLaserGrid::Init(double rMin, double rMax, double zMin, double zMax) {
  if(fInit) 
    std::cout<<"WARNING: TpcLaserGrid::Init(): ReInitialization of geometry information!"
	     <<" Be careful what you do ..."<<std::endl;
  fZMin = zMin;
  fZMax = zMax;
  fRMin = rMin;
  fRMax = rMax;
  fInit = true;
}

const std::map<unsigned int, TpcLaser*>*
TpcLaserGrid::GetLasers() const {
  if(!fInit) 
    std::cerr<<"ERROR: TpcLaserGrid::GetLasers(): TpcLaserGrid was not initialized yet! "
	     <<"\nReturning empty list ..."<<std::endl;
  return &fLaserMap;
}


const std::map<unsigned int, GFAbsTrackRep*>*
TpcLaserGrid::GetTracks() const {
  if(!fInit) 
    std::cerr<<"ERROR: TpcLaserGrid::GetTracks(): TpcLaserGrid was not initialized yet! "
	     <<"\nReturning empty list ..."<<std::endl;
  return &fTrackMap;
}