FairRadLenManager.cxx

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file FairRadLenManager.cxx

// -------------------------------------------------------------------------
// -----                      FairRadLenManager source file                   -----
// -----                  Created 14/01/08  by M. Al-Turany            -----
// -------------------------------------------------------------------------


#include <iostream>
#include "FairRadLenPoint.h"
#include "FairRadLenManager.h"
#include "FairRootManager.h"
#include "TLorentzVector.h"
#include "TParticle.h"
#include "TVirtualMC.h"

using namespace std;

ClassImp(FairRadLenManager)

FairRadLenManager* FairRadLenManager::fgInstance = NULL;

FairRadLenManager* FairRadLenManager::Instance()
{
  return fgInstance;
}

FairRadLenManager::FairRadLenManager()
  : fPointCollection(new TClonesArray("FairRadLenPoint")),
    fTrackID(0),
    fVolumeID(0),
    fPosIn(TLorentzVector(0,0,0,0)),
    fPosOut(TLorentzVector(0,0,0,0)),
    fMomIn(TLorentzVector(0,0,0,0)),
    fMomOut(TLorentzVector(0,0,0,0)),
    fTime(0),
    fLength(0),
    fELoss(0),
    fA(0),
    fZmat(0),
    fDensity(0),
    fRadl(0),
    fAbsl(0)
{
  if(NULL == fgInstance) {
    fgInstance = this;
    //    fPointCollection=new TClonesArray("FairRadLenPoint");
  }
}

FairRadLenManager::~FairRadLenManager()
{
  fgInstance = NULL;
  fPointCollection->Delete();
  delete fPointCollection;
}

void FairRadLenManager::Init()
{
  FairRootManager::Instance()->Register("RadLen","RadLenPoint", fPointCollection, kTRUE);
}

void FairRadLenManager::Reset()
{
  fPointCollection->Delete();
}

void FairRadLenManager::AddPoint(Int_t& ModuleId)
{
  if ( gMC->IsTrackEntering() ) {
    fELoss  = 0.;
    Int_t copyNo;
    fVolumeID = gMC->CurrentVolID(copyNo);
    fTime   = gMC->TrackTime() * 1.0e09;
    fLength = gMC->TrackLength();
    gMC->TrackPosition(fPosIn);
    gMC->TrackMomentum(fMomIn);
    //    Int_t MatId=  gMC->CurrentMaterial(fA, fZmat, fDensity, fRadl, fAbsl);
    gMC->CurrentMaterial(fA, fZmat, fDensity, fRadl, fAbsl);
  }
  fELoss += gMC->Edep();
  if ( gMC->IsTrackExiting()    ||
       gMC->IsTrackStop()       ||
       gMC->IsTrackDisappeared()   ) {
    FairRadLenPoint* p=0;
    fTrackID  = gMC->GetStack()->GetCurrentTrackNumber();
    gMC->TrackPosition(fPosOut);
    gMC->TrackMomentum(fMomOut);
    TClonesArray& clref = *fPointCollection;
    Int_t tsize = clref.GetEntriesFast();
    p=new(clref[tsize]) FairRadLenPoint(fTrackID, ModuleId,
                                        TVector3(fPosIn.X(),fPosIn.Y(),fPosIn.Z()),
                                        TVector3(fMomIn.X(),fMomIn.Y(),fMomIn.Z()),
                                        fTime,  fLength, fELoss,
                                        TVector3(fPosOut.X(),fPosOut.Y(),fPosOut.Z()),
                                        TVector3(fMomOut.X(),fMomOut.Y(),fMomOut.Z()),
                                        fA, fZmat, fDensity, fRadl);
  }

}