ChargePolicy.hpp

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

// This file is part of the Acts project.
//
// Copyright (C) 2016-2018 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
namespace Acts {

class ChargedPolicy {
 public:
  ChargedPolicy(double charge) : m_dCharge(charge) {}

  bool operator==(const ChargedPolicy& rhs) const {
    return m_dCharge == rhs.m_dCharge;
  }

  bool operator!=(const ChargedPolicy& rhs) const { return !(*this == rhs); }

  double getCharge() const { return m_dCharge; }

  void setCharge(double charge) { m_dCharge = charge; }

  void flipSign() { m_dCharge *= -1.; }

 private:
  double m_dCharge;  
};

class NeutralPolicy {
 public:
  bool operator==(const NeutralPolicy& /*other*/) const { return true; }

  bool operator!=(const NeutralPolicy& rhs) const { return !(*this == rhs); }

  double getCharge() const { return 0.; }
};
}  // namespace Acts