quantumnumbers.h

Go to the documentation of this file.

/*
 *
 *    Copyright (c) 2014-2018
 *      SMASH Team
 *
 *    GNU General Public License (GPLv3 or later)
 *
 */

#ifndef SRC_INCLUDE_QUANTUMNUMBERS_H_
#define SRC_INCLUDE_QUANTUMNUMBERS_H_

#include <string>

#include "particles.h"

namespace smash {

class QuantumNumbers {
 public:
  QuantumNumbers()
      : momentum_(0., 0., 0., 0.),
        charge_(0),
        isospin3_(0),
        strangeness_(0),
        charmness_(0),
        bottomness_(0),
        baryon_number_(0) {}

  QuantumNumbers(const FourVector& m, const int q, const int i3, const int s,
                 const int c, const int b, const int B)
      : momentum_(m),
        charge_(q),
        isospin3_(i3),
        strangeness_(s),
        charmness_(c),
        bottomness_(b),
        baryon_number_(B) {}

  explicit QuantumNumbers(const Particles& particles) : QuantumNumbers() {
    for (const ParticleData& data : particles) {
      add_values(data);
    }
  }

  explicit QuantumNumbers(const ParticleList& part) : QuantumNumbers() {
    for (const auto& p : part) {
      add_values(p);
    }
  }

  void add_values(const ParticleData& p) {
    momentum_ += p.momentum();
    charge_ += p.pdgcode().charge();
    isospin3_ += p.pdgcode().isospin3();
    strangeness_ += p.pdgcode().strangeness();
    charmness_ += p.pdgcode().charmness();
    bottomness_ += p.pdgcode().bottomness();
    baryon_number_ += p.pdgcode().baryon_number();
  }

  FourVector momentum() const { return momentum_; }

  int charge() const { return charge_; }
  int isospin3() const { return isospin3_; }
  int strangeness() const { return strangeness_; }
  int charmness() const { return charmness_; }
  int bottomness() const { return bottomness_; }
  int baryon_number() const { return baryon_number_; }

  bool operator==(const QuantumNumbers& rhs) const {
    // clang-format off
    return (momentum_ == rhs.momentum_ &&
            charge_ == rhs.charge_ &&
            isospin3_ == rhs.isospin3_ &&
            strangeness_ == rhs.strangeness_ &&
            charmness_ == rhs.charmness_ &&
            bottomness_ == rhs.bottomness_ &&
            baryon_number_ == rhs.baryon_number_);
    // clang-format on
  }
  bool operator!=(const QuantumNumbers& rhs) const { return !(*this == rhs); }

  QuantumNumbers operator-(const QuantumNumbers& rhs) const {
    return {momentum_ - rhs.momentum_,          charge_ - rhs.charge_,
            isospin3_ - rhs.isospin3_,          strangeness_ - rhs.strangeness_,
            charmness_ - rhs.charmness_,        bottomness_ - rhs.bottomness_,
            baryon_number_ - rhs.baryon_number_};
  }

  std::string report_deviations(const Particles& particles) const {
    QuantumNumbers current_values(particles);
    return report_deviations(current_values);
  }

  std::string report_deviations(const QuantumNumbers& rhs) const;

 private:
  FourVector momentum_;

  int charge_;
  int isospin3_;
  int strangeness_;
  int charmness_;
  int bottomness_;
  int baryon_number_;
};

}  // namespace smash

#endif  // SRC_INCLUDE_QUANTUMNUMBERS_H_