IsoDepositDirection.h

Go to the documentation of this file.

#ifndef RecoCandidate_IsoDepositDirection_H
#define RecoCandidate_IsoDepositDirection_H 

#include <cmath>
#include <sstream>
#include <iostream>

#include "DataFormats/Math/interface/deltaR.h"
#include "DataFormats/Math/interface/deltaPhi.h"

namespace reco {
  namespace isodeposit {

    class  Direction {
    public:

      struct Distance { 
    float deltaR; 
    float relativeAngle; 
    bool operator < (const Distance & rd2) const { return deltaR < rd2.deltaR; }; 
      };
  
      Direction(double eta = 0., double phi = 0.) : theEta(eta), thePhi(phi) {
    while( thePhi < 0.0 ) thePhi += 2*M_PI; 
    while( thePhi >= 2*M_PI ) thePhi -= 2*M_PI;
      };

    double eta() const { return theEta; }
    double phi() const { return thePhi; } 
    double theta() const { return acos(tanh(theEta)); } 

    inline bool operator==(const Direction & d2) {
      if ( this == &d2 ) return true;
      if ( deltaR(d2) < 1.e-4) return true;
      return false; 
    }

    inline double deltaR2(const Direction & dir2) const { 
      return reco::deltaR2(*this, dir2);
    } 
    inline double deltaR(const Direction & dir2) const { 
      return reco::deltaR(*this, dir2);
    } 
  
    Distance operator- (const Direction & dir2) const {
      Distance result;
      double dR    = deltaR(dir2);
      double dEta = theEta-dir2.eta();
      double dPhi = reco::deltaPhi(thePhi,dir2.phi());

      result.relativeAngle = (dR > 1.e-4) ? atan2(dPhi,dEta) : 0.;
      result.deltaR = dR;
      return result;
    }

    Direction operator+ (const Distance & relDir) const {
      double eta = theEta + relDir.deltaR*cos(relDir.relativeAngle);
      double phi = thePhi + relDir.deltaR*sin(relDir.relativeAngle);
      return Direction(eta,phi);
    }

    std::string print() const {
      std::ostringstream str;
      str<<" (Eta="<<theEta<< "," << "Phi=" <<thePhi<<")";
      return str.str();
    }

    private:
    float  theEta;
    float  thePhi;
    };

  } //namespace isodeposit
} //namespace reco

#endif