source: trunk/src/MathUtils.h@ 1696

Last change on this file since 1696 was 1570, checked in by Malte Marquarding, 15 years ago

Ticket #167: c++ part of running polynomial smoothing

  • Property svn:eol-style set to native
  • Property svn:keywords set to Author Date Id Revision
File size: 3.6 KB
RevLine 
[37]1//#---------------------------------------------------------------------------
2//# MathUtilities.h: General math operations
3//#---------------------------------------------------------------------------
4//# Copyright (C) 2004
[125]5//# ATNF
[37]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//#
[1033]29//# $Id: MathUtils.h 1570 2009-06-29 02:04:00Z MalteMarquarding $
[37]30//#---------------------------------------------------------------------------
[125]31#ifndef MATHUTILS_H
32#define MATHUTILS_H
[37]33
[465]34#include <string>
35#include <vector>
[81]36#include <casa/aips.h>
37#include <casa/Arrays/Vector.h>
[465]38#include <casa/BasicSL/String.h>
[37]39
[125]40namespace mathutil {
41
[176]42// Hanning smoothing
[861]43/**
44 * Hanning smooth a masked vector
45 * @param out the smoothed vector
46 * @param outmask the smoothed mask
47 * @param in the input vector
48 * @param mask the input mask
49 * @param relaxed a weighting scheme
50 * @param ignoreOther drop every second channel (NYI)
51 */
[1570]52void hanning(casa::Vector<casa::Float>& out,
[1325]53 casa::Vector<casa::Bool>& outmask,
[1570]54 const casa::Vector<casa::Float>& in,
[1325]55 const casa::Vector<casa::Bool>& mask,
[1033]56 casa::Bool relaxed=casa::False,
57 casa::Bool ignoreOther=casa::False);
[37]58
[1373]59/**
60 * Apply a running median to a masked vector.
[1570]61 * Edge solution: The first and last hwidth channels will be replicated
[1373]62 * from the first/last value from a full window.
63 * @param out the smoothed vector
64 * @param outmask the smoothed mask
65 * @param in the input vector
66 * @param mask the input mask
67 * @param hwidth half-width of the smoothing window
68 */
[1570]69 void runningMedian(casa::Vector<casa::Float>& out,
70 casa::Vector<casa::Bool>& outflag,
71 const casa::Vector<casa::Float>& in,
72 const casa::Vector<casa::Bool>& flag,
73 float hwidth);
74
75 void polyfit(casa::Vector<casa::Float>& out,
[1373]76 casa::Vector<casa::Bool>& outmask,
[1570]77 const casa::Vector<casa::Float>& in,
[1373]78 const casa::Vector<casa::Bool>& mask,
[1570]79 float hwidth, int order);
[1373]80
[1570]81
[176]82// Generate specified statistic
[829]83float statistics(const casa::String& which,
[1033]84 const casa::MaskedArray<casa::Float>& data);
[176]85
86// Replace masked value by zero
[829]87void replaceMaskByZero(casa::Vector<casa::Float>& data,
[1033]88 const casa::Vector<casa::Bool>& mask);
[452]89
[861]90/**
[1373]91 * Convert casa implementations to stl
92 * @param in casa string
93 * @return a std vector of std strings
[861]94 */
[465]95std::vector<std::string> tovectorstring(const casa::Vector<casa::String>& in);
[452]96
[861]97/**
[1373]98 * convert stl implementations to casa versions
[861]99 * @param in
100 * @return
101 */
[465]102casa::Vector<casa::String> toVectorString(const std::vector<std::string>& in);
103
[1030]104}
[125]105
[37]106#endif
Note: See TracBrowser for help on using the repository browser.