//////////////////////////////////////////////////////////
// Mapper for TFT
// chamberID layerID tubeIDtot
//
// author: Isabella Garzia
//////////////////////////////////////////////////////////


#include "PndFtsMapCreator.h"
#include "PndFtsTube.h"
#include "PndGeoFtsPar.h"

#include "FairGeoNode.h"
#include "FairGeoVolume.h"
#include "FairGeoTransform.h"
#include "FairGeoVector.h"
#include "FairGeoRotation.h"
#include "FairGeoTube.h"

#include "TGeoTube.h"
#include "TVector3.h"
#include "TObjArray.h"
#include "TString.h"
#include "TGeoVolume.h"
#include "TGeoTube.h"
#include "TClonesArray.h"
#include "FairRootManager.h"

#include <iostream>

using namespace std;

PndFtsMapCreator* PndFtsMapCreator::fgMapperInstance = 0;

PndFtsMapCreator* PndFtsMapCreator::Instance() {
  if (0 == fgMapperInstance)
    fgMapperInstance = new PndFtsMapCreator();
  return fgMapperInstance;
}

PndFtsMapCreator::PndFtsMapCreator() {
  // Geometry loading                                                                    
  FairRootManager* ioman = FairRootManager::Instance();
  TFile *infile = ioman->GetInFile();
  //TGeoManager *geoMan = (TGeoManager*) infile->Get("FAIRGeom");             

}



//PndFtsMapCreator::PndFtsMapCreator(){}
	// to use in PndFts
PndFtsMapCreator::PndFtsMapCreator(Int_t geoType){
  fGeoType = geoType;
  if(fGeoType != 1) Info("PndFtsMapCreator","Geometry %i not supported by map", fGeoType); // CHECK
}

// crete geometry from parameters file
PndFtsMapCreator::PndFtsMapCreator(PndGeoFtsPar *ftsPar)
{
  fFtsParameters = ftsPar;
 // set general par
  SetGeneralParameters();
  // choose geometry type
  fGeoType = ftsPar->GetGeometryType(); // classic, optimized, average, detailed, CAD

  if(fGeoType != 1) Info("PndFtsMapCreator","Geometry %i not supported by map", fGeoType); // CHECK
 }

PndFtsMapCreator::~PndFtsMapCreator(){}

void PndFtsMapCreator::SetGeneralParameters() {   //  CHECK whether it depends on geometry or not
  fTubeInRad = fFtsParameters->GetTubeInRad();   // tube inner radius
  fTubeOutRad = fFtsParameters->GetTubeOutRad(); // tube outer radius
}

// during simulation (for tubeID)
Int_t PndFtsMapCreator::GetTubeIDFromPath(TString path) {
  if(fGeoType == 1) return GetTubeIDFromPathGeoType1(path);
  return -999;
}


// ===== GEO TYPE 1 =====
	// sensitive volume is gas
	// when reading the parameters we use fts01tube to retrieve geometrical information
	// OK
	// during simulation

Int_t PndFtsMapCreator::GetTubeIDFromPathGeoType1(TString path){
  TString tmpstring = GetNameFromPathGeoType1(path);
  return GetTubeIDFromNameGeoType1(tmpstring);
}




Int_t PndFtsMapCreator::GetChamberIDFromPath(TString path){
  TString pathstring=path;
 if(pathstring.Contains("fts01assembly_1")){
   //std::cout<<"chamber 1"<<std::endl;
    return 1;
  }
  if(pathstring.Contains("fts01assembly_2")){
    //std::cout<<"chamber 2"<<std::endl;
    return 2;
  }
  if(pathstring.Contains("fts031")||pathstring.Contains("fts032")||pathstring.Contains("fts033")||pathstring.Contains("fts034")){
    //std::cout<<"chamber 3"<<std::endl;
    return 3;
  }
  if(pathstring.Contains("fts035")||pathstring.Contains("fts036")||pathstring.Contains("fts037")||pathstring.Contains("fts038")){
    //std::cout<<"chamber 4"<<std::endl;
    return 4;
  }
if(pathstring.Contains("fts05tube")){
  //std::cout<<"chamber 5"<<std::endl;
    return 5;
  }
if(pathstring.Contains("fts06tube")){
  //std::cout<<"chamber 6"<<std::endl;
    return 6;
  }

}

Int_t PndFtsMapCreator::GetTubeIDFromNameGeoType1(TString name){

  TString tmpstring = name;

  if(tmpstring.Contains("#")) {
    int start = tmpstring.Index("#") + 1;
    tmpstring = tmpstring(start, tmpstring.Sizeof());
  }
  else{
    int start = tmpstring.Index("e") + 1;
    tmpstring = tmpstring(start, tmpstring.Sizeof());
  }

  return tmpstring.Atoi();
}	

// name as in geo file: from path "_" to name "#"
TString PndFtsMapCreator::GetNameFromPathGeoType1(TString path){
  TString tmpstring = path;

  //std::cout<<"tmpstring original###### = "<<tmpstring<<std::endl;

// cut /cave_1/fts01assembly_0/
  TString cavename;
  cavename = "/cave_1/ftsXXassembly_X/"; 
  tmpstring = tmpstring(cavename.Sizeof() - 1, tmpstring.Sizeof());

 // cut /fts01gas...
  int index = tmpstring.Index("/");
  tmpstring = tmpstring(0, index);

  // replace _ with #
  //  if(tmpstring.Contains("_0")) {
  //    tmpstring.Chop(); // cut "0"
  //   tmpstring.Chop(); // cut "_"
  // }
  if(tmpstring.Contains("down") && !tmpstring.Contains("fts03")){
    tmpstring.Replace(13, 1, "#");
    //std::cout<<"tmpstring = "<<tmpstring<<std::endl;
  }
  if(tmpstring.Contains("fts03") && tmpstring.Contains("down")){
    tmpstring.Replace(14, 1, "#");
    //std::cout<<"tmpstring = "<<tmpstring<<std::endl;
  }
 if(tmpstring.Contains("up") && !tmpstring.Contains("fts03")){
    tmpstring.Replace(11, 1, "#");
    //std::cout<<"tmpstring = "<<tmpstring<<std::endl;
  }
 if(tmpstring.Contains("up") && tmpstring.Contains("fts03")){
    tmpstring.Replace(12, 1, "#");
    //std::cout<<"tmpstring = "<<tmpstring<<std::endl;
  }
 if(!tmpstring.Contains("up") && !tmpstring.Contains("down") && tmpstring.Contains("fts03")){
    tmpstring.Replace(10, 1, "#");
    //std::cout<<"tmpstring = "<<tmpstring<<std::endl;
  }
 if(!tmpstring.Contains("up") && !tmpstring.Contains("down") && !tmpstring.Contains("fts03")){
    tmpstring.Replace(9, 1, "#");
    //std::cout<<"tmpstring = "<<tmpstring<<std::endl;
 }

  return tmpstring;
}

Int_t PndFtsMapCreator::GetLayerID(Int_t chamberid, Int_t tudeid,TString path ){

  Int_t tube=tudeid;
  Int_t chamber=chamberid;
  TString tmpstring = path;

  Int_t i=1;
  if(chamberid==1 || chamberid==2){
    if(chamberid==2){i=i+8;}
    while(tube>128){
      tube=tube-128;
      i++;
    }
  }
 if(chamberid==3 || chamberid==4){
   if(chamberid==3){i=i+16;}
   else{i=i+24;}
   if(tmpstring.Contains("fts032")){tube=tube+384;}
   if(tmpstring.Contains("fts033")){tube=tube+768;}
   if(tmpstring.Contains("fts034")){tube=tube+1152;}
   //if(tmpstring.Contains("fts035")){tube=tube+}
   if(tmpstring.Contains("fts036")){tube=tube+384;}
   if(tmpstring.Contains("fts037")){tube=tube+768;}
   if(tmpstring.Contains("fts038")){tube=tube+1152;}
    while(tube>192){
      tube=tube-192;
      i++;
    }
  }
 if(chamberid==5){
   i=i+32;
    while(tube>400){
      tube=tube-400;
      i++;
    }
  }
 if(chamberid==6){
   i=i+40;
    while(tube>592){
      tube=tube-592;
      i++;
    }
  }
 return i;

}

Int_t PndFtsMapCreator::GetTubeIDTot(Int_t chamberid, Int_t layerid, Int_t tubeid,TString path ){

  Int_t chamber=chamberid;
  Int_t layer=layerid;
  Int_t tubeID=tubeid;
  TString tmpstring=path;
  Int_t tube=0;
  Int_t totTubeID=0;

 if(chamberid==1){tube=tubeID;}
 if(chamberid==2){tube=tubeID+1024;}
 //if(chamberid==3){tube=tubeID+2240;} //2048
 if(chamberid==3){
   if(layer==17||layer==18){tube=tubeID+2240;}
   if(layer==19||layer==20){tube=tubeID+2648;}
   if(layer==21||layer==22){tube=tubeID+3056;}
   if(layer==23||layer==24){tube=tubeID+3464;}
 }
 if(chamberid==4){
   if(layer==25||layer==26){tube=tubeID+3872;}
   if(layer==27||layer==28){tube=tubeID+4280;}
   if(layer==29||layer==30){tube=tubeID+4688;}
   if(layer==31||layer==32){tube=tubeID+5096;}
 }
 //if(chamberid==4){tube=tubeID+3584;}
 if(chamberid==5){tube=tubeID+5120;}
 if(chamberid==6){tube=tubeID+8320;}

  //up e down hanno lo stesso id-> devo slittare il tutto per dare a ciascun tubo
  //un suo numero identificativo
  if(chamber==1 || chamberid==2){
     if(tmpstring.Contains("down") || tube>70){
       if(tmpstring.Contains("down_1/fts")){tube=tube+58;}
       totTubeID=tube+12*layer;
     }
     if(tmpstring.Contains("up")){
       if(tmpstring.Contains("up_1/fts")){tube=tube+58;}
       totTubeID=tube+12*(layer-1);
     }
     if(tube<=58){totTubeID=tube;}
  }
  if(chamberid==3 || chamberid==4){
    if(tmpstring.Contains("down_1/fts")){totTubeID=tube+90+12;}
    if(tmpstring.Contains("up_1/fts")){totTubeID=tube+90;}
    if(!tmpstring.Contains("down") && !tmpstring.Contains("up")){
      if(layer%2==1 && tubeid<=90){totTubeID=tube;} //odd layer
      if(layer%2==1 && tubeid>90){totTubeID=tube+12;}
      if(layer%2!=1 && tubeid<=90){totTubeID=tube+12;}
      if(layer%2!=1 && tubeid>90){totTubeID=tube+24;}
    }
    if(tmpstring.Contains("down")&& !tmpstring.Contains("down_1/fts")){
      if(layer%2==1){totTubeID=tube+12;} //odd layer
      else{totTubeID=tube+24;}
    }
    if(tmpstring.Contains("up")&& !tmpstring.Contains("up_1/fts")){
      if(layer%2==1){totTubeID=tube;} //odd layer
      else{totTubeID=tube+12;}
    }
  }

  if(chamber==5){
     if(tmpstring.Contains("down") || tube>5329){
       if(tmpstring.Contains("down_1/fts")){tube=tube+191;}
       totTubeID=tube+(12*32)+(18*(layer-32));
     }
     if(tmpstring.Contains("up")){
       if(tmpstring.Contains("up_1/fts")){tube=tube+191;}
       totTubeID=tube+(12*32)+(18*(layer-33));
     }
     if(tube<=5311){totTubeID=tube+(12*32);}
  }

 if(chamber==6){
     if(tmpstring.Contains("down") || tube>8625){
       if(tmpstring.Contains("down_1/fts")){tube=tube+287;}
       totTubeID=tube+(12*32)+(18*(layer-32));
     }
     if(tmpstring.Contains("up")){
       if(tmpstring.Contains("up_1/fts")){tube=tube+287;}
       totTubeID=tube+(12*32)+(18*(layer-33));
     }
     if(tube<=8607){totTubeID=tube+(12*32)+(18*8);}
  }
   
 return totTubeID;  

}

ClassImp(PndFtsMapCreator)