DedxSaveWireGain.cxx

Go to the documentation of this file.

#include "GaudiKernel/MsgStream.h"
 
#include <sstream>
#include <string>
#include "TTree.h"
#include "TH1F.h"
#include "TF1.h"
#include "TFile.h"
#include "TStyle.h"
#include "TCanvas.h"
 
#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()"<<endreq;
 
    ReadRecFileList();
}
 
void DedxSaveWireGain::FillHists()
{
    MsgStream log(msgSvc(), name());
    log<<MSG::INFO<<"DedxSaveWireGain::FillHists()"<<endreq;
    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();
}