#include <PrintMaterialBudgetInfo.h>

Inheritance diagram for PrintMaterialBudgetInfo:

Public Member Functions
	PrintMaterialBudgetInfo (edm::ParameterSet const &p)

	~PrintMaterialBudgetInfo ()

Public Member Functions inherited from SimWatcher
	SimWatcher ()

virtual	~SimWatcher ()

Public Member Functions inherited from Observer< const BeginOfJob * >
	Observer ()

void	slotForUpdate (const BeginOfJob *iT)

virtual	~Observer ()

Public Member Functions inherited from Observer< const BeginOfRun * >
	Observer ()

void	slotForUpdate (const BeginOfRun *iT)

virtual	~Observer ()

Private Member Functions
void	dumpElementMassFraction (std::ostream &elementOut=std::cout)

void	dumpHeader (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	dumpLaTeXFooter (std::ostream &out=std::cout)

void	dumpLaTeXHeader (std::ostream &out=std::cout)

void	printInfo (G4VPhysicalVolume pv, G4LogicalVolume lv, unsigned int leafDepth, std::ostream &weightOut=std::cout, std::ostream &texOut=std::cout)

std::string	stringLaTeXSuperscript (std::string stringname)

std::string	stringLaTeXUnderscore (std::string stringname)

void	update (const BeginOfJob *job)
	This routine will be called when the appropriate signal arrives. More...

void	update (const BeginOfRun *run)
	This routine will be called when the appropriate signal arrives. More...

Private Attributes
std::vector< std::string >	elementNames

std::ofstream	elementOutputFile

std::vector< double >	elementTotalWeight

std::vector< double >	elementWeightFraction

G4NavigationHistory	fHistory

unsigned int	levelFound

std::string	name

int	nchar

std::ofstream	texOutputFile

mpvpv	thePVTree

G4VPhysicalVolume *	theTopPV

bool	volumeFound

std::ofstream	weightOutputFile

Additional Inherited Members

Detailed Description

Definition at line 24 of file PrintMaterialBudgetInfo.h.

Constructor & Destructor Documentation

PrintMaterialBudgetInfo::PrintMaterialBudgetInfo ( edm::ParameterSet const & p )

Definition at line 32 of file PrintMaterialBudgetInfo.cc.

References gather_cfg::cout, elementNames, elementOutputFile, elementTotalWeight, elementWeightFraction, edm::ParameterSet::getUntrackedParameter(), name, nchar, texOutputFile, volumeFound, and weightOutputFile.

                                                                          {
   name  = p.getUntrackedParameter<std::string>("Name","*");
   nchar = name.find("*");
   name.assign(name,0,nchar);
   std::cout << "PrintMaterialBudget selected volume " << name << std::endl;
   volumeFound = false;
   std::string weightFileName = name+".weight";
   weightOutputFile.open( weightFileName.c_str() );
   std::string elementFileName = name+".element";
   elementOutputFile.open( elementFileName.c_str() );
   std::string texFileName = name+"_table.tex";
   texOutputFile.open( texFileName.c_str() );
   std::cout << "PrintMaterialBudget output file " << weightFileName  << std::endl;
   std::cout << "PrintMaterialBudget output file " << elementFileName << std::endl;
   std::cout << "PrintMaterialBudget output file " << texFileName     << std::endl;
   elementNames.clear();
   elementTotalWeight.clear();
   elementWeightFraction.clear();
 }

PrintMaterialBudgetInfo::~PrintMaterialBudgetInfo ( )

Definition at line 52 of file PrintMaterialBudgetInfo.cc.

52 {}

Member Function Documentation

void PrintMaterialBudgetInfo::dumpElementMassFraction ( std::ostream & elementOut = std::cout )

private

Definition at line 222 of file PrintMaterialBudgetInfo.cc.

References python.multivaluedict::append(), elementNames, elementTotalWeight, elementWeightFraction, and findQualityFiles::size.

Referenced by update().

                                                                              {
   // calculate mass fraction
   double totalWeight   = 0.0;
   double totalFraction = 0.0;
   for(unsigned int iElement = 0; iElement<(unsigned int)elementTotalWeight.size(); iElement++) {
     totalWeight+=elementTotalWeight[iElement];
   }
   // calculate element mass fractions
   for(unsigned int iElement = 0; iElement<(unsigned int)elementTotalWeight.size(); iElement++) {
     elementWeightFraction[iElement] = elementTotalWeight[iElement]/totalWeight;
     totalFraction+=elementWeightFraction[iElement];
   }
   // header
   elementOut << "Element"        << "\t\t"
          << "Index"          << "\t"
          << "Total Mass"     << "\t"
          << "Mass Fraction " << "\t"
          << std::endl;
   // dump
   for(unsigned int iElement = 0; iElement<(unsigned int)elementTotalWeight.size(); iElement++) {
     if(elementNames[iElement]!="rr") {
       if(elementNames[iElement].size()<8) elementNames[iElement].append("\t");
       elementOut << elementNames[iElement]                          << "\t"
          << iElement                                        << "\t"
          << G4BestUnit(elementTotalWeight[iElement],"Mass") << "\t"
          << elementWeightFraction[iElement]
          << std::endl;
     }
   }
   elementOut << "\n\t\tTotal Weight without Air " << G4BestUnit(totalWeight,"Mass")
          << "\tTotal Fraction "   << totalFraction
          << std::endl;
 }

void PrintMaterialBudgetInfo::dumpHeader ( std::ostream & out = std::cout )

private

Definition at line 79 of file PrintMaterialBudgetInfo.cc.

Referenced by update().

                                                          {
   out << "Geom."      << "\t"
       << "Volume"     << "\t" << "\t"
       << "Copy"       << "\t"
       << "Solid"      << "\t" << "\t"
       << "Material"   << "\t"
       << "Density"    << "\t" << "\t"
       << "Mass"       << "\t" << "\t"
       << std::endl;
   out << "Level"     << "\t"
       << "Name"      << "\t" << "\t"
       << "Number"    << "\t"
       << "Name"      << "\t" << "\t"
       << "Name"      << "\t" << "\t"
       << "[g/cm3]"   << "\t" << "\t"
       << "[g]    "   << "\t" << "\t"
       << std::endl;
 }

void PrintMaterialBudgetInfo::dumpHierarchyLeaf	(	G4VPhysicalVolume *	pv,
		G4LogicalVolume *	lv,
		unsigned int	leafDepth,
		std::ostream &	weightOut = `std::cout`,
		std::ostream &	texOut = `std::cout`
	)

private

Definition at line 132 of file PrintMaterialBudgetInfo.cc.

References levelFound, name, nchar, printInfo(), and volumeFound.

Referenced by update().

                                              {
   
   if( volumeFound && ( leafDepth <= levelFound ) ) return; 
   if( volumeFound && ( leafDepth >  levelFound ) ) printInfo(pv, lv, leafDepth, weightOut, texOut);
   
   // choose mother volume
   std::string lvname = lv->GetName();
   lvname.assign(lvname,0,nchar);
   if (lvname == name) {
     volumeFound = true;
     levelFound  = leafDepth;
     printInfo(pv, lv, leafDepth, weightOut, texOut);
     texOut << "      \\hline" << std::endl;
   }
   
   //----- Get LV daughters from list of PV daughters
   mmlvpv lvpvDaughters;
   std::set< G4LogicalVolume* > lvDaughters;
   int NoDaughters = lv->GetNoDaughters();
   while ((NoDaughters--)>0)
     {
       G4VPhysicalVolume* pvD = lv->GetDaughter(NoDaughters);
       lvpvDaughters.insert(mmlvpv::value_type(pvD->GetLogicalVolume(), pvD));
       lvDaughters.insert(pvD->GetLogicalVolume());
     }
   
   std::set< G4LogicalVolume* >::const_iterator scite;
   mmlvpv::const_iterator mmcite;
   
   //----- Dump daughters PV and LV
   for (scite = lvDaughters.begin(); scite != lvDaughters.end(); scite++) {
     std::pair< mmlvpv::iterator, mmlvpv::iterator > mmER = lvpvDaughters.equal_range(*scite);    
     //----- Dump daughters PV of this LV
     for (mmcite = mmER.first ; mmcite != mmER.second; mmcite++) 
       dumpHierarchyLeaf((*mmcite).second, *scite, leafDepth+1, weightOut, texOut );
   }
   
 }

void PrintMaterialBudgetInfo::dumpLaTeXFooter ( std::ostream & out = std::cout )

private

Definition at line 125 of file PrintMaterialBudgetInfo.cc.

Referenced by update().

                                                               {
   out << "      \\hline"      << std::endl
       << "    \\end{tabular}" << std::endl
       << "  \\end{center}"    << std::endl
       << "\\end{table}"       << std::endl;
 }

void PrintMaterialBudgetInfo::dumpLaTeXHeader ( std::ostream & out = std::cout )

private

Definition at line 98 of file PrintMaterialBudgetInfo.cc.

References name.

Referenced by update().

                                                               {
   out << "\\begin{table}[h!]"                                     << std::endl
       << "  \\caption{\\textsf {" << name << "} volume list.}"               << std::endl
       << "  \\label{tab: " << name << "}"                         << std::endl
       << "  \\begin{center}"                                      << std::endl
       << "    \\begin{tabular}{ccccccc}"                          << std::endl
       << "      \\hline"                                          << std::endl;
   out << "      Geom."      << "\t & "
       << "      Volume"     << "\t & "
       << "      Copy"       << "\t & "
       << "      Solid"      << "\t & "
       << "      Material"   << "\t & "
       << "      Density"    << "\t & "
       << "      Mass"       << "\t \\\\ "
       << std::endl;
   out << "      Level"      << "\t & "
       << "      Name"       << "\t & "
       << "      Number"     << "\t & "
       << "      Name"       << "\t & "
       << "      Name"       << "\t & "
       << "                " << "\t & "
       << "                " << "\t \\\\ "
       << std::endl
       << "      \\hline\\hline"
       << std::endl;
 }

void PrintMaterialBudgetInfo::printInfo	(	G4VPhysicalVolume *	pv,
		G4LogicalVolume *	lv,
		unsigned int	leafDepth,
		std::ostream &	weightOut = `std::cout`,
		std::ostream &	texOut = `std::cout`
	)

private

Definition at line 174 of file PrintMaterialBudgetInfo.cc.

References elementNames, elementTotalWeight, stringLaTeXSuperscript(), stringLaTeXUnderscore(), and CommonMethods::weight().

Referenced by dumpHierarchyLeaf().

                                                                 {
   
   double density = lv->GetMaterial()->GetDensity();
   double weight  = lv->GetMass(false,false);
   
   std::string volumeName = lv->GetName();
   if(volumeName.size()<8) volumeName.append("\t");
   
   std::string solidName = lv->GetSolid()->GetName();
   if(solidName.size()<8) solidName.append("\t");
     
   std::string materialName = lv->GetMaterial()->GetName();
   if(materialName.size()<8) materialName.append("\t");
   
   //----- dump info 
   weightOut << leafDepth                                            << "\t"
         << volumeName                                           << "\t"
         << pv->GetCopyNo()                                      << "\t"
         << solidName                                            << "\t"
         << materialName                                         << "\t"
         << G4BestUnit(density,"Volumic Mass")                   << "\t"
         << G4BestUnit(weight,"Mass")                            << "\t"
         << std::endl;
   //
   texOut << "\t" 
      << leafDepth                                                  << "\t & "
      << stringLaTeXUnderscore(volumeName)                          << "\t & "
      << pv->GetCopyNo()                                            << "\t & "
      << stringLaTeXUnderscore(solidName)                           << "\t & "
      << stringLaTeXUnderscore(materialName)                        << "\t & "
      << stringLaTeXSuperscript(G4BestUnit(density,"Volumic Mass")) << "\t & "
      << stringLaTeXSuperscript(G4BestUnit(weight,"Mass"))          << "\t \\\\ "
      << std::endl;
   //  
   for(unsigned int iElement = 0; iElement<(unsigned int)lv->GetMaterial()->GetNumberOfElements(); iElement++) {
     // exclude Air in element weight fraction computation
     if(materialName.find("Air")) {
       std::string elementName = lv->GetMaterial()->GetElement(iElement)->GetName();
       double elementMassFraction = lv->GetMaterial()->GetFractionVector()[iElement];
       double elementWeight = weight*elementMassFraction;
       unsigned int elementIndex = (unsigned int)lv->GetMaterial()->GetElement(iElement)->GetIndex();
       elementNames[elementIndex]        = elementName;
       elementTotalWeight[elementIndex] += elementWeight;
     }
   }
 }

std::string PrintMaterialBudgetInfo::stringLaTeXSuperscript ( std::string stringname )

private

Definition at line 272 of file PrintMaterialBudgetInfo.cc.

References i.

Referenced by printInfo().

                                                                               {
   // To replace 'm3' with 'm$^3$' to compile LaTeX output
   std::string stringoutput = stringname.substr(0,1);
   
   for (unsigned int  i=1; i<stringname.length() ; i++) {
     if (stringname.substr(i-1,1) == "m" && stringname.substr(i,1) == "3") {
       stringoutput += "$^3$";
     } else {
       stringoutput += stringname.substr(i,1);
     }
   }
     
   return stringoutput;
   
 }

std::string PrintMaterialBudgetInfo::stringLaTeXUnderscore ( std::string stringname )

private

Definition at line 256 of file PrintMaterialBudgetInfo.cc.

References i.

Referenced by printInfo().

                                                                              {
   // To replace '\' with '\_' to compile LaTeX output
   std::string stringoutput;
   
   for (unsigned int  i=0; i<stringname.length() ; i++) {
     if (stringname.substr(i,1) == "_") {
       stringoutput += "\\_";
     } else {
       stringoutput += stringname.substr(i,1);
     }
   }
   
   return stringoutput;
   
 }

void PrintMaterialBudgetInfo::update ( const BeginOfJob * )

inlineprivatevirtual

This routine will be called when the appropriate signal arrives.

Implements Observer< const BeginOfJob * >.

Definition at line 32 of file PrintMaterialBudgetInfo.h.

32 {};

void PrintMaterialBudgetInfo::update ( const BeginOfRun * )

privatevirtual

This routine will be called when the appropriate signal arrives.

Implements Observer< const BeginOfRun * >.

Definition at line 54 of file PrintMaterialBudgetInfo.cc.

References dumpElementMassFraction(), dumpHeader(), dumpHierarchyLeaf(), dumpLaTeXFooter(), dumpLaTeXHeader(), elementNames, elementOutputFile, elementTotalWeight, elementWeightFraction, texOutputFile, theTopPV, and weightOutputFile.

                                                           {
   
   // Physical Volume
   theTopPV = G4TransportationManager::GetTransportationManager()->GetNavigatorForTracking()->GetWorldVolume();
   // Logical Volume
   G4LogicalVolume*  lv = theTopPV->GetLogicalVolume();
   unsigned int leafDepth = 0;
   // the first time fill the vectors of elements
   if( elementNames.size()==0 && elementTotalWeight.size()==0 && elementWeightFraction.size()==0) {
     for(unsigned int iElement = 0;
     iElement < lv->GetMaterial()->GetElement(iElement)->GetElementTable()->size();
     iElement++) { // first element in table is 0
       elementNames.push_back("rr");
       elementTotalWeight.push_back(0);
       elementWeightFraction.push_back(0);
     }
   }
   dumpHeader(weightOutputFile);
   dumpLaTeXHeader(texOutputFile);
   dumpHierarchyLeaf(theTopPV, lv, leafDepth, weightOutputFile, texOutputFile);
   dumpElementMassFraction(elementOutputFile);
   dumpLaTeXFooter(texOutputFile);
   //
 }