ecalAnalysisMaterial.cxx

Go to the documentation of this file.

#include "ecalAnalysisMaterial.h"
 
#include "FairRootManager.h"
#include "ShipMCTrack.h"
#include "FairRadLenPoint.h"
 
 
#include "TTree.h"
#include "TClonesArray.h"
#include "TLorentzVector.h"
 
#include <iostream>
 
using std::cout;
using std::cerr;
using std::endl;
using std::list;
 
void ecalAnalysisMaterial::Exec(Option_t* option)
{
  fEvent++;
  if (fVerbose>0)
    Info("Exec", "Event %d.", fEvent);
  InitTree();
  ShipMCTrack* tr;
  FairRadLenPoint* p;
  Int_t i;
  Int_t k;
  Int_t n;
  Double_t z=3535;
  TVector3 newp;
  TVector3 oldp(0.0, 0.0, 0.0);
  TVector3 d;
  if (fMC->GetEntriesFast()!=1)
  {
    cout << "Size of MC points arrays differs from one" << endl;
  }
  tr=(ShipMCTrack*)fMC->At(0);
  fX=tr->GetPx();
  fY=tr->GetPy();
  fZ=tr->GetPz();
  fX*=z/fZ;
  fY*=z/fZ;
  fZ*=z/fZ;
//  if (TMath::Abs(fX)>594||TMath::Abs(fY)>474) return;
//  if (TMath::Abs(fX)<42&&TMath::Abs(fY)<42) return;
  n=fRadLen->GetEntriesFast();
  fX0=0; fIntL=0;
  for(i=0;i<400;i++) fX0z[i]=0;
  for(i=0;i<n;i++)
  {
    p=(FairRadLenPoint*)fRadLen->At(i);
    newp=p->GetPositionOut();
//    cout << d.Mag() << "  " << p->GetRadLength() << " " << newp.Z() << "  " << fX0 << endl;
    if (newp.Z()>3535) break;
    k=newp.Z()/10;
    d=newp-oldp;
//    cout << p->GetLength() << "   " << p->GetRadLength() << " " << fEvent << "    " <<p->GetXOut() << "   " << p->GetYOut() << "  " << p->GetZOut() << endl;
    fX0+=d.Mag()/p->GetRadLength();
    fX0z[k]+=d.Mag()/p->GetRadLength();
    oldp=newp;
  }
  fTree->Fill();
}
 
 
void ecalAnalysisMaterial::InitTree()
{
  if (fTree) return;
  fTree=new TTree("rad","Radiation lenght tree");
  fTree->Branch("ev", &fEvent, "ev/I");
  fTree->Branch("x", &fX, "x/D");
  fTree->Branch("y", &fY, "Y/D");
  fTree->Branch("z", &fZ, "z/D");
  fTree->Branch("x0", &fX0, "x0/D");
  fTree->Branch("x0z", &fX0z, "x0z[400]/D");
  fTree->Branch("intl", &fIntL, "intl/D");
}
 
ecalAnalysisMaterial::ecalAnalysisMaterial()
  : FairTask(),
    fTree(NULL),
    fEvent(0),
    fX(0.),
    fY(0.),
    fZ(0.),
    fX0(0.),
    fIntL(0.),
    fMC(NULL),
    fRadLen(NULL)
{
}
 
ecalAnalysisMaterial::ecalAnalysisMaterial(const char* name, const Int_t iVerbose)
  : FairTask(name, iVerbose),
    fTree(NULL),
    fEvent(0),
    fX(0.),
    fY(0.),
    fZ(0.),
    fX0(0.),
    fIntL(0.),
    fMC(NULL),
    fRadLen(NULL)
{
}
 
InitStatus ecalAnalysisMaterial::Init()
{
  FairRootManager* fManager=FairRootManager::Instance();
  if (!fManager)
  {
    Fatal("Init", "Can't find a Root Manager.");
    return kFATAL;
  }
  fMC=(TClonesArray*)fManager->GetObject("MCTrack");
  if (!fMC)
  {
    Fatal("Init", "Can't find an array of MC tracks.");
    return kFATAL;
  }
  fRadLen=(TClonesArray*)fManager->GetObject("RadLen");
  if (!fRadLen)
  {
    Fatal("Init", "Can't find an array of radiation length points.");
    return kFATAL;
  }
  fTree=NULL;
 
  return kSUCCESS;
}
 
void ecalAnalysisMaterial::Finish()
{
  if (fTree)
    fTree->Write();
}
 
ClassImp(ecalAnalysisMaterial)