RegSelEtaPhiModule.h

Go to the documentation of this file.

// emacs: this is -*- c++ -*-
/*
  Copyright (C) 2002-2018 CERN for the benefit of the ATLAS collaboration
*/
//
//   @file    RegSelEtaPhiModule.h        
//
//            a "module" with a region in eta-phi space with max and min
//            coordinates in both eta and phi and a payload.
//            designed so can easily test whether two "modules"
//            overlap, and objects can be sorted by payload.
//  
//
//   $Id: RegSelEtaPhiModule.h, v0.0   Sat 25 Jun 2011 10:30:34 BST sutt $
 
 
#ifndef REGSELLUT_REGSELETAPHIMODULE_H
#define REGSELLUT_REGSELETAPHIMODULE_H
 
#include <iostream>
#include <cmath>
 
 
 
class EtaPhiBase {
 
public:
 
  EtaPhiBase(double etamin, double etamax, double phimin, double phimax) 
    : m_boundary(false) {
 
    static const double twopi = 2*M_PI; 
    
    m_eta[0] = etamin;    
    m_eta[1] = etamax;
   
    m_phi[0] = phimin;
    m_phi[1] = phimax;
    
    if ( m_phi[0]<0 ) m_phi[0] += twopi;
    if ( m_phi[1]<0 ) m_phi[1] += twopi;
 
    if ( m_phi[0]>twopi ) m_phi[0] -= twopi;
    if ( m_phi[1]>twopi ) m_phi[1] -= twopi;
 
    if ( m_phi[0]>m_phi[1] ) m_boundary = true;
  } 
 
  virtual ~EtaPhiBase() = default;
 
  double etamin() const { return m_eta[0]; }
  double etamax() const { return m_eta[1]; }
 
  double phimin() const { return m_phi[0]; }
  double phimax() const { return m_phi[1]; }
 
 
  bool overlap(const EtaPhiBase& e) const {
 
    static const double twopi = 2*M_PI; 
 
    //    struct timeval timer = simpletimer_start();
 
    if ( ! ( e.etamin()<etamax() && e.etamax()>etamin() ) ) return false;    
 
    bool inphi = false;
 
    double phimin1 = phimin();
    double phimax1 = phimax();
 
    double phimin2 = e.phimin();
    double phimax2 = e.phimax();
 
    // check for phi wrapping and boundary intersection
    if ( phimin1>phimax1 ) phimax1 += twopi;
    if ( phimin2>phimax2 ) phimax2 += twopi;
 
    if ( ( phimin1<phimax2         && phimax1>phimin2 )         ||
     ( phimin1<(phimax2+twopi) && phimax1>(phimin2+twopi) ) ||
     ( (phimin1+twopi)<phimax2 && (phimax1+twopi)>phimin2 ) ) inphi = true; 
 
    //    double time = simpletimer_stop( timer );
 
    return inphi; 
  }
 
protected:
 
  double  m_eta[2];
  double  m_phi[2];
 
  bool    m_boundary;
 
};
 
 
 
inline std::ostream& operator<<( std::ostream& s, const EtaPhiBase& e ) { 
  s << "[ eta=" << e.etamin() << " - " << e.etamax();
  s << "\tphi=" << 180*e.phimin()/M_PI << " - " << 180*e.phimax()/M_PI;
  s<< " ]";
  return s;
}
 
 
 
 
 
template<class T>
class TRegSelEtaPhiModule : public EtaPhiBase {
 
public:
 
  TRegSelEtaPhiModule(double etamin, double etamax, double phimin, double phimax, const T& t) : 
    EtaPhiBase(etamin, etamax, phimin, phimax), m_t(t) { }
  
  virtual ~TRegSelEtaPhiModule() { } 
 
  const T& payload() const { return m_t; }
  T&       payload()       { return m_t; }
  
  template<class E> bool operator<(const TRegSelEtaPhiModule<E>& e)  const {  return payload()<e.payload();  }
  template<class E> bool operator>(const TRegSelEtaPhiModule<E>& e)  const {  return payload()>e.payload(); }
  template<class E> bool operator==(const TRegSelEtaPhiModule<E>& e) const {  return payload()==e.payload(); }
  template<class E> bool operator!=(const TRegSelEtaPhiModule<E>& e) const {  return payload()!=e.payload(); }
 
private:
 
  T  m_t;
 
};
 
 
template<class T>
inline std::ostream& operator<<( std::ostream& s, const TRegSelEtaPhiModule<T>& e ) { 
  s << "[ eta= " << e.etamin() << " - " << e.etamax();
  s << "\tphi= " << 180*e.phimin()/M_PI << " - " << 180*e.phimax()/M_PI;
  s << "\tpay= " << e.payload();
  s << " ]";
  return s;
}
 
 
 
#endif  // REGSELLUT_REGSELETAPHIMODULE_H