forked from jbengtsson/thor-scsi-lib
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
[TASK] non linear kicker integration
work in progress
- Loading branch information
1 parent
1d3a484
commit 44ed5d8
Showing
11 changed files
with
494 additions
and
48 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,2 @@ | ||
| // check that syntax is correct | ||
| #include <thor_scsi/custom/nonlinear_kicker.h> |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,83 @@ | ||
| #ifndef _THOR_SCSI_ELEMENTS_NONLINEAR_KICKER_H_ | ||
| #define _THOR_SCSI_ELEMENTS_NONLINEAR_KICKER_H_ | ||
|
|
||
| #include <thor_scsi/elements/field_kick.h> | ||
| #include <thor_scsi/custom/nonlinear_kicker_interpolation.h> | ||
|
|
||
| #include <iomanip> | ||
|
|
||
| namespace thor_scsi::custom { | ||
| template<class C, typename = typename C::complex_type, typename = typename C::double_type> // CK for complex knob | ||
| class NonLinearKickerTypeWithKnob : public thor_scsi::elements::FieldKickKnobbed<C> { | ||
| protected: | ||
| using complex_type = typename C::complex_type; | ||
| using double_type = typename C::double_type; | ||
| using nonlinearkicker_knobbed = thor_scsi::custom::NonLinearKickerInterpolationKnobbed<C>; | ||
| public: | ||
| /** | ||
| * | ||
| * \verbatim embed:rst:leading-asterisk | ||
| * | ||
| * .. Warning:: | ||
| * | ||
| * fix memory handling of interpolation object | ||
| * add move constructor | ||
| * | ||
| * \endverbatim | ||
| */ | ||
| inline NonLinearKickerTypeWithKnob(const Config &config) : thor_scsi::elements::FieldKickKnobbed<C>(config){ | ||
| std::vector<thor_scsi::custom::aircoil_filament_t> off {{200e0, 200e0, 0e0}}; | ||
| auto tmp = std::make_shared<nonlinearkicker_knobbed>(off); | ||
| this->intp = std::move(std::dynamic_pointer_cast<thor_scsi::core::Field2DInterpolationKnobbed<C>>(tmp)); | ||
| /* | ||
| std::cerr << "NonLinearKicker Type: " << std::setw(10) << name << " interpolation object " << this->intp << std::endl; | ||
| auto parent = dynamic_cast<FieldKick*>(this); | ||
| std::cerr << "FieldKick: interpolation object " << parent->getFieldInterpolator() << std::endl; | ||
| */ | ||
| } | ||
|
|
||
| const char* type_name(void) const override { return "NonLinearKicker"; }; | ||
|
|
||
| inline auto getFieldInterpolator(void) const { | ||
| std::stringstream strm; | ||
| strm << __FILE__ << ":mpole.getFieldInterpolator: this->intp " << static_cast<void *>(this->intp.get()); | ||
| auto p = std::dynamic_pointer_cast<nonlinearkicker_knobbed>(this->intp); | ||
| strm << " p " << static_cast<void *>(p.get()); | ||
| if(!p){ | ||
| strm<<" cast failed!"; | ||
| throw std::runtime_error(strm.str()); | ||
| } | ||
| auto cp = std::const_pointer_cast<nonlinearkicker_knobbed>(p); | ||
| strm << " cp " << static_cast<void *>(cp.get()); | ||
| if(!cp){ | ||
| strm<<" cast failed!"; | ||
| throw std::runtime_error(strm.str()); | ||
| } | ||
| // std::cerr << strm.str() << std::endl; | ||
| return cp; | ||
| } | ||
|
|
||
| inline void setFieldInterpolator(std::shared_ptr<thor_scsi::custom::NonLinearKickerInterpolationKnobbed<C>> intp) { | ||
| auto p = std::dynamic_pointer_cast<thor_scsi::core::Field2DInterpolationKnobbed<C>>(intp); | ||
| if(!p){ | ||
| throw std::runtime_error("Could not cast multipole to Field2DInterpolation"); | ||
| } | ||
| this->intp = p; | ||
| } | ||
|
|
||
| //thor_scsi::core::TwoDimensionalMultipoles* intp; | ||
| }; | ||
|
|
||
|
|
||
| typedef class NonLinearKickerTypeWithKnob<core::StandardDoubleType> NonLinearKickerType; | ||
| typedef class NonLinearKickerTypeWithKnob<core::TpsaVariantType> NonLinearKickerTypeTpsa; | ||
|
|
||
| } // Name space | ||
|
|
||
| #endif // _THOR_SCSI_ELEMENTS_NONLINEAR_KICKER_H_ | ||
| /* | ||
| * Local Variables: | ||
| * mode: c++ | ||
| * c-file-style: "python" | ||
| * End: | ||
| */ |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.