FemtoSourceModel.h

/*
 * FemtoSourceModel.h
 *
 *  Created on: 26-11-2013
 *      Author: Daniel Wielanek
 *      E-mail: daniel.wielanek@gmail.com
 *      Warsaw University of Technology, Faculty of Physics
 */
 
#ifndef HALFEMTOSOURCEMODEL_H_
#define HALFEMTOSOURCEMODEL_H_
 
#include <TRandom.h>
#include <TString.h>
#include <TVector3.h>
 
namespace Hal {
  class FemtoPair;
  class Package;
  class FemtoSourceDensity;
 
  class FemtoSourceModel : public TObject {
  private:
    const Int_t fParametersNo;
    Double_t* fParams = {nullptr};  //[fParametersNo]
    TString* fParameterNames = {nullptr};  //[fParametersNo]
  protected:
    TRandom* fRandom = {nullptr};
    FemtoSourceDensity* fDensity = {nullptr};
    Double_t fRout;
    Double_t fRside;
    Double_t fRlong;
    TString fModelName;
    void SetParName(Int_t no, TString name) { fParameterNames[no] = name; };
    // FemtoSourceModel();
  public:
    enum class ENumProperty { kNonAnalytical, kAnalytical1d, kAnalytical3d, kFullyAnalytical };
    FemtoSourceModel(Int_t nparams = 1);
    FemtoSourceModel(const FemtoSourceModel& model);
    inline Double_t GetROut() const { return fRout; };
    inline Double_t GetRSide() const { return fRside; };
    inline Double_t GetRLong() const { return fRlong; };
    void SetSeed(UInt_t seed) { fRandom->SetSeed(seed); };
    inline void SetParameter(Int_t par_no, Double_t par_val) { fParams[par_no] = par_val; };
    void SetParameterByName(TString name, Double_t par);
    virtual void GenerateCoordinates(FemtoPair* Pair) = 0;
    virtual void SetRadius(Double_t /*radii*/) {};
    Int_t GetNpar() const { return fParametersNo; };
    TString GetParamName(Int_t n) const;
    Double_t GetParameter(Int_t n) const { return fParams[n]; };
    virtual Bool_t Init() { return kTRUE; }
    virtual void Print(Option_t* option = "") const;
    virtual FemtoSourceModel* MakeCopy() const = 0;
    FemtoSourceDensity* GetDensityProb() const { return fDensity; }
    ENumProperty GetModelNumProp() const;
    virtual Package* Report() const;
    virtual ~FemtoSourceModel();
    ClassDef(FemtoSourceModel, 1)
  };
  class FemtoSourceModel1D : public FemtoSourceModel {
  protected:
    FemtoSourceModel1D(Int_t par);
 
  public:
    FemtoSourceModel1D();
    FemtoSourceModel1D(const FemtoSourceModel1D& model);
    void SetRadius(Double_t radii) { SetParameter(0, radii); };
    virtual FemtoSourceModel* MakeCopy() const = 0;
    virtual ~FemtoSourceModel1D();
    ClassDef(FemtoSourceModel1D, 1)
  };
  class FemtoSourceModel3D : public FemtoSourceModel {
  protected:
    FemtoSourceModel3D(Int_t no);
 
  public:
    FemtoSourceModel3D();
    FemtoSourceModel3D(const FemtoSourceModel3D& model);
    void SetOutRadius(Double_t out) { SetParameter(0, out); };
    void SetSideRadius(Double_t side) { SetParameter(1, side); };
    void SetLongRadius(Double_t longr) { SetParameter(2, longr); };
    virtual void SetRadius(Double_t radii);
    virtual FemtoSourceModel* MakeCopy() const = 0;
    virtual ~FemtoSourceModel3D();
    ClassDef(FemtoSourceModel3D, 1)
  };
}  // namespace Hal
 
#endif /* HALFEMTOSOURCEMODEL_H_ */