TMBTrackCalJet.cpp

Go to the documentation of this file.

//                                                                      //
//  TMBTrackCalJet                                                      //
//                                                                      //
//  Ariel Schwartzman
//                                                                      //

#include "tmb_tree/TMBTrackCalJet.hpp"
#include "tmb_tree/TMBTrack.hpp"
#include "tmb_tree/TMBVertex.hpp"

ClassImp(TMBTrackCalJet);



TMBTrackCalJet::TMBTrackCalJet() 
{
  //
}

TMBTrackCalJet::TMBTrackCalJet(TRef jet, TRefArray* trackCalList, 
                               Float_t C, Float_t Cin, 
                               Float_t Cout, Float_t Csub, 
                               std::vector<Float_t> Efraction, 
                               std::vector<Int_t> Ncells) : 
  _q(999.9)
{
  SetPxPyPzE(999.9, 999.9, 999.9, 999.9);
  
  _jet     = jet;
  
  _trackCalList.Clear();
  if (trackCalList) {
    for (Int_t i=0; i<trackCalList->GetLast()+1; i++) {
      _trackCalList.Add( trackCalList->At(i) );
    }
  }
  
  _Corr = C;
  _Cin  = Cin;
  _Cout = Cout;
  _Csub = Csub;

  _Efraction.clear();
  for(Int_t i=0; i < (Int_t )Efraction.size(); i++) {
    _Efraction.push_back(Efraction[i]);
  }
  
  _Ncells.clear();
  for(Int_t i=0; i < (Int_t )Ncells.size(); i++) {
    _Ncells.push_back(Ncells[i]);
  }


  TMBJet* Jet = (TMBJet*) this->GetJet();
  if(Jet) {
    Jet->ActAsUnCorrected();
    SetPxPyPzE(Jet->Px()*_Corr, 
               Jet->Py()*_Corr, 
               Jet->Pz()*_Corr,
               Jet->E()*_Corr);
    _q  = Jet->charge();
  } else {
    std::cout << "TMBTrackCalJet::ERROR no matching TMBJet" << std::endl;
  }

}


TMBTrackCalJet::~TMBTrackCalJet() 
{
  //
}

/*
void TMBTrackCalJet::MatchTracks(float& Ein, int& ntrkin, 
                                 float& Eout, int& ntrkout, 
                                 float track_cone_size,
                                 float ptMin, float ptMax, 
                                 float dcaRmax, float dcaZmax,
                                 int cftMin, int smtMin)
{
  Ein = 0; ntrkin = 0;
  Eout = 0; ntrkout = 0;

  int np=_trackCalList.GetLast()+1;
  for(int i=0; i<np; ++i){
    TMBTrackCal* tmbtrkcal = (TMBTrackCal*) this->GetTrackCal(i);
    TMBTrack*  tmbtrk = (TMBTrack*) tmbtrkcal->GetChargedTrack();
    TMBVertex* tmbvtx = (TMBVertex*) tmbtrkcal->GetVertex();

    float pT = tmbtrk->Pt();
    float ztrk  = tmbtrk->trpars(1);
    int smt = tmbtrk->nsmt();
    int cft = tmbtrk->ncft();
    float dcaZ = tmbtrkcal->getDCAz();
    float dcaR = tmbtrkcal->getDCAr();
    float DR_PV = this->GetDR(i);
    float EoverP = tmbtrkcal->getE015(15)/E;
    float expectedTrackCalEnergy = tmbtrkcal->SinglePionResponse();

    if(Vertex==0 && pT>ptMin && pT<ptMax && 
       dcaR<dcaRmax && dcaZ<dcaZmax && 
       DR_PV<track_cone_size && cft>=cftMin && smt>=smtMin) {
      
      //in
      if(_Efraction[i]>0) {
        ntrkin ++;
        if(EoverP>0.5) Ein += E - expectedTrackCalEnergy;
        else Ein += p;
      } else {
        //if(DR_CH>0.5) {
        ntrkout++;
        Eout += E;
        //}
      }
    }
  }

}
*/


void TMBTrackCalJet::print(std::ostream& os) const
{
  TMBJet* tmbjet = (TMBJet*) this->GetJet();

  os << std::endl;
  os << "TrackCalJet::print" << std::endl;
  os << "pT;  eta, phi = " << tmbjet->Pt() << ";   ";
  os << tmbjet->Eta() << ", " << tmbjet->Phi() << std::endl;
   
  int np=_trackCalList.GetLast()+1;
  os << np << " tracks: ";
  os << std::endl;
  os << " properties: " << std::endl;
  for(int i=0; i<np; ++i){
    TMBTrackCal* tmbtrkcal = (TMBTrackCal*) this->GetTrackCal(i);
    TMBTrack*  tmbtrk = (TMBTrack*) tmbtrkcal->GetChargedTrack();
    TMBVertex* tmbvtx = (TMBVertex*) tmbtrkcal->GetVertex();

    float pt = tmbtrk->Pt();
    float p  = tmbtrk->P();
    float Deta  = tmbtrkcal->PtWeighteDetectorEta();
    float ztrk  = tmbtrk->trpars(1);
    int smt = tmbtrk->nsmt();
    int cft = tmbtrk->ncft();

    os << "   pT = " << pt << ", P = " << p << ", deta = " << Deta << ", ztrk = "
       << ztrk << std::endl;
    os << "   phi = " << tmbtrk->Phi() << std::endl;
    os << "   smt = " << smt << ", cft = " << cft << std::endl;
    os << "   Etrackcal = " << tmbtrkcal->getE015(15) << std::endl;
    os << "   Efraction = " << _Efraction[i] << std::endl;
    os << "   Ncells    = " << _Ncells[i] << std::endl;
    os << "   DR        = " << this->GetDR(i) << std::endl;
    os << "   dr1       = " << this->GetDR(i,1) << std::endl;
    os << "   dr15      = " << this->GetDR(i,15) << std::endl;
    os << "   vtx Z     = " << tmbvtx->vz() << std::endl;
    os << "   dca(Z,R)  = " << tmbtrkcal->getDCAz() << ", ";
    os << tmbtrkcal->getDCAr() << std::endl;
    os << std::endl;
  }
  os << std::endl;
  os << "Correction factor = " << _Corr << std::endl;
  os << "Corrected jet pT: "<< std::endl;
  os << "    in       = " << this->Pt()/_Corr*_Cin << std::endl;
  os << "    out      = " << this->Pt()/_Corr*_Cout << std::endl;
  os << "    full     = " << this->Pt()/_Corr*_Corr << std::endl;
  os << "    subtract = " << this->Pt()/_Corr*_Csub << std::endl;
  os << std::endl;
  os << "---------------------------------------------------" << std::endl;
  os << std::endl;
}

---