Changeset 452 for trunk/src/MathUtils2.cc

Timestamp:

02/16/05 12:26:03 (19 years ago)

Author:

mar637

Message:

• Added extendLastArrayAxis

File:

• trunk/src/MathUtils2.cc (modified) (2 diffs)

trunk/src/MathUtils2.cc

@@ -31 +31 @@
 
 #include <casa/aips.h>
+#include <casa/Arrays/Array.h>
+#include <casa/Arrays/ArrayIter.h>
 #include <casa/Arrays/Vector.h>
 #include <casa/Arrays/VectorSTLIterator.h>
@@ -108 +110 @@
 }
 
-
- 
-
-
-
+template <class T>
+void mathutil::extendLastArrayAxis(casa::Array<T>& out, 
+                                   const casa::Array<T>& in,
+                                   const T& initVal)
+{
+  
+  IPosition ipin = in.shape();
+  IPosition ipout = ipin; // copy?
+  // extend the axis by 1
+  uInt axis = in.ndim()-1;
+  ipout[axis] = ipin[axis]+1;
+  out.resize(ipout);
+  out = initVal;
+  ArrayIterator<T> itout(out,axis);
+  ReadOnlyArrayIterator<T> itin(in,axis);
+  while ( !itin.pastEnd() ) {
+    itout.array() = itin.array();  // copy vector by vector
+    itin.next(); itout.next();
+  }
+}