Geant4 Cross Reference

Cross-Referencing   Geant4
Geant4/examples/extended/parameterisations/Par04/include/Par04InferenceMessenger.hh

Version: [ ReleaseNotes ] [ 1.0 ] [ 1.1 ] [ 2.0 ] [ 3.0 ] [ 3.1 ] [ 3.2 ] [ 4.0 ] [ 4.0.p1 ] [ 4.0.p2 ] [ 4.1 ] [ 4.1.p1 ] [ 5.0 ] [ 5.0.p1 ] [ 5.1 ] [ 5.1.p1 ] [ 5.2 ] [ 5.2.p1 ] [ 5.2.p2 ] [ 6.0 ] [ 6.0.p1 ] [ 6.1 ] [ 6.2 ] [ 6.2.p1 ] [ 6.2.p2 ] [ 7.0 ] [ 7.0.p1 ] [ 7.1 ] [ 7.1.p1 ] [ 8.0 ] [ 8.0.p1 ] [ 8.1 ] [ 8.1.p1 ] [ 8.1.p2 ] [ 8.2 ] [ 8.2.p1 ] [ 8.3 ] [ 8.3.p1 ] [ 8.3.p2 ] [ 9.0 ] [ 9.0.p1 ] [ 9.0.p2 ] [ 9.1 ] [ 9.1.p1 ] [ 9.1.p2 ] [ 9.1.p3 ] [ 9.2 ] [ 9.2.p1 ] [ 9.2.p2 ] [ 9.2.p3 ] [ 9.2.p4 ] [ 9.3 ] [ 9.3.p1 ] [ 9.3.p2 ] [ 9.4 ] [ 9.4.p1 ] [ 9.4.p2 ] [ 9.4.p3 ] [ 9.4.p4 ] [ 9.5 ] [ 9.5.p1 ] [ 9.5.p2 ] [ 9.6 ] [ 9.6.p1 ] [ 9.6.p2 ] [ 9.6.p3 ] [ 9.6.p4 ] [ 10.0 ] [ 10.0.p1 ] [ 10.0.p2 ] [ 10.0.p3 ] [ 10.0.p4 ] [ 10.1 ] [ 10.1.p1 ] [ 10.1.p2 ] [ 10.1.p3 ] [ 10.2 ] [ 10.2.p1 ] [ 10.2.p2 ] [ 10.2.p3 ] [ 10.3 ] [ 10.3.p1 ] [ 10.3.p2 ] [ 10.3.p3 ] [ 10.4 ] [ 10.4.p1 ] [ 10.4.p2 ] [ 10.4.p3 ] [ 10.5 ] [ 10.5.p1 ] [ 10.6 ] [ 10.6.p1 ] [ 10.6.p2 ] [ 10.6.p3 ] [ 10.7 ] [ 10.7.p1 ] [ 10.7.p2 ] [ 10.7.p3 ] [ 10.7.p4 ] [ 11.0 ] [ 11.0.p1 ] [ 11.0.p2 ] [ 11.0.p3, ] [ 11.0.p4 ] [ 11.1 ] [ 11.1.1 ] [ 11.1.2 ] [ 11.1.3 ] [ 11.2 ] [ 11.2.1 ] [ 11.2.2 ] [ 11.3.0 ]

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