TofDataSet.cxx

Go to the documentation of this file.

#include "tofcalgsec/TofDataSet.h"
 
Record::Record() {
  initial();
  return;
}
 
 
Record::~Record() {}
 
 
void Record::initial() {
  m_run     = 0;
  m_event   = -1;
  m_tofid   = -1;
  m_qleft   = -99.0;
  m_qright  = -99.0;
  m_tleft   = -99.0;
  m_tright  = -99.0;
  m_zrhit   = -999.0;
  m_texp    = -99.0;
  m_path    = -99.0;
  m_phi     = -99.0;
  m_theta   = -99.0;
  m_p       = -9.0;
  m_t0      = -99.0;
  m_q0      = -99.0;
  m_hitcase = -1;
  return;
}
 
 
Record::Record( RecBTofCalHit* hit ) {
  initial();
  if( hit ) {
    m_run     = hit->run();
    m_event   = hit->event();
    m_tofid   = hit->mod();
    m_qleft   = hit->adc1();
    m_qright  = hit->adc2();
    m_tleft   = hit->tdc1();
    m_tright  = hit->tdc2();
    m_zrhit   = hit->zHit();
    m_texp    = hit->tpred();
    m_path    = hit->path();
    m_phi     = hit->deltaPhi();
    m_theta   = hit->sinTheta();
    m_p       = hit->p();
    m_hitcase = hit->qual();
  }
  return;
}
 
 
Record::Record( RecETofCalHit* hit ) {
  initial();
  if( hit ) {
    m_run     = hit->run();
    m_event   = hit->event();
    m_tofid   = hit->mod();
    m_qleft   = hit->adc();
    m_tleft   = hit->tdc();
    m_zrhit   = hit->rHit();
    m_texp    = hit->tpred();
    m_path    = hit->path();
    m_phi     = hit->deltaPhi();
    m_theta   = hit->cosTheta();
    m_p       = hit->p();
    m_hitcase = hit->qual();
  }
  return;
}
 
 
Record::Record( const rootRecord& one ) {
  initial();
  if( &one ) {
    m_run     = one.run;
    m_event   = one.event;
    m_tofid   = one.tofid;
    m_qleft   = one.qleft;
    m_tleft   = one.tleft;
    if( one.hitcase >= 0 && one.hitcase <=2 ) {
      m_qright  = one.qright;
      m_tright  = one.tright;
    }
    m_zrhit   = one.zrhit;
    m_texp    = one.texp;
    m_path    = one.path;
    m_phi     = one.phi;
    m_theta   = one.theta;
    m_p       = one.p;
    m_hitcase = one.hitcase;
  }
  return;
}
 
 
void TofDataSet::setData( TTree* t, bool isBarrel ) {
  if( t && t->GetEntries()>0 ) {
    rootRecord item;
    t->SetBranchAddress( "run",     &item.run     );
    t->SetBranchAddress( "event",   &item.event   );
    t->SetBranchAddress( "tofid",   &item.tofid   );
    t->SetBranchAddress( "qleft",   &item.qleft   );
    t->SetBranchAddress( "qright",  &item.qright  );
    t->SetBranchAddress( "tleft",   &item.tleft   );
    t->SetBranchAddress( "tright",  &item.tright  );
    t->SetBranchAddress( "zrhit",   &item.zrhit   );
    t->SetBranchAddress( "texp",    &item.texp    );
    t->SetBranchAddress( "path",    &item.path    );
    t->SetBranchAddress( "phi",     &item.phi     );
    t->SetBranchAddress( "theta",   &item.theta   );
    t->SetBranchAddress( "p",       &item.p       );
    t->SetBranchAddress( "hitcase", &item.hitcase );
 
    for( unsigned int i=0; i<t->GetEntries(); i++ ) {
      t->GetEntry(i);
      if( isBarrel && item.hitcase>=0 && item.hitcase<=2 ) {
    Record *r = new Record( item );
    if( r->cutBarrel() ) {
      unsigned int tofID = item.tofid;
      barrelData[tofID]->push_back(r);
    }
    else {
      delete r;
    }
      }
      else if( !isBarrel && ( item.hitcase==3 || item.hitcase==4 ) ) {
    Record *r = new Record( item );
    if( r->cutEndcap() ) {
      unsigned int tofID = item.tofid;
      endcapData[tofID]->push_back(r);
    }
    else {
      delete r;
    }
      }
    }
  }
  else {
    std::cerr << "Error: a invalid tree or a blank tree, When converting a tree to TofDataSet,exit" << std::endl;
  }
 
  return;
}
 
 
TofDataSet::TofDataSet() {
  for( unsigned int i=0; i<NBarrel; i++ ) {
    barrelData[i] = new RecordSet;
  }
  for( unsigned int i=0; i<NEndcap; i++ ) {
    endcapData[i] = new RecordSet;
  }
  return;
}
 
 
TofDataSet::~TofDataSet() {
  for( unsigned int i=0; i<NBarrel; i++ ) {
    barrelData[i]->clear();
    delete barrelData[i];
  }
  for( unsigned int i=0; i<NEndcap; i++ ) {
    endcapData[i]->clear();
    delete endcapData[i];
  }
  return;
}
 
 
void TofDataSet::setBarrelDataFiles( std::vector<std::string>& barrelFiles ) {
  TChain* data_barrel = new TChain("btrk");
  if( !data_barrel ) {
    std::cerr << " tofcalgsec Error Msg: creating a tree[barrel] fails in TofDataSet()"<<std::endl;
    throw "Error Msg: creating a tree fails in TofDataSet() ";
  }
  std::cout<<"begin reading barrel data file ... "<<std::endl;
  try{
    for( std::vector<std::string>::iterator it=barrelFiles.begin(); it!=barrelFiles.end(); it++ ) {
      std::cout << " Add file : " << (*it) << std::endl;
      data_barrel->Add( (*it).c_str() );
    }
  }
  catch(...){
    std::cerr << "tofcalgsec Error Msg : in TofDataSet::setDataFiles(std::vector<std::string>&) " << std::endl;
    return;
  }
  setData( data_barrel, true );
  delete data_barrel;
  return;
}
 
 
void TofDataSet::setEndcapDataFiles( std::vector<std::string>& endcapFiles ) {
  TChain* data_endcap = new TChain("etrk");
  if( !data_endcap ) {
    std::cerr << " tofcalgsec Error Msg: creating a tree[endcap] fails in TofDataSet()"<<std::endl;
    throw "Error Msg: creating a tree fails in TofDataSet() ";
  }
  std::cout<<"begin reading endcap data file ... "<<std::endl;
  try{
    for( std::vector<std::string>::iterator it=endcapFiles.begin(); it!=endcapFiles.end(); it++ ) {
      std::cout << " Add file : " << (*it) << std::endl;
      data_endcap->Add( (*it).c_str() );
    }
  }
  catch(...){
    std::cerr << "tofcalgsec Error Msg : in TofDataSet::setDataFiles(std::vector<std::string>&) " << std::endl;
    return;
  }
  setData( data_endcap, false );
  delete data_endcap;
  return;
}
 
 
void TofDataSet::setBarrelData( RecBTofCalHitCol& bhitcol ) {
  RecBTofCalHitCol::iterator iter = bhitcol.begin();
  for( ; iter!=bhitcol.end(); iter++ ) {
    int tofid = (*iter)->mod();
    if( tofid<0 || tofid>175 ) continue;
 
    Record *r = new Record( (*iter) );
    if( r->cutBarrel() ) {
      barrelData[tofid]->push_back(r);
    }
    else {
      delete r;
    }
  }
  return;
}
 
 
void TofDataSet::setEndcapData( RecETofCalHitCol& ehitcol ) {
  RecETofCalHitCol::iterator iter = ehitcol.begin();
  for( ; iter!=ehitcol.end(); iter++ ) {
    int tofid = (*iter)->mod();
    if( tofid<0 || tofid>95 ) continue;
 
    Record *r = new Record( (*iter) );
    if( r->cutEndcap() ) {
      endcapData[tofid]->push_back(r);
    }
    else {
      delete r;
    }
  }
  return;
}