InnerTrackerChargeInRange.h

Go to the documentation of this file.

#ifndef NSL_SELECTIONS_TRACKER_INNERTRACKERCHARGEINRANGE_H
#define NSL_SELECTIONS_TRACKER_INNERTRACKERCHARGEINRANGE_H

// our headers
#include "NSL/Selection.h"

namespace NSL {
namespace Selections {

namespace InnerTracker {
class ChargeInRange : public Selection {
public:
  ChargeInRange(float min, float max, NAIA::TrTrack::ChargeRecoType recoType);
};
} // namespace InnerTracker

namespace { // hidden
namespace it_base = ::NSL::Selections::InnerTracker;
}

namespace PG {
namespace InnerTracker {

namespace { // hidden
std::pair<float, float> chargeLimits(unsigned int charge) {
  switch (charge) {
  case 1:
  case 2:
    return {static_cast<float>(charge) - 0.3f, static_cast<float>(charge) + 0.5f};
  default:
    return {static_cast<float>(charge) - 0.45f, static_cast<float>(charge) + 0.45f};
  }
}
} // namespace

class ChargeInRange : public it_base::ChargeInRange {
public:
  explicit ChargeInRange(unsigned int charge, NAIA::TrTrack::ChargeRecoType recoType)
      : it_base::ChargeInRange(chargeLimits(charge).first, chargeLimits(charge).second, recoType) {}
};

} // namespace InnerTracker
} // namespace PG

namespace Common {
namespace InnerTracker {

namespace { // hidden
std::pair<float, float> chargeLimits(unsigned int charge) {
  switch (charge) {
  case 1:
  case 2:
    return {static_cast<float>(charge) - 0.3f, static_cast<float>(charge) + 0.5f};
  default:
    return {static_cast<float>(charge) - ((charge >= 14.0f) ? 0.5f : 0.0075f * pow(charge, 1.414f) + 0.198f),
            static_cast<float>(charge) + ((charge >= 14.0f) ? 0.5f : 0.0075f * pow(charge, 1.414f) + 0.198f)};
  }
}
} // namespace

class ChargeInRange : public it_base::ChargeInRange {
public:
  explicit ChargeInRange(unsigned int charge, NAIA::TrTrack::ChargeRecoType recoType)
      : it_base::ChargeInRange(chargeLimits(charge).first, chargeLimits(charge).second, recoType) {}
};
} // namespace InnerTracker
} // namespace Common

} // namespace Selections
} // namespace NSL

#endif