Changeset 136 for trunk

Timestamp:

12/23/04 13:58:08 (20 years ago)

Author:

kil064

Message:

add function statistics

Location:

trunk/src

Files:

• MathUtils.cc (modified) (2 diffs)
• MathUtils.h (modified) (1 diff)

trunk/src/MathUtils.cc

@@ -33 +33 @@
 #include <casa/Arrays/Vector.h>
 #include <casa/Arrays/VectorSTLIterator.h>
+#include <casa/Arrays/MaskArrMath.h>
+#include <casa/BasicSL/String.h>
 
 #include "MathUtils.h"
@@ -90 +92 @@
   }
 }
+
+
+float mathutil::statistics (const std::string& which,  const MaskedArray<Float>& data)
+{
+   String str(which);
+   str.upcase();
+   if (str.contains(String("MIN"))) {
+      return min(data); 
+   } else if (str.contains(String("MAX"))) {
+      return max(data);
+   } else if (str.contains(String("SUMSQ"))) {
+      return sumsquares(data);
+   } else if (str.contains(String("SUM"))) {
+      return sum(data);
+   } else if (str.contains(String("MEAN"))) {
+      return mean(data);
+   } else if (str.contains(String("VAR"))) {
+      return variance(data); 
+   } else if (str.contains(String("STDDEV"))) {
+      return stddev(data);
+   } else if (str.contains(String("AVDEV"))) {
+      return avdev(data);
+   } else if (str.contains(String("RMS"))) {
+      uInt n = data.nelementsValid();
+      return sqrt(sumsquares(data)/n);
+   } else if (str.contains(String("MED"))) {
+      return median(data);
+   }
+}
+  

trunk/src/MathUtils.h

@@ -42 +42 @@
              casa::Bool relaxed=False, casa::Bool ignoreOther=False);
 
+float statistics (const std::string& which,  const casa::MaskedArray<casa::Float>& data);
 };