IPGlasmaWrapper.h

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

#include <memory>

#include "JetScapeModuleBase.h"
#include "InitialState.h"
#include "JetScapeLogger.h"
#include "IPGlasma.h"

using namespace Jetscape;

class IPGlasmaWrapper : public Jetscape::InitialState {
  // this is wrapper class to read external files that
  // stores initial number of binary collisions and corresponding
  // configurations
public:
  IPGlasmaWrapper();
  ~IPGlasmaWrapper();

  //void Init();

  void Exec();

  void Clear();

  void InitTask();

  virtual void Write(weak_ptr<JetScapeWriter> w);

  double GetNcoll() { return(static_cast<double>(ncoll_)); };

  void ReadNbcList(std::string filename);

  void SampleABinaryCollisionPoint(double &x, double &y);

private:
  std::unique_ptr<IPGlasma> IPGlasma_ptr_;
  int dim_x_, dim_y_;

  std::vector<double> binary_collision_x_;
  std::vector<double> binary_collision_y_;
  std::shared_ptr<std::uniform_int_distribution<int>> rand_int_ptr_;

  int ncoll_ = -1;

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

#endif  // IPGLASMAWRAPPER_H