CCollimator.h

Go to the documentation of this file.

#ifndef CLASSIC_CCollimator_HH
#define CLASSIC_CCollimator_HH

#include "AbsBeamline/PluginElement.h"

class ParticleMatterInteractionHandler;

// Class CCollimator
// ------------------------------------------------------------------------
//  Class CCollimator defines the abstract interface for a collimator.

class CCollimator: public PluginElement {

public:
    explicit CCollimator(const std::string &name);

    CCollimator();
    CCollimator(const CCollimator &rhs);
    void operator=(const CCollimator &) = delete;
    virtual ~CCollimator();

    virtual void accept(BeamlineVisitor &) const override;
    virtual void goOnline(const double &kineticEnergy) override;
    virtual ElementBase::ElementType getType() const override;
    virtual void getDimensions(double &zBegin, double &zEnd) const override;

    // bool checkCollimator(Vector_t r, Vector_t rmin, Vector_t rmax);

    void print();

    void setDimensions(double xstart, double xend,
                       double ystart, double yend,
                       double zstart, double zend,
                       double width);
    using PluginElement::setDimensions;

    double getZStart() ;
    double getZEnd() ;
    double getWidth() ;
private:
    virtual void doInitialise(PartBunchBase<double, 3> *bunch) override;
    virtual bool doCheck(PartBunchBase<double, 3> *bunch, const int turnnumber, const double t, const double tstep) override;
    virtual void doSetGeom() override;
    virtual void doFinalise() override;
    virtual bool doPreCheck(PartBunchBase<double, 3>*) override;
    virtual bool doFinaliseCheck(PartBunchBase<double, 3> *bunch, bool flagNeedUpdate) override;

    bool informed_m = false; 

    double zstart_m;
    double zend_m;
    double width_m;
    double rmax_m; 

    ParticleMatterInteractionHandler *parmatint_m = nullptr;
};

inline
double CCollimator::getZStart() {
    return zstart_m;
}

inline
double CCollimator::getZEnd() {
    return zend_m;
}

inline
double CCollimator::getWidth() {
    return width_m;
}
#endif // CLASSIC_CCollimator_HH