Trim filtered table to original time range in TabOpLowPassFilter

opensim-org · Nov 16, 2024 · 9edc0bf · 9edc0bf
1 parent 2300a90
commit 9edc0bf
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 0 deletions.
diff --git a/OpenSim/Simulation/TableProcessor.h b/OpenSim/Simulation/TableProcessor.h
@@ -194,8 +194,12 @@ class OSIMSIMULATION_API TabOpLowPassFilter : public TableOperator {
                     "Expected cutoff frequency to be positive, but got {}.",
                     get_cutoff_frequency());
 
+            const auto& times = table.getIndependentColumn();
+            double initialTime = times.front();
+            double finalTime = times.back();
             TableUtilities::filterLowpass(
                     table, get_cutoff_frequency(), true);
+            table.trim(initialTime, finalTime);
         }
     }
 };

diff --git a/OpenSim/Simulation/Test/testTableProcessor.cpp b/OpenSim/Simulation/Test/testTableProcessor.cpp
@@ -76,3 +76,13 @@ TEST_CASE("TableProcessor") {
         }
     }
 }
+
+TEST_CASE("TabOpLowPassFilter retains original time range") {
+    TimeSeriesTable table(std::vector<double>{1, 2, 3, 4, 5, 6},
+            SimTK::Test::randMatrix(6, 2), std::vector<std::string>{"a", "b"});
+    TableProcessor proc = TableProcessor(table) | TabOpLowPassFilter(6);
+    TimeSeriesTable out = proc.process();
+    CHECK(out.getNumRows() == 6);
+    CHECK(out.getIndependentColumn().front() == 1);
+    CHECK(out.getIndependentColumn().back() == 6);
+}