source: trunk/src/python_SDMath.cc@ 376

Last change on this file since 376 was 311, checked in by kil064, 20 years ago

replace velocity alignment function by frequency alignment one

  • Property svn:eol-style set to native
  • Property svn:keywords set to Author Date Id Revision
File size: 3.8 KB
Line 
1//#---------------------------------------------------------------------------
2//# python_SDMath.cc: python exposure of c++ SDMath class
3//#---------------------------------------------------------------------------
4//# Copyright (C) 2004
5//# ATNF
6//#
7//# This program is free software; you can redistribute it and/or modify it
8//# under the terms of the GNU General Public License as published by the Free
9//# Software Foundation; either version 2 of the License, or (at your option)
10//# any later version.
11//#
12//# This program is distributed in the hope that it will be useful, but
13//# WITHOUT ANY WARRANTY; without even the implied warranty of
14//# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General
15//# Public License for more details.
16//#
17//# You should have received a copy of the GNU General Public License along
18//# with this program; if not, write to the Free Software Foundation, Inc.,
19//# 675 Massachusetts Ave, Cambridge, MA 02139, USA.
20//#
21//# Correspondence concerning this software should be addressed as follows:
22//# Internet email: Malte.Marquarding@csiro.au
23//# Postal address: Malte Marquarding,
24//# Australia Telescope National Facility,
25//# P.O. Box 76,
26//# Epping, NSW, 2121,
27//# AUSTRALIA
28//#
29//# $Id:
30//#---------------------------------------------------------------------------
31#include <boost/python.hpp>
32
33#include <casa/aips.h>
34#include <casa/Containers/Block.h>
35#include <casa/Utilities/CountedPtr.cc>
36
37#include "SDMathWrapper.h"
38#include "SDMath.h"
39
40using namespace casa;
41using namespace boost::python;
42
43namespace asap {
44 namespace SDMathWrapper {
45 SDMemTableWrapper SDMathWrapper::average(boost::python::tuple tp,
46 const std::vector<bool>& mask,
47 bool scanAv,
48 const std::string& weightStr) {
49 int n;
50 n = extract<int>(tp.attr("__len__")());
51 Block<CountedPtr<asap::SDMemTable> > b(n);
52 for (int i=0;i< n;++i) {
53 SDMemTableWrapper sdmw =
54 extract<SDMemTableWrapper>( tp.attr("__getitem__")(i) );
55 b[i] = sdmw.getCP();
56 }
57 Vector<Bool> msk(mask);
58//
59 SDMath sdm;
60 return SDMemTableWrapper(sdm.average(b, msk, Bool(scanAv), weightStr));
61 };
62 } // namespace SDMathWrapper
63
64 namespace python {
65 void python_SDMath() {
66 def("b_operate", &SDMathWrapper::binaryOperate);
67//
68 def("quotient", &SDMathWrapper::quotient);
69//
70 def("scale", &SDMathWrapper::scale);
71 def("scale_insitu", &SDMathWrapper::scaleInSitu);
72//
73 def("add", &SDMathWrapper::add);
74 def("add_insitu", &SDMathWrapper::addInSitu);
75//
76 def("smooth", &SDMathWrapper::smooth);
77 def("smooth_insitu", &SDMathWrapper::smoothInSitu);
78//
79 def("convertflux", &SDMathWrapper::convertFlux);
80 def("convertflux_insitu", &SDMathWrapper::convertFluxInSitu);
81//
82 def("gainel", &SDMathWrapper::gainElevation);
83 def("gainel_insitu", &SDMathWrapper::gainElevationInSitu);
84//
85 def("freq_align", &SDMathWrapper::frequencyAlignment);
86 def("freq_align_insitu", &SDMathWrapper::frequencyAlignmentInSitu);
87//
88 def("opacity", &SDMathWrapper::opacity);
89 def("opacity_insitu", &SDMathWrapper::opacityInSitu);
90//
91 def("average", &SDMathWrapper::average);
92//
93 def("averagepol", &SDMathWrapper::averagePol);
94 def("averagepol_insitu", &SDMathWrapper::averagePolInSitu);
95//
96 def("bin", &SDMathWrapper::bin);
97 def("bin_insitu", &SDMathWrapper::binInSitu);
98//
99 def("resample", &SDMathWrapper::resample);
100 def("resample_insitu", &SDMathWrapper::resampleInSitu);
101//
102 def("stats", &SDMathWrapper::statistic);
103 };
104
105 } // python
106} // asap
107
Note: See TracBrowser for help on using the repository browser.