//-----------------------------------------------------------
// File and Version Information:
// $Id$
//
// Description:
//      Conversion of FOPI RPC hits ("RpcTrack") to
//      planar Genfit hits
//
// Environment:
//      Software developed for the GEM-TPC in FOPI
//
// Author List:
//      Felix Boehmer, E18   TUM          (original author)
//
//-----------------------------------------------------------


// This Class' Header ------------------
#include "FopiRpcInitTask.h"

#include "TClonesArray.h"
#include "TMath.h"
#include "RpcPixHit.h"
#include "RpcTrack.h"

#include <iostream>


// Class Member definitions -----------

ClassImp(FopiRpcInitTask)

FopiRpcInitTask::FopiRpcInitTask()
  : fHitBranchName("RpcTrack"), 
    fOutBranchName("RpcPixHit"),
    fVerbose(false),
    fHitArray(NULL),
    fOutArray(NULL), 
    fPersistence(kFALSE),
    fConstantOffset(kFALSE),
    fUseConstSigZ(kFALSE),
    fHitArrID(-1),
    fConstSigPhi(4.6/TMath::Sqrt(12.))
//_weightedPlaneConstruction(kFALSE),
//    _cutCov(kFALSE)
{
  initModuleLookup();
}   



FopiRpcInitTask::~FopiRpcInitTask()
{;}



InitStatus
FopiRpcInitTask::Init()
{
  //Get ROOT Manager
  FairRootManager* ioman= FairRootManager::Instance();
  if(ioman==NULL){
    Error("FopiRpcInitTask::Init","RootManager not instantiated!");
    return kERROR;
  }

  // get input arrays    
  fHitArray=(TClonesArray*) ioman->GetObject(fHitBranchName);
  if(fHitArray==NULL){
    Error("FopiRpcInitTask::Init","RPC Hit-Array not found!");
    return kERROR;
  }

  // create and register output array
  fOutArray = new TClonesArray("RpcPixHit");
  ioman->Register(fOutBranchName,"GenFit",fOutArray,fPersistence);
  fHitArrID = FairRootManager::Instance()->GetBranchId(fOutBranchName);
  
  
  return kSUCCESS;
}


void
FopiRpcInitTask::Exec(Option_t* opt)
{
  
  // Reset output Arrays
  if(fOutArray==NULL) Fatal("FopiRpcInitTask::Exec()","No OutTrackArray");
  fOutArray->Delete();
  
  if(fVerbose) std::cout<<"FopiRpcInitTask::Exec() ... \n";

  //create recohits from CdcHits:
  int nRpcHits = fHitArray->GetEntriesFast();
  for(int irpc=0; irpc<nRpcHits; irpc++) {
    RpcTrack* ir = (RpcTrack*) (*fHitArray)[irpc];
    
    int module = ir->GetSuperMod();
    double zPos = ir->GetDetX();   //this is the detector Z measurement
    double zPosSig = ir->GetDetXSig();
    double x = ir->GetGlobX();
    double y = ir->GetGlobY();
    
    double z = 0.;
    
    if(fConstantOffset) {
      z = zPos + _zOffset;
    }
    
    else {
      double rad = TMath::Sqrt(x*x + y*y);
      double theta = ir->GetTheta()*TMath::Pi()/180.;
      z = rad / TMath::Tan(theta);
    }
        
    
    TVector3 pos; pos.SetXYZ(x,y,z);
    TVector3 dir; dir.SetXYZ(0.,0.,1.);
    
    int nOut = fOutArray->GetEntriesFast();
    double phiMod;

    bool valid = true;
    
    if(fModLookup.count(module)) 
      phiMod = fModLookup[module]; 
    else {
      std::cout<<"FopiRpcInitTask::Exec() Warning: Invalid module number! "
	       <<"Skipping this point..."<<std::endl;
      phiMod = 1.;
      valid = false;
    }
      
    
    TVector3 norm; norm.SetXYZ(1.,0.,0.);
    norm.SetPhi(phiMod*TMath::Pi()/180.);

    if(fUseConstSigZ)
      zPosSig = fConstSigZ;
    
    RpcPixHit* out = new((*fOutArray)[nOut]) RpcPixHit(pos,
						       dir,
						       norm,
						       zPosSig,
						       fConstSigPhi);
    out->setValid(valid);  
    out->setHitBranchID(fHitArrID);
    out->setHitID(nOut);
	
  }
  
  if(fVerbose) std::cout<<"FopiRpcInitTask::Exec() Set up "<<nRpcHits<<" RPC hits."<<std::endl;

}

void
FopiRpcInitTask::initModuleLookup() {
  //obtained from geometry
  fModLookup[0] = -5.625000;
  fModLookup[1] = 5.625000;
  fModLookup[2] = 16.875000;
  fModLookup[3] = 28.125000;
  fModLookup[4] = 39.375000;
  fModLookup[5] = 50.625000;
  fModLookup[6] = 61.875000;
  fModLookup[7] = 73.125000;
  fModLookup[8] = 84.375000;
  fModLookup[9] = 95.625000;
  fModLookup[10] = 106.875000;
  fModLookup[11] = 118.125000;
  fModLookup[12] = 129.375000;
  fModLookup[13] = 140.625000;
  fModLookup[14] = 151.875000;
  fModLookup[15] = 163.125000;
  fModLookup[16] = 174.375000;
  fModLookup[17] = 185.625000;
  fModLookup[18] = 196.875000;
  fModLookup[19] = 208.125000;
  fModLookup[20] = 219.375000;
  fModLookup[21] = 230.625000;
  fModLookup[22] = 241.875000;
  fModLookup[23] = 253.125000;
  fModLookup[24] = 264.375000;
  fModLookup[25] = 275.625000;
  fModLookup[26] = 286.875000;
  fModLookup[27] = 298.125000;
  fModLookup[28] = 309.375000;
  fModLookup[29] = 320.625000;
  fModLookup[30] = 331.875000;
  fModLookup[31] = 343.125000;
}