convert2simroot.cxx

Go to the documentation of this file.

//=====================================================
// this script is used for TOF simulation
//=====================================================
#include "TFile.h"
#include "TTree.h"
#include <assert.h>
#include <fstream>
#include <iostream>
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();
}