Changeset 1779 for branches/mergetest/src/STSelector.cpp

Timestamp:

07/29/10 19:13:46 (14 years ago)

Author:

Kana Sugimoto

Message:

New Development: Yes

JIRA Issue: No (test merging alma branch)

Ready for Test: Yes

Interface Changes: Yes

What Interface Changed:

Test Programs:

Put in Release Notes: No

Module(s):

Description:

Location:

branches/mergetest

Files:

• . (modified) (1 prop)
• src/STSelector.cpp (modified) (6 diffs)

branches/mergetest/src/STSelector.cpp

@@ -87 +87 @@
 }
 
+void STSelector::setTypes( const std::vector< int >& types )
+{
+  setint("SRCTYPE", types);
+}
+
 void STSelector::setint(const std::string& key, const std::vector< int >& val)
 {
@@ -116 +121 @@
 }
 
+void STSelector::setRows( const std::vector< int >& rows )
+{
+  rowselection_ = rows;
+}
+
+// Table STSelector::apply( const Table& tab )
+// {
+//   if ( empty() ) {
+//     return sort(tab);
+//   }
+//   TableExprNode query;
+//   intidmap::const_iterator it;
+//   for (it = intselections_.begin(); it != intselections_.end(); ++it) {
+//     TableExprNode theset(Vector<Int>( (*it).second ));
+//     if ( query.isNull() ) {
+//       query = tab.col((*it).first).in(theset);
+//     } else {
+//       query = tab.col((*it).first).in(theset) && query;
+//     }
+//   }
+//   stringidmap::const_iterator it1;
+//   for (it1 = stringselections_.begin(); it1 != stringselections_.end(); ++it1) {
+//     TableExprNode theset(mathutil::toVectorString( (*it1).second ));
+//     if ( query.isNull() ) {
+//       query = tab.col((*it1).first).in(theset);
+//     } else {
+//       query = tab.col((*it1).first).in(theset) && query;
+//     }
+//   }
+//   // add taql query
+//   if ( taql_.size() > 0 ) {
+//     Table tmpt = tab;
+//     std::string pytaql = "USING STYLE PYTHON " + taql_;
+
+//     if ( !query.isNull() ) { // taql and selection
+//       tmpt = tableCommand(pytaql, tab(query));
+//     } else { // taql only
+//       tmpt = tableCommand(pytaql, tab);
+//     }
+//     return sort(tmpt);
+//   } else {
+//     if ( query.isNull() ) {
+//       return sort(tab);
+//     } else {
+//       return sort(tab(query));
+//     }
+//   }
+// }
 Table STSelector::apply( const Table& tab )
 {
   if ( empty() ) {
     return sort(tab);
+  }
+  Table basetab = tab;
+  // Important!! Be sure to apply row selection first. 
+  if (rowselection_.size() > 0){
+    //Vector<Int> intrownrs(rowselection_);
+    Vector<uInt> rownrs( rowselection_.size() );
+    convertArray(rownrs, Vector<Int> ( rowselection_ ));
+    basetab = tab( rownrs );
+    ///TableExprNode theset(Vector<Int>( rowselection_ ));
+    ///query = tab.nodeRownr().in(theset);
   }
   TableExprNode query;
@@ -126 +189 @@
     TableExprNode theset(Vector<Int>( (*it).second ));
     if ( query.isNull() ) {
-      query = tab.col((*it).first).in(theset);
+      //query = tab.col((*it).first).in(theset);
+      query = basetab.col((*it).first).in(theset);
     } else {
-      query = tab.col((*it).first).in(theset) && query;
+      //query = tab.col((*it).first).in(theset) && query;
+      query = basetab.col((*it).first).in(theset) && query;
     }
   }
@@ -135 +200 @@
     TableExprNode theset(mathutil::toVectorString( (*it1).second ));
     if ( query.isNull() ) {
-      query = tab.col((*it1).first).in(theset);
+      //query = tab.col((*it1).first).in(theset);
+      query = basetab.col((*it1).first).in(theset);
     } else {
-      query = tab.col((*it1).first).in(theset) && query;
+      //query = tab.col((*it1).first).in(theset) && query;
+      query = basetab.col((*it1).first).in(theset) && query;
     }
   }
   // add taql query
   if ( taql_.size() > 0 ) {
-    Table tmpt = tab;
+    //Table tmpt = tab;
+    Table tmpt = basetab;
     std::string pytaql = "USING STYLE PYTHON " + taql_;
 
     if ( !query.isNull() ) { // taql and selection
-      tmpt = tableCommand(pytaql, tab(query));
+      //tmpt = tableCommand(pytaql, tab(query));
+      tmpt = tableCommand(pytaql, basetab(query));
     } else { // taql only
-      tmpt = tableCommand(pytaql, tab);
+      //tmpt = tableCommand(pytaql, tab);
+      tmpt = tableCommand(pytaql, basetab);
     }
     return sort(tmpt);
   } else {
     if ( query.isNull() ) {
-      return sort(tab);
+      //return sort(tab);
+      return sort(basetab);
     } else {
-      return sort(tab(query));
+      //return sort(tab(query));
+      return sort(basetab(query));
     }
   }
@@ -191 +263 @@
 {
   return getint("CYCLENO");
+}
+
+std::vector< int > asap::STSelector::getTypes( ) const
+{
+  return getint("SRCTYPE") ;
 }
 
@@ -227 +304 @@
 bool asap::STSelector::empty( ) const
 {
-  return (intselections_.empty() && taql_.size() == 0 );
+  //return (intselections_.empty() && taql_.size() == 0 );
+  return (intselections_.empty() && taql_.size() == 0 && rowselection_.size() == 0);
 }