TrackerLayerChargeInRange.h

Go to the documentation of this file.

#ifndef NSL_SELECTIONS_TRACKER_TRACKERLAYERCHARGEINRANGE_H
#define NSL_SELECTIONS_TRACKER_TRACKERLAYERCHARGEINRANGE_H

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

namespace NSL {
namespace Selections {

namespace TrackerLayer {

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

namespace { // hidden
namespace tl_base = ::NSL::Selections::TrackerLayer;
}

namespace PG {
namespace TrackerLayer {

namespace { // hidden
std::pair<float, float> chargeLimits(unsigned int charge, unsigned int layer) {

  if (layer == 1) {
    switch (charge) {
    case 1:
    case 2:
      return {static_cast<float>(charge) - 0.4f, static_cast<float>(charge) + 0.9f};
    default:
      return {0.0f, static_cast<float>(charge) + 0.8f};
    }
  }
  else if (layer == 9) {
    switch (charge) {
    case 1:
    case 2:
      return {static_cast<float>(charge) - 0.4f, static_cast<float>(charge) + 0.9f};
    default:
      return {static_cast<float>(charge) - 0.5f, static_cast<float>(charge) + 1.0f};
    }
  }
  else return {0,0};
}
} // namespace

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

} // namespace TrackerLayer
} // namespace PG

namespace Common {
namespace TrackerLayer {

namespace { // hidden
std::pair<float, float> chargeLimits(unsigned int charge, unsigned int layer) {

  if (layer == 1) {
    switch (charge) {
    case 1:
    case 2:
      return {static_cast<float>(charge) - 0.4f, static_cast<float>(charge) + 0.9f};
    case 9:
    case 11:
    case 13:
    case 10:
    case 12:
    case 14:
    case 16:
      return {static_cast<float>(charge) - 0.0585f * pow(charge, 1.15f) - 0.35f,
              static_cast<float>(charge) + 0.0334f * pow(charge, 1.15f) + 0.20f};
    default:
      return {static_cast<float>(charge) - 0.46f - (charge - 3.0f) * 0.1f,
              static_cast<float>(charge) + ((charge < 5.0f) ? 0.65f : 0.65f + (charge - 5.0f) * 0.03f)};
    }
  }
  if (layer == 9) {
    switch (charge) {
    case 1:
    case 2:
      return {static_cast<float>(charge) - 0.4f, static_cast<float>(charge) + 0.9f};
    case 9:
    case 11:
    case 13:
    case 10:
    case 12:
    case 14:
    case 16:
      return {static_cast<float>(charge) - 0.0284f * pow(charge, 1.15f) - 0.17f,
              static_cast<float>(charge) + 0.0585f * pow(charge, 1.15f) + 0.35f};
    default:
      return {static_cast<float>(charge) - 0.5f, static_cast<float>(charge) + 1.5f + (charge - 3.0f) * 0.06f};
    }
  }
  else return {0,0};
}
} // namespace

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

} // namespace Selections
} // namespace NSL

#endif