PHPythia6.h

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

#ifndef __PHPYTHIA6_H__
#define __PHPYTHIA6_H__

#include <fun4all/SubsysReco.h>
#include <phhepmc/PHHepMCGenHelper.h>

#include <string>
#include <vector>

class PHCompositeNode;
class PHHepMCGenEvent;
class PHPy6GenTrigger; 

namespace HepMC {
  class GenEvent;
};

class PHPythia6: public SubsysReco {

public:

  PHPythia6(const std::string &name = "PHPythia6");
  virtual ~PHPythia6();

  int Init(PHCompositeNode *topNode);

  int process_event(PHCompositeNode *topNode);

  int ResetEvent(PHCompositeNode *topNode);

  int End(PHCompositeNode *topNode);

  void set_config_file( const std::string cfg_file ) { _configFile = cfg_file; }

  void print_config() const;

  void beam_vertex_parameters(double beamX,
            double beamY,
            double beamZ,
            double beamXsigma,
            double beamYsigma,
            double beamZsigma) {

    set_vertex_distribution_mean(beamX, beamY, beamZ, 0);
    set_vertex_distribution_width(beamXsigma, beamYsigma, beamZsigma, 0);
  }

  void save_ascii( std::string fname = "pythia_hepmc.dat" )
  {
    _save_ascii = true;
    _filename_ascii = fname;
  }

  void register_trigger(PHPy6GenTrigger *theTrigger);
  void set_trigger_OR() { _triggersOR = true; } // default true
  void set_trigger_AND() { _triggersAND = true; }

  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); }
private:

  int ReadConfig(const std::string cfg_file = "");
  int CreateNodeTree(PHCompositeNode *topNode);

  void IntegerTest(double number );

  int _eventcount;
  int _geneventcount;

  // Pythia6 configuration file
  std::string _configFile;

  bool _save_ascii;

  std::string _filename_ascii;

  // event selection
  std::vector<PHPy6GenTrigger*> _registeredTriggers;
  bool _triggersOR;
  bool _triggersAND;
 
  void initPythia();
  PHHepMCGenHelper hepmc_helper;

};

#endif  /* __PHPYTHIA6_H__ */