DedxSaveWireGain.cxx

Go to the documentation of this file.

#include "GaudiKernel/MsgStream.h"
 
#include "TCanvas.h"
#include "TF1.h"
#include "TFile.h"
#include "TH1F.h"
#include "TStyle.h"
#include "TTree.h"
#include <sstream>
#include <string>
 
#include "DedxCalibAlg/DedxSaveWireGain.h"
DECLARE_COMPONENT( DedxSaveWireGain )
using namespace std;
 
DedxSaveWireGain::DedxSaveWireGain( const std::string& name, ISvcLocator* pSvcLocator )
    : DedxCalib( name, pSvcLocator ) {
  declareProperty( "PMin", pMin = 0.2 );
  declareProperty( "PMax", pMax = 2.3 );
  f_save     = new TFile( "wiregain.root", "recreate" );
  m_wiregain = new TTree( "wiregain", "wiregain" );
}
 
void DedxSaveWireGain::BookHists() {
  MsgStream log( msgSvc(), name() );
  log << MSG::INFO << "DedxSaveWireGain::BookHists()" << endmsg;
 
  ReadRecFileList();
}
 
void DedxSaveWireGain::FillHists() {
  MsgStream log( msgSvc(), name() );
  log << MSG::INFO << "DedxSaveWireGain::FillHists()" << endmsg;
  int    run, evt;
  double gain;
  m_wiregain->Branch( "run", &run, "run/I" );
  m_wiregain->Branch( "evt", &evt, "evt/I" );
  m_wiregain->Branch( "gain", &gain, "gain/D" );
 
  TFile* f;
  TTree* n102;
  string runlist;
 
  double dedx  = 0;
  float  runNO = 0, evtNO = 0, runFlag = 0, pathlength = 0, wid = 0, layid = 0, dd_in = 0,
        driftdist = 0, eangle = 0, zhit = 0, costheta = 0, tes = 0, ptrk = 0;
  cout << "m_recFileVector.size()= " << m_recFileVector.size() << endl;
  for ( int i = 0; i < m_recFileVector.size(); i++ )
  {
    runlist = m_recFileVector[i];
    cout << "runlist: " << runlist.c_str() << endl;
    f    = new TFile( runlist.c_str() );
    n102 = (TTree*)f->Get( "n102" );
    n102->SetBranchAddress( "adc_raw", &dedx );
    n102->SetBranchAddress( "path_rphi", &pathlength );
    n102->SetBranchAddress( "wire", &wid );
    n102->SetBranchAddress( "layer", &layid );
    n102->SetBranchAddress( "doca_in", &dd_in );
    n102->SetBranchAddress( "driftdist", &driftdist );
    n102->SetBranchAddress( "eangle", &eangle );
    n102->SetBranchAddress( "zhit", &zhit );
    n102->SetBranchAddress( "runNO", &runNO );
    n102->SetBranchAddress( "evtNO", &evtNO );
    n102->SetBranchAddress( "runFlag", &runFlag );
    n102->SetBranchAddress( "costheta1", &costheta );
    n102->SetBranchAddress( "t01", &tes );
    n102->SetBranchAddress( "ptrk1", &ptrk );
 
    cout << "entries in this file" << n102->GetEntries() << endl;
    for ( int j = 0; j < n102->GetEntries(); j++ )
    {
      n102->GetEntry( j );
      if ( tes > 1400 ) continue;
      if ( ptrk > pMax || ptrk < pMin ) continue;
      if ( layid < 8 )
      {
        if ( dedx < MinADCValuecut || dedx > MaxADCValuecut || pathlength > RPhi_PathMaxCut ||
             pathlength < Iner_RPhi_PathMinCut || driftdist > Iner_DriftDistCut )
          continue;
      }
      else
      {
        if ( dedx < MinADCValuecut || dedx > MaxADCValuecut || pathlength > RPhi_PathMaxCut ||
             pathlength < Out_RPhi_PathMinCut || driftdist > Out_DriftDistCut )
          continue;
      }
      dedx = exsvc->StandardHitCorrec( 0, runFlag, 2, runNO, evtNO, pathlength, wid, layid,
                                       dedx, dd_in, eangle, costheta );
      dedx = exsvc->StandardTrackCorrec( 0, runFlag, 2, runNO, evtNO, dedx, costheta, tes );
      run  = (int)runNO;
      evt  = (int)evtNO;
      gain = dedx;
      // cout << "run " << run << " evt " << evt << " gain " << gain << endl;
      m_wiregain->Fill();
    } // end of loop in n102 entries
  }   // end of loop in rec files
} // end of this function
 
void DedxSaveWireGain::WriteHists() {
  f_save->cd();
  m_wiregain->Write();
  f_save->Save();
  f_save->Close();
}