Changeset 1534 for trunk/src/LineCatalog.cpp

Timestamp:

03/24/09 12:33:56 (15 years ago)

Author:

Malte Marquarding

Message:

Fix for Ticket #90; treating integer as double columns now

File:

• trunk/src/LineCatalog.cpp (modified) (3 diffs)

trunk/src/LineCatalog.cpp

@@ -62 +62 @@
 void LineCatalog::setPattern(const std::string& name, const std::string& stype)
 {
-  //std::string mode = stype+"('"+name+"')";
-  std::string taql = "SELECT FROM $1 WHERE Column1 == pattern('"+name+"')";
-  cerr << taql << endl;
+  std::string mode = stype+"('"+name+"')";
+  std::string taql = "SELECT FROM $1 WHERE Column1 == " + mode;
   Table tmp = tableCommand(taql, table_);
   if (tmp.nrow() > 0) table_ = tmp.sort("Column2");
@@ -110 +109 @@
       }
   }
-  /// @todo implement me
   return String(oss);
 }
 
-/*!
-    \fn asap::LineCatalog::getName(int row)
- */
+
 std::string LineCatalog::getName(uint row) const
 {
@@ -123 +119 @@
 }
 
-double asap::LineCatalog::getFrequency(uint row) const
+double LineCatalog::getFrequency(uint row) const
 {
-  ROScalarColumn<Double> col(table_, "Column2");
-  return col(row);
+  return getDouble("Column2", row);
 }
 
-double asap::LineCatalog::getStrength(uint row) const
+double LineCatalog::getStrength(uint row) const
 {
-  ROScalarColumn<Double> col(table_, "Column3");
-  return col(row);
+  return getDouble("Column4", row);
 }
 
+double LineCatalog::getDouble(const std::string& colname, uint row) const {
+  DataType dtype = table_.tableDesc().columnDesc(colname).dataType();
+  if (dtype == TpDouble) {
+    ROScalarColumn<Double> col(table_, colname);
+    return col(row);
+  } else if (dtype == TpInt) {
+    ROScalarColumn<Int> col(table_, colname);
+    return Double(col(row));
+  } else {
+    throw AipsError("Column " + colname + "doesn't contain numerical values." );
+  }
+}
 
 } // namespace