Stepper.h

Go to the documentation of this file.

//
// Class Stepper
//   Time integrator base class
//
// Copyright (c) 2017, Matthias Frey, 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 STEPPER_H
#define STEPPER_H
 
#include "Algorithms/PartBunchBase.h"
#include "Algorithms/Vektor.h"
 
#include <functional>
 
template <typename FieldFunction, typename ... Arguments>
class Stepper {
    
public:
    
    Stepper(const FieldFunction& fieldfunc) : fieldfunc_m(fieldfunc) { }
    
    virtual bool advance(PartBunchBase<double, 3>* bunch,
                         const size_t& i,
                         const double& t,
                         const double dt,
                         Arguments& ... args) const
    {
        bool isGood = doAdvance_m(bunch, i, t, dt, args...);
 
        bool isNaN = false;
        for (int j = 0; j < 3; ++j) {
            if (std::isnan(bunch->R[i](j)) ||
                std::isnan(bunch->P[i](j)) ||
                std::abs(bunch->R[i](j)) > 1.0e10 ||
                std::abs(bunch->P[i](j)) > 1.0e10) {
                isNaN = true;
                break;
            }
        }
 
        bool isBad = (!isGood || isNaN);
        if ( isBad ) {
            bunch->Bin[i] = -1;
        }
        return isBad;
    };
    virtual ~Stepper() {};
 
protected:
    const FieldFunction& fieldfunc_m;
 
private:
    virtual bool doAdvance_m(PartBunchBase<double, 3>* bunch,
                             const size_t& i,
                             const double& t,
                             const double dt,
                             Arguments& ... args) const = 0;
};
 
#endif