hadronic/EventSmear.h

Go to the documentation of this file.

 
#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_