src/hadronization/ColoredHadronization.h

0001 /*******************************************************************************
0002  * Copyright (c) The JETSCAPE Collaboration, 2018
0003  *
0004  * Modular, task-based framework for simulating all aspects of heavy-ion collisions
0005  *
0006  * For the list of contributors see AUTHORS.
0007  *
0008  * Report issues at https://github.com/JETSCAPE/JETSCAPE/issues
0009  *
0010  * or via email to bugs.jetscape@gmail.com
0011  *
0012  * Distributed under the GNU General Public License 3.0 (GPLv3 or later).
0013  * See COPYING for details.
0014  ******************************************************************************/
0015
0016 #ifndef COLOREDHADRONIZATION_H
0017 #define COLOREDHADRONIZATION_H
0018
0019 #include "HadronizationModule.h"
0020 #include "Pythia8/Pythia.h"
0021
0022 using namespace Jetscape;
0023
0024 class ColoredHadronization : public HadronizationModule<ColoredHadronization> {
0025 public:
0026   ColoredHadronization();
0027   virtual ~ColoredHadronization();
0028
0029   void Init();
0030   void DoHadronization(vector<vector<shared_ptr<Parton>>> &shower,
0031                        vector<shared_ptr<Hadron>> &hOut,
0032                        vector<shared_ptr<Parton>> &pOut);
0033   void WriteTask(weak_ptr<JetScapeWriter> w);
0034
0035 private:
0036   double p_fake;
0037
0038   // Allows the registration of the module so that it is available to be used by the Jetscape framework.
0039   static RegisterJetScapeModule<ColoredHadronization> reg;
0040
0041 protected:
0042   static Pythia8::Pythia pythia;
0043 };
0044
0045 #endif // COLOREDHADRONIZATION_H