Add spatial crop utility

Author: umgnunes

CMakeLists.txt

@@ -51,6 +51,14 @@ FetchContent_Declare(
 )
 FetchContent_MakeAvailable(sepia)
 
+# tarsier
+FetchContent_Declare(
+  tarsier
+  GIT_REPOSITORY https://github.com/neuromorphic-paris/tarsier
+  GIT_TAG        master
+)
+FetchContent_MakeAvailable(tarsier)
+
 # Source
 add_subdirectory(${${LIB_NAME}_SOURCE_DIR})
 

README.md

@@ -38,6 +38,8 @@ The following are dependencies that are automatically downloaded during the buil
 
 - sepia: <https://github.com/neuromorphic-paris/sepia>
 
+- tarsier: <https://github.com/neuromorphic-paris/tarsier>
+
 - googletest (only downloaded if tests are enabled): <https://github.com/google/googletest.git>
 
 ### General

src/batch_size.cpp

@@ -2,6 +2,7 @@
 
 #include "event_batch.hpp"
 #include "pontella.hpp"
+#include "select_rectangle.hpp"
 #include "sepia.hpp"
 
 int
@@ -16,6 +17,10 @@ main(int argc, char* argv[])
   {
     uint64_t t_decay_first;
     float weight_thresh;
+    uint16_t left;
+    uint16_t right;
+    uint16_t bottom;
+    uint16_t top;
   };
 
   return pontella::main(
@@ -26,16 +31,40 @@ main(int argc, char* argv[])
        "                                        defaults to 10000",
        "    -e e, --weight-threshold e      sets the weight threshold",
        "                                        defaults to 0.1",
+       "    -cl cl, --crop-left cl          sets the crop's left side "
+       "coordinate",
+       "                                        defaults to 0",
+       "    -cr cr, --crop-right cr         sets the crop's right side "
+       "coordinate",
+       "                                        defaults to context width",
+       "    -cb cb, --crop-bottom cb        sets the crop's bottom side "
+       "coordinate",
+       "                                        defaults to 0",
+       "    -ct ct, --crop-top ct           sets the crop's top side "
+       "coordinate",
+       "                                        defaults to context height",
        "    -h, --help                      shows this help message"},
-      argc, argv, 1, {{"time-decay-first", {"t"}}, {"weight-threshold", {"e"}}},
+      argc, argv, 1,
+      {{"time-decay-first", {"t"}},
+       {"weight-threshold", {"e"}},
+       {"crop-left", {"cl"}},
+       {"crop-right", {"cr"}},
+       {"crop-bottom", {"cb"}},
+       {"crop-top", {"ct"}}},
       {}, [&](pontella::command command) {
         const std::string& filename = command.arguments[0];
+        const auto header =
+            sepia::read_header(sepia::filename_to_ifstream(filename));
 
         Arguments arguments;
         arguments.t_decay_first =
             extract_argument(command, "time-decay-first", 10000);
         arguments.weight_thresh =
             extract_argument(command, "weight-threshold", 0.1);
+        arguments.left = extract_argument(command, "crop-left", 0);
+        arguments.right = extract_argument(command, "crop-right", header.width);
+        arguments.bottom = extract_argument(command, "crop-bottom", 0);
+        arguments.top = extract_argument(command, "crop-top", header.height);
 
         Decay event_decay;
         auto handle_global_decay = [&](Decay decay) { event_decay = decay; };
@@ -55,11 +84,15 @@ main(int argc, char* argv[])
         auto batch = make_batch<Event>(arguments.weight_thresh, event_decay,
                                        handle_batch);
 
+        auto crop = tarsier::make_select_rectangle<Event>(
+            arguments.left, arguments.bottom, arguments.right - arguments.left,
+            arguments.top - arguments.bottom, [&](Event event) {
+              global_decay(event);
+              batch(event);
+            });
+
         sepia::join_observable<Type>(sepia::filename_to_ifstream(filename),
-                                     [&](Event event) {
-                                       global_decay(event);
-                                       batch(event);
-                                     });
+                                     crop);
 
         if (batch.batch().size() > 0)
         {

src/batch_timestamp.cpp

@@ -2,6 +2,7 @@
 
 #include "event_batch.hpp"
 #include "pontella.hpp"
+#include "select_rectangle.hpp"
 #include "sepia.hpp"
 
 int
@@ -16,6 +17,10 @@ main(int argc, char* argv[])
   {
     uint64_t t_decay_first;
     float weight_thresh;
+    uint16_t left;
+    uint16_t right;
+    uint16_t bottom;
+    uint16_t top;
   };
 
   return pontella::main(
@@ -27,16 +32,40 @@ main(int argc, char* argv[])
        "                                        defaults to 10000",
        "    -e e, --weight-threshold e      sets the weight threshold",
        "                                        defaults to 0.1",
+       "    -cl cl, --crop-left cl          sets the crop's left side "
+       "coordinate",
+       "                                        defaults to 0",
+       "    -cr cr, --crop-right cr         sets the crop's right side "
+       "coordinate",
+       "                                        defaults to context width",
+       "    -cb cb, --crop-bottom cb        sets the crop's bottom side "
+       "coordinate",
+       "                                        defaults to 0",
+       "    -ct ct, --crop-top ct           sets the crop's top side "
+       "coordinate",
+       "                                        defaults to context height",
        "    -h, --help                      shows this help message"},
-      argc, argv, 1, {{"time-decay-first", {"t"}}, {"weight-threshold", {"e"}}},
+      argc, argv, 1,
+      {{"time-decay-first", {"t"}},
+       {"weight-threshold", {"e"}},
+       {"crop-left", {"cl"}},
+       {"crop-right", {"cr"}},
+       {"crop-bottom", {"cb"}},
+       {"crop-top", {"ct"}}},
       {}, [&](pontella::command command) {
         const std::string& filename = command.arguments[0];
+        const auto header =
+            sepia::read_header(sepia::filename_to_ifstream(filename));
 
         Arguments arguments;
         arguments.t_decay_first =
             extract_argument(command, "time-decay-first", 10000);
         arguments.weight_thresh =
             extract_argument(command, "weight-threshold", 0.1);
+        arguments.left = extract_argument(command, "crop-left", 0);
+        arguments.right = extract_argument(command, "crop-right", header.width);
+        arguments.bottom = extract_argument(command, "crop-bottom", 0);
+        arguments.top = extract_argument(command, "crop-top", header.height);
 
         Decay event_decay;
         auto handle_global_decay = [&](Decay decay) { event_decay = decay; };
@@ -56,11 +85,15 @@ main(int argc, char* argv[])
         auto batch = make_batch<Event>(arguments.weight_thresh, event_decay,
                                        handle_batch);
 
+        auto crop = tarsier::make_select_rectangle<Event>(
+            arguments.left, arguments.bottom, arguments.right - arguments.left,
+            arguments.top - arguments.bottom, [&](Event event) {
+              global_decay(event);
+              batch(event);
+            });
+
         sepia::join_observable<Type>(sepia::filename_to_ifstream(filename),
-                                     [&](Event event) {
-                                       global_decay(event);
-                                       batch(event);
-                                     });
+                                     crop);
 
         if (batch.batch().size() > 0)
         {

src/event_batch/CMakeLists.txt

@@ -2,5 +2,5 @@
 add_library(${LIB_NAME} INTERFACE)
 
 target_include_directories(${LIB_NAME} INTERFACE ${${LIB_NAME}_INCLUDE_DIR}
-  ${pontella_SOURCE_DIR}/source ${sepia_SOURCE_DIR}/source)
+  ${pontella_SOURCE_DIR}/source ${sepia_SOURCE_DIR}/source ${tarsier_SOURCE_DIR}/source)
 target_link_libraries(${LIB_NAME} INTERFACE pthread)