convert2simroot.cxx

Go to the documentation of this file.

//=====================================================
// this script is used for TOF simulation
//=====================================================
#include "TTree.h"
#include "TFile.h"
#include <assert.h>
#include <iostream>
#include <fstream>
using namespace std;
 
bool sim_conv2root(){
  TTree* t=new TTree("SimConstants","threshold and gain constants");
  double blth=50., bhth=50., elth=50., ehth=50.;
  double bgain=5.E5, egain=2.5E5;
  double bconst=1.0, econst=1.0;
  double eswitch=0;//swtich for endcap noise sim: 0-no sigma smear ; 1-only add smear in east endcap sim; 2-add smear in both east and west endcap sim;
  t->Branch("BarLowThres",  &blth,   "BarLowThres/D" );
  t->Branch("BarHighThres", &bhth,   "BarHighThres/D");
  t->Branch("EndLowThres",  &elth,   "EndLowThres/D" );
  t->Branch("EndHighThres", &ehth,   "EndHighThres/D");
  t->Branch("BarPMTGain",   &bgain,  "BarPMTGain/D"  );
  t->Branch("EndPMTGain",   &egain,  "EndPMTGain/D"  );
  t->Branch("BarConstant",  &bconst, "BarConst/D"    );
  t->Branch("EndConstant",  &econst, "EndConst/D"    );
  t->Branch("EndNoiseSwitch",&eswitch, "EndNoiseSwitch/D"    );
 
  ifstream inf;
  bool is_open=false;
  const char* file="SimConstants.txt";
  inf.open(file,ios::in);
  if( inf.good() ) {
    is_open=true;
  }
  else {
    std::cerr<<"file: "<<file <<" can't be found!"<<std::endl;
  }
  try{
    if( is_open ){
      inf >> blth >> bhth;
      inf >> elth >> ehth;
      inf >> bgain;
      inf >> egain;
      inf >> bconst;
      inf >> econst;
      inf >> eswitch;
      t->Fill();
    }
  }
 
  TFile f("TofSimConst.root","RECREATE");
  t->Write();
  f.Close();
  delete t;
  t=NULL;
 
  return true;
}
 
bool barrel_conv2root(){
  double gain=0.0, ratio=0.0, length=1.0;
  TTree* t=new TTree("BTofSim","barrel attenuation length and gain");
  t->Branch("Gain",        &gain,   "Gain/D");
  t->Branch("Ratio",       &ratio,  "Ratio/D");
  t->Branch("AttenLength", &length, "AttenLength/D");
 
  double a1overa2 = 1.0, a2 = 1.0;
  double qmean = 0.0, qsigma = 0.0;
  ifstream inf1, inf2;
  bool is_open1=false, is_open2=false;
  const char* file1="calib_barrel_atten.txt";
  const char* file2="calib_barrel_q0.txt";
  inf1.open(file1,ios::in);
  inf2.open(file2,ios::in);
  if( inf1.good() ) {
    is_open1=true;
  }
  else {
    std::cerr<<"file: "<<file1 <<" can't be found!"<<std::endl;
  }
  if( inf2.good() ) {
    is_open2=true;
  }
  else {
    std::cerr<<"file: "<<file2 <<" can't be found!"<<std::endl;
  }
 
  try{
    if( is_open1 && is_open2 ) {
      for( unsigned int i=0; i<176; i++ ) {
    inf1 >> length >> ratio;
    if( fabs(length)<1.0e-6 ) { length = 1.0; }
    length = 1.0/length;
 
    inf2 >> gain >> qmean >> qsigma;
 
    t->Fill();
      }
    }
  } catch(...) {
    return false;
  }
 
  inf1.close();
  inf2.close();
 
  TFile f("BarTofSim.root","RECREATE");
  t->Write();
  f.Close();
  delete t;
  t=NULL;
 
  return true;
}
 
bool endcap_conv2root(){
  double gain=0.0, length=1.0, noisesmear=0.;
  double scin, mean, errmean, sigma, errsigma;
  double mc_sigma;
  TTree* t=new TTree("ETofSim","endcap attenuation length and gain");
  t->Branch("Gain",        &gain,      "Gain/D"       );
  t->Branch("AttenLength", &length,    "AttenLength/D");
  t->Branch("NoiseSmear",  &noisesmear,"NoiseSmear/D" );
 
  double length2=1.0;
  ifstream inf1, inf2, inf3;
  bool is_open1=false, is_open2=false, is_open3=false;
  const char* file1="calib_endcap_atten.txt";
  const char* file2="tEndRes.txt";
  const char* file3="Endcap_mc_tEndRes.txt";
  inf1.open(file1,ios::in);
  inf2.open(file2,ios::in);
  inf3.open(file3,ios::in);
  if(inf1.good())
    is_open1=true;
  else{
    std::cerr<<"File: "<<file1<<" can't be opened"<<std::endl;
  }
  if(inf2.good())
    is_open2=true;
  else{
    std::cerr<<"File: "<<file2<<" can't be opened"<<std::endl;
  }
  if(inf3.good())
    is_open3=true;
  else{
    std::cerr<<"File: "<<file3<<" can't be opened"<<std::endl;
  }
  try{
    if( is_open1 && is_open2 && is_open3) {
      for(int i=0;i<96;i++) {
    inf1 >> gain >> length >> length2;
    inf2 >> scin >> mean >> errmean >> sigma >> errsigma; 
    inf3 >> mc_sigma; 
    if ( mc_sigma > sigma ) {noisesmear = 0.;}
    else { noisesmear = sqrt(sigma*sigma - mc_sigma*mc_sigma);}
    t->Fill();
      }
    }
  } catch(...){
    return false;
  }
 
  inf1.close();
  inf2.close();
  inf3.close();
 
  TFile f("EndTofSim.root","RECREATE");
  t->Write();
  f.Close();
  delete t;
  t=NULL;
 
  return true;
}
 
bool join(){
  
  TFile f1("TofSimConst.root");
  TTree* t1=(TTree*)f1.Get("SimConstants");
  if(0==t1) { std::cerr<<" can't get tree :SimConstants in file TofSimConst.root"<<std::endl; return false;}
  TFile f2("BarTofSim.root");
  TTree* t2=(TTree*)f2.Get("BTofSim");
  if(0==t2) { std::cerr<<" can't get tree :BTofSim in file BarTofSim.root"<<std::endl; return false;}
  TFile f3("EndTofSim.root");
  TTree* t3=(TTree*)f3.Get("ETofSim");
  if(0==t3) { std::cerr<<" can't get tree :ETofSim in file EndTofSim.root"<<std::endl; return false;}
 
  TFile fnew("TofSimConst.root","RECREATE");
  t1->Write();
  t2->Write();
  t3->Write();
  f1.Close();
  f2.Close();
  f3.Close();
  fnew.Close();
  t1=NULL;
  t2=NULL;
  t3=NULL;
 
  return true;
}
 
int main(){
  sim_conv2root();
  barrel_conv2root();
  endcap_conv2root();
  join();
}