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 "OPALTypes.h"
 
#include "PartBunch/PartBunch.h"
 
#include <functional>
 
template <typename FieldFunction, typename... Arguments>
class Stepper {
public:
    Stepper(const FieldFunction& fieldfunc) : fieldfunc_m(fieldfunc) {}
 
    virtual bool advance(
            PartBunch_t* 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) {
            // \todo  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) {
            // \todo  bunch->Bin(i) = -1;
        }
        return isBad;
    };
    virtual ~Stepper() {};
 
protected:
    const FieldFunction& fieldfunc_m;
 
private:
    virtual bool doAdvance_m(
            PartBunch_t* bunch, const size_t& i, const double& t, const double dt,
            Arguments&... args) const = 0;
};
 
#endif