add angle conditioning

• extend Geometry interface with localDirection(G4FastTrack const& aFastTrack) - to compute the local direction in a right handed coordinate system where the z-axis points into the calorimeter, normal to the calorimeter planes
• add localDir to ModelInterface::prepareInput(...) and ModelInterface::convertOutput(...) to allow for angular conditioning on one or two angles
• implement this in RegularGridBIBAEModel
  • using condition angle as local theta (polar angle) and rotate output of inference model by local azimuth angle phi
• use angular conditioning in Par04VAE
• add property CorrectForAngle to geometry models PolyhedraBarrelGeometry and EndcapGeometry
  • use true for GAN and Par04VAE, false for BIB-AE

include/DDML/EndcapGeometry.h

@@ -18,7 +18,9 @@ namespace ddml {
   class EndcapGeometry : public GeometryInterface {
     
   public:
-    EndcapGeometry(){} ;
+    EndcapGeometry(){
+      initialize() ;
+    } ;
 
     /// initialize the plugin - after properties have been set
     void initialize() ;
@@ -29,21 +31,26 @@ namespace ddml {
     void declareProperties( dd4hep::sim::Geant4Action* plugin ) {
 
       plugin->declareProperty("Detector" , this->_detector ) ;
-
+      plugin->declareProperty("CorrectForAngles" , this->_correctForAngles ) ;
     }
   
+    /** compute local direction in coordinate system that has the z-axis pointing into the calorimeter,
+     *  normal to the layers
+     */
+    G4ThreeVector localDirection(G4FastTrack const& aFastTrack)  const ;
+
     /** convert the local spacepoints to global spacepoints
      */
     virtual void localToGlobal(G4FastTrack const& aFastTrack,
-			       std::vector<SpacePointVec>& spacepoints ) ;
+			       std::vector<SpacePointVec>& spacepoints ) const  ;
     
     
   private:
     std::vector<float> _caloLayerDistances ={} ;
-    bool _isInitialized = false ;
 
     /// model properties for plugin
     std::string _detector = { "EcalEndcap" } ;
+    bool _correctForAngles = false  ;
   };
 
 } // namespace