.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/PartBunch.h"
 
class PartBunch : public PartBunch_t {
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 runTests();
 
    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 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<double, 3>& hr, const Vector_t<double, 3>& origin);
 
    void resizeMesh();
 
    BConds<double, 3, Mesh_t, Center_t> bc_m;
    BConds<Vector_t<double, 3>, 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