prepare_data.cxx File Reference

#include "TCanvas.h"
#include "TFile.h"
#include "TTree.h"
#include "TH1F.h"
#include <sstream>
#include <iostream>
#include <string>
#include <cstring>
#include <vector>
#include <cmath>
#include "TCut.h"

Go to the source code of this file.

Macros
#define	mass_e   0.511e-3
#define	mass_mu   105.658e-3
#define	mass_pi   139.570e-3
#define	mass_k   493.677e-3
#define	mass_p   938.272e-3

Functions
void	collect (TString filename, int type, TTree *tree)
void	setregion ()
int	main ()

Variables
float	ptrk
float	dEdx_meas
float	costheta
float	charge
float	bg
float	mass
int	nhits
float	dEdx_hit [100]

Macro Definition Documentation

mass_e

#define mass_e   0.511e-3

Definition at line 22 of file prepare_data.cxx.

mass_k

#define mass_k   493.677e-3

Definition at line 25 of file prepare_data.cxx.

mass_mu

#define mass_mu   105.658e-3

Definition at line 23 of file prepare_data.cxx.

mass_p

#define mass_p   938.272e-3

Definition at line 26 of file prepare_data.cxx.

mass_pi

#define mass_pi   139.570e-3

Definition at line 24 of file prepare_data.cxx.

Function Documentation

collect()

void collect	(	TString	filename,
		int	type,
		TTree *	tree
	)

Definition at line 66 of file prepare_data.cxx.

                                                  {
  TFile *ff = new TFile(filename);
  TTree* trk = (TTree*)ff->Get("n103");
  trk->SetBranchAddress("dEdx_meas",  &dEdx_meas);
  trk->SetBranchAddress("ptrk",     &ptrk);
  trk->SetBranchAddress("costheta", &costheta);
  trk->SetBranchAddress("charge",    &charge);
  trk->SetBranchAddress("ndedxhit",  &nhits);
  trk->SetBranchAddress("dEdx_hit",   dEdx_hit);
 
  //initialize variables
  switch (type) {
  case 0:
    mass  = mass_p;
    break;
  case 1:
    mass  = mass_k;
    break;
  case 2:
    mass  = mass_pi;
    break;
  case 3:
    mass  = mass_mu;
    break;
  case 4:
    mass  = mass_e;
    break;
  default:
    cout << "Error: no such sample type!" << endl;
  }
 
  int m_ntot = trk->GetEntries();    
  // if(m_ntot>500000) m_ntot = 500000;
  for (int i = 0; i < m_ntot; i++){
    trk->GetEntry(i);
    if (type == 0 && dEdx_meas < 500) continue; // type 0 is proton!
    if (type == 4 && dEdx_meas > 1200) continue; // type 4 is electron!
    if(ptrk <= 0 || ptrk > 2.3) continue;
    if(fabs(costheta) >= 0.93) continue;
    bg = ptrk/mass;
    t->Fill();
  }
  ff->Close();
}// end of collect()

Referenced by main().

main()

int main

(

)

Definition at line 34 of file prepare_data.cxx.

          {
  TFile *f = new TFile("pre_data.root","recreate");
  TTree *t = new TTree("n103","predata");
  t->Branch("dEdx_meas", &dEdx_meas, "dEdx_meas/f");
  t->Branch("costheta", &costheta, "costheta/f");
  t->Branch("charge", &charge, "charge/f");
  t->Branch("bg", &bg, "bg/f");
  t->Branch("nhits", &nhits, "nhits/I");
  t->Branch("dEdx_hit", dEdx_hit, "dEdx_hit[nhits]/f");
 
  // collect all data here
  cout << "begin to collect data" << endl;
  collect("../hadron_track/proton/proton.root", 0, t);
  cout << "proton collected" << endl;
  collect("../hadron_track/kaon/kaon.root", 1, t);
  cout << "kaon collected" << endl;
  collect("../hadron_track/pion/pion.root", 2, t);
  cout << "pion collected" << endl;
  collect("../hadron_track/muon/muon.root", 3, t);
  cout << "muon collected" << endl;
  collect("../hadron_track/electron/electron.root", 4, t);
  cout << "electron collected" << endl;
 
  f->cd();
  t->Write();
  f->Close();
 
  //  setregion();
 
  return 0;
}

setregion()

void setregion

(

)

Definition at line 111 of file prepare_data.cxx.

                { // use greedy algorithm, step by step to check
  TFile *f = new TFile("pre_data.root");
  TTree *t = (TTree*)f->Get("n103");
  bool small_flag(true);
  double step(0.1); // for beta-gamma
  double current(0.0); // current position in beta-gamma
  int accu(0); // accumulated events
  double stop(4600);
  int threshold(1000);
  while(current<stop){
    cout << "now: " << current << endl;
    if(small_flag){
      accu = t->GetEntries(Form("bg>=%f && bg<%f", current, current+step));
      current += step;
      if(accu > threshold){
    cout << "point: " << current << "  events: " << accu << endl;
    small_flag = false;
      }
    }// end of if(small_flag)
    else{
      accu = t->GetEntries(Form("bg>=%f && bg<%f", current, current+step));
      current += step;
      if(accu < threshold){
    cout << "point: " << current << "  events: " << accu << endl;
    small_flag = true;
      }
    }// end of else(small_flag)
  }// end of while
  f->Close();
}

Variable Documentation

bg

float bg

Definition at line 28 of file prepare_data.cxx.

Referenced by HadronPrep::bgCosThetaFits(), HadronPrep::bgFits(), collect(), combineDigits(), combineMdcDigits(), WidgetParameterization::dedxPrediction(), drawmom(), f_group(), HadronCalibration::fitBGCurve(), HadronCalibration::fitSigmaVsNHit(), HadronCalibration::fitSigmaVsSin(), ElectronGenerator::generateEvents(), WidgetGenerator::generateEvents(), histgen(), main(), MixerAlg::mixDigi(), HadronCalibration::plotBGCurve(), HadronCalibration::plotEfficiency(), ElectronCorrection::process(), rebg(), root_to_csv(), CalibData::TofSimConstBase::setBarPMTGain(), TofConst::setBarPMTGain(), and WidgetGenerator::simulateDedx().

charge

float charge

Definition at line 28 of file prepare_data.cxx.

Referenced by Pdt::addParticle(), FastVertexFit::addTrack(), BesTofDigitizerEcV4::StripStruct::avalanche(), HadronPrep::bgFits(), BesTofDigitizerBrV2::BirksLaw(), BesTofDigitizerEcV3::BirksLaw(), KalmanKinematicFit::BuildVirtualParticle(), KinematicFit::BuildVirtualParticle(), VertexFit::BuildVirtualParticle(), MdcUtilitySvc::cellTrackPassed(), collect(), BesEvent::ConstructMdcTrackFromRec(), BesEvent::ConstructMucTrackFromRec(), HoughHitList::continousHit(), EmcDigiCnv::DataObjectToTObject(), MdcDigiCnv::DataObjectToTObject(), MucDigiCnv::DataObjectToTObject(), TofDigiCnv::DataObjectToTObject(), HltRawCnv::DataObjectToTObject(), EvtBsquark::decay(), Dedx_Helix::Dedx_Helix(), CDChargedVisible< H >::defaultKinematicData(), VFHelix::del4MDelA(), Dedx_Helix::del4MDelA(), KalmanFit::Helix::del4MDelA(), Helix::del4MDelA(), Ext_Helix::del4MDelA(), VFHelix::del4MXDelA(), Dedx_Helix::del4MXDelA(), KalmanFit::Helix::del4MXDelA(), Helix::del4MXDelA(), Ext_Helix::del4MXDelA(), VFHelix::delMDelA(), Dedx_Helix::delMDelA(), KalmanFit::Helix::delMDelA(), Helix::delMDelA(), Ext_Helix::delMDelA(), BesTofDigitizerEcV4::Digitize(), TPerfectFinder::doit(), RawDataUtil::EmcChargeChannel(), RawDataUtil::EmcChargeMeasure(), DQADtag::execute(), BhabhaPreSelect::execute(), EventPreSelect::execute(), CosmicGenerator::execute(), EsTimeAlg::execute(), KalFitAlg::execute(), MdcDedxRecon::execute(), MucRecTrkExt::execute(), TrkExtAlg::execute(), FarmMonitorAlg::execute(), Single::execute(), Ext_Helix::Ext_Helix(), TofCheckData::Fill(), DedxCalibCostheta::FillHists(), DedxCalibDocaEAng::FillHists(), DedxCalibEAng::FillHists(), DedxCalibMomentum::FillHists(), DedxCalibTzero::FillHists(), KalFitAlg::fillTds_ip(), SecondVertexFit::Fit(), Pdt::geantId(), DedxCalibEvent::genNtuple(), RecMucTrack::GetMdcExtTrack(), utility::getp4(), KalmanFit::Helix::Helix(), Helix::Helix(), histgen(), HoughMap::HoughMap(), HoughPeak::HoughPeak(), BesTofDigitizerEcV4::StripStruct::initial(), TRawData::initialize(), MdcDedxRecon::kaltrackrec(), Pdt::lookup(), Pdt::lundId(), main(), TTrackManager::makeTds(), RawDataUtil::MdcChargeChannel(), MdcDedxRecon::mdctrackrec(), DTagTool::p4(), DedxCorrecSvc::PathL(), Pdt::pdgId(), BesTofDigitizerEcV4::StripStruct::print(), BesMdcSD::ProcessHits(), BesTofSD::ProcessHits(), KinematicFit::pull(), RootRawEvtReader::readEmc(), Pdt::readMCppTable(), RootRawEvtReader::readMdc(), RootRawEvtReader::readMuc(), RootRawEvtReader::readTof(), McTestAlg::RetrieveTof(), ChargedDReconstruction::saveDpInfo(), DsReconstruction::saveDsInfo(), BesRawDataWriter::SaveEmcDigits(), LambdaCReconstruction::saveLcInfo(), BesRawDataWriter::SaveMdcDigits(), BesRawDataWriter::SaveTofDigits(), Emc2DCrystal::SetCharge(), Mdc2DWire::SetCharge(), Tof2DScin::SetCharge(), BesTofHit::SetCharge(), Event::McTrajectory::setCharge(), Hough2D::setCharge(), Hough3D::setCharge(), HoughTrack::setCharge(), TMcHitTof::SetCharge(), BesGeoTrack::SetCharge(), EmcROOTGeo::SetHits(), MdcROOTGeo::SetHits(), TofROOTGeo::SetHits(), Event::EmcMcHit::setPDGCharge(), BesEmcTruthHit::SetPDGCharge(), BesTruthTrack::SetPDGCharge(), EmcDigiCnv::TObjectToDataObject(), MdcDigiCnv::TObjectToDataObject(), MucDigiCnv::TObjectToDataObject(), TofDigiCnv::TObjectToDataObject(), MdcKalTrackCnv::TObjectToDataObject(), MdcTrackCnv::TObjectToDataObject(), HltRawCnv::TObjectToDataObject(), RecMdcKalTrackCnv::TObjectToDataObject(), RecMdcTrackCnv::TObjectToDataObject(), RawDataUtil::TofChargeChannel(), Track2Helix(), VFHelix::VFHelix(), and KalmanVertexFit::wTrack().

costheta

float costheta

Definition at line 28 of file prepare_data.cxx.

Referenced by AngularSam::amps(), EvtPsi3Sdecay::AngSam(), EvtRexc::angularSampling(), EvtConExc::angularSampling(), EvtNT3::AR1(), EvtNT3::AR2(), EvtNT3::AR3(), EvtConExc::baryon_sampling(), DedxCorrecSvc::CellCorrec(), collect(), KalFitAlg::complete_track(), DedxCorrecSvc::CosthetaCorrec(), EvtAngSam::decay(), EvtAngSam3::decay(), EvtAngSamLab::decay(), EvtAngSamX::decay(), EvtHAngSam3::decay(), EvtJ2BB1::decay(), BesTofDigitizerBrV2::DirectPh(), McCor::execute(), DQA_EMC::execute(), DQA_MUC::execute(), MdcDedxRecon::execute(), SD0Tag::execute(), TofRec::execute(), f_real_work(), DedxCalibCostheta::FillHists(), DedxCalibDocaEAng::FillHists(), DedxCalibEAng::FillHists(), DedxCalibLayerGain::FillHists(), DedxCalibMomentum::FillHists(), DedxCalibRunByRun::FillHists(), DedxCalibTzero::FillHists(), DedxCalibWireGain::FillHists(), DedxSaveWireGain::FillHists(), EvtConExc::findMaxXS(), BesPrimaryGeneratorAction::GeneratePrimaries(), DedxCalibEvent::genNtuple(), DedxCorrecSvc::HadronCorrec(), ElectronCorrection::HadronCorrection(), histgen(), DedxPID::interpolation(), DTagTool::isGoodTrack(), DSemilepAlg::isGoodTrack(), DTagTool::isGoodTrack_loose(), EvtConExc::ISR_ang_integrate(), KalFitAlg::kalman_fitting_anal(), KalFitAlg::kalman_fitting_calib(), KalFitAlg::kalman_fitting_csmalign(), KalFitAlg::kalman_fitting_MdcxReco_Csmc_Sew(), MdcDedxRecon::kaltrackrec(), DedxCorrecSvc::LayerCorrec(), main(), MdcDedxRecon::mdctrackrec(), EvtConExc::meson_sampling(), EvtRexc::meson_sampling(), LocalKaonSelector::operator()(), LocalPionSelector::operator()(), LocalProtonSelector::operator()(), EvtConExc::photonSampling(), DedxCorrecSvc::SaturCorrec(), TofTrack::setExtTrack(), DedxCorrecSvc::StandardCorrec(), DedxCorrecSvc::StandardHitCorrec(), DedxCorrecSvc::StandardTrackCorrec(), t(), EvtConExc::VP_sampling(), and EvtRexc::VP_sampling().

dEdx_hit

float dEdx_hit[100]

Definition at line 30 of file prepare_data.cxx.

Referenced by collect(), DedxCalibEvent::genNtuple(), and main().

dEdx_meas

float dEdx_meas

Definition at line 28 of file prepare_data.cxx.

Referenced by collect(), MdcDedxRecon::execute(), DedxCalibEvent::genNtuple(), histgen(), and main().

mass

float mass

Definition at line 28 of file prepare_data.cxx.

Referenced by collect().

nhits

int nhits

Definition at line 29 of file prepare_data.cxx.

Referenced by MdcSeg::addHits(), collect(), KalFitAlg::complete_track(), RecMdcKalTrackCnv::DataObjectToTObject(), doLineFit(), MdcMergeDups::dumpRecMdcTrack(), MdcTrkRecon::dumpTdsTrack(), DstEventMaker::execute(), TrkReco::execute(), f_cals(), ResiAlign::fillHist(), FillValidRecMdcTrackHisto(), TrkHelixFitter::fit(), ElectronCalibration::fitRunGains(), HadronCalibration::fitSigmaVsNHit(), FTTrack::get_nhits(), KalFitAlg::kalman_fitting_csmalign(), KalFitAlg::kalman_fitting_MdcxReco_Csmc_Sew(), main(), TTrackManager::makeTds(), TrkHotListEmpty::operator=(), MdcPrintSvc::printRecMdcTrack(), ElectronCorrection::process(), DstExtTrack::SetMucKalData(), TRecMdcKalTrack::setNhits(), RecMdcTrack::setNhits(), DstMdcKalTrack::setNhits(), RecMdcKalTrack::setNhits(), WidgetGenerator::simulateReconstruction(), RecMdcDedxCnv::TObjectToDataObject(), RecMdcKalTrackCnv::TObjectToDataObject(), RecMdcTrackCnv::TObjectToDataObject(), and ElectronCalibration::TwoDCorrection().

ptrk

float ptrk

Definition at line 28 of file prepare_data.cxx.

Referenced by TrackPool::AddMissTrack(), TrackPool::AddTrack(), TrackPool::AddTrackVertex(), DedxPID::cal_par(), collect(), DedxPID::CorrDedx(), event_validate(), Gam4pikp::execute(), Ppjrhopi::execute(), PipiJpsi::execute(), Rhopi::execute(), DQADtag::execute(), inclks::execute(), inclkstar::execute(), incllambda::execute(), inclphi::execute(), DQAJpsi2PPbarAlg::execute(), DQAKsKpi::execute(), DQAKsKpiDEDX::execute(), DQAPi2p2::execute(), DQARhopi::execute(), DQASelBhabha::execute(), DQASelDimu::execute(), DQASelHadron::execute(), EmcSelBhaEvent::execute(), CalibEventSelect::execute(), EventPreSelect::execute(), MdcDedxRecon::execute(), rhopi::execute(), Signal::execute(), TestV::execute(), DedxCalibCostheta::FillHists(), DedxCalibDocaEAng::FillHists(), DedxCalibEAng::FillHists(), DedxCalibLayerGain::FillHists(), DedxCalibMomentum::FillHists(), DedxCalibTzero::FillHists(), DedxCalibWireGain::FillHists(), DedxSaveWireGain::FillHists(), DedxCalibEvent::genNtuple(), EmcBhabhaEvent::getDepoMCShowerEnergy_lab(), getP4(), histgen(), MucPID::init(), DTagTool::isMuon(), DedxPID::iterate(), DedxPID::offsetCorr(), Tof1PID::offsetTof1(), Tof2PID::offsetTof2(), TofCPID::offsetTofC(), TofEPID::offsetTofE(), HTrackParameter::p(), DedxPID::particleIDCalculation(), EmcPID::particleIDCalculation(), MucPID::particleIDCalculation(), Tof1PID::particleIDCalculation(), Tof2PID::particleIDCalculation(), TofCorrPID::particleIDCalculation(), TofCPID::particleIDCalculation(), TofEPID::particleIDCalculation(), DedxCalib::set_dEdx(), DedxPID::sigmaCorr(), TofCPID::sigmaTofC(), TofEPID::sigmaTofE(), and HTrackParameter::wTrack().