TrackStatePropMask.hpp

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file TrackStatePropMask.hpp

// This file is part of the Acts project.
//
// Copyright (C) 2019 CERN for the benefit of the Acts project
//
// This Source Code Form is subject to the terms of the Mozilla Public
// License, v. 2.0. If a copy of the MPL was not distributed with this
// file, You can obtain one at http://mozilla.org/MPL/2.0/.

#pragma once

#include <limits>
#include <type_traits>

namespace Acts {

enum struct TrackStatePropMask : uint8_t {
  None = 0,
  Predicted = 1 << 0,
  Filtered = 1 << 1,
  Smoothed = 1 << 2,
  Jacobian = 1 << 3,

  Uncalibrated = 1 << 4,
  Calibrated = 1 << 5,

  All = std::numeric_limits<uint8_t>::max(),  // should be all ones
};

constexpr TrackStatePropMask operator|(TrackStatePropMask lhs,
                                       TrackStatePropMask rhs) {
  return static_cast<TrackStatePropMask>(
      static_cast<std::underlying_type<TrackStatePropMask>::type>(lhs) |
      static_cast<std::underlying_type<TrackStatePropMask>::type>(rhs));
}

constexpr TrackStatePropMask operator&(TrackStatePropMask lhs,
                                       TrackStatePropMask rhs) {
  return static_cast<TrackStatePropMask>(
      static_cast<std::underlying_type<TrackStatePropMask>::type>(lhs) &
      static_cast<std::underlying_type<TrackStatePropMask>::type>(rhs));
}

constexpr TrackStatePropMask operator^(TrackStatePropMask lhs,
                                       TrackStatePropMask rhs) {
  return static_cast<TrackStatePropMask>(
      static_cast<std::underlying_type<TrackStatePropMask>::type>(lhs) ^
      static_cast<std::underlying_type<TrackStatePropMask>::type>(rhs));
}

constexpr TrackStatePropMask operator~(TrackStatePropMask op) {
  return static_cast<TrackStatePropMask>(
      ~static_cast<std::underlying_type<TrackStatePropMask>::type>(op));
}

constexpr TrackStatePropMask& operator|=(TrackStatePropMask& lhs,
                                         TrackStatePropMask rhs) {
  lhs = static_cast<TrackStatePropMask>(
      static_cast<std::underlying_type<TrackStatePropMask>::type>(lhs) |
      static_cast<std::underlying_type<TrackStatePropMask>::type>(rhs));
  return lhs;
}

constexpr TrackStatePropMask& operator&=(TrackStatePropMask& lhs,
                                         TrackStatePropMask rhs) {
  lhs = static_cast<TrackStatePropMask>(
      static_cast<std::underlying_type<TrackStatePropMask>::type>(lhs) &
      static_cast<std::underlying_type<TrackStatePropMask>::type>(rhs));
  return lhs;
}

constexpr TrackStatePropMask& operator^=(TrackStatePropMask& lhs,
                                         TrackStatePropMask rhs) {
  lhs = static_cast<TrackStatePropMask>(
      static_cast<std::underlying_type<TrackStatePropMask>::type>(lhs) ^
      static_cast<std::underlying_type<TrackStatePropMask>::type>(rhs));
  return lhs;
}

}  // namespace Acts