StiScanTreeMaker.cxx

Go to the documentation of this file.

#include "StiScan/StiScanTreeMaker.h"

#include "TBranch.h"

#include "StEvent/StEnumerations.h"
#include "Sti/StiToolkit.h"
#include "Sti/StiTrackContainer.h"
#include "StiMaker/StiMaker.h"

#include "StiScan/StiScanEvent.h"

ClassImp(StiScanTreeMaker)


StiScanTreeMaker::StiScanTreeMaker(const std::string &name) : StiTreeMaker(name, "stiscan")
{
}


void StiScanTreeMaker::SetEventTree()
{
   fEvent = new StiScanEvent();
   fTree->Branch("e.", "StiScanEvent", &fEvent, 64000, 99);
}


Int_t StiScanTreeMaker::Make()
{
   // Fill event with information from Sti tracks
   StiMaker* stiMaker = (StiMaker*) GetMaker("Sti");
   assert(stiMaker);

   StiToolkit *stiToolkit = stiMaker->getToolkit();
   StiTrackContainer *stiTrackContainer = stiToolkit->getTrackContainer();

   EReturnCodes retCode = fEvent->Fill(*stiTrackContainer);

   fTree->Fill();

   return retCode;
}