source: trunk/src/python_SDMath.cc@ 778

//#---------------------------------------------------------------------------
//# python_SDMath.cc: python exposure of c++ SDMath class
//#---------------------------------------------------------------------------
//# Copyright (C) 2004
//# ATNF
//#
//# This program is free software; you can redistribute it and/or modify it
//# under the terms of the GNU General Public License as published by the Free
//# Software Foundation; either version 2 of the License, or (at your option)
//# any later version.
//#
//# This program is distributed in the hope that it will be useful, but
//# WITHOUT ANY WARRANTY; without even the implied warranty of
//# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General
//# Public License for more details.
//#
//# You should have received a copy of the GNU General Public License along
//# with this program; if not, write to the Free Software Foundation, Inc.,
//# 675 Massachusetts Ave, Cambridge, MA 02139, USA.
//#
//# Correspondence concerning this software should be addressed as follows:
//#        Internet email: Malte.Marquarding@csiro.au
//#        Postal address: Malte Marquarding,
//#                        Australia Telescope National Facility,
//#                        P.O. Box 76,
//#                        Epping, NSW, 2121,
//#                        AUSTRALIA
//#
//# $Id:
//#---------------------------------------------------------------------------
#include <boost/python.hpp>
#include <vector>

#include <casa/aips.h>
#include <casa/Containers/Block.h>
#include <casa/Utilities/CountedPtr.cc>

#include "SDMathWrapper.h"
#include "SDMath.h"

using namespace casa;
using namespace boost::python;

namespace asap {
  namespace SDMathWrapper {
    SDMemTableWrapper SDMathWrapper::average(boost::python::tuple tp,
                                             const std::vector<bool>& mask,
                                             bool scanAv,
                                             const std::string& weightStr) {
      int n;
      n = extract<int>(tp.attr("__len__")());
      std::vector<CountedPtr<asap::SDMemTable> > b(n);
      //Block<CountedPtr<asap::SDMemTable> > b(n);
      for (int i=0;i< n;++i) {
        SDMemTableWrapper sdmw =
          extract<SDMemTableWrapper>( tp.attr("__getitem__")(i) );
        b[i] = sdmw.getCP();
      }
      Vector<Bool> msk(mask);

      return SDMemTableWrapper(sdmath.average(b, msk, Bool(scanAv), weightStr));
    };
  } // namespace SDMathWrapper

  namespace python {
    void python_SDMath() {
      def("b_operate", &SDMathWrapper::binaryOperate);

      def("quotient", &SDMathWrapper::quotient);

      def("scale", &SDMathWrapper::scale);
      def("scale_insitu", &SDMathWrapper::scaleInSitu);

      def("add", &SDMathWrapper::add);
      def("add_insitu", &SDMathWrapper::addInSitu);

      def("smooth", &SDMathWrapper::smooth);
      def("smooth_insitu", &SDMathWrapper::smoothInSitu);

      def("convertflux", &SDMathWrapper::convertFlux);
      def("convertflux_insitu", &SDMathWrapper::convertFluxInSitu);

      def("gainel", &SDMathWrapper::gainElevation);
      def("gainel_insitu", &SDMathWrapper::gainElevationInSitu);

      def("freq_align", &SDMathWrapper::frequencyAlignment);
      def("freq_align_insitu", &SDMathWrapper::frequencyAlignmentInSitu);

      def("opacity", &SDMathWrapper::opacity);
      def("opacity_insitu", &SDMathWrapper::opacityInSitu);

      def("average", &SDMathWrapper::average);

      def("averagepol", &SDMathWrapper::averagePol);
      def("averagepol_insitu", &SDMathWrapper::averagePolInSitu);

      def("bin", &SDMathWrapper::bin);
      def("bin_insitu", &SDMathWrapper::binInSitu);

      def("resample", &SDMathWrapper::resample);
      def("resample_insitu", &SDMathWrapper::resampleInSitu);

      def("stats", &SDMathWrapper::statistic);

      def("_rotate_xyphase", &SDMathWrapper::rotateXYPhaseInSitu);
      def("_rotate_linpolphase", &SDMathWrapper::rotateLinPolPhaseInSitu);

      def("_frequency_switch", &SDMathWrapper::frequencySwitch);
      def("_frequency_switch_insitu", &SDMathWrapper::frequencySwitchInSitu);
    };

  } // python
} // asap