1 /** Copyright © 2019 Université de Genève, LMU Munich - Faculty of Physics, IAP-CNRS/Sorbonne Université
3 * This library is free software; you can redistribute it and/or modify it under
4 * the terms of the GNU Lesser General Public License as published by the Free
5 * Software Foundation; either version 3.0 of the License, or (at your option)
8 * This library is distributed in the hope that it will be useful, but WITHOUT
9 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
10 * FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more
13 * You should have received a copy of the GNU Lesser General Public License
14 * along with this library; if not, write to the Free Software Foundation, Inc.,
15 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
18 * @file FrameModel.icpp
19 * @date September 1, 2015
20 * @author Nikolaos Apostolakos
23 namespace ModelFitting {
25 template <typename PsfType>
26 FrameModelPsfContainer<PsfType>::FrameModelPsfContainer(size_t): PsfType() {}
28 template <typename PsfType>
29 FrameModelPsfContainer<PsfType>::FrameModelPsfContainer(PsfType psf, size_t): PsfType(std::move(psf)) {}
31 template <typename PsfType>
32 FrameModelPsfContextContainer<PsfType>::FrameModelPsfContextContainer(size_t n_extended_models)
33 : PsfType(), m_psf_contexts(n_extended_models) {}
35 template <typename PsfType>
36 FrameModelPsfContextContainer<PsfType>::FrameModelPsfContextContainer(PsfType psf, size_t n_extended_models)
37 : PsfType(std::move(psf)), m_psf_contexts(n_extended_models) {}
39 template <typename PsfType, typename ImageType>
40 FrameModel<PsfType, ImageType>::FrameModel(double pixel_scale, std::size_t width, std::size_t height,
41 std::vector<ConstantModel> constant_model_list,
42 std::vector<PointModel> point_model_list,
43 std::vector<std::shared_ptr<ExtendedModel<ImageType>>> extended_model_list,
45 : m_pixel_scale{pixel_scale}, m_width{width}, m_height{height},
46 m_constant_model_list{std::move(constant_model_list)},
47 m_point_model_list{std::move(point_model_list)},
48 m_extended_model_list{std::move(extended_model_list)},
49 m_psf{std::move(psf), m_extended_model_list.size()} {
52 template <typename PsfType, typename ImageType>
53 FrameModel<PsfType, ImageType>::FrameModel(double pixel_scale, std::size_t width, std::size_t height,
54 std::vector<ConstantModel> constant_model_list,
55 std::vector<PointModel> point_model_list,
56 std::vector<std::shared_ptr<ExtendedModel<ImageType>>> extended_model_list)
57 : m_pixel_scale{pixel_scale}, m_width{width}, m_height{height},
58 m_constant_model_list{std::move(constant_model_list)},
59 m_point_model_list{std::move(point_model_list)},
60 m_extended_model_list{std::move(extended_model_list)},
61 m_psf{m_extended_model_list.size()} {
64 template <typename PsfType, typename ImageType>
65 FrameModel<PsfType, ImageType>::~FrameModel() = default;
69 template <typename ImageType>
70 void addConstantModels(ImageType& image, const std::vector<ConstantModel>& model_list) {
71 using Traits = ImageTraits<ImageType>;
72 for (auto& model : model_list) {
73 double value = model.getValue();
74 for(auto it=Traits::begin(image); it!=Traits::end(image); ++it) {
80 template <typename ImageType, typename PsfType>
81 void addPointModels(ImageType& image, const std::vector<PointModel>& model_list,
82 const PsfType& psf, double pixel_scale) {
83 using Traits = ImageTraits<ImageType>;
84 auto scale_factor = psf.getPixelScale() / pixel_scale;
85 for (auto& model : model_list) {
86 Traits::addImageToImage(image, psf.getScaledKernel(model.getValue()), scale_factor, model.getX(), model.getY());
90 template <typename ImageType, typename PsfType>
91 void addExtendedModels(ImageType& image, const std::vector<std::shared_ptr<ExtendedModel<ImageType>>>& model_list,
92 PsfType& psf, double pixel_scale) {
93 using Traits = ImageTraits<ImageType>;
94 auto scale_factor = psf.getPixelScale() / pixel_scale;
95 for (size_t i = 0; i < model_list.size(); ++i) {
96 auto& model = model_list[i];
97 std::size_t width = std::ceil(model->getWidth() / psf.getPixelScale() + psf.getSize());
101 std::size_t height = std::ceil(model->getHeight() / psf.getPixelScale() + psf.getSize());
105 auto extended_image = model->getRasterizedImage(psf.getPixelScale(), width, height);
106 psf.convolve(i, extended_image);
107 Traits::addImageToImage(image, extended_image, scale_factor, model->getX(), model->getY());
111 } // end of namespace _impl
113 template <typename PsfType, typename ImageType>
114 void FrameModel<PsfType, ImageType>::recomputeImage() {
115 using Traits = ImageTraits<ImageType>;
116 m_model_image.reset(new ImageType(Traits::factory(m_width, m_height)));
117 rasterToImage(*m_model_image);
120 template <typename PsfType, typename ImageType>
121 const ImageType& FrameModel<PsfType, ImageType>::getImage() {
123 return *m_model_image;
126 template <typename PsfType, typename ImageType>
127 void FrameModel<PsfType, ImageType>::rasterToImage(ImageType &model_image) {
128 _impl::addConstantModels(model_image, m_constant_model_list);
129 _impl::addPointModels(model_image, m_point_model_list, m_psf, m_pixel_scale);
130 _impl::addExtendedModels(model_image, m_extended_model_list, m_psf, m_pixel_scale);
133 template <typename PsfType, typename ImageType>
134 auto FrameModel<PsfType, ImageType>::begin() -> const_iterator {
136 return ImageTraits<ImageType>::begin(*m_model_image);
139 template <typename PsfType, typename ImageType>
140 auto FrameModel<PsfType, ImageType>::end() -> const_iterator {
141 return ImageTraits<ImageType>::end(*m_model_image);
144 template <typename PsfType, typename ImageType>
145 std::size_t FrameModel<PsfType, ImageType>::size() const {
146 return m_width * m_height;
149 } // end of namespace ModelFitting