MaterialEffects.h

/* Copyright 2008-2014, Technische Universitaet Muenchen,
   Authors: Christian Hoeppner & Sebastian Neubert & Johannes Rauch
 
   This file is part of GENFIT.
 
   GENFIT is free software: you can redistribute it and/or modify
   it under the terms of the GNU Lesser General Public License as published
   by the Free Software Foundation, either version 3 of the License, or
   (at your option) any later version.
 
   GENFIT is distributed in the hope that it will be useful,
   but WITHOUT ANY WARRANTY; without even the implied warranty of
   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
   GNU Lesser General Public License for more details.
 
   You should have received a copy of the GNU Lesser General Public License
   along with GENFIT.  If not, see <http://www.gnu.org/licenses/>.
*/
 
#ifndef genfit_MaterialEffects_h
#define genfit_MaterialEffects_h
 
#include "RKTools.h"
#include "AbsMaterialInterface.h"
 
#include <iostream>
#include <vector>
 
#include <TVector3.h>
 
 
namespace genfit {
 
class MaterialEffects {
 
 private:
 
  MaterialEffects();
  virtual ~MaterialEffects();
 
  static MaterialEffects* instance_;
 
 
public:
 
  static MaterialEffects* getInstance();
  static void destruct();
 
  void init(AbsMaterialInterface* matIfc);
  bool isInitialized() { return materialInterface_ != nullptr; }
 
  void setNoEffects(bool opt = true) {noEffects_ = opt;}
 
  void setEnergyLossBetheBloch(bool opt = true) {energyLossBetheBloch_ = opt; noEffects_ = false;}
  void setNoiseBetheBloch(bool opt = true) {noiseBetheBloch_ = opt; noEffects_ = false;}
  void setNoiseCoulomb(bool opt = true) {noiseCoulomb_ = opt; noEffects_ = false;}
  void setEnergyLossBrems(bool opt = true) {energyLossBrems_ = opt; noEffects_ = false;}
  void setNoiseBrems(bool opt = true) {noiseBrems_ = opt; noEffects_ = false;}
  void ignoreBoundariesBetweenEqualMaterials(bool opt = true) {ignoreBoundariesBetweenEqualMaterials_ = opt;}
  void setMagCharge(double magCharge) {mag_charge_ = magCharge;} 
 
  void setMscModel(const std::string& modelName);
 
 
  double effects(const std::vector<RKStep>& steps,
                 int materialsFXStart,
                 int materialsFXStop,
                 const double& mom,
                 const int& pdg,
                 M7x7* noise = nullptr);
 
  void stepper(const RKTrackRep* rep,
               M1x7& state7,
               const double& mom, // momentum
               double& relMomLoss, // relative momloss for the step will be added
               const int& pdg,
               Material& currentMaterial,
               StepLimits& limits,
               bool varField = true);
 
  void setDebugLvl(unsigned int lvl = 1);
 
 
  void drawdEdx(int pdg = 11);
 
 private:
 
  void getParticleParameters();
 
  void getMomGammaBeta(double Energy,
                       double& mom, double& gammaSquare, double& gamma, double& betaSquare) const;
 
 
  double momentumLoss(double stepSign, double mom, bool linear);
 
  double dEdx(double Energy);
 
 
  double dEdxBetheBloch(double betaSquare, double gamma, double gammasquare) const;
 
 
  void noiseBetheBloch(M7x7& noise, double mom, double betaSquare, double gamma, double gammaSquare) const;
 
 
  void noiseCoulomb(M7x7& noise,
                    const M1x3& direction, double momSquare, double betaSquare) const;
 
 
  double dEdxBrems(double mom) const;
 
 
  void noiseBrems(M7x7& noise, double momSquare, double betaSquare) const;
 
 
 
  bool noEffects_;
 
  bool energyLossBetheBloch_;
  bool noiseBetheBloch_;
  bool noiseCoulomb_;
  bool energyLossBrems_;
  bool noiseBrems_;
 
  bool ignoreBoundariesBetweenEqualMaterials_;
 
  const double me_; // electron mass (GeV)
 
  double stepSize_; // stepsize
 
  // cached values for energy loss and noise calculations
  double dEdx_; // Runkge Kutta dEdx
  double E_; // Runge Kutta Energy
  double matDensity_;
  double matZ_;
  double matA_;
  double radiationLength_;
  double mEE_; // mean excitation energy
 
  int pdg_;
  double charge_;
  double mag_charge_; // in units of e+
  double mass_;
 
  int mscModelCode_; 
 
  AbsMaterialInterface* materialInterface_;
 
  unsigned int debugLvl_;
 
  // ClassDef(MaterialEffects, 1);
 
};
 
} /* End of namespace genfit */
#endif // genfit_MaterialEffects_h