HardProcess.h

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

/*******************************************************************************
 * Copyright (c) The JETSCAPE Collaboration, 2018
 *
 * Modular, task-based framework for simulating all aspects of heavy-ion collisions
 * 
 * For the list of contributors see AUTHORS.
 *
 * Report issues at https://github.com/JETSCAPE/JETSCAPE/issues
 *
 * or via email to bugs.jetscape@gmail.com
 *
 * Distributed under the GNU General Public License 3.0 (GPLv3 or later).
 * See COPYING for details.
 ******************************************************************************/

#ifndef HARDPROCESS_H
#define HARDPROCESS_H

#include "InitialState.h"
#include "JetScapeModuleBase.h"
#include "JetClass.h"
#include <vector>

namespace Jetscape {

class HardProcess : public JetScapeModuleBase {

public:
  HardProcess();

  virtual ~HardProcess();

  virtual void Init();

  virtual void Exec();

  virtual void Clear();

  virtual void WriteTask(weak_ptr<JetScapeWriter> w);

  virtual void CollectHeader(weak_ptr<JetScapeWriter> w);

  // connect the InitialState module with hard process
  std::shared_ptr<InitialState> ini;

  int GetNHardPartons() { return hp_list.size(); }

  shared_ptr<Parton> GetPartonAt(int i) { return hp_list[i]; }

  vector<shared_ptr<Parton>> &GetPartonList() { return hp_list; }

  void AddParton(shared_ptr<Parton> p) { hp_list.push_back(p); }

  // Slots ...
  void GetHardPartonList(vector<shared_ptr<Parton>> &plist) { plist = hp_list; }

  virtual double GetSigmaGen() { return 1; };
  virtual double GetSigmaErr() { return 0; };

  virtual double GetPtHat() { return 0; };

  virtual double GetEventWeight() { return 1; };

  void AddHadron(shared_ptr<Hadron> h) { hd_list.push_back(h); }

  // Slots ...
  void GetHadronList(vector<shared_ptr<Hadron>> &hlist) { hlist = hd_list; }

  vector<shared_ptr<Hadron>> &GetHadronList() { return hd_list; }

  int GetNHadrons() { return hd_list.size(); }

    std::string printer;
    
private:
  // Think of always using unique_ptr for any vector in jetscape framework !???
  // To be discussed ...
  vector<shared_ptr<Parton>> hp_list;

  // A vector of Hadrons generated by Pythia
  vector<shared_ptr<Hadron>> hd_list;
    

};

} // end namespace Jetscape

#endif