LiquefierBase.h

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

#include "JetClass.h"
#include "sigslot.h"
#include "FluidCellInfo.h"

#include <array>
#include <vector>
#include "RealType.h"

namespace Jetscape {

class Droplet {
private:
  std::array<Jetscape::real, 4> xmu;
  std::array<Jetscape::real, 4> pmu;

public:
  Droplet() = default;
  Droplet(std::array<Jetscape::real, 4> x_in,
          std::array<Jetscape::real, 4> p_in) {
    xmu = x_in;
    pmu = p_in;
  }
  ~Droplet(){};

  std::array<Jetscape::real, 4> get_xmu() const { return (xmu); }

  std::array<Jetscape::real, 4> get_pmu() const { return (pmu); }
};

class LiquefierBase {
private:
  std::vector<Droplet> dropletlist;
  bool GetHydroCellSignalConnected;
  const int drop_stat;
  const int miss_stat;
  const int neg_stat;
  const Jetscape::real hydro_source_abs_err;

public:
  LiquefierBase();
  ~LiquefierBase() { Clear(); }

  void add_a_droplet(Droplet droplet_in) { dropletlist.push_back(droplet_in); }

  int get_drop_stat() const { return (drop_stat); }
  int get_miss_stat() const { return (miss_stat); }
  int get_neg_stat() const { return (neg_stat); }

  Droplet get_a_droplet(const int idx) const { return (dropletlist[idx]); }

  void check_energy_momentum_conservation(const std::vector<Parton> &pIn,
                                          std::vector<Parton> &pOut);
  void filter_partons(std::vector<Parton> &pOut);
  void add_hydro_sources(std::vector<Parton> &pIn, std::vector<Parton> &pOut);

  sigslot::signal5<double, double, double, double,
                   std::unique_ptr<FluidCellInfo> &,
                   sigslot::multi_threaded_local>
      GetHydroCellSignal;

  const bool get_GetHydroCellSignalConnected() {
    return GetHydroCellSignalConnected;
  }

  void set_GetHydroCellSignalConnected(bool m_GetHydroCellSignalConnected) {
    GetHydroCellSignalConnected = m_GetHydroCellSignalConnected;
  }

  int get_dropletlist_size() const { return (dropletlist.size()); }

  Jetscape::real get_dropletlist_total_energy() const;

  virtual void smearing_kernel(Jetscape::real tau, Jetscape::real x,
                               Jetscape::real y, Jetscape::real eta,
                               const Droplet drop_i,
                               std::array<Jetscape::real, 4> &jmu) const {
    jmu = {0, 0, 0, 0};
  }

  void get_source(Jetscape::real tau, Jetscape::real x, Jetscape::real y,
                  Jetscape::real eta, std::array<Jetscape::real, 4> &jmu) const;

  virtual void Clear();
};

}; // namespace Jetscape

#endif // LIQUEFIERBASE_H