FemtoSourceDensity.h

/*
 * FemtoSourceDensity.h
 *
 *  Created on: 3 sty 2023
 *      Author: Daniel Wielanek
 *      E-mail: daniel.wielanek@gmail.com
 *      Warsaw University of Technology, Faculty of Physics
 */
#ifndef HAL_ANALYSIS_FEMTO_BASE_SOURCE_MODELS_FEMTOSOURCEDENSITY_H_
#define HAL_ANALYSIS_FEMTO_BASE_SOURCE_MODELS_FEMTOSOURCEDENSITY_H_
 
#include <RtypesCore.h>
#include <TVector3.h>
 
namespace Hal {
  class FemtoSourceDensity : public TObject {
    Int_t fNParams;
    Bool_t fAna1d;
    Bool_t fAna3d;
    Double_t* fParams;  //[fNParams]
  public:
    FemtoSourceDensity(Int_t params = 0, Bool_t ana1d = kFALSE, Bool_t ana3d = kTRUE) :
      fNParams(TMath::Min(1, params)), fAna1d(ana1d), fAna3d(ana3d), fParams(nullptr) {
      fParams = new Double_t[fNParams];
    };
    FemtoSourceDensity(const FemtoSourceDensity& other);
    FemtoSourceDensity& operator=(const FemtoSourceDensity& other);
    virtual void SetBaseParameters(const Double_t* params);
    virtual Double_t GetProbDensity3d(const TVector3& /*vec*/, const Double_t* /*params*/) const { return 0; };
    virtual Double_t GetProbDensity1d(const Double_t /*r*/, const Double_t* /*params*/) const { return 0; };
    virtual Double_t GetRelativeProbDensity3d(const TVector3& vec, const Double_t* params);
    virtual Double_t GetRelativeProbDensity1d(const Double_t r, const Double_t* params) const;
    Bool_t IsAna1d() const { return fAna1d; }
    Bool_t IsAna3d() const { return fAna3d; }
    virtual ~FemtoSourceDensity() {
      if (fParams) delete[] fParams;
    };
    ClassDef(FemtoSourceDensity, 1)
  };
}  // namespace Hal
 
#endif /* HAL_ANALYSIS_FEMTO_BASE_SOURCE_MODELS_FEMTOSOURCEDENSITY_H_ */