source: branches/asap-3.x/src/MathUtils.h@ 2749

Last change on this file since 2749 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
Line 
1//#---------------------------------------------------------------------------
2//# MathUtilities.h: General math operations
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: MathUtils.h 1570 2009-06-29 02:04:00Z MalteMarquarding $
30//#---------------------------------------------------------------------------
31#ifndef MATHUTILS_H
32#define MATHUTILS_H
33
34#include <string>
35#include <vector>
36#include <casa/aips.h>
37#include <casa/Arrays/Vector.h>
38#include <casa/BasicSL/String.h>
39
40namespace mathutil {
41
42// Hanning smoothing
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 */
52void hanning(casa::Vector<casa::Float>& out,
53 casa::Vector<casa::Bool>& outmask,
54 const casa::Vector<casa::Float>& in,
55 const casa::Vector<casa::Bool>& mask,
56 casa::Bool relaxed=casa::False,
57 casa::Bool ignoreOther=casa::False);
58
59/**
60 * Apply a running median to a masked vector.
61 * Edge solution: The first and last hwidth channels will be replicated
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 */
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,
76 casa::Vector<casa::Bool>& outmask,
77 const casa::Vector<casa::Float>& in,
78 const casa::Vector<casa::Bool>& mask,
79 float hwidth, int order);
80
81
82// Generate specified statistic
83float statistics(const casa::String& which,
84 const casa::MaskedArray<casa::Float>& data);
85
86// Replace masked value by zero
87void replaceMaskByZero(casa::Vector<casa::Float>& data,
88 const casa::Vector<casa::Bool>& mask);
89
90/**
91 * Convert casa implementations to stl
92 * @param in casa string
93 * @return a std vector of std strings
94 */
95std::vector<std::string> tovectorstring(const casa::Vector<casa::String>& in);
96
97/**
98 * convert stl implementations to casa versions
99 * @param in
100 * @return
101 */
102casa::Vector<casa::String> toVectorString(const std::vector<std::string>& in);
103
104}
105
106#endif
Note: See TracBrowser for help on using the repository browser.