lsst.astshim gee0fe1808a+052faf71bd
Mapping.h
Go to the documentation of this file.
1/*
2 * LSST Data Management System
3 * Copyright 2017 AURA/LSST.
4 *
5 * This product includes software developed by the
6 * LSST Project (http://www.lsst.org/).
7 *
8 * This program is free software: you can redistribute it and/or modify
9 * it under the terms of the GNU General Public License as published by
10 * the Free Software Foundation, either version 3 of the License, or
11 * (at your option) any later version.
12 *
13 * This program is distributed in the hope that it will be useful,
14 * but WITHOUT ANY WARRANTY; without even the implied warranty of
15 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
16 * GNU General Public License for more details.
17 *
18 * You should have received a copy of the LSST License Statement and
19 * the GNU General Public License along with this program. If not,
20 * see <https://www.lsstcorp.org/LegalNotices/>.
21 */
22#ifndef ASTSHIM_MAPPING_H
23#define ASTSHIM_MAPPING_H
24
25#include <memory>
26#include <vector>
27
28#include "ndarray.h"
29
30#include "astshim/base.h"
32#include "astshim/Object.h"
33
34namespace ast {
35
36class ParallelMap;
37class SeriesMap;
38
59class Mapping : public Object {
60 friend class Object;
61
62public:
63 virtual ~Mapping() {}
64
66 Mapping(Mapping const &) = default;
67 Mapping(Mapping &&) = default;
68 Mapping &operator=(Mapping const &) = delete;
69 Mapping &operator=(Mapping &&) = default;
70
72 std::shared_ptr<Mapping> copy() const { return std::static_pointer_cast<Mapping>(copyPolymorphic()); }
73
77 int getNIn() const { return getI("NIn"); }
78
82 int getNOut() const { return getI("NOut"); }
83
87 bool getIsSimple() const { return getI("IsSimple"); }
88
95 bool isInverted() const { return getB("Invert"); }
96
100 bool getIsLinear() const { return getB("IsLinear"); }
101
105 bool getReport() const { return getB("Report"); }
106
114 bool hasForward() const { return getB("TranForward"); }
115
123 bool hasInverse() const { return getB("TranInverse"); }
124
138 std::shared_ptr<Mapping> inverted() const;
139
164 Array2D linearApprox(PointD const &lbnd, PointD const &ubnd, double tol) const;
165
177 SeriesMap then(Mapping const &next) const;
178
195 ParallelMap under(Mapping const &next) const;
196
215 double rate(PointD const &at, int ax1, int ax2) const {
216 detail::assertEqual(at.size(), "at.size", static_cast<std::size_t>(getNIn()), "nIn");
217 double result = astRate(getRawPtr(), const_cast<double *>(at.data()), ax1, ax2);
218 assertOK();
219 return result;
220 }
221
225 void setReport(bool report) { setB("Report", report); }
226
248 std::shared_ptr<Mapping> simplified() const {
249 AstObject *rawSimpMap = reinterpret_cast<AstObject *>(astSimplify(getRawPtr()));
250 assertOK(rawSimpMap);
251 return Object::fromAstObject<Mapping>(rawSimpMap, true);
252 }
253
260 void applyForward(ConstArray2D const &from, Array2D const &to) const { _tran(from, true, to); }
261
268 Array2D applyForward(ConstArray2D const &from) const {
269 Array2D to = ndarray::allocate(getNOut(), from.getSize<1>());
270 _tran(from, true, to);
271 return to;
272 }
273
280 std::vector<double> applyForward(std::vector<double> const &from) const {
281 auto fromArr = arrayFromVector(from, getNIn());
282 std::vector<double> to(fromArr.getSize<1>() * getNOut());
283 auto toArr = arrayFromVector(to, getNOut());
284 _tran(fromArr, true, toArr);
285 return to;
286 }
287
294 void applyInverse(ConstArray2D const &from, Array2D const &to) const { _tran(from, false, to); }
295
302 Array2D applyInverse(ConstArray2D const &from) const {
303 Array2D to = ndarray::allocate(getNIn(), from.getSize<1>());
304 _tran(from, false, to);
305 return to;
306 }
307
314 std::vector<double> applyInverse(std::vector<double> const &from) const {
315 auto fromArr = arrayFromVector(from, getNOut());
316 std::vector<double> to(fromArr.getSize<1>() * getNIn());
317 auto toArr = arrayFromVector(to, getNIn());
318 _tran(fromArr, false, toArr);
319 return to;
320 }
321
358 void tranGridForward(PointI const &lbnd, PointI const &ubnd, double tol, int maxpix,
359 Array2D const &to) const {
360 _tranGrid(lbnd, ubnd, tol, maxpix, true, to);
361 }
362
369 Array2D tranGridForward(PointI const &lbnd, PointI const &ubnd, double tol, int maxpix, int nPts) const {
370 Array2D to = ndarray::allocate(nPts, getNOut());
371 _tranGrid(lbnd, ubnd, tol, maxpix, true, to);
372 return to;
373 }
374
380 void tranGridInverse(PointI const &lbnd, PointI const &ubnd, double tol, int maxpix,
381 Array2D const &to) const {
382 _tranGrid(lbnd, ubnd, tol, maxpix, false, to);
383 }
384
390 Array2D tranGridInverse(PointI const &lbnd, PointI const &ubnd, double tol, int maxpix, int nPts) const {
391 Array2D to = ndarray::allocate(nPts, getNIn());
392 _tranGrid(lbnd, ubnd, tol, maxpix, false, to);
393 return to;
394 }
395
396protected:
400 explicit Mapping(AstMapping *rawMap) : Object(reinterpret_cast<AstObject *>(rawMap)) {
401 assertOK();
402 if (!astIsAMapping(getRawPtr())) {
403 std::ostringstream os;
404 os << "this is a " << getClassName() << ", which is not a Mapping";
405 throw std::invalid_argument(os.str());
406 }
407 }
408
409 // Protected implementation of deep-copy.
410 virtual std::shared_ptr<Object> copyPolymorphic() const override {
411 return std::static_pointer_cast<Mapping>(copyImpl<Mapping, AstMapping>());
412 }
413
427 template <typename Class>
428 std::shared_ptr<Class> decompose(int i, bool copy) const;
429
430private:
438 void _tran(ConstArray2D const &from, bool doForward, Array2D const &to) const;
439
443 void _tranGrid(PointI const &lbnd, PointI const &ubnd, double tol, int maxpix, bool doForward,
444 Array2D const &to) const;
445};
446
447} // namespace ast
448
449#endif
Definition: Mapping.h:59
bool hasForward() const
Definition: Mapping.h:114
std::shared_ptr< Mapping > simplified() const
Definition: Mapping.h:248
bool isInverted() const
Definition: Mapping.h:95
void tranGridInverse(PointI const &lbnd, PointI const &ubnd, double tol, int maxpix, Array2D const &to) const
Definition: Mapping.h:380
std::shared_ptr< Mapping > copy() const
Return a deep copy of this object.
Definition: Mapping.h:72
Array2D tranGridForward(PointI const &lbnd, PointI const &ubnd, double tol, int maxpix, int nPts) const
Definition: Mapping.h:369
void tranGridForward(PointI const &lbnd, PointI const &ubnd, double tol, int maxpix, Array2D const &to) const
Definition: Mapping.h:358
SeriesMap then(Mapping const &next) const
Definition: Mapping.cc:37
Array2D applyForward(ConstArray2D const &from) const
Definition: Mapping.h:268
bool getReport() const
Definition: Mapping.h:105
Mapping(Mapping const &)=default
Copy constructor: make a deep copy.
void applyInverse(ConstArray2D const &from, Array2D const &to) const
Definition: Mapping.h:294
Mapping(AstMapping *rawMap)
Definition: Mapping.h:400
void applyForward(ConstArray2D const &from, Array2D const &to) const
Definition: Mapping.h:260
std::shared_ptr< Class > decompose(int i, bool copy) const
Definition: Mapping.cc:64
ParallelMap under(Mapping const &next) const
Definition: Mapping.cc:39
Array2D linearApprox(PointD const &lbnd, PointD const &ubnd, double tol) const
Definition: Mapping.cc:49
bool getIsSimple() const
Definition: Mapping.h:87
Array2D applyInverse(ConstArray2D const &from) const
Definition: Mapping.h:302
Mapping & operator=(Mapping const &)=delete
Array2D tranGridInverse(PointI const &lbnd, PointI const &ubnd, double tol, int maxpix, int nPts) const
Definition: Mapping.h:390
double rate(PointD const &at, int ax1, int ax2) const
Definition: Mapping.h:215
bool getIsLinear() const
Definition: Mapping.h:100
int getNOut() const
Definition: Mapping.h:82
virtual ~Mapping()
Definition: Mapping.h:63
Mapping(Mapping &&)=default
void setReport(bool report)
Definition: Mapping.h:225
std::vector< double > applyInverse(std::vector< double > const &from) const
Definition: Mapping.h:314
virtual std::shared_ptr< Object > copyPolymorphic() const override
Definition: Mapping.h:410
bool hasInverse() const
Definition: Mapping.h:123
int getNIn() const
Definition: Mapping.h:77
std::shared_ptr< Mapping > inverted() const
Definition: Mapping.cc:41
std::vector< double > applyForward(std::vector< double > const &from) const
Definition: Mapping.h:280
Mapping & operator=(Mapping &&)=default
Definition: Object.h:51
std::string getClassName() const
Definition: Object.h:139
bool getB(std::string const &attrib) const
Definition: Object.h:348
int getI(std::string const &attrib) const
Definition: Object.h:400
void setB(std::string const &attrib, bool value)
Definition: Object.h:452
AstObject const * getRawPtr() const
Definition: Object.h:292
Definition: ParallelMap.h:53
Definition: SeriesMap.h:53
void assertEqual(T1 val1, std::string const &descr1, T2 val2, std::string const &descr2)
Definition: utils.h:48
AST wrapper classes and functions.
Definition: attributes_channel.dox:1
std::vector< int > PointI
Definition: base.h:50
void assertOK(AstObject *rawPtr1=nullptr, AstObject *rawPtr2=nullptr)
Definition: base.cc:49
std::vector< double > PointD
Definition: base.h:57
ConstArray2D arrayFromVector(std::vector< double > const &vec, int nAxes)
Definition: base.cc:65
ndarray::Array< double, 2, 2 > Array2D
Definition: base.h:42
ndarray::Array< const double, 2, 2 > ConstArray2D
Definition: base.h:46
def next(self)
Definition: fitsChanContinued.py:105