ecalAnalysisSimple.cxx

Go to the documentation of this file.

#include "ecalAnalysisSimple.h"
 
#include "FairRootManager.h"
 
#include "ecalStructure.h"
#include "ecalCell.h"
#include "ecalPoint.h"
 
#include "TTree.h"
#include "TClonesArray.h"
 
#include <iostream>
#include <fstream>
#include <list>
 
using namespace std;
 
void ecalAnalysisSimple::Exec(Option_t* option)
{
  Int_t n=fTracks->GetEntries();
  Int_t i;
  ecalPoint* t;
  ecalCell* cell;
  ecalCell* mcell;
  TVector3 m;
  list<ecalCell*> cells;
  list<ecalCell*>::const_iterator p;
//  TVector3 m1;
 
  fEv++;
  InitTree();
 
  for(i=0;i<n;i++)
  {
    t=(ecalPoint*)fTracks->At(i);
    fX=t->GetX();
    fY=t->GetY();
    t->Momentum(m);
    fP=m.Mag();
    fPX=m.Px();
    fPY=m.Py();
    fPZ=m.Pz();
 
//    m1=m.Unit();
    cell=fStr->GetCell(fX, fY);
    if (!cell) continue;
    mcell=cell;
    cell->GetNeighborsList(cells);
    for(p=cells.begin();p!=cells.end();++p)
      if ((*p)->GetEnergy()>mcell->GetEnergy())
    mcell=(*p);
 
    mcell->GetNeighborsList(cells);
    for(p=cells.begin();p!=cells.end();++p)
      if ((*p)->GetEnergy()>mcell->GetEnergy())
    break;
 
    if (p!=cells.end()) continue;
 
    fCX=mcell->GetCenterX();
    fCY=mcell->GetCenterY();
    fCE=mcell->GetEnergy();
    fCellNum=mcell->GetCellNumber();
    fADC=mcell->ADC();
    fOE=fCE;
    for(p=cells.begin();p!=cells.end();++p)
      fOE+=(*p)->GetEnergy();
 
    fTree->Fill();
  }
 
}
 
void ecalAnalysisSimple::InitTree()
{
  if (fTree) return;
  fTree=new TTree("calib", "calib");
  fTree->Branch("px", &fPX, "px/D");
  fTree->Branch("py", &fPY, "py/D");
  fTree->Branch("pz", &fPZ, "pz/D");
  fTree->Branch("p" , &fP , "p/D");
  fTree->Branch("x" , &fX , "x/D");
  fTree->Branch("y" , &fY , "y/D");
  fTree->Branch("cx", &fCX, "cx/D");
  fTree->Branch("cy", &fCY, "cy/D");
  fTree->Branch("ce", &fCE, "ce/D");
  fTree->Branch("oe", &fOE, "oe/D");
  fTree->Branch("ev", &fEv, "ev/I");
  fTree->Branch("cn", &fCellNum, "cn/I");
  fTree->Branch("adc", &fADC, "adc/I");
}
 
ecalAnalysisSimple::ecalAnalysisSimple(const char* name, const Int_t iVerbose)
  : FairTask(name, iVerbose),
    fTree(NULL),
    fX(0.),
    fY(0.),
    fCX(0.),
    fCY(0.),
    fP(0.),
    fCE(0.),
    fOE(0.),
    fPX(0.),
    fPY(0.),
    fPZ(0.),
    fEv(0),
    fCellNum(0),
    fADC(0),
    fStr(NULL),
    fTracks(NULL)
{
}
 
ecalAnalysisSimple::ecalAnalysisSimple()
  : FairTask(),
    fTree(NULL),
    fX(0.),
    fY(0.),
    fCX(0.),
    fCY(0.),
    fP(0.),
    fCE(0.),
    fOE(0.),
    fPX(0.),
    fPY(0.),
    fPZ(0.),
    fEv(0),
    fCellNum(0),
    fADC(0),
    fStr(NULL),
    fTracks(NULL)
{
}
 
InitStatus ecalAnalysisSimple::Init()
{
  FairRootManager* fManager=FairRootManager::Instance();
  fStr=(ecalStructure*)fManager->GetObject("EcalStructure");
  if (!fStr)
  {
    Fatal("Init()", "Can't find calorimeter structure. ");
    return kFATAL;
  }
  fTracks=(TClonesArray*)fManager->GetObject("EcalPoint");
  if (!fTracks)
  {
    Fatal("Init()", "Can't find array of reconstructed tracks. ");
    return kFATAL;
  }
  return kSUCCESS;
}
 
void ecalAnalysisSimple::Finish()
{
  if (fTree)
    fTree->Write();
}
 
ClassImp(ecalAnalysisSimple)