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