convElec2root.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_conv2root(){
  TTree* t1=new TTree("QElecBarParEast","barrel parameters of east end");
  TTree* t2=new TTree("QElecBarParWest","barrel parameters of west end");
  TTree* t3=new TTree("BarBoardNum","barrel electronics number");
  double p1[11]={0.};
  double p2[11]={0.};
  char brname[50]; //branch name
  char ptname[50]; //pattern name
  for(int i=0;i<11;i++){
    sprintf(brname,"P%d",i);
    sprintf(ptname,"p%d/D",i);
    t1->Branch(brname,&p1[i],ptname);
    t2->Branch(brname,&p2[i],ptname);
  }
 
  double num[4]={0.};
  sprintf(brname,"Board");
  sprintf(ptname,"board/D");
  t3->Branch(brname,&num[0],ptname);
  sprintf(brname,"Crate");
  sprintf(ptname,"crate/D");
  t3->Branch(brname,&num[1],ptname);
  sprintf(brname,"Fee");
  sprintf(ptname,"fee/D");
  t3->Branch(brname,&num[2],ptname);
  sprintf(brname,"Channel");
  sprintf(ptname,"channel/D");
  t3->Branch(brname,&num[3],ptname);
 
  const unsigned int N = 2;
  bool is_open[N]={false};
  ifstream infile[N];
  char inTxtFile[N][500]={"ElecBarEast.txt","ElecBarWest.txt"};
  for(int m=0;m<N;m++){
    infile[m].open(inTxtFile[m],ios::in);
    if(infile[m].good())
      is_open[m]=true;
    else
      std::cerr<<"file: "<<inTxtFile[N]<<" can't be found!"<<std::endl;
  }
  try{
    for(int k=0;k<176;k++){
      //------------set p-------------------------
      if(is_open[0]){
    for( unsigned int j=0; j<4; j++ ) {
      infile[0] >> num[j];
    }
    for( unsigned int j=0; j<9; j++ ) {
      infile[0] >> p1[j];
    }
      }
      t1->Fill();
      t3->Fill();
      if(is_open[1]){
    for( unsigned int j=0; j<4; j++ ) {
      infile[1] >> num[j];
    }
    for( unsigned int j=0; j<9; j++ ) {
      infile[1] >> p2[j];
    }
      }
      t2->Fill();
      t3->Fill();
    }
  }
  catch(...) {
    return false;
  }
 
  TFile f("BarTofQElecPar.root","RECREATE");
  t1->Write();
  t2->Write();
  t3->Write();
  f.Close();
  delete t1;
  delete t2;
  delete t3;
  t1=NULL;
  t2=NULL;
  t3=NULL;
  return true;
}
 
bool endcap_conv2root(){
  TTree* t1=new TTree("QElecEndPar","endcap parameters of east end");
  TTree* t2=new TTree("EndBoardNum","endcap electronics number");
  double p[11]={0.};
  char brname[50]; //branch name
  char ptname[50]; //pattern name
  for(int i=0;i<11;i++){
    sprintf(brname,"P%d",i);
    sprintf(ptname,"p%d/D",i);
    t1->Branch(brname,&p[i],ptname);
  }
 
  double num[4]={0.};
  sprintf(brname,"Board");
  sprintf(ptname,"board/D");
  t2->Branch(brname,&num[0],ptname);
  sprintf(brname,"Crate");
  sprintf(ptname,"crate/D");
  t2->Branch(brname,&num[1],ptname);
  sprintf(brname,"Fee");
  sprintf(ptname,"fee/D");
  t2->Branch(brname,&num[2],ptname);
  sprintf(brname,"Channel");
  sprintf(ptname,"channel/D");
  t2->Branch(brname,&num[3],ptname);
 
  bool is_open=false;
  ifstream infile;
  char inTxtFile[500]="ElecEndcap.txt";
  infile.open(inTxtFile,ios::in);
  if(infile.good())
    is_open=true;
  else
    std::cerr<<"file: "<<inTxtFile<<" can't be found!"<<std::endl;
  try{
    for(int k=0;k<96;k++){
      //------------set p-------------------------
      if(is_open){
    for( unsigned int j=0; j<4; j++ ) {
      infile >> num[j];
    }
    for( unsigned int j=0; j<9; j++ ) {
      infile >> p[j];
    }
      }
      t1->Fill();
      t2->Fill();
    }
  }
  catch(...) {
    return false;
  }
 
  TFile f("EndTofQElecPar.root","RECREATE");
  t1->Write();
  t2->Write();
  f.Close();
  delete t1;
  delete t2;
  t1=NULL;
  t2=NULL;
  return true;
}
 
 
bool sim_barrel_conv2root(){
  TTree* t1=new TTree("SimQElecBarParEast","barrel parameters of east end");
  TTree* t2=new TTree("SimQElecBarParWest","barrel parameters of west end");
  double p1[11]={0.};
  double p2[11]={0.};
  char brname[50]; //branch name
  char ptname[50]; //pattern name
  for(int i=0;i<11;i++){
    sprintf(brname,"P%d",i);
    sprintf(ptname,"p%d/D",i);
    t1->Branch(brname,&p1[i],ptname);
    t2->Branch(brname,&p2[i],ptname);
  }
 
  const unsigned int N = 2;
  bool is_open[N]={false};
  ifstream infile[N];
  char inTxtFile[N][500]={"ElecBarEast-Curve.txt","ElecBarWest-Curve.txt"};
  for(int m=0;m<N;m++){
    infile[m].open(inTxtFile[m],ios::in);
    if(infile[m].good())
      is_open[m]=true;
    else
      std::cerr<<"file: "<<inTxtFile[N]<<" can't be found!"<<std::endl;
  }
  try{
    for(int k=0;k<176;k++){
      //------------set p-------------------------
      if(is_open[0]){
    for( unsigned int j=0; j<11; j++ ) {
      infile[0] >> p1[j];
    }
      }
      if(is_open[1]){
    for( unsigned int j=0; j<11; j++ ) {
      infile[1] >> p2[j];
    }
      }
      t1->Fill();
      t2->Fill();
    }
  }
  catch(...) {
    return false;
  }
 
  TFile f("SimBarTofQElecPar.root","RECREATE");
  t1->Write();
  t2->Write();
  f.Close();
  delete t1;
  delete t2;
  t1=NULL;
  t2=NULL;
  return true;
}
 
bool sim_endcap_conv2root(){
  TTree* t=new TTree("SimQElecEndPar","endcap parameters of east end");
  double p[11]={0.};
  char brname[50]; //branch name
  char ptname[50]; //pattern name
  for(int i=0;i<11;i++){
    sprintf(brname,"P%d",i);
    sprintf(ptname,"p%d/D",i);
    t->Branch(brname,&p[i],ptname);
  }
 
  bool is_open=false;
  ifstream infile;
  char inTxtFile[500]="ElecEndcap-Curve.txt";
  infile.open(inTxtFile,ios::in);
  if(infile.good())
    is_open=true;
  else
    std::cerr<<"file: "<<inTxtFile<<" can't be found!"<<std::endl;
  try{
    for(int k=0;k<96;k++){
      //------------set p-------------------------
      if(is_open){
    for( unsigned int j=0; j<11; j++ ) {
      infile >> p[j];
    }
      }
      t->Fill();
    }
  }
  catch(...) {
    return false;
  }
 
  TFile f("SimEndTofQElecPar.root","RECREATE");
  t->Write();
  f.Close();
  delete t;
  t=NULL;
  return true;
}
 
 
bool join() {
  TFile f1("EndTofQElecPar.root");
  TTree* t11=(TTree*)f1.Get("QElecEndPar");
  if(0==t11) { std::cerr<<" can't get tree :QElecEndPar in file EndTofQElecPar.root"<<std::endl; return false;}
  TTree* t12=(TTree*)f1.Get("EndBoardNum");
  if(0==t12) { std::cerr<<" can't get tree :EndBoardNum in file EndTofQElecPar.root"<<std::endl; return false;}
 
  TFile f2("BarTofQElecPar.root");
  TTree* t21=(TTree*)f2.Get("QElecBarParEast");
  if(0==t21) { std::cerr<<" can't get tree :QElecBarParEast in file BarTofQElecPar.root"<<std::endl; return false;}
  TTree* t22=(TTree*)f2.Get("QElecBarParWest");
  if(0==t22) { std::cerr<<" can't get tree :QElecBarParWest in file BarTofQElecPar.root"<<std::endl; return false;}
  TTree* t23=(TTree*)f2.Get("BarBoardNum");
  if(0==t23) { std::cerr<<" can't get tree :BarBoardNum in file BarTofQElecPar.root"<<std::endl; return false;}
 
  TFile f3("SimEndTofQElecPar.root");
  TTree* t31=(TTree*)f3.Get("SimQElecEndPar");
  if(0==t31) { std::cerr<<" can't get tree :SimQElecEndPar in file SimEndTofQElecPar.root"<<std::endl; return false;}
 
  TFile f4("SimBarTofQElecPar.root");
  TTree* t41=(TTree*)f4.Get("SimQElecBarParEast");
  if(0==t41) { std::cerr<<" can't get tree :SimQElecBarParEast in file SimBarTofQElecPar.root"<<std::endl; return false;}
  TTree* t42=(TTree*)f4.Get("SimQElecBarParWest");
  if(0==t42) { std::cerr<<" can't get tree :SimQElecBarParWest in file SimBarTofQElecPar.root"<<std::endl; return false;}
 
  TFile fnew("TofQElecConst.root","RECREATE");
  t11->Write();
  t12->Write();
  t21->Write();
  t22->Write();
  t23->Write();
  t31->Write();
  t41->Write();
  t42->Write();
  f1.Close();
  f2.Close();
  f3.Close();
  f4.Close();
  fnew.Close();
  t11=NULL;
  t12=NULL;
  t21=NULL;
  t22=NULL;
  t23=NULL;
  t31=NULL;
  t41=NULL;
  t42=NULL;
  return true;
}
 
bool main() {
  barrel_conv2root();
  endcap_conv2root();
  sim_barrel_conv2root();
  sim_endcap_conv2root();
  join();
}