Afterburner.h

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file Afterburner.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.
 ******************************************************************************/
// This is a general basic class for hadronic afterburner

#ifndef AFTERBURNER_H
#define AFTERBURNER_H

#include "JetScapeModuleBase.h"
#include "SoftParticlization.h"
#include "sigslot.h"

namespace Jetscape {

class Afterburner : public JetScapeModuleBase {
public:
  Afterburner() {
    VERBOSE(8);
    SetId("Afterburner");
  }

  ~Afterburner() {
    VERBOSE(8);
    disconnect_all();
  }

  virtual void Init();
  virtual void Exec();

protected:
  std::vector<std::vector<std::shared_ptr<Hadron>>> GatherAfterburnerHadrons();
  std::vector<std::vector<std::shared_ptr<Hadron>>> GetSoftParticlizationHadrons();
  std::vector<std::shared_ptr<Hadron>> GetFragmentationHadrons();

  std::vector<std::vector<std::shared_ptr<Hadron>>> dummy;
  std::uniform_real_distribution<double> ZeroOneDistribution;
  // rng for the Kaon-L / Kaon-S switch to K0 / Anti-K0
  std::shared_ptr<std::uniform_int_distribution<int>> rand_int_ptr_;
};

} // end namespace Jetscape

#endif // AFTERBURNER_H