SourceXtractorPlusPlus  0.12
Please provide a description of the project.
FrameModel.icpp
Go to the documentation of this file.
1 /** Copyright © 2019 Université de Genève, LMU Munich - Faculty of Physics, IAP-CNRS/Sorbonne Université
2  *
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)
6  * any later version.
7  *
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
11  * details.
12  *
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
16  */
17 /**
18  * @file FrameModel.icpp
19  * @date September 1, 2015
20  * @author Nikolaos Apostolakos
21  */
22 
23 namespace ModelFitting {
24 
25 template <typename PsfType>
26 FrameModelPsfContainer<PsfType>::FrameModelPsfContainer(size_t): PsfType() {}
27 
28 template <typename PsfType>
29 FrameModelPsfContainer<PsfType>::FrameModelPsfContainer(PsfType psf, size_t): PsfType(std::move(psf)) {}
30 
31 template <typename PsfType>
32 FrameModelPsfContextContainer<PsfType>::FrameModelPsfContextContainer(size_t n_extended_models)
33  : PsfType(), m_psf_contexts(n_extended_models) {}
34 
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) {}
38 
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,
44  PsfType psf)
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()} {
50 }
51 
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()} {
62 }
63 
64 template <typename PsfType, typename ImageType>
65 FrameModel<PsfType, ImageType>::~FrameModel() = default;
66 
67 namespace _impl {
68 
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) {
75  *it += value;
76  }
77  }
78 }
79 
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());
87  }
88 }
89 
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());
98  if (width%2 == 0) {
99  ++width;
100  }
101  std::size_t height = std::ceil(model->getHeight() / psf.getPixelScale() + psf.getSize());
102  if (height%2 == 0) {
103  ++height;
104  }
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());
108  }
109 }
110 
111 } // end of namespace _impl
112 
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);
118 }
119 
120 template <typename PsfType, typename ImageType>
121 const ImageType& FrameModel<PsfType, ImageType>::getImage() {
122  recomputeImage();
123  return *m_model_image;
124 }
125 
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);
131 }
132 
133 template <typename PsfType, typename ImageType>
134 auto FrameModel<PsfType, ImageType>::begin() -> const_iterator {
135  recomputeImage();
136  return ImageTraits<ImageType>::begin(*m_model_image);
137 }
138 
139 template <typename PsfType, typename ImageType>
140 auto FrameModel<PsfType, ImageType>::end() -> const_iterator {
141  return ImageTraits<ImageType>::end(*m_model_image);
142 }
143 
144 template <typename PsfType, typename ImageType>
145 std::size_t FrameModel<PsfType, ImageType>::size() const {
146  return m_width * m_height;
147 }
148 
149 } // end of namespace ModelFitting
150