PartonPrinter.h

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file PartonPrinter.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 PARTONPRINTER_H
#define PARTONPRINTER_H

#include "JetScapeModuleBase.h"
#include "PartonShower.h"
#include <vector>
#include <string>
#include <fstream>

namespace Jetscape {

class PartonPrinter : public JetScapeModuleBase {

public:
  PartonPrinter();
  virtual ~PartonPrinter();

  virtual void Init();
  virtual void Exec() final;
  virtual void Clear();
  std::ofstream
      dist_output; 

  void GetFinalPartons(shared_ptr<PartonShower>
                           pShower /*, vector<shared_ptr<Parton>>& fPartons*/);

  void GetFinalPartons2(shared_ptr<PartonShower> pShower);

  void GetPartonsAtTime(shared_ptr<PartonShower> pShower,
                        vector<shared_ptr<Parton>> &fPartons, double time);

  void PrintFinalPartons(vector<vector<shared_ptr<Parton>>> &fPartons) {
    fPartons = pFinals;
  };

private:
  vector<vector<shared_ptr<Parton>>> pFinals;

  // Allows the registration of the module so that it is available to be used by the Jetscape framework.
  static RegisterJetScapeModule<PartonPrinter> reg;
};

} // end namespace Jetscape

#endif