TRestGeant4BiasingVolume.h

 
#ifndef RestCore_TRestGeant4BiasingVolume
#define RestCore_TRestGeant4BiasingVolume
 
#include <iostream>
 
#include "TObject.h"
#include "TString.h"
#include "TVector3.h"
 
class TRestGeant4BiasingVolume : public TObject {
   protected:
    TVector3 fVolumePosition;
    Double_t fVolumeSize;
    TString fBiasingVolumeType;
    Double_t fBiasingFactor;
    TVector2 fEnergyRange;
    TString fVolumeType;
 
   public:
    inline Double_t GetBiasingFactor() const { return fBiasingFactor; }
    inline Double_t GetBiasingVolumeSize() const { return fVolumeSize; }
    inline TString GetBiasingVolumeType() const { return fVolumeType; }
    inline TVector3 GetBiasingVolumePosition() const { return fVolumePosition; }
    inline TVector2 GetEnergyRange() const { return fEnergyRange; }
    inline Double_t GetMaxEnergy() const { return fEnergyRange.Y(); }
    inline Double_t GetMinEnergy() const { return fEnergyRange.X(); }
 
    void SetBiasingVolumeSize(Double_t size) { fVolumeSize = size; }
    void SetBiasingVolumeType(TString type) { fVolumeType = type; }
    void SetBiasingVolumePosition(TVector3 pos) { fVolumePosition = pos; }
    void SetBiasingFactor(Double_t factor) { fBiasingFactor = factor; }
    void SetEnergyRange(TVector2 eRange) { fEnergyRange = eRange; }
 
    // Check if it is inside the sphere
    Int_t isInside(Double_t x, Double_t y, Double_t z) {
        if (fVolumeType == "virtualBox") {
            if (x < fVolumeSize / 2. && x > -fVolumeSize / 2.)
                if (y < fVolumeSize / 2. && y > -fVolumeSize / 2.)
                    if (z < fVolumeSize / 2. && z > -fVolumeSize / 2.) return 1;
        }
 
        if (fVolumeType == "virtualSphere") {
            Double_t r2 = x * x + y * y + z * z;
            if (r2 < fVolumeSize * fVolumeSize) return 1;
        }
        return 0;
    }
 
    void PrintBiasingVolume();
 
    // Constructor
    TRestGeant4BiasingVolume();
    // Destructor
    virtual ~TRestGeant4BiasingVolume();
 
    ClassDef(TRestGeant4BiasingVolume, 2);  // REST event superclass
};
#endif