LBNFDeDxMap.hh

Go to the documentation of this file.

#ifndef LBNFDEDXMAP_HH
#define LBNFDEDXMAP_HH
#include <fstream>
#include <map>
#include <vector>
#include <string>
#include <fstream>
#include <cmath>
#include "globals.hh"
#include "G4Step.hh"

class LBNFDeDxCell {

  public:
    bool keyedByZ;
    int nEntries;
    std::string G4Name; // not necessarily the full name
    std::string MARSName; // not necessarily the full name
    // provide 
    double zCenter; // all dimension in mm
    double zWidth;
    double rCenter;
    double rWidth;
    double phiCenter;
    double phiWidth;
    double eDep;
    double eDepSq;
    
};
class LBNFDeDxEntryInCell {
  public:
   int tag;
   int pid;
   double length;
   double rPre;
   double zPre;
   double rPost;
   double zPost;
   double eDep;
   double eKin;   
};

class LBNFDeDxMap {

   public:

     explicit LBNFDeDxMap(const std::string &aName);
     void fill(const G4Step* aStep);
     void dumpASCII(const std::string &aDirectory, int nEvt);
   
   private:
     std::string fName;
     double fPhiKeyFact;
     double fZKeyFact;
     double fRKeyFact;
     unsigned int fPhiKeyMult; 
     unsigned int fNumPhiSlices;
     std::map<unsigned int, LBNFDeDxCell> fMap;
     std::vector<std::string> G4NamesInMapRTag;
     std::vector<std::string> G4NamesInMapZTag;
     std::vector<double> zStarts;
     std::vector<double> zWidths;
     std::vector<double> rStarts;
     std::vector<double> rWidths;
     std::vector<LBNFDeDxEntryInCell> fVectSpecialCell;
     
     void defineInnerCHornA(); 
     void defineInnerCHornB();
     unsigned int getKey(const G4Step* aStep);
     
     inline unsigned int getTagFromR(double r, double rStart, double rWidth, double f ) const {
       size_t kPhi = static_cast<size_t>(f*fNumPhiSlices/(2.0*M_PI));
       size_t iRC = static_cast<size_t>((r - rStart)/rWidth);
       const double rc = rStart + iRC*rWidth;
       unsigned int  aKey = kPhi*fPhiKeyMult + static_cast<unsigned int>(fRKeyFact*rc);
//       std::cerr << " getTagFromR, z = " << r 
//                 << " params " << rStart << " / " << rWidth <<  " key " << aKey << std::endl;
                 return aKey;
     } 
     inline unsigned int getTagFromZ(double z, double zStart, double zWidth, double f ) const {
       size_t kPhi = static_cast<size_t>(f*fNumPhiSlices/(2.0*M_PI));
       size_t iZC = static_cast<size_t>((z - zStart)/zWidth);
       const double zc = zStart + iZC*zWidth;
       unsigned int aKey = kPhi*fPhiKeyMult + static_cast<unsigned int>(fZKeyFact*zc);
//       std::cerr << " getTagFromZ, z = " << z 
//                 << " params " << zStart << " / " << zWidth <<  " key " << aKey << std::endl;
                 return aKey;
     } 
};
#endif