PrintMaterialBudgetInfo.h

Go to the documentation of this file.

#ifndef SimG4CorePrintGeomInfo_PrintMaterialBudgetInfo_H
#define SimG4CorePrintGeomInfo_PrintMaterialBudgetInfo_H

#include "SimG4Core/Watcher/interface/SimWatcher.h"
#include "SimG4Core/Notification/interface/Observer.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"

#include "G4NavigationHistory.hh"

#include <iostream>
#include <fstream>
#include <vector>
#include <string>

class BeginOfJob;
class BeginOfRun;
class G4LogicalVolume;
class G4VPhysicalVolume;
class G4VSolid;

typedef std::map< G4VPhysicalVolume*, G4VPhysicalVolume*, std::less<G4VPhysicalVolume*> > mpvpv;
typedef std::multimap< G4LogicalVolume*, G4VPhysicalVolume*, std::less<G4LogicalVolume*> > mmlvpv;

class PrintMaterialBudgetInfo : public SimWatcher,
                public Observer<const BeginOfJob*>,
                public Observer<const BeginOfRun*> {

public:
  PrintMaterialBudgetInfo(edm::ParameterSet const & p);
  ~PrintMaterialBudgetInfo() override;
private:
  void update(const BeginOfJob* job) override {};
  void update(const BeginOfRun* run) override;
  void dumpHeader(std::ostream& out = std::cout);
  void dumpLaTeXHeader(std::ostream& out = std::cout);
  void dumpHierarchyLeaf(G4VPhysicalVolume* pv, G4LogicalVolume* lv, unsigned int leafDepth,
             std::ostream& weightOut = std::cout, std::ostream& texOut = std::cout);
  void printInfo(G4VPhysicalVolume* pv, G4LogicalVolume* lv, unsigned int leafDepth,
         std::ostream& weightOut = std::cout, std::ostream& texOut = std::cout);
  void dumpElementMassFraction(std::ostream& elementOut = std::cout);
  void dumpLaTeXFooter(std::ostream& out = std::cout);
  
private:
  std::string              name;
  int                      nchar;
  mpvpv                    thePVTree;
  G4VPhysicalVolume*       theTopPV; 
  G4NavigationHistory      fHistory;
  bool                     volumeFound;
  unsigned int             levelFound;
  std::ofstream            weightOutputFile;
  std::ofstream            elementOutputFile;
  std::ofstream            texOutputFile;
  std::vector<std::string> elementNames;
  std::vector<double>      elementTotalWeight;
  std::vector<double>      elementWeightFraction;
  //
  std::string stringLaTeXUnderscore(std::string stringname);
  std::string stringLaTeXSuperscript(std::string stringname);
};

#endif