StBET4pMakerImp.cxx

// $Id: StBET4pMakerImp.cxx,v 1.7 2010/04/24 04:15:35 pibero Exp $
// Copyright (C) 2008 Tai Sakuma <sakuma@bnl.gov>
#include "StBET4pMakerImp.h"

#include "StjTPC.h"
#include "StjBEMC.h"
#include "StjEEMC.h"

#include "StjTrackListCut.h"
#include "StjTowerEnergyListCut.h"

#include "StjAbstractTowerEnergyCorrectionForTracks.h"

#include <iostream>

using namespace std;

StBET4pMakerImp::StBET4pMakerImp(StjTPC* tpc, StjTrackListCut* tpcCut,
                                 StjBEMC* bemc, StjTowerEnergyListCut* bemcCut,
                                 StjAbstractTowerEnergyCorrectionForTracks* correctTowerEnergyForTracks,
                                 StjEEMC* eemc)
  : _tpc(tpc), _bemc(bemc), _eemc(eemc)
  , _tpcCut(tpcCut), _bemcCut(bemcCut)
  , _correctTowerEnergyForTracks(correctTowerEnergyForTracks)
{

}

void StBET4pMakerImp::Init()
{
  _tpc->Init();
  _bemc->Init();
  _eemc->Init();
}

std::pair<StjTrackList, StjTowerEnergyList> StBET4pMakerImp::getTrackAndEnergyList()
{
  StjTrackList trackList = _tpc->getTrackList();
  StjTowerEnergyList bemcEnergyList = _bemc->getEnergyList();
  StjTowerEnergyList eemcEnergyList = _eemc->getEnergyList();

  trackList = (*_tpcCut)(trackList);
  bemcEnergyList = (*_bemcCut)(bemcEnergyList);

  StjTowerEnergyList energyList;
  energyList.insert(energyList.end(),bemcEnergyList.begin(),bemcEnergyList.end());
  energyList.insert(energyList.end(),eemcEnergyList.begin(),eemcEnergyList.end());

  energyList = (*_correctTowerEnergyForTracks)(energyList,trackList);

  return make_pair(trackList,energyList);
}