Add In/Out Format in NodeExport.

aidge_core/export_utils/node_export.py

@@ -137,6 +137,7 @@ class ExportNode(ABC):
         self.attributes["in_name"] = [None] * self.attributes["nb_in"]
         self.attributes["in_dims"] = [None] * self.attributes["nb_in"]
         self.attributes["in_dformat"] = [None] * self.attributes["nb_in"]
+        self.attributes["in_format"] = [None] * self.attributes["nb_in"]
         self.attributes["in_dtype"] = [None] * self.attributes["nb_in"]
         self.attributes["in_cdtype"] = [None] * self.attributes["nb_in"]
         self.attributes["in_chan"] = [None] * self.attributes["nb_in"]
@@ -146,6 +147,7 @@ class ExportNode(ABC):
         self.attributes["out_name"] = [None] * self.attributes["nb_out"]
         self.attributes["out_dims"] = [None] * self.attributes["nb_out"]
         self.attributes["out_dformat"] = [None] * self.attributes["nb_out"]
+        self.attributes["out_format"] = [None] * self.attributes["nb_out"]
         self.attributes["out_dtype"] = [None] * self.attributes["nb_out"]
         self.attributes["out_cdtype"] = [None] * self.attributes["nb_out"]
         self.attributes["out_chan"] = [None] * self.attributes["nb_out"]
@@ -174,6 +176,7 @@ class ExportNode(ABC):
                 self.attributes["in_name"][idx] = f"{self.attributes['name']}_input_{idx}" if parent_node is None else f"{parent_node.name()}_output_{out_id}"
                 self.attributes["in_dims"][idx] = tensor.dims()
                 self.attributes["in_dformat"][idx] = tensor.dformat()
+                self.attributes["in_format"][idx] = aidge_core.format_as(tensor.dformat())
                 self.attributes["in_dtype"][idx] = tensor.dtype()
                 self.attributes["in_cdtype"][idx] = data_conversion.aidge2c(
                     tensor.dtype())
@@ -190,6 +193,7 @@ class ExportNode(ABC):
                 self.attributes["out_name"][idx] = f"{self.attributes['name']}_output_{idx}"
                 self.attributes["out_dims"][idx] = tensor.dims()
                 self.attributes["out_dformat"][idx] = tensor.dformat()
+                self.attributes["out_format"][idx] = aidge_core.format_as(tensor.dformat())
                 self.attributes["out_dtype"][idx] = tensor.dtype()
                 self.attributes["out_cdtype"][idx] = data_conversion.aidge2c(
                     tensor.dtype())

python_binding/data/pybind_Data.cpp

@@ -45,5 +45,8 @@ void init_Data(py::module& m){
     py::class_<Data, std::shared_ptr<Data>>(m,"Data");
 
 
+    m.def("format_as", (const char* (*)(DataType)) &format_as, py::arg("dt"));
+    m.def("format_as", (const char* (*)(DataFormat)) &format_as, py::arg("df"));
+
 }
 }