PartBunch.h

Go to the documentation of this file.

//
// Class PartBunch
//   Particle Bunch.
//   A representation of a particle bunch as a vector of particles.
//
// 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 OPAL_PartBunch_HH
#define OPAL_PartBunch_HH
 
#include "Algorithms/PartBunchBase.h"
 
class PartBunch: public PartBunchBase<double, 3> {
 
public:
    typedef IpplParticleBase<Layout_t> pbase_t;
    enum { Dim = Dimension };
 
public:
 
    //  Construct empty bunch.
    explicit PartBunch(const PartData *ref);
 
    PartBunch() = delete;
    PartBunch(const PartBunch &) = delete;
    PartBunch &operator=(const PartBunch &) = delete;
 
    ~PartBunch();
 
//     pbase_t* clone();
 
    void initialize(FieldLayout_t *fLayout);
 
    void do_binaryRepart();
 
    double getRho(int x, int y, int z);
 
    /*
 
      Mesh and Field Layout related functions
 
    */
 
    // MATTHIAS CHECK
    const Mesh_t &getMesh() const;
 
//     void setMesh(Mesh_t* mesh);
 
    // MATTHIAS CHECK
    Mesh_t &getMesh();
 
//     void setFieldLayout(FieldLayout_t* fLayout);
 
    // MATTHIAS CHECK
    FieldLayout_t &getFieldLayout();
 
    void setBCAllPeriodic();
    void setBCAllOpen();
 
    void setBCForDCBeam();
 
    /*
      Compatibility function push_back
 
    */
 
    VectorPair_t getEExtrema();
 
    void computeSelfFields();
 
    void computeSelfFields(int b);
 
    void computeSelfFields_cycl(double gamma);
    void computeSelfFields_cycl(int b);
 
    void resetInterpolationCache(bool clearCache = false);
 
    void swap(unsigned int i, unsigned int j);
 
    Field_t rho_m;
 
    VField_t  eg_m;
 
    Inform &print(Inform &os);
 
private:
 
    void updateDomainLength(Vektor<int, 3>& grid);
 
    void updateFields(const Vector_t& hr, const Vector_t& origin);
 
    void resizeMesh();
 
    BConds<double, 3, Mesh_t, Center_t> bc_m;
    BConds<Vector_t, 3, Mesh_t, Center_t> vbc_m;
 
 
    bool interpolationCacheSet_m;
 
    ParticleAttrib<CacheDataCIC<double, 3U> > interpolationCache_m;
 
    //FIXME
    ParticleLayout<double, 3> & getLayout() {
        return pbase_m->getLayout();
    }
 
    //FIXME
    const ParticleLayout<double, 3>& getLayout() const {
        return pbase_m->getLayout();
    }
};
 
 
 
inline
double PartBunch::getRho(int x, int y, int z) {
    return rho_m[x][y][z].get();
}
 
inline
const Mesh_t &PartBunch::getMesh() const {
    const Layout_t* layout = static_cast<const Layout_t*>(&getLayout());
    return layout->getLayout().getMesh();
}
 
inline
Mesh_t &PartBunch::getMesh() {
    Layout_t* layout = static_cast<Layout_t*>(&getLayout());
    return layout->getLayout().getMesh();
}
 
// inline
 
inline Inform &operator<<(Inform &os, PartBunch &p) {
    return p.print(os);
}
 
#endif // OPAL_PartBunch_HH