CSRIGFWakeFunction.h

Go to the documentation of this file.

//
// Class CSRIGFWakeFunction
//
// 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 CSRIGFWAKEFUNCTION_HH
#define CSRIGFWAKEFUNCTION_HH
 
#include "Solvers/WakeFunction.h"
 
#include <cstddef>
#include <memory>
#include <string>
#include <vector>
#include <utility>
 
class Filter;
class ElementBase;
 
class CSRIGFWakeFunction: public WakeFunction {
public:
    CSRIGFWakeFunction(const std::string& name, std::vector<Filter*> filters, const unsigned int& N);
 
    void apply(PartBunchBase<double, 3>* bunch) override;
 
    void initialize(const ElementBase* ref) override;
 
    virtual WakeType getType() const override;
 
private:
    void calculateLineDensity(PartBunchBase<double, 3>* bunch, std::pair<double, double>& meshInfo);
 
    void calculateContributionInside(std::size_t sliceNumber, double angleOfSlice, double meshSpacing);
    void calculateContributionAfter(std::size_t sliceNumber, double angleOfSlice, double meshSpacing);
    void calculateGreenFunction(PartBunchBase<double, 3>* bunch, double meshSpacing);
    double calcPsi(const double& psiInitial, const double& x, const double& Ds) const;
 
    std::vector<Filter*> filters_m;
    std::shared_ptr<Filter> defaultFilter_m;
    Filter *diffOp_m;
 
    LineDensity lineDensity_m;
    LineDensity dlineDensitydz_m;
 
    // Longitudinal CSR field.
    std::vector<double> Ez_m;
 
    // Chi used to calculate the Green's function
    std::vector<double> Chi_m;
 
    // Green's function
    std::vector<double> Grn_m;
 
    // Retarded angle Psi; (for after dipole, Stupakov)
    std::vector<double> Psi_m;
 
    // Start position of CSR wake.
    double Begin_m;
 
    // Start position of equivalent hard edge dipole that approximates actual
    // dipole.
    double FieldBegin_m;
 
    // Effective length of dipole.
    double Length_m;
 
    // Radius of curvature of effective dipole.
    double bendRadius_m;
 
    std::string bendName_m;
 
    double totalBendAngle_m;
};
 
#endif //CSRIGFWAKEFUNCTION_HH