Geant4 Cross Reference |
1 // 1 // 2 // ******************************************* 2 // ******************************************************************** 3 // * License and Disclaimer 3 // * License and Disclaimer * 4 // * 4 // * * 5 // * The Geant4 software is copyright of th 5 // * The Geant4 software is copyright of the Copyright Holders of * 6 // * the Geant4 Collaboration. It is provided 6 // * the Geant4 Collaboration. It is provided under the terms and * 7 // * conditions of the Geant4 Software License 7 // * conditions of the Geant4 Software License, included in the file * 8 // * LICENSE and available at http://cern.ch/ 8 // * LICENSE and available at http://cern.ch/geant4/license . These * 9 // * include a list of copyright holders. 9 // * include a list of copyright holders. * 10 // * 10 // * * 11 // * Neither the authors of this software syst 11 // * Neither the authors of this software system, nor their employing * 12 // * institutes,nor the agencies providing fin 12 // * institutes,nor the agencies providing financial support for this * 13 // * work make any representation or warran 13 // * work make any representation or warranty, express or implied, * 14 // * regarding this software system or assum 14 // * regarding this software system or assume any liability for its * 15 // * use. Please see the license in the file 15 // * use. Please see the license in the file LICENSE and URL above * 16 // * for the full disclaimer and the limitatio 16 // * for the full disclaimer and the limitation of liability. * 17 // * 17 // * * 18 // * This code implementation is the result 18 // * This code implementation is the result of the scientific and * 19 // * technical work of the GEANT4 collaboratio 19 // * technical work of the GEANT4 collaboration. * 20 // * By using, copying, modifying or distri 20 // * By using, copying, modifying or distributing the software (or * 21 // * any work based on the software) you ag 21 // * any work based on the software) you agree to acknowledge its * 22 // * use in resulting scientific publicati 22 // * use in resulting scientific publications, and indicate your * 23 // * acceptance of all terms of the Geant4 Sof 23 // * acceptance of all terms of the Geant4 Software license. * 24 // ******************************************* 24 // ******************************************************************** 25 // 25 // 26 26 27 // Author: Ivana Hrivnacova, 18/06/2013 (ivan 27 // Author: Ivana Hrivnacova, 18/06/2013 (ivana@ipno.in2p3.fr) 28 28 29 #include "G4HnMessenger.hh" 29 #include "G4HnMessenger.hh" 30 #include "G4HnManager.hh" 30 #include "G4HnManager.hh" 31 #include "G4AnalysisUtilities.hh" 31 #include "G4AnalysisUtilities.hh" >> 32 #include "G4AnalysisMessengerHelper.hh" 32 33 33 #include "G4UIcommand.hh" 34 #include "G4UIcommand.hh" 34 #include "G4UIparameter.hh" 35 #include "G4UIparameter.hh" >> 36 #include "G4UIcmdWithAnInteger.hh" 35 #include "G4UIcmdWithABool.hh" 37 #include "G4UIcmdWithABool.hh" 36 #include "G4UIcmdWithAString.hh" << 37 38 38 using namespace G4Analysis; 39 using namespace G4Analysis; 39 40 40 //____________________________________________ << 41 #include <iostream> >> 42 41 G4HnMessenger::G4HnMessenger(G4HnManager& mana 43 G4HnMessenger::G4HnMessenger(G4HnManager& manager) 42 : fManager(manager), << 44 : G4UImessenger(), 43 fHnType(manager.GetHnType()), << 45 fManager(manager), 44 fHnDimension(std::stoi(manager.GetHnType() << 46 fHelper(nullptr), 45 { << 47 fSetHnAsciiCmd(nullptr), >> 48 fSetHnActivationCmd(nullptr), >> 49 fSetHnActivationAllCmd(nullptr), >> 50 fSetHnPlottingCmd(nullptr), >> 51 fSetHnPlottingAllCmd(nullptr) >> 52 { >> 53 G4String hnType = fManager.GetHnType(); >> 54 hnType.toLower(); >> 55 fHelper = G4Analysis::make_unique<G4AnalysisMessengerHelper>(hnType); >> 56 46 SetHnAsciiCmd(); 57 SetHnAsciiCmd(); 47 SetHnActivationCmd(); 58 SetHnActivationCmd(); 48 SetHnActivationToAllCmd(); 59 SetHnActivationToAllCmd(); 49 SetHnPlottingCmd(); 60 SetHnPlottingCmd(); 50 SetHnPlottingToAllCmd(); 61 SetHnPlottingToAllCmd(); 51 SetHnFileNameCmd(); << 52 SetHnFileNameToAllCmd(); << 53 << 54 auto maxDim = (fHnDimension < kMaxDim) ? fHn << 55 for (unsigned int idim = 0; idim < maxDim; + << 56 fSetAxisLogCmd.push_back(CreateSetAxisLogC << 57 } << 58 } 62 } 59 63 60 //____________________________________________ 64 //_____________________________________________________________________________ 61 G4HnMessenger::~G4HnMessenger() = default; << 65 G4HnMessenger::~G4HnMessenger() >> 66 {} 62 67 63 // 68 // 64 // private functions 69 // private functions 65 // 70 // 66 71 67 //____________________________________________ 72 //_____________________________________________________________________________ 68 G4String G4HnMessenger::GetObjectType() const << 69 { << 70 return (fHnType[0] == 'h') ? << 71 fHnType.substr(1,1) + "D histogram" : << 72 fHnType.substr(1,1) + "D profile"; << 73 } << 74 << 75 //____________________________________________ << 76 void G4HnMessenger::AddIdParameter(G4UIcommand << 77 { << 78 auto htId = new G4UIparameter("id", 'i', fal << 79 htId->SetGuidance("Histogram id"); << 80 htId->SetParameterRange("id>=0"); << 81 command.SetParameter(htId); << 82 } << 83 << 84 //____________________________________________ << 85 void G4HnMessenger::AddOptionParameter(G4UIcom << 86 { << 87 auto param = new G4UIparameter(optionName, ' << 88 auto guidance = GetObjectType() + " " + opti << 89 param->SetGuidance(guidance.c_str()); << 90 param->SetDefaultValue("true"); << 91 command.SetParameter(param); << 92 } << 93 << 94 //____________________________________________ << 95 void G4HnMessenger::SetHnAsciiCmd() 73 void G4HnMessenger::SetHnAsciiCmd() 96 { 74 { 97 fSetAsciiCmd = << 75 fSetHnAsciiCmd 98 CreateCommand<G4UIcommand>("setAscii", "Pr << 76 = G4Analysis::make_unique<G4UIcmdWithAnInteger>(fHelper->Update("/analysis/HNTYPE_/setAscii"), this); 99 << 77 fSetHnAsciiCmd->SetGuidance( 100 AddIdParameter(*fSetAsciiCmd); << 78 fHelper->Update("Print NDIM_D LOBJECT of given id on ascii file.")); 101 AddOptionParameter(*fSetAsciiCmd, "hnAscii") << 79 102 << 80 fSetHnAsciiCmd->SetParameterName("id",false); >> 81 fSetHnAsciiCmd->SetRange("id>=0"); >> 82 fSetHnAsciiCmd->AvailableForStates(G4State_PreInit, G4State_Idle); 103 } 83 } 104 84 105 //____________________________________________ 85 //_____________________________________________________________________________ 106 void G4HnMessenger::SetHnActivationCmd() 86 void G4HnMessenger::SetHnActivationCmd() 107 { 87 { 108 fSetActivationCmd = << 88 auto hnId = new G4UIparameter("id", 'i', false); 109 CreateCommand<G4UIcommand>("setActivation" << 89 hnId->SetGuidance(fHelper->Update("OBJECT id")); 110 << 90 hnId->SetParameterRange("id>=0"); 111 AddIdParameter(*fSetActivationCmd); << 91 112 AddOptionParameter(*fSetActivationCmd, "hnAc << 92 auto hnActivation = new G4UIparameter("hnActivation", 's', true); 113 } << 93 hnActivation->SetGuidance(fHelper->Update("OBJECT activation")); >> 94 hnActivation->SetDefaultValue("none"); >> 95 >> 96 fSetHnActivationCmd >> 97 = G4Analysis::make_unique<G4UIcommand>(fHelper->Update("/analysis/HNTYPE_/setActivation"), this); >> 98 fSetHnActivationCmd->SetGuidance( >> 99 fHelper->Update("Set activation for the NDIM_D LOBJECT of given id")); >> 100 fSetHnActivationCmd->SetParameter(hnId); >> 101 fSetHnActivationCmd->SetParameter(hnActivation); >> 102 fSetHnActivationCmd->AvailableForStates(G4State_PreInit, G4State_Idle); >> 103 } 114 104 115 //____________________________________________ 105 //_____________________________________________________________________________ 116 void G4HnMessenger::SetHnActivationToAllCmd() 106 void G4HnMessenger::SetHnActivationToAllCmd() 117 { 107 { 118 fSetActivationAllCmd = << 108 fSetHnActivationAllCmd 119 CreateCommand<G4UIcmdWithABool>( << 109 = G4Analysis::make_unique<G4UIcmdWithABool>(fHelper->Update("/analysis/HNTYPE_/setActivationToAll"), this); 120 "setActivationToAll", "Set activation to << 110 fSetHnActivationAllCmd->SetGuidance( 121 fSetActivationAllCmd->SetParameterName("Acti << 111 fHelper->Update("Set activation to all NDIM_D LOBJECTs")); 122 } << 112 fSetHnActivationAllCmd->SetParameterName("Activation",false); 123 << 113 } >> 114 124 //____________________________________________ 115 //_____________________________________________________________________________ 125 void G4HnMessenger::SetHnPlottingCmd() 116 void G4HnMessenger::SetHnPlottingCmd() 126 { 117 { 127 fSetPlottingCmd = << 118 auto hnId = new G4UIparameter("id", 'i', false); 128 CreateCommand<G4UIcommand>("setPlotting", << 119 hnId->SetGuidance(fHelper->Update("OBJECT id")); 129 << 120 hnId->SetParameterRange("id>=0"); 130 AddIdParameter(*fSetPlottingCmd); << 121 131 AddOptionParameter(*fSetPlottingCmd, "hnPlot << 122 auto hnPlotting = new G4UIparameter("hnPlotting", 's', true); 132 } << 123 hnPlotting->SetGuidance(fHelper->Update("(In)Activate OBJECT plotting")); >> 124 hnPlotting->SetDefaultValue("none"); >> 125 >> 126 fSetHnPlottingCmd >> 127 = G4Analysis::make_unique<G4UIcommand>(fHelper->Update("/analysis/HNTYPE_/setPlotting"), this); >> 128 fSetHnPlottingCmd->SetGuidance( >> 129 fHelper->Update("(In)Activate batch plotting of the NDIM_D LOBJECT of given id")); >> 130 fSetHnPlottingCmd->SetParameter(hnId); >> 131 fSetHnPlottingCmd->SetParameter(hnPlotting); >> 132 fSetHnPlottingCmd->AvailableForStates(G4State_PreInit, G4State_Idle); >> 133 } 133 134 134 //____________________________________________ 135 //_____________________________________________________________________________ 135 void G4HnMessenger::SetHnPlottingToAllCmd() 136 void G4HnMessenger::SetHnPlottingToAllCmd() 136 { 137 { 137 fSetPlottingAllCmd = << 138 fSetHnPlottingAllCmd 138 CreateCommand<G4UIcmdWithABool>( << 139 = G4Analysis::make_unique<G4UIcmdWithABool>(fHelper->Update("/analysis/HNTYPE_/setPlottingToAll"), this); 139 "setPlottingToAll", "(In)Activate batch << 140 fSetHnPlottingAllCmd->SetGuidance( 140 fSetPlottingAllCmd->SetParameterName("Plotti << 141 fHelper->Update("(In)Activate batch plotting of all NDIM_D LOBJECTs")); 141 } << 142 fSetHnPlottingAllCmd->SetParameterName("Plotting",false); 142 << 143 } 143 //____________________________________________ << 144 void G4HnMessenger::SetHnFileNameCmd() << 145 { << 146 fSetFileNameCmd = << 147 CreateCommand<G4UIcommand>("setFileName", << 148 << 149 AddIdParameter(*fSetFileNameCmd); << 150 << 151 auto param = new G4UIparameter("hnFileName", << 152 auto guidance = GetObjectType() + " output f << 153 param->SetGuidance(guidance.c_str()); << 154 fSetFileNameCmd->SetParameter(param); << 155 } << 156 << 157 //____________________________________________ << 158 void G4HnMessenger::SetHnFileNameToAllCmd() << 159 { << 160 fSetFileNameAllCmd = << 161 CreateCommand<G4UIcmdWithAString>( << 162 "setFileNameToAll", "Set output file nam << 163 fSetFileNameAllCmd->SetParameterName("FileN << 164 } << 165 << 166 //____________________________________________ << 167 std::unique_ptr<G4UIcommand> << 168 G4HnMessenger::CreateSetAxisLogCommand(unsigne << 169 { << 170 G4String xyz{"XYZ"}; << 171 auto axis = xyz.substr(idim, 1); << 172 << 173 G4String commandName = "set" + axis + "axisL << 174 G4String guidance = "Activate " + axis + "- << 175 << 176 auto command = CreateCommand<G4UIcommand>(st << 177 command->AvailableForStates(G4State_PreInit, << 178 << 179 // Add Id parameter << 180 AddIdParameter(*command); << 181 << 182 auto parAxisLog = new G4UIparameter("axis", << 183 guidance = GetObjectType() + " " + std::move << 184 parAxisLog->SetGuidance(guidance.c_str()); << 185 command->SetParameter(parAxisLog); << 186 << 187 return command; << 188 } << 189 144 190 // 145 // 191 // public methods 146 // public methods 192 // 147 // 193 148 194 //____________________________________________ 149 //_____________________________________________________________________________ 195 void G4HnMessenger::SetNewValue(G4UIcommand* c 150 void G4HnMessenger::SetNewValue(G4UIcommand* command, G4String newValues) 196 { 151 { 197 // process "All" commands first << 152 if ( command == fSetHnAsciiCmd.get() ) { 198 if ( command == fSetActivationAllCmd.get() ) << 153 auto id = fSetHnAsciiCmd->GetNewIntValue(newValues); 199 fManager.SetActivation(fSetActivationAllCm << 154 fManager.SetAscii(id, true); 200 return; << 155 } 201 } << 156 else if ( command == fSetHnActivationCmd.get() ) { 202 << 157 // tokenize parameters in a vector 203 if ( command == fSetPlottingAllCmd.get() ) { << 158 std::vector<G4String> parameters; 204 fManager.SetPlotting(fSetPlottingAllCmd->G << 159 G4Analysis::Tokenize(newValues, parameters); 205 return; << 160 // check consistency 206 } << 161 if ( G4int(parameters.size()) == command->GetParameterEntries() ) { 207 << 162 auto counter = 0; 208 if ( command == fSetFileNameAllCmd.get() ) { << 163 auto id = G4UIcommand::ConvertToInt(parameters[counter++]); 209 fManager.SetFileName(newValues); << 164 auto activation = G4UIcommand::ConvertToBool(parameters[counter++]); 210 return; << 165 fManager.SetActivation(id, activation); 211 } << 212 << 213 // Tokenize parameters in a vector << 214 std::vector<G4String> parameters; << 215 G4Analysis::Tokenize(newValues, parameters); << 216 // check consistency << 217 if ( parameters.size() != command->GetParame << 218 // Should never happen but let's check any << 219 G4Analysis::Warn( << 220 "Got wrong number of \"" + command->GetC << 221 "\" parameters: " + std::to_string(param << 222 " instead of " + std::to_string(command- << 223 fkClass, "WarnAboutParameters"); << 224 return; << 225 } << 226 << 227 auto counter = 0; << 228 auto id = G4UIcommand::ConvertToInt(paramete << 229 << 230 if ( command == fSetAsciiCmd.get() ) { << 231 fManager.SetAscii(id, G4UIcommand::Convert << 232 return; << 233 } << 234 << 235 if ( command == fSetActivationCmd.get() ) { << 236 fManager.SetActivation(id, G4UIcommand::Co << 237 return; << 238 } << 239 << 240 if ( command == fSetPlottingCmd.get() ) { << 241 fManager.SetPlotting(id, G4UIcommand::Conv << 242 return; << 243 } << 244 << 245 if ( command == fSetFileNameCmd.get() ) { << 246 fManager.SetFileName(id, parameters[counte << 247 return; << 248 } << 249 << 250 auto maxDim = (fHnDimension < kMaxDim) ? fHn << 251 for (unsigned int idim = 0; idim < maxDim; + << 252 if ( command == fSetAxisLogCmd[idim].get() << 253 auto axisLog = G4UIcommand::ConvertToBoo << 254 fManager.SetAxisIsLog(idim, id, axisLog) << 255 return; << 256 } 166 } 257 } << 167 else { 258 } << 168 // Should never happen but let's check anyway for consistency >> 169 fHelper->WarnAboutParameters(command, parameters.size()); >> 170 } >> 171 } >> 172 else if ( command == fSetHnActivationAllCmd.get() ) { >> 173 auto activation = fSetHnActivationAllCmd->GetNewBoolValue(newValues); >> 174 fManager.SetActivation(activation); >> 175 } >> 176 else if ( command == fSetHnPlottingCmd.get() ) { >> 177 // tokenize parameters in a vector >> 178 std::vector<G4String> parameters; >> 179 G4Analysis::Tokenize(newValues, parameters); >> 180 // check consistency >> 181 if ( G4int(parameters.size()) == command->GetParameterEntries() ) { >> 182 auto counter = 0; >> 183 auto id = G4UIcommand::ConvertToInt(parameters[counter++]); >> 184 auto activation = G4UIcommand::ConvertToBool(parameters[counter++]); >> 185 fManager.SetPlotting(id, activation); >> 186 } >> 187 else { >> 188 // Should never happen but let's check anyway for consistency >> 189 fHelper->WarnAboutParameters(command, parameters.size()); >> 190 } >> 191 } >> 192 else if ( command == fSetHnPlottingAllCmd.get() ) { >> 193 auto activation = fSetHnPlottingAllCmd->GetNewBoolValue(newValues); >> 194 fManager.SetPlotting(activation); >> 195 } >> 196 } 259 197