Skip to content

add angle conditioning

Frank Gaede requested to merge angle_cond into main
  • 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
Edited by Frank Gaede

Merge request reports