EstTofCaliSvc/EstTofCaliSvc-00-00-14/test/convert2root.cxx

Go to the documentation of this file.

//=====================================================
// this script is used to convert result in text file//
// to root files. it is based on files named that start
// by "calib_barrel" .it means thest result from barrel.         //
//=====================================================
#include "TTree.h"
#include "TFile.h"
#include <assert.h>
#include <iostream>
#include <fstream>
using namespace std;
bool barrel_con2root(){
  //this method so bothered
 
  TTree* t0=new TTree("EsTimeOffset","EsTimeOffset");
  double offset[2]={0.};;
  char estbrname[20];
  char estptname[20];
  for(int i = 0; i<2; i++){
    sprintf(estbrname, "Offset%d",i);
    sprintf(estptname, "Offset%d/D",i);
    t0->Branch(estbrname, &offset[i], estptname);
  }
 
  ifstream inf0;
  ifstream inf01;
  inf0.open("calib_EsTimeOffset.txt",ios::in);
  inf01.open("EsTimeOffset.txt", ios::in);
  if(inf0.good() && inf01.good()) {
    try{
      inf0 >> offset[0] ;
      inf01 >> offset[1] ;
      t0->Fill();
    }catch(...) {
      return false;
    }
  }
  else {
    std::cerr<<"file: calib_estimeOffset.txt or est_estimeOffset can't be found!"<<std::endl;
  }
  inf0.close();
  inf01.close();
 
  TTree* t=new TTree("BarTofPar","barrel parameters");
  //create a full tree;
  double p[20]={0.};;
  char brname[20];//branch name
  char ptname[20];//pattern name
  for(int i=0;i<20;i++){
    sprintf(brname,"P%d",i);
    sprintf(ptname,"p%d/D",i);
    t->Branch(brname,&p[i],ptname);
  }
  double w[4];
  for(int i=0;i<4;i++){
    sprintf(brname,"W%d",i);
    sprintf(ptname,"w%d/D",i);
    t->Branch(brname,&w[i],ptname);
  }
  double Atten[8]={0.};
  for(int i=0;i<8;i++){
    sprintf(brname, "Atten%d",i);
    sprintf(ptname, "Atten%d/D",i);
    t->Branch(brname, &Atten[i], ptname);
  }
  double Speed[2]={0.},Q0[1]={0.};
  t->Branch("Speed0",&Speed[0],"Speed0/D");
  t->Branch("Speed1",&Speed[1],"Speed1/D");
  t->Branch("Q0",Q0,"Q0[1]/D");
  double fitfun[20]={0.};;
  char fbrname1[5];//branch name
  char fptname1[5];//pattern name
  char fbrname2[5];//branch name
  char fptname2[5];//pattern name
  char fbrname3[10];//branch name
  char fptname3[10];//pattern name
  for(int i=0;i<5;i++){
    sprintf(fbrname1,"FLeft%d",i);
    sprintf(fptname1,"fleft%d/D",i);
    t->Branch(fbrname1,&fitfun[i],fptname1);
    sprintf(fbrname2,"FRight%d",i);
    sprintf(fptname2,"fright%d/D",i);
    t->Branch(fbrname2,&fitfun[5+i],fptname2);
  }
  for(int i=0;i<10;i++){
    sprintf(fbrname3,"FCounter%d",i);
    sprintf(fptname3,"fcounter%d/D",i);
    t->Branch(fbrname3,&fitfun[10+i],fptname3);
  }
  const int N=10;
  ifstream inf[N];
  bool is_open[N]={false};
  const char* filelist[N]={"calib_barrel_left.txt","calib_barrel_right.txt","calib_barrel_w.txt","calib_barrel_atten.txt","calib_barrel_veff.txt","calib_barrel_sigma.txt","calib_barrel_left1.txt","calib_barrel_right1.txt","calib_barrel_common.txt","calib_barrel_q0.txt"};
  for(int m=0;m<N;m++){
    inf[m].open(filelist[m],ios::in);
    if(inf[m].good())
      is_open[m]=true;
    else
      std::cerr<<"file: "<<filelist[N]<<" can't be found!"<<std::endl;
  }
  try{
    for(int k=0;k<176;k++){
      //------------set p-------------------------
      if(is_open[0]&&is_open[1]){
    for(int j=0;j<7;j++){
      inf[0]>>p[j];
      //        cout<<p[j]<<' ';
    }
    for(int l=10;l<17;l++)
      inf[1]>>p[l];
      }
      //--------------w pars---------------------
      if(is_open[2]){
    for(int j=0;j<4;j++)
      inf[2]>>w[j];
      }
      //-----------atten lenght------------------
      if(is_open[3]){
    for(int j=0;j<2;j++){
      inf[3]>>Atten[j];
    }
    Atten[0]=1./Atten[0];
      }
      //-------------speed-----------------------
      if(is_open[4]){
    for(int j=0;j<2;j++){
      inf[4]>>Speed[j];
    }
    Speed[0]=1./Speed[0];
      }
      if(is_open[5]){
    for(int j=0; j<4; j++) {
      inf[5] >> fitfun[j];
    }
    for(int j=5; j<9; j++) {
      inf[5] >> fitfun[j];
    }
    for(int j=10; j<13; j++) {
      inf[5] >> fitfun[j];
    }
      }
      //--------------Q0-----------------
      if(is_open[9]){
    for(int j=2;j<5;j++){
      inf[9] >> Atten[j];
    }
      }
      //-------------Q0----------------------------
      //   for(int j=0;j<1;j++)
      //        inf[5]>>Q0[j];
      //-------------fill-----------------------
      t->Fill();
    }
  }catch(...) {
    return false;
  }
 
  TTree* t1=new TTree("BarTofParInner","barrel inner layer parameters");
  //create a full tree;
  double p1[20]={0.};;
  char brname1[20];//branch name
  char ptname1[20];//pattern name
  for(int i=0;i<20;i++){
    sprintf(brname1,"P%d",i);
    sprintf(ptname1,"p%d/D",i);
    t1->Branch(brname1,&p1[i],ptname1);
  }
  try{
    for(int k=0;k<88;k++){
      //------------set p-------------------------
      if(is_open[6]&&is_open[7]){
    for(int j=0;j<7;j++){
      inf[6]>>p1[j];
      //        cout<<p[j]<<' ';
    }
    for(int l=10;l<17;l++)
      inf[7]>>p1[l];
      }
      t1->Fill();
    }
  } catch(...) {
    return false;
  }
 
  TTree* t2=new TTree("BarTofParCommon","common part of barrel");
  double t0Offset[2]={0.},sigmaCorr[8]={0.};
  for(int i=0;i<2;i++){
    sprintf(brname,"t0offset%d",i);
    sprintf(ptname,"t0offset%d/D",i);
    t2->Branch(brname,&t0Offset[i],ptname);
  }
  for(int i=0;i<8;i++){
    sprintf(brname,"sigmaCorr%d",i);
    sprintf(ptname,"sigmaCorr%d/D",i);
    t2->Branch(brname,&sigmaCorr[i],ptname);
  }
  try{
    if(is_open[8]){
      inf[8] >> sigmaCorr[0] >> sigmaCorr[1] >> t0Offset[0] >> t0Offset[1];
    }
    t2->Fill();
  } catch(...) {
    return false;
  }
  for(int m=0;m<N;m++)
    inf[m].close();
  //-------------------------------------------
  TFile f("BarTofPar.root","RECREATE");
  t0->Write();
  t->Write();
  t1->Write();
  t2->Write();
  f.Close();
  delete t0;
  delete t;
  delete t1;
  delete t2;
  t0=NULL;
  t=NULL;
  t1=NULL;
  t2=NULL;
  return true;
}
//=====================================================
// this script is used to convert result in text file//
// to root files. it is based on files named that start
// by "calib_endcap_" .it means thest result from endcap.        //
//=====================================================
bool endcap_con2root(){
  //this method so bothered
 
  TTree* t0=new TTree("EsTimeOffset","EsTimeOffset");
  double offset[2]={0.};
  char estbrname[20];
  char estptname[20];
  for(int i = 0; i<2; i++){
    sprintf(estbrname, "Offset%d",i);
    sprintf(estptname, "Offset%d/D",i);
    t0->Branch(estbrname, &offset[i], estptname);
  }
 
  ifstream inf0;
  ifstream inf01;
  inf0.open("calib_estimeOffset.txt",ios::in);
  inf01.open("est_estimeOffset.txt", ios::in);
  if(inf0.good()&& inf01.good()) {
    try{
      inf0 >> offset[0] ;
      inf01 >> offset[1] ;
      t0->Fill();
    }catch(...) {
      return false;
    }
  }
  else {
    std::cerr<<"file: calib_estimeOffset.txt or est_estimeOffset can't be found!"<<std::endl;
  }
  inf0.close();
  inf01.close();
 
  TTree* t=new TTree("EndTofPar","endcap parameters");
  //create a full tree;
  double p[8]={0.};;
  char brname[8];//branch name
  char ptname[8];//pattern name
  for(int i=0;i<8;i++){
    sprintf(brname,"P%d",i);
    sprintf(ptname,"p%d/D",i);
    t->Branch(brname,&p[i],ptname);
  }
  double Atten[5]={0.},Speed[3]={0.},Q0[1]={0.};
  t->Branch("Atten0",&Atten[0],"Atten0/D");
  t->Branch("Atten1",&Atten[1],"Atten1/D");
  t->Branch("Atten2",&Atten[2],"Atten2/D");
  t->Branch("Atten3",&Atten[3],"Atten3/D");
  t->Branch("Atten4",&Atten[4],"Atten4/D");
  t->Branch("Speed0",&Speed[0],"Speed0/D");
  t->Branch("Speed1",&Speed[1],"Speed1/D");
  t->Branch("Speed2",&Speed[2],"Speed2/D");
  t->Branch("Q0",Q0,"Q0[1]/D");
  const int N=3;
  ifstream inf[N];
  bool is_open[N]={false};
  const char* filelist[N]={"calib_endcap_left.txt","calib_endcap_atten.txt","calib_endcap_veff.txt"};
  for(int m=0;m<N;m++){
    inf[m].open(filelist[m],ios::in);
    if(inf[m].good())
      is_open[m]=true;
    else{
      std::cerr<<"File: "<<filelist[m]<<" can't be opened"<<std::endl;
    }
  }
  try{
    for(int k=0;k<96;k++){
      //------------set p-------------------------
      if(is_open[0]&&is_open[1]){
    for(int j=0;j<5;j++){
      inf[0]>>p[j];
      //        cout<<p[j]<<' ';
    }
      }
      //-----------atten lenght------------------
      if(is_open[1]){
    for(int j=0;j<3;j++){
      inf[1]>>Atten[j];
    }
    Atten[0]=1./Atten[0];
      }
      //-------------speed-----------------------
      if(is_open[2]){
    for(int j=0;j<3;j++){
      inf[2]>>Speed[j];
    }
    Speed[0]=1./Speed[0];
      }
      //-------------Q0----------------------------
      //   for(int j=0;j<1;j++)
      //        inf[5]>>Q0[j];
      //-------------fill-----------------------
      t->Fill();
    }
  } catch(...){
    return false;
  }
  for(int m=0;m<N;m++)
    inf[m].close();
  //-------------------------------------------
  TFile f("EndTofPar.root","RECREATE");
  t0->Write();
  t->Write();
  f.Close();
  delete t0;
  delete t;
  t=NULL;
  t=NULL;
  return true;
}
bool join(){
 
  TTree* t0=new TTree("EsTimeOffset","EsTimeOffset");
  double offset[2]={0.};
  char estbrname[20];
  char estptname[20];
  for(int i = 0; i<2; i++){
    sprintf(estbrname, "Offset%d",i);
    sprintf(estptname, "Offset%d/D",i);
    t0->Branch(estbrname, &offset[i], estptname);
  }
 
  ifstream inf0;
  ifstream inf01;
  inf0.open("calib_estimeOffset.txt",ios::in);
  inf01.open("est_estimeOffset.txt", ios::in);
  if(inf0.good()&& inf01.good()) {
    try{
      inf0 >> offset[0] ;
      inf01 >> offset[1] ;
      t0->Fill();
    }catch(...) {
      return false;
    }
  }
  else {
    std::cerr<<"file: calib_estimeOffset.txt or est_estimeOffset can't be found!"<<std::endl;
  }
  inf0.close();
  inf01.close();
 
  TFile f("EndTofPar.root");
  TTree* t=(TTree*)f.Get("EndTofPar");
  if(0==t) { std::cerr<<" can't get tree :EndTofPar in file EndTofPar.root"<<std::endl; return false;}
  TFile f1("BarTofPar.root");
  TTree* t1=(TTree*)f1.Get("BarTofPar");
  if(0==t1) { std::cerr<<" can't get tree :BarTofPar in file BarTofPar.root"<<std::endl; return false;}
  TTree* t2=(TTree*)f1.Get("BarTofParInner");
  if(0==t2) { std::cerr<<" can't get tree :BarTofPar Inner Layer in file BarTofPar.root"<<std::endl; return false;}
  TTree* t3=(TTree*)f1.Get("BarTofParCommon");
  if(0==t3) { std::cerr<<" can't get tree :BarTofPar Common in file BarTofPar.root"<<std::endl; return false;}
  TFile fnew("TofCalConst.root","RECREATE");
  t0->Write();
  t->Write();
  t1->Write();
  t2->Write();
  t3->Write();
  f.Close();
  f1.Close();
  fnew.Close();
  t0=NULL;
  t=NULL;
  t1=NULL;
  t2=NULL;
  t3=NULL;
  return true;
}
int main(){
  barrel_con2root();
  endcap_con2root();
  join();
}