AmrPartBunch.h

Go to the documentation of this file.

//
// Class AmrPartBunch
//   This class is used to represent a bunch in AMR mode.
//
// Copyright (c) 2017 - 2019, Matthias Frey, Paul Scherrer Institut, Villigen PSI, Switzerland
// All rights reserved
//
// Implemented as part of the PhD thesis
// "Precise Simulations of Multibunches in High Intensity Cyclotrons"
//
// 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 AMR_PART_BUNCH_H
#define AMR_PART_BUNCH_H
 
#include "Algorithms/PartBunchBase.h"
#include "Amr/AmrObject.h"
 
class AmrPartBunch : public PartBunchBase<double, 3>
{
public:
    typedef AmrParticle_t pbase_t;
    
public:
 
    AmrPartBunch(const PartData *ref);
 
    AmrPartBunch(const PartData *ref, pbase_t* pbase_p);
 
    ~AmrPartBunch();
 
    pbase_t *getAmrParticleBase();
 
    const pbase_t *getAmrParticleBase() const;
 
    void initialize(FieldLayout_t *fLayout);
    
    // does actually another repartition
    void do_binaryRepart();
    
    Vector_t get_hr() const;
    
    void set_meshEnlargement(double dh);
    
    VectorPair_t getEExtrema();
    
    double getRho(int x, int y, int z);
    
    FieldLayout_t &getFieldLayout();
    
    
    void boundp();
    
    void computeSelfFields();
    
    void computeSelfFields(int bin);
    
    void computeSelfFields_cycl(double gamma);
    
    void computeSelfFields_cycl(int bin);
    
    void setSolver(FieldSolver *fs) {
        PartBunchBase<double, 3>::setSolver(fs);
        this->amrobj_mp = fs->getAmrObject();
    }
    
    virtual void setBinCharge(int /*bin*/, double /*q*/) { };
    virtual void setBinCharge(int /*bin*/) { };
    
    /*
     * AmrPartBunch only
     */
    
    const AmrObject* getAmrObject() const {
        return this->amrobj_mp;
    }
    
    PoissonSolver *getFieldSolver() {
        return fs_m->solver_m;
    }
    
    const PoissonSolver *getFieldSolver() const {
        return fs_m->solver_m;
    }
    
    void setBaseLevelMeshSpacing(const Vector_t& hr) {
        for (int i = 0; i < 3; ++i)
            hr_m[i] = hr[i];
    }
 
    void setAmrDomainRatio(const std::vector<double>& ratio);
 
    void gatherLevelStatistics();
    
    const size_t& getLevelStatistics(int l) const;
    
    
    void updateLorentzFactor(int bin=0);
    
    void updateLorentzFactor(double gamma);
    
    //FIXME BCs
    void setBCAllPeriodic() {}
    void setBCAllOpen() {}
    void setBCForDCBeam() {}
    
    
private:
    void updateFieldContainers_m();
    
    void updateDomainLength(Vektor<int, 3>& grid);
    
    void updateFields(const Vector_t& hr, const Vector_t& origin);
    
private:
    
    /* pointer to AMR object that is part
     * of solver_m (AmrPoissonSolver) in src/Structure/FieldSolver.h
     */
    AmrObject *amrobj_mp;
    pbase_t *amrpbase_mp;
    
    /* We need this due to H5PartWrapper etc, but it's always nullptr.
     * Thus, don't use it.
     */
    FieldLayout_t* fieldlayout_m;
    
    std::unique_ptr<size_t[]> globalPartPerLevel_m;
};
 
#endif