00001
00002
00003
00004
00005
00006
00007
00008
00009
00010
00011
00012
00013
00014
00015
00016
00017
00018
00019
00020
00021
00022
00023
00024
00025
00026
00027
00028 #ifndef CASA_ARRAYPARTMATH_H
00029 #define CASA_ARRAYPARTMATH_H
00030
00031 #include <casacore/casa/aips.h>
00032 #include <casacore/casa/Arrays/ArrayMath.h>
00033 #include <casacore/casa/Arrays/ArrayMathBase.h>
00034
00035 namespace casacore {
00036
00037
00038
00039
00040
00041
00042
00043
00044
00045
00046
00047
00048
00049
00050
00051
00052
00053
00054
00055
00056
00057
00058
00059
00060
00061
00062
00063
00064
00065
00066
00067
00068
00069
00070
00071
00072
00073
00074
00075
00076
00077
00078
00079
00080
00081
00082
00083
00084
00085
00086
00087
00088
00089
00090
00091
00092
00093
00094
00095
00096
00097
00098
00099
00100
00101
00102
00103
00104
00105
00106
00107 template<class T> Array<T> partialSums (const Array<T>& array,
00108 const IPosition& collapseAxes);
00109 template<class T> Array<T> partialSumSqrs (const Array<T>& array,
00110 const IPosition& collapseAxes);
00111 template<class T> Array<T> partialProducts (const Array<T>& array,
00112 const IPosition& collapseAxes);
00113 template<class T> Array<T> partialMins (const Array<T>& array,
00114 const IPosition& collapseAxes);
00115 template<class T> Array<T> partialMaxs (const Array<T>& array,
00116 const IPosition& collapseAxes);
00117 template<class T> Array<T> partialMeans (const Array<T>& array,
00118 const IPosition& collapseAxes);
00119 template<class T> inline Array<T> partialVariances (const Array<T>& array,
00120 const IPosition& collapseAxes)
00121 {
00122 return partialVariances (array, collapseAxes,
00123 partialMeans (array, collapseAxes));
00124 }
00125 template<class T> Array<T> partialVariances (const Array<T>& array,
00126 const IPosition& collapseAxes,
00127 const Array<T>& means);
00128 template<class T> inline Array<T> partialStddevs (const Array<T>& array,
00129 const IPosition& collapseAxes)
00130 {
00131 return sqrt (partialVariances (array, collapseAxes,
00132 partialMeans (array, collapseAxes)));
00133 }
00134 template<class T> inline Array<T> partialStddevs (const Array<T>& array,
00135 const IPosition& collapseAxes,
00136 const Array<T>& means)
00137 {
00138 return sqrt (partialVariances (array, collapseAxes, means));
00139 }
00140 template<class T> inline Array<T> partialAvdevs (const Array<T>& array,
00141 const IPosition& collapseAxes)
00142 {
00143 return partialAvdevs (array, collapseAxes,
00144 partialMeans (array, collapseAxes));
00145 }
00146 template<class T> Array<T> partialAvdevs (const Array<T>& array,
00147 const IPosition& collapseAxes,
00148 const Array<T>& means);
00149 template<class T> Array<T> partialRmss (const Array<T>& array,
00150 const IPosition& collapseAxes);
00151 template<class T> Array<T> partialMedians (const Array<T>& array,
00152 const IPosition& collapseAxes,
00153 Bool takeEvenMean=False,
00154 Bool inPlace=False);
00155 template<class T> Array<T> partialMadfms (const Array<T>& array,
00156 const IPosition& collapseAxes,
00157 Bool takeEvenMean=False,
00158 Bool inPlace=False);
00159 template<class T> Array<T> partialFractiles (const Array<T>& array,
00160 const IPosition& collapseAxes,
00161 Float fraction,
00162 Bool inPlace=False);
00163 template<class T> Array<T> partialInterFractileRanges (const Array<T>& array,
00164 const IPosition& collapseAxes,
00165 Float fraction,
00166 Bool inPlace=False);
00167 template<class T> Array<T> partialInterHexileRanges (const Array<T>& array,
00168 const IPosition& collapseAxes,
00169 Bool inPlace=False)
00170 { return partialInterFractileRanges (array, collapseAxes, 1./6., inPlace); }
00171 template<class T> Array<T> partialInterQuartileRanges (const Array<T>& array,
00172 const IPosition& collapseAxes,
00173 Bool inPlace=False)
00174 { return partialInterFractileRanges (array, collapseAxes, 0.25, inPlace); }
00175
00176
00177
00178
00179
00180
00181
00182 template<typename T> class SumFunc : public ArrayFunctorBase<T> {
00183 public:
00184 virtual ~SumFunc() {}
00185 virtual T operator() (const Array<T>& arr) const { return sum(arr); }
00186 };
00187 template<typename T> class SumSqrFunc : public ArrayFunctorBase<T> {
00188 public:
00189 virtual ~SumSqrFunc() {}
00190 virtual T operator() (const Array<T>& arr) const { return sumsqr(arr); }
00191 };
00192 template<typename T> class ProductFunc : public ArrayFunctorBase<T> {
00193 public:
00194 virtual ~ProductFunc() {}
00195 virtual T operator() (const Array<T>& arr) const { return product(arr); }
00196 };
00197 template<typename T> class MinFunc : public ArrayFunctorBase<T> {
00198 public:
00199 virtual ~MinFunc() {}
00200 virtual T operator() (const Array<T>& arr) const { return min(arr); }
00201 };
00202 template<typename T> class MaxFunc : public ArrayFunctorBase<T> {
00203 public:
00204 virtual ~MaxFunc() {}
00205 virtual T operator() (const Array<T>& arr) const { return max(arr); }
00206 };
00207 template<typename T> class MeanFunc : public ArrayFunctorBase<T> {
00208 public:
00209 virtual ~MeanFunc() {}
00210 virtual T operator() (const Array<T>& arr) const { return mean(arr); }
00211 };
00212 template<typename T> class VarianceFunc : public ArrayFunctorBase<T> {
00213 public:
00214 virtual ~VarianceFunc() {}
00215 virtual T operator() (const Array<T>& arr) const { return variance(arr); }
00216 };
00217 template<typename T> class StddevFunc : public ArrayFunctorBase<T> {
00218 public:
00219 virtual ~StddevFunc() {}
00220 virtual T operator() (const Array<T>& arr) const { return stddev(arr); }
00221 };
00222 template<typename T> class AvdevFunc : public ArrayFunctorBase<T> {
00223 public:
00224 virtual ~AvdevFunc() {}
00225 virtual T operator() (const Array<T>& arr) const { return avdev(arr); }
00226 };
00227 template<typename T> class RmsFunc : public ArrayFunctorBase<T> {
00228 public:
00229 virtual ~RmsFunc() {}
00230 virtual T operator() (const Array<T>& arr) const { return rms(arr); }
00231 };
00232 template<typename T> class MedianFunc : public ArrayFunctorBase<T> {
00233 public:
00234 explicit MedianFunc (Bool sorted=False, Bool takeEvenMean=True,
00235 Bool inPlace = False)
00236 : itsSorted(sorted), itsTakeEvenMean(takeEvenMean), itsInPlace(inPlace) {}
00237 virtual ~MedianFunc() {}
00238 virtual T operator() (const Array<T>& arr) const
00239 { return median(arr, itsTmp, itsSorted, itsTakeEvenMean, itsInPlace); }
00240 private:
00241 Bool itsSorted;
00242 Bool itsTakeEvenMean;
00243 Bool itsInPlace;
00244 mutable Block<T> itsTmp;
00245 };
00246 template<typename T> class MadfmFunc {
00247 public:
00248 explicit MadfmFunc(Bool sorted = False, Bool takeEvenMean = True,
00249 Bool inPlace = False)
00250 : itsSorted(sorted), itsTakeEvenMean(takeEvenMean), itsInPlace(inPlace) {}
00251 virtual ~MadfmFunc() {}
00252 virtual T operator()(const Array<T>& arr) const
00253 { return madfm(arr, itsTmp, itsSorted, itsTakeEvenMean, itsInPlace); }
00254 private:
00255 Bool itsSorted;
00256 Bool itsTakeEvenMean;
00257 Bool itsInPlace;
00258 mutable Block<Float> itsTmp;
00259 };
00260 template<typename T> class FractileFunc : public ArrayFunctorBase<T> {
00261 public:
00262 explicit FractileFunc (Float fraction,
00263 Bool sorted = False, Bool inPlace = False)
00264 : itsFraction(fraction), itsSorted(sorted), itsInPlace(inPlace) {}
00265 virtual ~FractileFunc() {}
00266 virtual T operator() (const Array<T>& arr) const
00267 { return fractile(arr, itsTmp, itsFraction, itsSorted, itsInPlace); }
00268 private:
00269 float itsFraction;
00270 Bool itsSorted;
00271 Bool itsInPlace;
00272 mutable Block<T> itsTmp;
00273 };
00274 template<typename T> class InterFractileRangeFunc {
00275 public:
00276 explicit InterFractileRangeFunc(Float fraction,
00277 Bool sorted = False, Bool inPlace = False)
00278 : itsFraction(fraction), itsSorted(sorted), itsInPlace(inPlace) {}
00279 virtual ~InterFractileRangeFunc() {}
00280 virtual T operator()(const Array<T>& arr) const
00281 { return interFractileRange(arr, itsTmp, itsFraction,
00282 itsSorted, itsInPlace); }
00283 private:
00284 float itsFraction;
00285 Bool itsSorted;
00286 Bool itsInPlace;
00287 mutable Block<Float> itsTmp;
00288 };
00289 template<typename T> class InterHexileRangeFunc: public InterFractileRangeFunc<T> {
00290 public:
00291 explicit InterHexileRangeFunc(Bool sorted = False, Bool inPlace = False)
00292 : InterFractileRangeFunc<T> (1./6., sorted, inPlace)
00293 {}
00294 virtual ~InterHexileRangeFunc() {}
00295 };
00296 template<typename T> class InterQuartileRangeFunc: public InterFractileRangeFunc<T> {
00297 public:
00298 explicit InterQuartileRangeFunc(Bool sorted = False, Bool inPlace = False)
00299 : InterFractileRangeFunc<T> (0.25, sorted, inPlace)
00300 {}
00301 virtual ~InterQuartileRangeFunc() {}
00302 };
00303
00304
00305
00306
00307
00308 template<typename T>
00309 inline Array<T> partialArrayMath (const Array<T>& a,
00310 const IPosition& collapseAxes,
00311 const ArrayFunctorBase<T>& funcObj)
00312 {
00313 Array<T> res;
00314 partialArrayMath (res, a, collapseAxes, funcObj);
00315 return res;
00316 }
00317 template<typename T, typename RES>
00318 void partialArrayMath (Array<RES>& res,
00319 const Array<T>& a,
00320 const IPosition& collapseAxes,
00321 const ArrayFunctorBase<T,RES>& funcObj);
00322
00323
00324
00325
00326
00327
00328
00329
00330
00331
00332
00333
00334
00335
00336 template<typename T>
00337 inline Array<T> boxedArrayMath (const Array<T>& a,
00338 const IPosition& boxSize,
00339 const ArrayFunctorBase<T>& funcObj)
00340 {
00341 Array<T> res;
00342 boxedArrayMath (res, a, boxSize, funcObj);
00343 return res;
00344 }
00345 template<typename T, typename RES>
00346 void boxedArrayMath (Array<RES>&,
00347 const Array<T>& array,
00348 const IPosition& boxSize,
00349 const ArrayFunctorBase<T,RES>& funcObj);
00350
00351
00352
00353
00354
00355
00356
00357
00358
00359
00360
00361
00362
00363
00364
00365
00366
00367
00368
00369
00370
00371
00372
00373
00374
00375 template<typename T>
00376 inline Array<T> slidingArrayMath (const Array<T>& a,
00377 const IPosition& halfBoxSize,
00378 const ArrayFunctorBase<T>& funcObj,
00379 Bool fillEdge=True)
00380 {
00381 Array<T> res;
00382 slidingArrayMath (res, a, halfBoxSize, funcObj, fillEdge);
00383 return res;
00384 }
00385 template<typename T, typename RES>
00386 void slidingArrayMath (Array<RES>& res,
00387 const Array<T>& array,
00388 const IPosition& halfBoxSize,
00389 const ArrayFunctorBase<T,RES>& funcObj,
00390 Bool fillEdge=True);
00391
00392
00393
00394
00395
00396
00397 void fillBoxedShape (const IPosition& shape, const IPosition& boxShape,
00398 IPosition& fullBoxShape, IPosition& resultShape);
00399
00400
00401 Bool fillSlidingShape (const IPosition& shape, const IPosition& halfBoxSize,
00402 IPosition& boxEnd, IPosition& resultShape);
00403
00404
00405 }
00406
00407 #ifndef CASACORE_NO_AUTO_TEMPLATES
00408 #include <casacore/casa/Arrays/ArrayPartMath.tcc>
00409 #endif //# CASACORE_NO_AUTO_TEMPLATES
00410 #endif