EventSmear.h

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

#ifndef INCLUDE_EICSMEAR_HADRONIC_EVENTSMEAR_H_
#define INCLUDE_EICSMEAR_HADRONIC_EVENTSMEAR_H_

#include <vector>

#include <Rtypes.h>  // For ClassDef macro
#include <TBranch.h>

#include "eicsmear/erhic/VirtualEvent.h"
#include "eicsmear/hadronic/EventMC.h"
#include "eicsmear/smear/EventFactory.h"
#include "eicsmear/smear/ParticleMCS.h"

namespace erhic {
namespace hadronic {

class EventSmear : public VirtualEvent {
 public:
  virtual ~EventSmear();

  EventSmear();

  virtual const Smear::ParticleMCS* GetTrack(UInt_t) const;

  virtual Smear::ParticleMCS* GetTrack(UInt_t);

  virtual UInt_t GetNTracks() const;

  virtual void AddLast(Smear::ParticleMCS*);

 protected:
  std::vector<Smear::ParticleMCS*> particles;

  ClassDef(erhic::hadronic::EventSmear, 1)
};

}  // namespace hadronic
}  // namespace erhic

namespace Smear {

class HadronicEventBuilder : public EventFactory<erhic::hadronic::EventSmear> {
 public:
  virtual ~HadronicEventBuilder() { }

  HadronicEventBuilder(const Detector& d, TBranch& mcBranch)
  : mDetector(d)
  , mMcEvent(NULL) {
    mcBranch.SetAddress(&mMcEvent);
  }

  virtual EventType* Create() {
    EventType* event = new EventType;
    for (unsigned j(0); j < mMcEvent->GetNTracks(); j++) {
      const erhic::VirtualParticle* ptr = mMcEvent->GetTrack(j);
      if (ptr) {
        event->AddLast(mDetector.Smear(*ptr));
      }  // if
    }  // for
    return event;
  }

 protected:
  Detector mDetector;
  erhic::hadronic::EventMC* mMcEvent;
};

}  // namespace Smear

#endif  // INCLUDE_EICSMEAR_HADRONIC_EVENTSMEAR_H_