From ebf6a358d8a8f01047684a530eadc90aed74878b Mon Sep 17 00:00:00 2001
From: mmeijerdfki <marc.meijer@dfki.de>
Date: Tue, 10 Sep 2024 14:39:51 +0200
Subject: [PATCH] add polygonSensorPosition to the msg abstractions

---
 tests/python/gRPC/util/msg_abs/msgs.py | 39 ++++++++++++++++----------
 1 file changed, 24 insertions(+), 15 deletions(-)

diff --git a/tests/python/gRPC/util/msg_abs/msgs.py b/tests/python/gRPC/util/msg_abs/msgs.py
index 88d93892e..1345a7559 100644
--- a/tests/python/gRPC/util/msg_abs/msgs.py
+++ b/tests/python/gRPC/util/msg_abs/msgs.py
@@ -26,6 +26,7 @@
 
 class EnumFbQuery(FrozenEnum):
     POLYGON = auto()  # def: None
+    POLYGONSENSORPOSITION = auto()  # def: None
     FULLY_ENCAPSULATED = auto()  # def: False
     IN_MAP_FRAME = auto()  # def: True
     TIMEINTERVAL = auto()  # def: None
@@ -52,6 +53,11 @@ def _set_enum_func_mapping(self) -> Dict[EnumFbQuery, MsgsFunctions]:
             EnumFbQuery.POLYGON: MsgsFunctions(
                 lambda: None, lambda: Dtypes.Fb.polygon2d(self.builder)
             ),
+            EnumFbQuery.POLYGONSENSORPOSITION: MsgsFunctions(
+                # TODO provide a example polygon, differing from POLYGON
+                lambda: None,
+                lambda: Dtypes.Fb.polygon2d(self.builder),
+            ),
             EnumFbQuery.FULLY_ENCAPSULATED: MsgsFunctions(
                 lambda: False, lambda: True
             ),
@@ -109,6 +115,9 @@ def datauuid(self) -> List[int]:
 
     def _assemble_datatype_instance(self):
         polygon = self.get_component(EnumFbQuery.POLYGON)
+        polygon_sensor_position = self.get_component(
+            EnumFbQuery.POLYGONSENSORPOSITION
+        )
         fully_encapsulated = self.get_component(EnumFbQuery.FULLY_ENCAPSULATED)
         in_map_frame = self.get_component(EnumFbQuery.IN_MAP_FRAME)
         timeinterval = self.get_component(EnumFbQuery.TIMEINTERVAL)
@@ -124,17 +133,18 @@ def _assemble_datatype_instance(self):
 
         return fbh.createQuery(
             self.builder,
-            timeinterval,
-            label,
-            must_have_all_labels,
-            projectuuid,
-            instanceuuid,
-            datauuid,
-            withoutdata,
-            polygon,
-            fully_encapsulated,
-            in_map_frame,
-            sort_by_time,
+            timeInterval=timeinterval,
+            labels=label,
+            mustHaveAllLabels=must_have_all_labels,
+            projectUuids=projectuuid,
+            instanceUuids=instanceuuid,
+            dataUuids=datauuid,
+            withoutData=withoutdata,
+            polygon2d=polygon,
+            polygon2dSensorPos=polygon_sensor_position,
+            fullyEncapsulated=fully_encapsulated,
+            inMapFrame=in_map_frame,
+            sortByTime=sort_by_time,
         )
 
 
@@ -163,10 +173,9 @@ def _assemble_datatype_instance(self):
         datatype = self.get_component(EnumFbQueryInstance.DATATYPE)
         query = self.get_component(EnumFbQueryInstance.QUERY)
 
-        QueryInstance.Start(self.builder)
-        QueryInstance.AddDatatype(self.builder, datatype)
-        QueryInstance.AddQuery(self.builder, query)
-        return QueryInstance.End(self.builder)
+        return fbh.createQueryInstance(
+            self.builder, datatype=datatype, query=query
+        )
 
 
 class DatatypeImplementations: