From 98387ddcacb8732375a5d2ec6a8f8d4598b6f5cd Mon Sep 17 00:00:00 2001
From: Mathieu Beligon <mathieu@feedly.com>
Date: Mon, 22 Mar 2021 12:58:51 -0400
Subject: [PATCH] [filter] (size) Add SizeObjectsFilter

---
 .../target_pipeline/objects_filters/size_filter.py   | 12 ++++++++++++
 1 file changed, 12 insertions(+)
 create mode 100644 src/polystar/target_pipeline/objects_filters/size_filter.py

diff --git a/src/polystar/target_pipeline/objects_filters/size_filter.py b/src/polystar/target_pipeline/objects_filters/size_filter.py
new file mode 100644
index 0000000..184a358
--- /dev/null
+++ b/src/polystar/target_pipeline/objects_filters/size_filter.py
@@ -0,0 +1,12 @@
+from dataclasses import dataclass
+
+from polystar.models.roco_object import ROCOObject
+from polystar.target_pipeline.objects_filters.objects_filter_abc import ObjectsFilterABC
+
+
+@dataclass
+class SmallObjectFilter(ObjectsFilterABC):
+    min_size: int
+
+    def validate_single(self, obj: ROCOObject) -> bool:
+        return obj.box.area >= self.min_size
-- 
GitLab