#define rootTree_cxx
#include "rootTree.h"
#include "string.h"
#include "iostream"


rootTree::rootTree(){

   debug = true;//you will get couts for each vble as filled

   tree = new TTree("data","SPData");
   file = new TFile("temp/SPDAQTree.root","recreate");
 
   tree->Branch("L1ANumber",&L1ANumber,"L1ANumber/I");
   tree->Branch("L1A_BXN",&L1A_BXN,"L1A_BXN/I");//L1A bunch crossing number
   tree->Branch("quality",&quality,"quality[8][16]/I");
   tree->Branch("cscID",&cscID,"cscID[8][16]/I");   
   tree->Branch("wireGP",&wireGP,"wireGP[8][16]/I");

   tree->Branch("SynchErrors",&SynchErrors,"SynchErrors[8]/I");
   tree->Branch("validPatternLinks",&validPatternLinks,"validPatternLinks[8]/I");
  tree->Branch("VQa",&VQa,"VQa[8]/I");
  tree->Branch("VQd",&VQd,"VQd[8]/I");
  
   tree->Branch("OC",&OC,"OC/I"); 
   tree->Branch("TC",&TC,"TC/I");   

   tree->Branch("CLCT_PAT_ID",&CLCT_PAT_ID,"CLCT_PAT_ID[8][16]/I");
   tree->Branch("CLCT_PAT",&CLCT_PAT,"CLCT_PAT[8][16]/I");
   tree->Branch("L_R_bit",&L_R_bit,"L_R_bit[8][16]/I");

   tree->Branch("ME_BXN",&ME_BXN,"ME_BXN[8][16]/I");
   tree->Branch("BC0",&BC0,"BC0[8][16]/I");

   tree->Branch("Phi",&Phi,"Phi[8][4]/I");
   tree->Branch("CHRG",&CHRG,"CHRG[8][4]/I");     
   tree->Branch("ETA",&ETA,"ETA[8][4]/I");
   tree->Branch("del12Phi",&del12Phi,"del12Phi[8][4]/I");     
   tree->Branch("del23Phi",&del23Phi,"del23Phi[8][4]/I");
   tree->Branch("sign",&sign,"sign[8][4]/I");  
   
   tree->Branch("MS_ID",&MS_ID,"MS_ID[8][4]/I");
   tree->Branch("MB_ID",&MB_ID,"MB_ID[8][4]/I");
   tree->Branch("ME1_ID",&ME1_ID,"ME1_ID[8][4]/I");
    tree->Branch("ME2_ID",&ME2_ID,"ME2_ID[8][4]/I");
    tree->Branch("ME3_ID",&ME3_ID,"ME3_ID[8][4]/I");
    tree->Branch("ME4_ID",&ME4_ID,"ME4_ID[8][4]/I");
    
   //DT-data:
   tree->Branch("quality_MB",&quality_MB,"quality_MB[8][3]/I");   
   tree->Branch("phi_bend_MB",&phi_bend_MB,"phi_bend_MB[8][3]/I");
   tree->Branch("BC0_MB",&BC0_MB,"BC0_MB[8][3]/I");
   tree->Branch("BXN0_MB",&BXN0_MB,"BXN0_MB[8][3]/I");
   tree->Branch("BXN1_MB",&BXN1_MB,"BXN1_MB[8][3]/I");
   tree->Branch("phi_MB",&phi_MB,"phi_MB[8][3]/I");
   tree->Branch("MB_BXN",&MB_BXN,"MB_BXN[8][3]/I");
   tree->Branch("FLAG",&FLAG,"FLAG[8][3]/I");
}


rootTree::~rootTree(){
   tree -> Write();
   delete tree ;
   delete file ;
   cout<<"tree written, job done..."<<endl;
}


TTree* rootTree::getTree(){
   return tree;
}


void rootTree::fillEventHeader(si_map HD1a_map, si_map HD1b_map, si_map HD1c_map, si_map HD1d_map, 
			si_map HD2a_map, si_map HD2b_map ,si_map HD2c_map, si_map HD2d_map){
			
   L1ANumber = HD1a_map["L1A"]; //L1A lsb
   L1ANumber += (HD1b_map["L1A"]<<12) ;	//L1A msb
   L1A_BXN = HD1d_map["L1A_BXN"];

   if (debug){
      cout<<endl<<"		---> EventHeader"<<endl;
      cout<<"L1ANumber == 0x"<<hex<<L1ANumber<<endl;
      cout<<"L1A_BXN == 0x"<<hex<<L1A_BXN<<endl;
   }
   return;		
}

void rootTree::fillBlockCounter(si_map BCa_map, si_map BCb_map, si_map BCc_map, si_map BCd_map){

   TC = 0;			
   TC = BCa_map["TC"] +(BCb_map["TC"]<<15) ; //track counter

   OC = 0;			
   OC = BCc_map["OC"] +(BCd_map["OC"]<<15) ; //orbit counter

   if (debug){
      cout<<endl<<"		---> Block Counters"<<endl;
      cout<<"Track Counter == 0x"<<hex<<TC<<endl;
      cout<<"Orbit Counter == 0x"<<hex<<OC<<endl;
   }
   return;		
}

void rootTree::fillDataBlockHeader(si_map BH1a_map, si_map BH1b_map, si_map BH1c_map, si_map BH1d_map, 
			si_map BH2a_map, si_map BH2b_map, si_map BH2c_map, si_map BH2d_map, int bx){

   validPatternLinks[bx] 	= BH1a_map["hexwd"];
   VQd[bx]			= BH1b_map["VQd"];
   VQa[bx]			= BH1b_map["VQa"];
   SynchErrors[bx]		= BH1c_map["hexwd"];
   
   if (debug){
      cout<<endl<<"		---> DataBlockHeader"<<endl;
      cout<<"valid pattern word == 0x"<<hex<<validPatternLinks[bx]<<endl;
      cout<<"Valid Quality MB-D == 0x"<<hex<<VQd[bx]<<endl;
      cout<<"Valid Quality MB-A == 0x"<<hex<<VQa[bx]<<endl;
      cout<<"Synch Error word == 0x"<<hex<<SynchErrors[bx]<<endl;
   }

   return;			
}
			

void rootTree::fillMEDataRecord(si_map MEa_map, si_map MEb_map, si_map MEc_map, si_map MEd_map,
			int bx, int LINK){

   CLCT_PAT[bx][LINK] 		= MEa_map["CLCT Pattern #"];
   quality[bx][LINK]		= MEa_map["Quality"];
   wireGP[bx][LINK]		= MEa_map["Wire Group ID"];
   CLCT_PAT_ID[bx][LINK]	= MEb_map["CLCT Pattern ID"];
   cscID[bx][LINK]		= MEb_map["CSC ID"];
   L_R_bit[bx][LINK]		= MEb_map["L/R"];
   BC0[bx][LINK]		= MEb_map["BC0"];
   ME_BXN[bx][LINK]		= MEc_map["ME_BXN"];

   if (debug){
      cout<<endl<<"		---> MEDataRecord, bx="<<bx<<" LINK="<<LINK<<endl;
      cout<<"CLCT_PAT == 0x"<<hex<<CLCT_PAT[bx][LINK]<<endl;
      cout<<"quality == 0x"<<hex<<quality[bx][LINK]<<endl;
      cout<<"wireGP == 0x"<<hex<<wireGP[bx][LINK]<<endl;
      cout<<"CLCT_PAT_ID == 0x"<<hex<<CLCT_PAT_ID[bx][LINK]<<endl;
      cout<<"cscID == 0x"<<hex<<cscID[bx][LINK]<<endl;
      cout<<"L_R_bit == 0x"<<hex<<L_R_bit[bx][LINK]<<endl;
      cout<<"BC0 == 0x"<<hex<<BC0[bx][LINK]<<endl;
      cout<<"ME bx number == 0x"<<hex<<ME_BXN[bx][LINK]<<endl;
   }

   return;			
}


void rootTree::fillMBDataRecord(si_map MBa_map, si_map MBb_map, si_map MBc_map, si_map MBd_map, 
			int bx, int whichMB){
//whichMB=1 for MB1 and 2 for MBd  

 quality_MB[bx][whichMB]		= MBa_map["Q"];
 phi_bend_MB[bx][whichMB]		= MBa_map["PHI BEND"];
 FLAG[bx][whichMB]			= MBa_map["FLAG"];
 BC0_MB[bx][whichMB]			= MBb_map["BC0"];
 BXN0_MB[bx][whichMB]			= MBb_map["BXN0"];
 BXN1_MB[bx][whichMB]			= MBb_map["BXN1"];
 phi_MB[bx][whichMB]			= MBb_map["PHI"];
 MB_BXN[bx][whichMB]			= MBc_map["MB_BXN"];


   if (debug){
      cout<<endl<<"		---> MBDataRecord, bx="<<bx<<" whichMB="<<whichMB<<endl;
      cout<<"quality_MB == 0x"<<hex<<quality_MB[bx][whichMB]<<endl;
      cout<<"phi_bend_MB == 0x"<<hex<<phi_bend_MB[bx][whichMB]<<endl;
      cout<<"FLAG == 0x"<<hex<<FLAG[bx][whichMB]<<endl;
      cout<<"BC0_MB == 0x"<<hex<<BC0_MB[bx][whichMB]<<endl;
      cout<<"BXN0_MB == 0x"<<hex<<BXN0_MB[bx][whichMB]<<endl;
      cout<<"BXN1_MB == 0x"<<hex<<BXN1_MB[bx][whichMB]<<endl;
      cout<<"phi_MB == 0x"<<hex<<phi_MB[bx][whichMB]<<endl;
      cout<<"BC0 == 0x"<<hex<<BC0[bx][whichMB]<<endl;
      cout<<"MB_BXN == 0x"<<hex<<MB_BXN[bx][whichMB]<<endl;
   }
   return;
}


void rootTree::fillSPDataRecord(si_map SPa_map, si_map SPb_map, si_map SPc_map, si_map SPd_map, 
			int bx, int mode){
   Phi[bx][mode]		= SPa_map["PHI"];
   sign[bx][mode]		= SPa_map["SIGN"];
   CHRG[bx][mode]		= SPa_map["CHRG"];
   ETA[bx][mode]		= SPa_map["ETA"];
   del12Phi[bx][mode]		= SPb_map["D12_PHI"];
   del23Phi[bx][mode]		= SPb_map["D23_PHI"];
   
   MS_ID[bx][mode]		= SPc_map["MS_ID"];
   MB_ID[bx][mode]		= SPc_map["MB_ID"];
   ME4_ID[bx][mode]		= SPc_map["ME4_ID"];
   ME3_ID[bx][mode]		= SPc_map["ME3_ID"];
   ME2_ID[bx][mode]		= SPc_map["ME2_ID"];
   ME1_ID[bx][mode]		= SPc_map["ME1_ID"];
   MB_TBIN[bx][mode]		= SPd_map["MB_TBIN"];
   ME4_TBIN[bx][mode]		= SPd_map["ME4_TBIN"];
   ME3_TBIN[bx][mode]		= SPd_map["ME3_TBIN"];
   ME2_TBIN[bx][mode]		= SPd_map["ME2_TBIN"];
   ME1_TBIN[bx][mode]		= SPd_map["ME1_TBIN"];
   
   

   if (debug){
      cout<<endl<<"		---> SPDataRecord, bx="<<bx<<" mode="<<mode<<endl;
      cout<<"Phi == 0x"<<hex<<Phi[bx][mode]<<endl;
      cout<<"sign == 0x"<<hex<<sign[bx][mode]<<endl;
      cout<<"CHRG == 0x"<<hex<<CHRG[bx][mode]<<endl;
      cout<<"ETA == 0x"<<hex<<ETA[bx][mode]<<endl;
      cout<<"del12Phi == 0x"<<hex<<del12Phi[bx][mode]<<endl;
      cout<<"del23Phi == 0x"<<hex<<del23Phi[bx][mode]<<endl;
      
      cout<<"MS_ID == 0x"<<hex<<MS_ID[bx][mode]<<endl;
      cout<<"MB_ID == 0x"<<hex<<MB_ID[bx][mode]<<endl;
      cout<<"ME4_ID == 0x"<<hex<<ME4_ID[bx][mode]<<endl;
      cout<<"ME3_ID == 0x"<<hex<<ME3_ID[bx][mode]<<endl;
      cout<<"ME2_ID == 0x"<<hex<<ME2_ID[bx][mode]<<endl;
      cout<<"ME1_ID == 0x"<<hex<<ME1_ID[bx][mode]<<endl;
      
      cout<<"MB_TBIN == 0x"<<hex<<MB_TBIN[bx][mode]<<endl;    
      cout<<"ME4_TBIN == 0x"<<hex<<ME4_TBIN[bx][mode]<<endl;         
      cout<<"ME3_TBIN == 0x"<<hex<<ME3_TBIN[bx][mode]<<endl;  
      cout<<"ME2_TBIN == 0x"<<hex<<ME2_TBIN[bx][mode]<<endl;         
      cout<<"ME1_TBIN == 0x"<<hex<<ME1_TBIN[bx][mode]<<endl;       
   }			
   return;			
}


void rootTree::fillEventTrailer(si_map TR1a_map, si_map  TR1b_map, si_map  TR1c_map, si_map TR1d_map,
			si_map  TR2a_map, si_map  TR2b_map, si_map  TR2c_map, si_map  TR2d_map){
	
   if (debug){
      cout<<endl<<"		---> SPEventTrailer"<<endl;
   }	
			
   return;			
}