Gabriele Cosmo
251 lines
8.1 KiB
C++
251 lines
8.1 KiB
C++
//
|
|
// ********************************************************************
|
|
// * License and Disclaimer *
|
|
// * *
|
|
// * The Geant4 software is copyright of the Copyright Holders of *
|
|
// * the Geant4 Collaboration. It is provided under the terms and *
|
|
// * conditions of the Geant4 Software License, included in the file *
|
|
// * LICENSE and available at http://cern.ch/geant4/license . These *
|
|
// * include a list of copyright holders. *
|
|
// * *
|
|
// * Neither the authors of this software system, nor their employing *
|
|
// * institutes,nor the agencies providing financial support for this *
|
|
// * work make any representation or warranty, express or implied, *
|
|
// * regarding this software system or assume any liability for its *
|
|
// * use. Please see the license in the file LICENSE and URL above *
|
|
// * for the full disclaimer and the limitation of liability. *
|
|
// * *
|
|
// * This code implementation is the result of the scientific and *
|
|
// * technical work of the GEANT4 collaboration. *
|
|
// * By using, copying, modifying or distributing the software (or *
|
|
// * any work based on the software) you agree to acknowledge its *
|
|
// * use in resulting scientific publications, and indicate your *
|
|
// * acceptance of all terms of the Geant4 Software license. *
|
|
// ********************************************************************
|
|
//
|
|
//
|
|
//
|
|
|
|
#include "G4VScoreWriter.hh"
|
|
|
|
#include "G4MultiFunctionalDetector.hh"
|
|
#include "G4SDParticleFilter.hh"
|
|
#include "G4VPrimitiveScorer.hh"
|
|
#include "G4VScoringMesh.hh"
|
|
|
|
#include <map>
|
|
#include <fstream>
|
|
|
|
G4VScoreWriter::G4VScoreWriter()
|
|
: fScoringMesh(nullptr), verboseLevel(0) {
|
|
fNMeshSegments[0] = fNMeshSegments[1] = fNMeshSegments[2] = 0;
|
|
}
|
|
|
|
G4VScoreWriter::~G4VScoreWriter() {
|
|
}
|
|
|
|
void G4VScoreWriter::SetScoringMesh(G4VScoringMesh * sm) {
|
|
fScoringMesh = sm;
|
|
fScoringMesh->GetNumberOfSegments(fNMeshSegments);
|
|
}
|
|
|
|
void G4VScoreWriter::DumpQuantityToFile(const G4String& psName,
|
|
const G4String& fileName,
|
|
const G4String& option) {
|
|
|
|
// change the option string into lowercase to the case-insensitive.
|
|
G4String opt = option;
|
|
std::transform(opt.begin(), opt.end(), opt.begin(), (int (*)(int))(tolower));
|
|
|
|
// confirm the option
|
|
if(opt.size() == 0) opt = "csv";
|
|
if(opt.find("csv") == std::string::npos &&
|
|
opt.find("sequence") == std::string::npos) {
|
|
G4cerr << "ERROR : DumpToFile : Unknown option -> "
|
|
<< option << G4endl;
|
|
return;
|
|
}
|
|
|
|
// open the file
|
|
std::ofstream ofile(fileName);
|
|
if(!ofile) {
|
|
G4cerr << "ERROR : DumpToFile : File open error -> "
|
|
<< fileName << G4endl;
|
|
return;
|
|
}
|
|
ofile << "# mesh name: " << fScoringMesh->GetWorldName() << G4endl;
|
|
|
|
|
|
// retrieve the map
|
|
MeshScoreMap fSMap = fScoringMesh->GetScoreMap();
|
|
|
|
|
|
MeshScoreMap::const_iterator msMapItr = fSMap.find(psName);
|
|
if(msMapItr == fSMap.end()) {
|
|
G4cerr << "ERROR : DumpToFile : Unknown quantity, \""
|
|
<< psName << "\"." << G4endl;
|
|
return;
|
|
}
|
|
|
|
|
|
std::map<G4int, G4StatDouble*> * score = msMapItr->second->GetMap();
|
|
ofile << "# primitive scorer name: " << msMapItr->first << std::endl;
|
|
|
|
|
|
G4double unitValue = fScoringMesh->GetPSUnitValue(psName);
|
|
G4String unit = fScoringMesh->GetPSUnit(psName);
|
|
G4String divisionAxisNames[3];
|
|
fScoringMesh->GetDivisionAxisNames(divisionAxisNames);
|
|
// index order
|
|
ofile << "# i" << divisionAxisNames[0]
|
|
<< ", i" << divisionAxisNames[1]
|
|
<< ", i" << divisionAxisNames[2];
|
|
// unit of scored value
|
|
ofile << ", total(value) ";
|
|
if(unit.size() > 0) ofile << "[" << unit << "]";
|
|
ofile << ", total(val^2), entry" << G4endl;
|
|
|
|
// "sequence" option: write header info
|
|
if(opt.find("sequence") != std::string::npos) {
|
|
ofile << fNMeshSegments[0] << " " << fNMeshSegments[1] << " " << fNMeshSegments[2]
|
|
<< G4endl;
|
|
}
|
|
|
|
// write quantity
|
|
long count = 0;
|
|
ofile << std::setprecision(16); // for double value with 8 bytes
|
|
for(int x = 0; x < fNMeshSegments[0]; x++) {
|
|
for(int y = 0; y < fNMeshSegments[1]; y++) {
|
|
for(int z = 0; z < fNMeshSegments[2]; z++) {
|
|
G4int idx = GetIndex(x, y, z);
|
|
|
|
if(opt.find("csv") != std::string::npos)
|
|
ofile << x << "," << y << "," << z << ",";
|
|
|
|
std::map<G4int, G4StatDouble*>::iterator value = score->find(idx);
|
|
if(value == score->end()) {
|
|
ofile << 0. << "," << 0. << "," << 0;
|
|
} else {
|
|
ofile << (value->second->sum_wx())/unitValue << ","
|
|
<< (value->second->sum_wx2())/unitValue/unitValue << ","
|
|
<< value->second->n();
|
|
}
|
|
|
|
if(opt.find("csv") != std::string::npos) {
|
|
ofile << G4endl;
|
|
} else if(opt.find("sequence") != std::string::npos) {
|
|
ofile << " ";
|
|
if(count++%5 == 4) ofile << G4endl;
|
|
}
|
|
|
|
} // z
|
|
} // y
|
|
} // x
|
|
ofile << std::setprecision(6);
|
|
|
|
// close the file
|
|
ofile.close();
|
|
|
|
}
|
|
|
|
void G4VScoreWriter::DumpAllQuantitiesToFile(const G4String& fileName,
|
|
const G4String& option) {
|
|
|
|
// change the option string into lowercase to the case-insensitive.
|
|
G4String opt = option;
|
|
std::transform(opt.begin(), opt.end(), opt.begin(), (int (*)(int))(tolower));
|
|
|
|
// confirm the option
|
|
if(opt.size() == 0) opt = "csv";
|
|
if(opt.find("csv") == std::string::npos &&
|
|
opt.find("sequence") == std::string::npos) {
|
|
G4cerr << "ERROR : DumpToFile : Unknown option -> "
|
|
<< option << G4endl;
|
|
return;
|
|
}
|
|
|
|
// open the file
|
|
std::ofstream ofile(fileName);
|
|
if(!ofile) {
|
|
G4cerr << "ERROR : DumpToFile : File open error -> "
|
|
<< fileName << G4endl;
|
|
return;
|
|
}
|
|
ofile << "# mesh name: " << fScoringMesh->GetWorldName() << G4endl;
|
|
|
|
// retrieve the map
|
|
MeshScoreMap fSMap = fScoringMesh->GetScoreMap();
|
|
MeshScoreMap::const_iterator msMapItr = fSMap.begin();
|
|
std::map<G4int, G4StatDouble*> * score;
|
|
for(; msMapItr != fSMap.end(); msMapItr++) {
|
|
|
|
G4String psname = msMapItr->first;
|
|
|
|
score = msMapItr->second->GetMap();
|
|
ofile << "# primitive scorer name: " << msMapItr->first << std::endl;
|
|
|
|
G4double unitValue = fScoringMesh->GetPSUnitValue(psname);
|
|
G4String unit = fScoringMesh->GetPSUnit(psname);
|
|
G4String divisionAxisNames[3];
|
|
fScoringMesh->GetDivisionAxisNames(divisionAxisNames);
|
|
// index order
|
|
ofile << "# i" << divisionAxisNames[0]
|
|
<< ", i" << divisionAxisNames[1]
|
|
<< ", i" << divisionAxisNames[2];
|
|
// unit of scored value
|
|
ofile << ", total(value) ";
|
|
if(unit.size() > 0) ofile << "[" << unit << "]";
|
|
ofile << ", total(val^2), entry" << G4endl;
|
|
|
|
|
|
// "sequence" option: write header info
|
|
if(opt.find("sequence") != std::string::npos) {
|
|
ofile << fNMeshSegments[0] << " " << fNMeshSegments[1] << " " << fNMeshSegments[2]
|
|
<< G4endl;
|
|
}
|
|
|
|
// write quantity
|
|
long count = 0;
|
|
ofile << std::setprecision(16); // for double value with 8 bytes
|
|
for(int x = 0; x < fNMeshSegments[0]; x++) {
|
|
for(int y = 0; y < fNMeshSegments[1]; y++) {
|
|
for(int z = 0; z < fNMeshSegments[2]; z++) {
|
|
G4int idx = GetIndex(x, y, z);
|
|
|
|
if(opt.find("csv") != std::string::npos)
|
|
ofile << x << "," << y << "," << z << ",";
|
|
|
|
std::map<G4int, G4StatDouble*>::iterator value = score->find(idx);
|
|
if(value == score->end()) {
|
|
ofile << 0. << "," << 0. << "," << 0;
|
|
} else {
|
|
ofile << (value->second->sum_wx())/unitValue << ","
|
|
<< (value->second->sum_wx2())/unitValue/unitValue << ","
|
|
<< value->second->n();
|
|
}
|
|
|
|
if(opt.find("csv") != std::string::npos) {
|
|
ofile << G4endl;
|
|
} else if(opt.find("sequence") != std::string::npos) {
|
|
ofile << " ";
|
|
if(count++%5 == 4) ofile << G4endl;
|
|
}
|
|
|
|
} // z
|
|
} // y
|
|
} // x
|
|
ofile << std::setprecision(6);
|
|
|
|
} // for(; msMapItr ....)
|
|
|
|
// close the file
|
|
ofile.close();
|
|
|
|
}
|
|
|
|
G4int G4VScoreWriter::GetIndex(G4int x, G4int y, G4int z) const {
|
|
//return x + y*fNMeshSegments[0] + z*fNMeshSegments[0]*fNMeshSegments[1];
|
|
return x*fNMeshSegments[1]*fNMeshSegments[2] +y*fNMeshSegments[2]+z;
|
|
}
|
|
|