Bracket.h

Go to the documentation of this file.

// Copyright 2019-2020 CERN and copyright holders of ALICE O2.
// See https://alice-o2.web.cern.ch/copyright for details of the copyright holders.
// All rights not expressly granted are reserved.
//
// This software is distributed under the terms of the GNU General Public
// License v3 (GPL Version 3), copied verbatim in the file "COPYING".
//
// In applying this license CERN does not waive the privileges and immunities
// granted to it by virtue of its status as an Intergovernmental Organization
// or submit itself to any jurisdiction.
 
 
#ifndef ALICEO2_BRACKET_H
#define ALICEO2_BRACKET_H
 
#include "GPUCommonDef.h"
 
#include <GPUCommonRtypes.h>
#ifndef GPUCA_GPUCODE_DEVICE
#include <string>
#include <sstream>
#endif
 
namespace o2
{
namespace math_utils
{
namespace detail
{
 
template <typename T = float>
class Bracket
{
 public:
  enum Relation : int { Below = -1,
                        Inside = 0,
                        Above = 1 };
 
  Bracket(T minv, T maxv);
  Bracket() = default;
  Bracket(const Bracket<T>& src) = default;
  Bracket(Bracket<T>&& src) = default;
  Bracket& operator=(const Bracket<T>& src) = default;
  Bracket& operator=(Bracket<T>&& src) = default;
  ~Bracket() = default;
 
  bool operator<(T other) const;
  bool operator>(T other) const;
 
  bool operator<(const Bracket<T>& other) const;
  bool operator>(const Bracket<T>& other) const;
  bool operator==(const Bracket<T>& other) const;
  bool operator!=(const Bracket<T>& other) const;
 
  void setMax(T v) GPUnoexcept();
  void setMin(T v) GPUnoexcept();
  void set(T minv, T maxv) GPUnoexcept();
 
  T& getMax();
  T& getMin();
 
  T getMax() const;
  T getMin() const;
  T mean() const;
  T delta() const;
 
  Bracket getOverlap(const Bracket<T>& other) const;
 
  void scale(T c);
  bool isZeroLength() const;
  bool isValid() const;
  bool isInvalid() const;
  void update(T v);
  Relation isOutside(const Bracket<T>& t) const;
  Relation isOutside(T t, T tErr) const;
  Relation isOutside(T t) const;
 
#ifndef GPUCA_GPUCODE_DEVICE
  std::string asString() const;
#endif
 
 private:
  T mMin{};
  T mMax{};
 
  ClassDefNV(Bracket, 2);
};
 
template <typename T>
Bracket<T>::Bracket(T minv, T maxv) : mMin(minv), mMax(maxv)
{
}
 
template <typename T>
inline bool Bracket<T>::operator<(T other) const
{
  return mMax < other;
}
 
template <typename T>
inline bool Bracket<T>::operator>(T other) const
{
  return mMin > other;
}
 
template <typename T>
inline bool Bracket<T>::operator<(const Bracket<T>& other) const
{
  return *this < other.mMin;
}
 
template <typename T>
inline bool Bracket<T>::operator>(const Bracket<T>& other) const
{
  return *this > other.mMax;
}
 
template <typename T>
inline bool Bracket<T>::operator==(const Bracket<T>& rhs) const
{
  return mMin == rhs.mMin && mMax == rhs.mMax;
}
 
template <typename T>
inline bool Bracket<T>::operator!=(const Bracket<T>& rhs) const
{
  return !(*this == rhs);
}
 
template <typename T>
inline void Bracket<T>::setMax(T v) GPUnoexcept()
{
  mMax = v;
}
 
template <typename T>
inline void Bracket<T>::setMin(T v) GPUnoexcept()
{
  mMin = v;
}
 
template <typename T>
inline void Bracket<T>::set(T minv, T maxv) GPUnoexcept()
{
  this->setMin(minv);
  this->setMax(maxv);
}
 
template <typename T>
inline T& Bracket<T>::getMax()
{
  return mMax;
}
 
template <typename T>
inline T& Bracket<T>::getMin()
{
  return mMin;
}
 
template <typename T>
inline T Bracket<T>::getMax() const
{
  return mMax;
}
 
template <typename T>
inline T Bracket<T>::getMin() const
{
  return mMin;
}
 
template <typename T>
inline T Bracket<T>::mean() const
{
  return (mMin + mMax) / 2;
}
 
template <typename T>
inline T Bracket<T>::delta() const
{
  return mMax - mMin;
}
 
template <typename T>
inline bool Bracket<T>::isValid() const
{
  return mMax >= mMin;
}
 
template <typename T>
inline bool Bracket<T>::isInvalid() const
{
  return mMin > mMax;
}
 
template <typename T>
inline bool Bracket<T>::isZeroLength() const
{
  return mMin == mMax;
}
 
template <typename T>
inline void Bracket<T>::update(T v)
{
  // update limits
  if (v > mMax) {
    mMax = v;
  }
  if (v < mMin) {
    mMin = v;
  }
}
 
template <typename T>
inline void Bracket<T>::scale(T c)
{
  mMin *= c;
  mMax *= c;
}
 
template <typename T>
inline Bracket<T> Bracket<T>::getOverlap(const Bracket<T>& other) const
{
  return {getMin() > other.getMin() ? getMin() : other.getMin(), getMax() < other.getMax() ? getMax() : other.getMax()};
}
 
template <typename T>
inline typename Bracket<T>::Relation Bracket<T>::isOutside(const Bracket<T>& t) const
{
  return t.mMax < mMin ? Below : (t.mMin > mMax ? Above : Inside);
}
 
template <typename T>
inline typename Bracket<T>::Relation Bracket<T>::isOutside(T t, T tErr) const
{
  return t + tErr < mMin ? Below : (t - tErr > mMax ? Above : Inside);
}
 
template <typename T>
inline typename Bracket<T>::Relation Bracket<T>::isOutside(T t) const
{
  return t < mMin ? Below : (t > mMax ? Above : Inside);
}
 
#ifndef GPUCA_GPUCODE_DEVICE
template <typename T>
std::string Bracket<T>::asString() const
{
  std::stringstream tmp;
  tmp << "[" << getMin() << ":" << getMax() << "]";
  return tmp.str();
}
#endif
 
} // namespace detail
} // namespace math_utils
} // namespace o2
 
#endif