AbstractObjects/OpalParticle.h

Go to the documentation of this file.

//
// Class OpalParticle
//   This class represents the canonical coordinates of a particle.
//
// Copyright (c) 2008 - 2020, Paul Scherrer Institut, Villigen PSI, Switzerland
// All rights reserved
//
// This file is part of OPAL.
//
// OPAL is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// You should have received a copy of the GNU General Public License
// along with OPAL. If not, see <https://www.gnu.org/licenses/>.
//
#ifndef OPALX_OpalParticle_HH
#define OPALX_OpalParticle_HH
 
#include "Ippl.h"
#include "OPALTypes.h"
 
//
#include <Kokkos_MathematicalConstants.hpp>
#include <Kokkos_MathematicalFunctions.hpp>
 
class OpalParticle {
public:
    // Particle coordinate numbers.
    enum { X, Y, L, INVALID };
 
    //  Construct particle with the given coordinates.
    OpalParticle(
            int64_t id, double x, double px, double y, double py, double z, double pz, double time,
            double q, double m);
 
    OpalParticle(
            int64_t id, Vector_t<double, 3> const& R, Vector_t<double, 3> const& P, double time,
            double q, double m);
 
    OpalParticle();
 
    void setX(double);
 
    void setPx(double);
 
    void setY(double);
 
    void setPy(double);
 
    void setZ(double);
 
    void setPz(double);
 
    void setR(Vector_t<double, 3> const&);
 
    void setP(Vector_t<double, 3> const&);
 
    void setTime(double t);
 
    int64_t getId() const;
 
    //  Access coordinate by index for constant particle.
    double operator[](unsigned int) const;
 
    double getX() const;
 
    double getPx() const;
 
    double getY() const;
 
    double getPy() const;
 
    double getZ() const;
 
    double getPz() const;
 
    const Vector_t<double, 3>& getR() const;
 
    const Vector_t<double, 3>& getP() const;
 
    double getTime() const;
 
    double getCharge() const;
 
    double getMass() const;
 
private:
    int64_t id_m;
    Vector_t<double, 3> R_m;
    Vector_t<double, 3> P_m;
    double time_m;
    double charge_m;
    double mass_m;
};
 
inline void OpalParticle::setX(double val) { R_m[X] = val; }
 
inline void OpalParticle::setY(double val) { R_m[Y] = val; }
 
inline void OpalParticle::setZ(double val) { R_m[L] = val; }
 
inline void OpalParticle::setPx(double val) { P_m[X] = val; }
 
inline void OpalParticle::setPy(double val) { P_m[Y] = val; }
 
inline void OpalParticle::setPz(double val) { P_m[L] = val; }
 
inline void OpalParticle::setR(Vector_t<double, 3> const& R) { R_m = R; }
 
inline void OpalParticle::setP(Vector_t<double, 3> const& P) { P_m = P; }
 
inline void OpalParticle::setTime(double t) { time_m = t; }
 
inline int64_t OpalParticle::getId() const { return id_m; }
 
inline double OpalParticle::operator[](unsigned int i) const {
    PAssert_LT(i, 6u);
    return i % 2 == 0 ? R_m[i / 2] : P_m[i / 2];
}
 
inline double OpalParticle::getX() const { return R_m[X]; }
 
inline double OpalParticle::getY() const { return R_m[Y]; }
 
inline double OpalParticle::getZ() const { return R_m[L]; }
 
inline double OpalParticle::getPx() const { return P_m[X]; }
 
inline double OpalParticle::getPy() const { return P_m[Y]; }
 
inline double OpalParticle::getPz() const { return P_m[L]; }
 
inline const Vector_t<double, 3>& OpalParticle::getR() const { return R_m; }
 
inline const Vector_t<double, 3>& OpalParticle::getP() const { return P_m; }
 
inline double OpalParticle::getTime() const { return time_m; }
 
inline double OpalParticle::getCharge() const { return charge_m; }
 
inline double OpalParticle::getMass() const { return mass_m; }
 
#endif  // OPALX_OpalParticle_HH