Hadronization.cc

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

/*******************************************************************************
 * 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.
 ******************************************************************************/

#include "Hadronization.h"
#include "JetScapeLogger.h"
#include <string>
#include <vector>
#include <iostream>
#include "JetScapeSignalManager.h"
#include "JetScapeWriterStream.h"

using namespace std;

namespace Jetscape {

Hadronization::Hadronization() {
    SetId("Hadronization");
    TransformPartonsConnected = false;
    HydroHyperSurfaceConnected_ = false;
    GetHydroCellSignalConnected_ = false;
}

Hadronization::~Hadronization() {}

void Hadronization::Clear() {
  VERBOSESHOWER(8);
  outHadrons.clear();
}

void Hadronization::Init() {
  JetScapeModuleBase::Init();

  // May need to read some configuration info from XML file here

  if (GetNumberOfTasks() < 1) {
    JSWARN << " : No valid Hadronization modules found ...";
    exit(-1);
  }

  JSINFO << "Found " << GetNumberOfTasks()
         << " Hadronization Tasks/Modules Initialize them ... ";
  JetScapeTask::InitTasks();
}

void Hadronization::DoHadronize() {
  VERBOSE(2) << "Get Recombination Partons...";

  if (inPartons.size() > 0) {
    VERBOSE(2) << "There are Partons ready for Recombination...";
    TransformPartons(inPartons, outHadrons, outPartons);
  } else {
    VERBOSE(2) << "There is no Parton ready for Recombination...";
  }
}

void Hadronization::Exec() {
  VERBOSE(2) << "Run Hadronization Exec...";
  VERBOSE(2) << "Found " << GetNumberOfTasks()
             << " Hadronization Tasks/Modules Execute them ... ";

  //this->outHadrons = make_shared<vector<shared_ptr<Hadron>>>();
  //this->outPartons = make_shared<vector<shared_ptr<Parton>>>();

  DoHadronize();
}

void Hadronization::WriteTask(weak_ptr<JetScapeWriter> w) {
  VERBOSE(4) << "In Hadronization::WriteTask";
  auto f = w.lock();
  if (!f)
    return;

  f->WriteComment("Hadronization module: " + GetId());

  if (GetHadrons().size() > 0) {
    f->WriteComment("Final State Hadrons");
    int i = 0;
    for (auto &h : GetHadrons()) {
      f->WriteWhiteSpace("[" + to_string(i) + "] H");
      f->Write(h);
      ++i;
    }
  } else {
    f->WriteComment("There are no Hadrons");
  }
}

void Hadronization::DeleteHadrons() {
  outHadrons.clear();
}

void Hadronization::DeleteRealHadrons() {
  outHadrons.erase(
    std::remove_if(
      outHadrons.begin(),
      outHadrons.end(),
      [](const std::shared_ptr<Hadron>& hadron) {
          return hadron->pstat() > 0;
      }
    ),
    outHadrons.end()
  );
}

} // namespace Jetscape