ECCE @ EIC Software
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Groups Pages
RE03UserScoreWriter.cc
Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file RE03UserScoreWriter.cc
1 //
2 // ********************************************************************
3 // * License and Disclaimer *
4 // * *
5 // * The Geant4 software is copyright of the Copyright Holders of *
6 // * the Geant4 Collaboration. It is provided under the terms and *
7 // * conditions of the Geant4 Software License, included in the file *
8 // * LICENSE and available at http://cern.ch/geant4/license . These *
9 // * include a list of copyright holders. *
10 // * *
11 // * Neither the authors of this software system, nor their employing *
12 // * institutes,nor the agencies providing financial support for this *
13 // * work make any representation or warranty, express or implied, *
14 // * regarding this software system or assume any liability for its *
15 // * use. Please see the license in the file LICENSE and URL above *
16 // * for the full disclaimer and the limitation of liability. *
17 // * *
18 // * This code implementation is the result of the scientific and *
19 // * technical work of the GEANT4 collaboration. *
20 // * By using, copying, modifying or distributing the software (or *
21 // * any work based on the software) you agree to acknowledge its *
22 // * use in resulting scientific publications, and indicate your *
23 // * acceptance of all terms of the Geant4 Software license. *
24 // ********************************************************************
25 //
28 //
29 //
30 
31 #include "RE03UserScoreWriter.hh"
32 
34 #include "G4SDParticleFilter.hh"
35 #include "G4VPrimitiveScorer.hh"
36 #include "G4VScoringMesh.hh"
37 
38 #include <map>
39 #include <fstream>
40 
41 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
43  : G4VScoreWriter()
44 {;}
45 
46 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
48 {;}
49 
50 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
52  const G4String & fileName,
53  const G4String & option) {
54  using MeshScoreMap = G4VScoringMesh::MeshScoreMap;
55  //
56  if(verboseLevel > 0) {
57  G4cout << "User-defined DumpQuantityToFile() method is invoked."
58  << G4endl;
59  G4cout << " -- to obtain a projection of the quantity <"
60  << psName
61  << "> onto the x-y plane --" << G4endl;
62  }
63 
64  // change the option string into lowercase to the case-insensitive.
65  G4String opt = option;
66  std::transform(opt.begin(), opt.end(), opt.begin(), (int (*)(int))(tolower));
67 
68  // confirm the option
69  if(opt.size() == 0) opt = "csv";
70 
71  // open the file
72  std::ofstream ofile(fileName);
73  if(!ofile) {
74  G4cerr << "ERROR : DumpToFile : File open error -> "
75  << fileName << G4endl;
76  return;
77  }
78  ofile << "# mesh name: " << fScoringMesh->GetWorldName() << G4endl;
79 
80 
81  // retrieve the map
82  MeshScoreMap scMap = fScoringMesh->GetScoreMap();
83 
84  MeshScoreMap::const_iterator msMapItr = scMap.find(psName);
85  if(msMapItr == scMap.end()) {
86  G4cerr << "ERROR : DumpToFile : Unknown quantity, \""
87  << psName << "\"." << G4endl;
88  return;
89  }
90  std::map<G4int, G4StatDouble*> * score = msMapItr->second->GetMap();
91  ofile << "# primitive scorer name: " << msMapItr->first << G4endl;
92 
93  // write header info
94  ofile << "# xy projection" << G4endl;
95  ofile << fNMeshSegments[0] << " " << fNMeshSegments[1] << " " << G4endl;
96 
97  // declare xy array
98  std::vector<double> projy;
99  for(int y = 0; y < fNMeshSegments[1]; y++) projy.push_back(0.);
100  std::vector<std::vector<double> > projxy;
101  for(int x = 0; x < fNMeshSegments[0]; x++) projxy.push_back(projy);
102  // accumulate
103  ofile << std::setprecision(16); // for double value with 8 bytes
104  for(int x = 0; x < fNMeshSegments[0]; x++) {
105  for(int y = 0; y < fNMeshSegments[1]; y++) {
106  for(int z = 0; z < fNMeshSegments[2]; z++) {
107 
108  G4int idx = GetIndex(x, y, z);
109 
110  std::map<G4int, G4StatDouble*>::iterator value = score->find(idx);
111  if(value != score->end()) projxy[x][y] += value->second->sum_wx();
112 
113  } // z
114  } // y
115  } // x
116 
117  // write quantity
118  ofile << std::setprecision(16); // for double value with 8 bytes
119  for(int x = 0; x < fNMeshSegments[0]; x++) {
120  for(int y = 0; y < fNMeshSegments[1]; y++) {
121 
122  ofile << x << "," << y << ",";
123  ofile << projxy[x][y] << G4endl;
124 
125  } // y
126  } // x
127  ofile << std::setprecision(6);
128 
129  // close the file
130  ofile.close();
131 
132 }
133