EventFactory.h

Go to the documentation of this file.

#ifndef INCLUDE_EICSMEAR_ERHIC_EVENTFACTORY_H_
#define INCLUDE_EICSMEAR_ERHIC_EVENTFACTORY_H_

#include <iostream>
#include <memory>
#include <string>

#include <TBranch.h>
#include <TTree.h>

#include "eicsmear/functions.h"
#include "eicsmear/erhic/VirtualEvent.h"
#include "eicsmear/erhic/EventPythia.h"
#include "eicsmear/smear/EventSmear.h"

namespace erhic {

class ParticleMC;

class VirtualEventFactory : public TObject {
 public:
  virtual ~VirtualEventFactory() { }

  virtual VirtualEvent* Create() = 0;

  virtual VirtualEvent* GetEvBufferPtr() { return 0; }

  virtual std::string EventName() const = 0;

  virtual TBranch* Branch(TTree&, const std::string&) {
    return NULL;
  }
  virtual void Fill(TBranch&) { }

  ClassDef(VirtualEventFactory, 1)
};

template<typename T>
class EventFromAsciiFactory : public VirtualEventFactory {
 public:
  EventFromAsciiFactory() { }

  virtual ~EventFromAsciiFactory() { }

  explicit EventFromAsciiFactory(std::istream& is)
  : mInput(&is)
  , mEvent(NULL) {
  }

  virtual T* Create();

  virtual std::string EventName() const;

  std::istream* mInput;  
  std::string mLine;  
  std::auto_ptr<T> mEvent;  

 protected:
  bool AtEndOfEvent() const;

  Int_t FinishEvent();

  bool AddParticle();

  // Warning: explicitly putting the erhic:: namespace before the class
  // name doesn't seen to work for template classes.
  ClassDef(EventFromAsciiFactory, 1)
};

}  // namespace erhic

#endif  // INCLUDE_EICSMEAR_ERHIC_EVENTFACTORY_H_