| 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 #ifdef USE_INFERENCE 26 #ifdef USE_INFERENCE
27 # ifndef PAR04INFERENCEMESSENGER_H << 27 #ifndef PAR04INFERENCEMESSENGER_H
28 # define PAR04INFERENCEMESSENGER_H << 28 #define PAR04INFERENCEMESSENGER_H
29 29
30 # include "G4UImessenger.hh" // for G4UIme << 30 #include "G4UImessenger.hh" // for G4UImessenger
31 << 31 #include <G4String.hh> // for G4String
32 # include <G4String.hh> // for G4String <<
33 class G4UIcmdWithADoubleAndUnit; 32 class G4UIcmdWithADoubleAndUnit;
34 class G4UIcmdWithAString; 33 class G4UIcmdWithAString;
35 class G4UIcmdWithAnInteger; 34 class G4UIcmdWithAnInteger;
36 class G4UIcommand; 35 class G4UIcommand;
37 class G4UIdirectory; 36 class G4UIdirectory;
38 class Par04InferenceSetup; 37 class Par04InferenceSetup;
39 38
40 /** 39 /**
41 * @brief Inference messenger. 40 * @brief Inference messenger.
42 * 41 *
43 * Provides UI commands to setup the inference 42 * Provides UI commands to setup the inference: name of the inference library,
44 * path to the ML model, size of the latent sp 43 * path to the ML model, size of the latent space, the size of the condition
45 * vector and flags for optimization and profi 44 * vector and flags for optimization and profiling. It allows to specify the
46 * mesh size (in cylindrical coordinates) that 45 * mesh size (in cylindrical coordinates) that was used in training dataset
47 * (full simulation). 46 * (full simulation).
48 * 47 *
49 */ 48 */
50 49
51 class Par04InferenceMessenger : public G4UImes << 50 class Par04InferenceMessenger : public G4UImessenger {
52 { << 51 public:
53 public: << 52 Par04InferenceMessenger(Par04InferenceSetup *);
54 Par04InferenceMessenger(Par04InferenceSetu << 53 ~Par04InferenceMessenger();
55 ~Par04InferenceMessenger(); << 54 /// Invokes appropriate methods based on the typed command
56 /// Invokes appropriate methods based on t << 55 virtual void SetNewValue(G4UIcommand *, G4String) final;
57 virtual void SetNewValue(G4UIcommand*, G4S << 56 /// Retrieves the current settings
58 /// Retrieves the current settings << 57 virtual G4String GetCurrentValue(G4UIcommand *) final;
59 virtual G4String GetCurrentValue(G4UIcomma << 58
60 << 59 private:
61 private: << 60 /// Inference to setup
62 /// Inference to setup << 61 Par04InferenceSetup *fInference = nullptr;
63 Par04InferenceSetup* fInference = nullptr; << 62 /// Command to set the directory common to all inference messengers in this
64 /// Command to set the directory common to << 63 /// example /Par04
65 /// example /Par04 << 64 G4UIdirectory *fExampleDir = nullptr;
66 G4UIdirectory* fExampleDir = nullptr; << 65 /// Command to set the directory for inference settings /Par04/inference
67 /// Command to set the directory for infer << 66 G4UIdirectory *fInferenceDir = nullptr;
68 G4UIdirectory* fInferenceDir = nullptr; << 67 /// Command to set the inference library
69 /// Command to set the inference library << 68 G4UIcmdWithAString *fInferenceLibraryCmd = nullptr;
70 G4UIcmdWithAString* fInferenceLibraryCmd = << 69 /// Command to set fModelPathNameCmd
71 /// Command to set fModelPathNameCmd << 70 G4UIcmdWithAString *fModelPathNameCmd = nullptr;
72 G4UIcmdWithAString* fModelPathNameCmd = nu << 71 /// Command to set the fSizeLatentVectorCmd
73 /// Command to set the fSizeLatentVectorCm << 72 G4UIcmdWithAnInteger *fSizeLatentVectorCmd = nullptr;
74 G4UIcmdWithAnInteger* fSizeLatentVectorCmd << 73 /// Command to set the fSizeConditionVectorCmd
75 /// Command to set the fSizeConditionVecto << 74 G4UIcmdWithAnInteger *fSizeConditionVectorCmd = nullptr;
76 G4UIcmdWithAnInteger* fSizeConditionVector << 75 /// Command to set the fProfileFlagCmd
77 /// Command to set the fProfileFlagCmd << 76 G4UIcmdWithAnInteger *fProfileFlagCmd = nullptr;
78 G4UIcmdWithAnInteger* fProfileFlagCmd = nu << 77 /// Command to set the fOptimizationFlagCmd
79 /// Command to set the fOptimizationFlagCm << 78 G4UIcmdWithAnInteger *fOptimizationFlagCmd = nullptr;
80 G4UIcmdWithAnInteger* fOptimizationFlagCmd << 79 /// Command to set the number of cells in the cylindrical readout mesh (along
81 /// Command to set the number of cells in << 80 /// rho axis)
82 /// rho axis) << 81 G4UIcmdWithAnInteger *fMeshNbRhoCellsCmd = nullptr;
83 G4UIcmdWithAnInteger* fMeshNbRhoCellsCmd = << 82 /// Command to set the number of cells in the cylindrical readout mesh (along
84 /// Command to set the number of cells in << 83 /// phi axis)
85 /// phi axis) << 84 G4UIcmdWithAnInteger *fMeshNbPhiCellsCmd = nullptr;
86 G4UIcmdWithAnInteger* fMeshNbPhiCellsCmd = << 85 /// Command to set the number of cells in the cylindrical readout mesh (along
87 /// Command to set the number of cells in << 86 /// z axis)
88 /// z axis) << 87 G4UIcmdWithAnInteger *fMeshNbZCellsCmd = nullptr;
89 G4UIcmdWithAnInteger* fMeshNbZCellsCmd = n << 88 /// Command to the size of cells in the cylindrical readout mesh (along rho
90 /// Command to the size of cells in the cy << 89 /// axis)
91 /// axis) << 90 G4UIcmdWithADoubleAndUnit *fMeshSizeRhoCellsCmd = nullptr;
92 G4UIcmdWithADoubleAndUnit* fMeshSizeRhoCel << 91 /// Command to the size of cells in the cylindrical readout mesh (along z
93 /// Command to the size of cells in the cy << 92 /// axis)
94 /// axis) << 93 G4UIcmdWithADoubleAndUnit *fMeshSizeZCellsCmd = nullptr;
95 G4UIcmdWithADoubleAndUnit* fMeshSizeZCells << 94 /// Commands to set execution provider flags
96 /// Commands to set execution provider fla << 95 /// GPU
97 /// GPU << 96 G4UIcmdWithAnInteger *fCudaFlagCmd = nullptr;
98 G4UIcmdWithAnInteger* fCudaFlagCmd = nullp << 97 /// Cuda Commands
99 /// Cuda Commands << 98 G4UIdirectory *fCudaOptionsDir = nullptr;
100 G4UIdirectory* fCudaOptionsDir = nullptr; << 99 G4UIcmdWithAString *fCudaDeviceIdCmd = nullptr;
101 G4UIcmdWithAString* fCudaDeviceIdCmd = nul << 100 G4UIcmdWithAString *fCudaGpuMemLimitCmd = nullptr;
102 G4UIcmdWithAString* fCudaGpuMemLimitCmd = << 101 G4UIcmdWithAString *fCudaArenaExtendedStrategyCmd = nullptr;
103 G4UIcmdWithAString* fCudaArenaExtendedStra << 102 G4UIcmdWithAString *fCudaCudnnConvAlgoSearchCmd = nullptr;
104 G4UIcmdWithAString* fCudaCudnnConvAlgoSear << 103 G4UIcmdWithAString *fCudaDoCopyInDefaultStreamCmd = nullptr;
105 G4UIcmdWithAString* fCudaDoCopyInDefaultSt << 104 G4UIcmdWithAString *fCudaCudnnConvUseMaxWorkspaceCmd = nullptr;
106 G4UIcmdWithAString* fCudaCudnnConvUseMaxWo <<
107 }; 105 };
108 106
109 # endif << 107 #endif
110 #endif 108 #endif
111 109