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 "TFile.h"
#include "TTree.h"
#include <assert.h>
#include <fstream>
#include <iostream>
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();
}