FlexibleCollimator.h

Go to the documentation of this file.

//
// Class FlexibleCollimator
//   Defines the abstract interface for a collimator.
//
// Copyright (c) 200x - 2021, 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 CLASSIC_FlexibleCollimator_HH
#define CLASSIC_FlexibleCollimator_HH
 
#include "AbsBeamline/Component.h"
#include "Utilities/MSLang.h"
#include "Utilities/MSLang/BoundingBox2D.h"
#include "Utilities/MSLang/QuadTree.h"
 
#include <memory>
 
class BeamlineVisitor;
class LossDataSink;
 
class FlexibleCollimator: public Component {
 
public:
 
    explicit FlexibleCollimator(const std::string& name);
 
    FlexibleCollimator();
    FlexibleCollimator(const FlexibleCollimator& rhs);
    virtual ~FlexibleCollimator();
 
    virtual void accept(BeamlineVisitor&) const override;
 
    virtual bool apply(const size_t& i, const double& t, Vector_t& E, Vector_t& B) override;
 
    virtual bool applyToReferenceParticle(const Vector_t& R, const Vector_t& P, const double& t, Vector_t& E, Vector_t& B) override;
 
    virtual bool checkCollimator(PartBunchBase<double, 3>* bunch, const int turnnumber, const double t, const double tstep);
 
    virtual void initialise(PartBunchBase<double, 3>* bunch, double& startField, double& endField) override;
 
    virtual void initialise(PartBunchBase<double, 3>* bunch);
 
    virtual void finalise() override;
 
    virtual bool bends() const override;
 
    virtual void goOnline(const double& kineticEnergy) override;
 
    virtual void goOffline() override;
 
    virtual ElementType getType() const override;
 
    virtual void getDimensions(double& zBegin, double& zEnd) const override;
 
    void print();
 
    unsigned int getLosses() const;
 
    void setDescription(const std::string& desc);
    std::string getDescription() const;
 
    bool isStopped(const Vector_t& R);
 
    void writeHolesAndQuadtree(const std::string& baseFilename) const;
 
private:
 
    // Not implemented.
    void operator=(const FlexibleCollimator&);
 
    std::string description_m;
    std::vector<std::shared_ptr<mslang::Base>> holes_m;
    mslang::BoundingBox2D bb_m;
    mslang::QuadTree tree_m;
 
    bool informed_m;
    unsigned int losses_m;
    std::unique_ptr<LossDataSink> lossDs_m;
 
    ParticleMatterInteractionHandler* parmatint_m;
};
 
inline
unsigned int FlexibleCollimator::getLosses() const {
    return losses_m;
}
 
inline
bool FlexibleCollimator::bends() const {
    return false;
}
 
inline
std::string FlexibleCollimator::getDescription() const {
    return description_m;
}
 
#endif // CLASSIC_FlexibleCollimator_HH