EventFactory.h

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

#ifndef INCLUDE_EICSMEAR_ERHIC_EVENTFACTORY_H_
#define INCLUDE_EICSMEAR_ERHIC_EVENTFACTORY_H_

#include <iostream>
#include <memory>
#include <string>
#include <vector>
#include <map>

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

#include "eicsmear/functions.h"
#include "eicsmear/erhic/VirtualEvent.h"
#include "eicsmear/erhic/EventHepMC.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 void FindFirstEvent() {}

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

  typedef std::pair <TString, TObject*> NamedObjects;
  std::vector<NamedObjects> mObjectsToWriteAtTheEnd;

  std::map<std::string, std::string> mAdditionalInformation;

  ClassDef(VirtualEventFactory, 3)
};

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

  virtual ~EventFromAsciiFactory() { }

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

  virtual T* Create();

  virtual std::string EventName() const;

  virtual void FindFirstEvent();

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

  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, 2)
};

}  // namespace erhic

#endif  // INCLUDE_EICSMEAR_ERHIC_EVENTFACTORY_H_