Fun4AllOscarInputManager.h

Go to the documentation of this file.

#ifndef FUN4ALLOSCARINPUTMANAGER_H__
#define FUN4ALLOSCARINPUTMANAGER_H__
#include "PHHepMCGenHelper.h"
 
#include <fun4all/Fun4AllInputManager.h>
 
#include <string>
#include <map>
#include <fstream>
#include <iostream>
 
// forward declaration of classes in namespace
namespace HepMC
{
    class IO_GenEvent;
    class GenEvent;
};
 
class PHCompositeNode;
class PHHepMCGenEvent;
 
 
class Fun4AllOscarInputManager : public Fun4AllInputManager
{
 public:
  Fun4AllOscarInputManager(const std::string &name = "DUMMY", const std::string &topnodename = "TOP");
  virtual ~Fun4AllOscarInputManager();
  int fileopen(const std::string &filenam);
  int fileclose();
  int run(const int nevents = 0);
  int isOpen() {return isopen;}
  void Print(const std::string &what = "ALL") const;
  int ResetEvent();
  int PushBackEvents(const int i);
  int skip(const int i){ return PushBackEvents(i);}
 
  // Effectivly turn off the synchronization checking
  //
  int SyncIt(const SyncObject* /*mastersync*/) {return Fun4AllReturnCodes::SYNC_OK;}
  int GetSyncObject(SyncObject** /*mastersync*/) {return Fun4AllReturnCodes::SYNC_NOOBJECT;}
  int NoSyncPushBackEvents(const int nevt) {return PushBackEvents(nevt);}
  int ConvertFromOscar();
 
 
  void set_vertex_distribution_function(PHHepMCGenHelper::VTXFUNC x, PHHepMCGenHelper::VTXFUNC y, PHHepMCGenHelper::VTXFUNC z, PHHepMCGenHelper::VTXFUNC t)
  {
    hepmc_helper.set_vertex_distribution_function(x, y, z, t);
  }
 
  void set_vertex_distribution_mean(const double x, const double y, const double z, const double t)
  {
    hepmc_helper.set_vertex_distribution_mean(x, y, z, t);
  }
 
  void set_vertex_distribution_width(const double x, const double y, const double z, const double t)
  {
    hepmc_helper.set_vertex_distribution_width(x, y, z, t);
  }
  //
  void set_reuse_vertex(int src_embedding_id)
  {
    hepmc_helper.set_reuse_vertex(src_embedding_id);
  }
 
  int get_embedding_id() const { return hepmc_helper.get_embedding_id(); }
  //
  void set_embedding_id(int id) { hepmc_helper.set_embedding_id(id); }
 protected:
  int OpenNextFile();
  int isopen;
  int events_total;
  int events_thisfile;
  std::string filename;
  std::string topNodeName;
  PHCompositeNode *topNode;
  HepMC::GenEvent *evt;
  //HepMC::GenEvent *tmpEvt;
  int skipEvents, skippedEvents;
 
  // some pointers for use in decompression handling
  std::ifstream *filestream; // holds compressed filestream
  std::istream *unzipstream; // feed into HepMc
  std::ifstream theOscarFile;
 
  PHHepMCGenHelper hepmc_helper;
  bool isCompressed;
};
 
#endif /* __FUN4ALLOSCARINPUTMANAGER_H__ */