OpDetGeo.h

Go to the documentation of this file.

#ifndef LARCOREALG_GEOMETRY_OPDETGEO_H
#define LARCOREALG_GEOMETRY_OPDETGEO_H

// LArSoft libraries
#include "larcorealg/Geometry/LocalTransformationGeo.h"
#include "larcoreobj/SimpleTypesAndConstants/geo_vectors.h"

// ROOT libraries
#include "TGeoMatrix.h" // TGeoHMatrix
#include "TGeoTube.h"
#include "TGeoBBox.h"
#include "TClass.h"

// C/C++ standard libraries
#include <vector>
#include <array>


// forward declarations
class TGeoNode;

namespace geo {

  class OpDetGeo {
  public:
    

    struct OpDetGeoCoordinatesTag {};
    
    using LocalPoint_t = geo::Point3DBase_t<OpDetGeoCoordinatesTag>;
    
    using LocalVector_t = geo::Vector3DBase_t<OpDetGeoCoordinatesTag>;
    
    
    OpDetGeo(std::vector<const TGeoNode*>& path, 
            int depth);

    void   GetCenter(double* xyz, double localz=0.0) const;
    geo::Point_t const& GetCenter() const { return fCenter; }
    double RMin() const;
    double RMax() const;
    double HalfL() const;
    double HalfW() const;
    double HalfH() const;
    double Length() const { return 2.0 * HalfL(); }
    double Width() const { return 2.0 * HalfW(); }
    double Height() const { return 2.0 * HalfH(); }
    double ThetaZ() const;  
    double ThetaZ(bool degrees) const; 

    double CosThetaFromNormal(geo::Point_t const& point) const;
    double CosThetaFromNormal(double const* xyz) const;

    double DistanceToPoint(geo::Point_t const& point) const;
    double DistanceToPoint(double const* xyz) const;



    void LocalToWorld(const double* opdet, double* world) const
      { fTrans.LocalToWorld(opdet, world); }
      
    geo::Point_t toWorldCoords(LocalPoint_t const& local) const
      { return fTrans.toWorldCoords(local); }
    
    void LocalToWorldVect(const double* opdet, double* world) const
      { fTrans.LocalToWorldVect(opdet, world); }
    
    geo::Vector_t toWorldCoords(LocalVector_t const& local) const
      { return fTrans.toWorldCoords(local); }
    
    void WorldToLocal(const double* world, double* opdet) const
      { fTrans.WorldToLocal(world, opdet); }
    
    LocalPoint_t toLocalCoords(geo::Point_t const& world) const
      { return fTrans.toLocalCoords(world); }
    
    void WorldToLocalVect(const double* world, double* opdet) const
      { fTrans.WorldToLocalVect(world, opdet); }
    
    LocalVector_t toLocalCoords(geo::Vector_t const& world) const
      { return fTrans.toLocalCoords(world); }
    
    
    const TGeoNode*     Node() const { return fOpDetNode; }
    
    TGeoShape const* Shape() const { return Node()->GetVolume()->GetShape(); }
    
    bool isTube() const { return asTube() != nullptr; }
    
    bool isBar() const { return (asBox() != nullptr) && !isTube(); }
    
    template <typename Stream>
    void PrintOpDetInfo
      (Stream&& out, std::string indent = "", unsigned int verbosity = 0) const;
    
    static constexpr unsigned int MaxVerbosity = 2;
    
  private:
    using LocalTransformation_t
      = geo::LocalTransformationGeo<TGeoHMatrix, LocalPoint_t, LocalVector_t>;
    
    LocalTransformation_t fTrans; 
    const TGeoNode* fOpDetNode;  
    geo::Point_t fCenter; 
    
    TGeoTube const* asTube() const
      { return dynamic_cast<TGeoTube const*>(Shape()); }
    
    TGeoBBox const* asBox() const
      { return dynamic_cast<TGeoBBox const*>(Shape()); }
    
  }; // class OpDetGeo
  
} // namespace geo


//------------------------------------------------------------------------------
//--- template implementation
//---
template <typename Stream>
void geo::OpDetGeo::PrintOpDetInfo(
  Stream&& out,
  std::string indent /* = "" */,
  unsigned int verbosity /* = 0 */
) const {
  
  //----------------------------------------------------------------------------
  out << "centered at " << GetCenter() << " cm";
  
  if (verbosity-- <= 0) return; // 0
  
  //----------------------------------------------------------------------------
  if (isTube()) {
    out << ", radius: " << RMax() << " cm";
    if (RMin() != 0.0) out << " (inner: " << RMin() << " cm)";
    out << ", length: " << Length() << " cm";
  }
  else if (isBar()) {
    out << ", bar size " << Width() << " x " << Height() << " x " << Length()
      << " cm";
  }
  else out << ", shape: '" << Shape()->IsA()->GetName() << "'"; 
  
  if (verbosity-- <= 0) return; // 1
  
  //----------------------------------------------------------------------------
  out << ", theta(z): " << ThetaZ() << " rad";
  
//  if (verbosity-- <= 0) return; // 2
  
  //----------------------------------------------------------------------------
  
} // geo::OpDetGeo::PrintOpDetInfo()


#endif // LARCOREALG_GEOMETRY_OPDETGEO_H