MagneticField/GeomBuilder/src/BaseVolumeHandle.h - master - githubmirror/cmssw

MagneticField/GeomBuilder/src/BaseVolumeHandle.h
224 строки7 KB
Carl Vuosalo
One blank line removed by code-format
13 сен 2019, 22:31
13 сен 2019, 22:31b20b626
#ifndef MagneticField_GeomBuilder_BaseVolumeHandle_H
#define MagneticField_GeomBuilder_BaseVolumeHandle_H

/** \class BaseVolumeHandle
 * A temporary container to cache info on a six-surface volume during
 * the processing. Used to sort, organise, and build shared planes.
 * One instance is created for each DDVolume. The parameters of the 
 * boundary surfaces are calculated during construction.
 *
 *  \author N. Amapane - INFN Torino (original developer)
 */

#include "DataFormats/GeometrySurface/interface/Surface.h"
#include "DataFormats/GeometrySurface/interface/ReferenceCounted.h"
#include "DetectorDescription/Core/interface/DDSolidShapes.h"
#include "MagneticField/VolumeGeometry/interface/VolumeSide.h"

class MagVolume6Faces;

namespace magneticfield {

  class BaseVolumeHandle {
  public:
    typedef Surface::GlobalPoint GlobalPoint;
    typedef Surface::LocalPoint LocalPoint;
    typedef Surface::LocalVector LocalVector;
    typedef SurfaceOrientation::GlobalFace Sides;

    BaseVolumeHandle(bool expand2Pi = false, bool debugVal = false);

    // Disallow Default/copy ctor
    // (we want to handle only pointers!!!)
    BaseVolumeHandle(const BaseVolumeHandle& v) = delete;

    virtual ~BaseVolumeHandle();

    /// Return the center of the volume
    const GlobalPoint& center() const;

    /// Distance of (x,y) plane from origin
    const double RN() const { return theRN; }

    /// Get the current surface on specified side.
    const Surface& surface(int which_side) const;

    const Surface& surface(Sides which_side) const;

    /// Find out if two surfaces are the same physical surface
    bool sameSurface(const Surface& s1, Sides which_side, float tolerance = 0.01);

    /// Assign a shared surface perorming sanity checks.
    bool setSurface(const Surface& s1, Sides which_side);

    /// if the specified surface has been matched.
    bool isPlaneMatched(int which_side) const { return isAssigned[which_side]; }

    int references(int which_side) const {  // FIXME!
      /*     return surfaces[which_side]->references(); */
      return 0;
    }

    /// Name of the volume
    std::string name;

    /// Name of magnetic field table file
    std::string magFile;

    /// volume number
    unsigned short volumeno;

    /// copy number
    unsigned short copyno;

    // Phi ranges: Used by: LessDPhiMax; bSector; bSlab::[min|max]Phi();
    // MagBSector, MagBRod

    /// Minimum value of phi covered by the volume
    // FIXME: actually returns phi of the point on median plane of the -phi
    // surface, except for trapezoids where the absoulte min has been implemented
    Geom::Phi<float> minPhi() const { return thePhiMin; }

    /// Maximum value of phi covered by the volume
    // FIXME: actually returns phi of the point on median plane of the +phi
    // surface
    Geom::Phi<float> maxPhi() const { return surface(SurfaceOrientation::phiplus).position().phi(); }

    /// Z limits.
    // ASSUMPTION: Computed on median Z plane, but this is not a problem since
    // all Z planes are orthogonal to the beam line in the current geometry.
    double minZ() const { return surface(SurfaceOrientation::zminus).position().z(); }
    double maxZ() const { return surface(SurfaceOrientation::zplus).position().z(); }

    /// Minimum R for any point within the volume
    double minR() const { return theRMin; }

    /// Position and rotation
    const GloballyPositioned<float>* placement() const { return refPlane; }

    /// The surfaces and they orientation, as required to build a MagVolume.
    virtual std::vector<VolumeSide> sides() const = 0;

    /// Pointer to the final MagVolume (must be set from outside)
    MagVolume6Faces* magVolume;

    bool toExpand() const { return expand; }

    /// Temporary hack to pass information on material. Will eventually be replaced!
    bool isIron() const { return isIronFlag; }

    /// The sector for which an interpolator for this class of volumes should be built
    int masterSector;

    /// Shape of the solid
    virtual DDSolidShape shape() const = 0;

  protected:
    typedef ConstReferenceCountingPointer<Surface> RCPS;

    // The volume's six surfaces.
    RCPS surfaces[6];
    // If the corresponding surface has been assigned to a shared surface.
    bool isAssigned[6];

    // Build phi, z surfaces (common for ddtubs and ddcons)
    void buildPhiZSurf(double startPhi, double deltaPhi, double zhalf, double rCentr);

    // Distance from the origin along the normal to the volume's zphi plane.
    double theRN;

    // Max and min radius for _any_ point within the volume
    // FIXME!
    double theRMin;
    double theRMax;
    Geom::Phi<float> thePhiMin;

    // The refPlane is the "main plane" for the solid. It corresponds to the
    // x,y plane in the DDD local frame, and defines a frame where the local
    // coordinates are the same as in DDD.
    GloballyPositioned<float>* refPlane;

    // the center of the volume
    GlobalPoint center_;

    // Flag this as a master volume out of wich a 2pi volume should be built
    // (e.g. central cylinder); this is taken into account by sides().
    bool expand;

    // Temporary hack to keep information on material. Will eventually be replaced!
    bool isIronFlag;

    const bool debug;
  };

  typedef std::vector<BaseVolumeHandle*> handles;

  // Extractors for precomputed_value_sort() (safe sorting)

  // To sort volumes in Z
  struct ExtractZ {
    double operator()(const BaseVolumeHandle* v) const { return v->center().z(); }
  };

  // To sort volumes in abs(Z)
  struct ExtractAbsZ {
    double operator()(const BaseVolumeHandle* v) const { return fabs(v->center().z()); }
  };

  // To sort volumes in phi (from -pi to pi).
  struct ExtractPhi {
    double operator()(const BaseVolumeHandle* v) const {
      // note that Geom::Phi is implicitly converted to double.
      // Periodicity is guaranteed.
      return v->center().phi();
    }
  };

  // To sort volumes based on max phi(from -pi to pi).
  struct ExtractPhiMax {
    double operator()(const BaseVolumeHandle* v) const {
      // note that Geom::Phi is implicitly converted to double.
      // Periodicity is guaranteed.
      return v->maxPhi();
    }
  };

  // To sort volumes in R
  struct ExtractR {
    double operator()(const BaseVolumeHandle* v) const { return v->center().perp(); }
  };

  // To sort volumes in RN (distance of (x,y) plane from origin)
  struct ExtractRN {
    double operator()(const BaseVolumeHandle* v) const { return v->RN(); }
  };

  // To sort angles within any range SMALLER THAN PI "counter-clockwise",
  // even if the angles cross the pi boundary.
  // CAVEAT: // The result is undefined if the input values cover a
  // range larger than pi!!!
  struct LessDPhi {
    bool operator()(double phi1, double phi2) const {
      // handle periodicity
      return ((Geom::Phi<float>(phi2) - Geom::Phi<float>(phi1)) > 0.);
    }
  };

  // Compare the Z of volumes.
  // Should be used ONLY for std::max_element and std::min_element
  // and NEVER for sorting (use precomputed_value_sort with ExtractZ instead)
  struct LessZ {
    bool operator()(const BaseVolumeHandle* v1, const BaseVolumeHandle* v2) const {
      if (v1->center().z() < v2->center().z())
        return true;
      return false;
    }
  };

  inline const char* const newln = "\n";
  // Newline for formatting debug messages
  // Intended to be similar to "endl"

}  // namespace magneticfield

#endif