GtRockGammaTool.h

Go to the documentation of this file.

/******************************************************************************
 * GtRockGammaTool - Tool for roughly generating gammas from rock
 *
 * Tool to generate gammas from radioactivity in the rock around the
 * pool.  It chooses a random point around the pool wall, top, or
 * bottom, and emits a gamma ray in a uniform random direction, in a
 * hemisphere defined away from the wall.
 *
 * GammaSpectrum: vector containing the bin content for the gamma
 * ray spectrum histogram. 
 * 
 * GammaBinEdges: vector containing the bin edges of the
 * histogram. Inputting this allows for dealing with uneven bin sizes.
 * The pdf does not need to be normalized. Obviously, GammaBinEdges
 * should have one more element than GammaSpectrum.
 *
 * Created by: dandwyer@caltech.edu 2011/05/16
 * Based on GtDiffuserBallTool by kboddy@caltech.edu
 *****************************************************************************/

#ifndef GTROCKGAMMATOOL_H
#define GTROCKGAMMATOOL_H

#include "GenTools/IHepMCEventMutator.h"
#include "GaudiAlg/GaudiTool.h"
#include "GaudiKernel/RndmGenerators.h"
#include "CLHEP/Vector/ThreeVector.h"

// Rectangular surface
class GtSourceRect {
 public:
  GtSourceRect(const CLHEP::Hep3Vector& origin,
               const CLHEP::Hep3Vector& edgeA,
               const CLHEP::Hep3Vector& edgeB);
  virtual ~GtSourceRect();
  // Center of rectangle
  virtual double area();
  // Center of rectangle
  virtual CLHEP::Hep3Vector origin();
  // Normal of rectangular plane
  virtual CLHEP::Hep3Vector normal();
  // Select random, uniform point on surface
  virtual CLHEP::Hep3Vector randomPoint(Rndm::Numbers& random);
 protected:
  double m_area;
  CLHEP::Hep3Vector m_origin;
  CLHEP::Hep3Vector m_edgeA;
  CLHEP::Hep3Vector m_edgeB;
  CLHEP::Hep3Vector m_normal;
};

// Octagonal surface: Based on rectangle, but with beveled corners
class GtSourceOct : virtual public GtSourceRect {
 public:
  GtSourceOct(const CLHEP::Hep3Vector& origin,
              const CLHEP::Hep3Vector& edgeA,
              const CLHEP::Hep3Vector& edgeB,
              const double& bevelLength);
  virtual ~GtSourceOct();
  virtual CLHEP::Hep3Vector randomPoint(Rndm::Numbers& random);
 protected:
  double m_bevelLength;
};

class GtRockGammaTool : public GaudiTool,
                        virtual public IHepMCEventMutator
{
 public: 
  GtRockGammaTool(const std::string& type,
                  const std::string& name, 
                  const IInterface* parent);
  virtual ~GtRockGammaTool();

  // GaudiTool interface
  virtual StatusCode initialize();
  virtual StatusCode finalize();

  // HepMCEventMutator interface
  virtual StatusCode mutate(HepMC::GenEvent& event);
 
 private:

  Rndm::Numbers m_uni;
  double m_totalSurfaceArea;
  std::vector<GtSourceRect*> m_walls;

  // User-defined PDF and associated bin edges for Gamma ray spectrum 
  std::vector<double> m_pdfGammas;
  std::vector<double> m_edgesGammas;
  std::vector<double> m_cdfGammas; // Transient storage of Cum. Dist. Func.

  StatusCode oneVertex(HepMC::GenEvent& event);
  double ConvertCdfRand(const double& rand, const std::vector<double>& cdf, 
                        const std::vector<double>& edges);
  unsigned int GetRandomWall();

};


#endif //GTROCKGAMMATOOL_H