feat(itkWasmExports): provide getMemoryInputArrayStore

Enable advanced usage of the input arrays in C++.
InsightSoftwareConsortium · Oct 27, 2023 · 19713e3 · 19713e3
1 parent 57a831b
commit 19713e3
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 4 deletions.
diff --git a/include/itkWasmExports.h b/include/itkWasmExports.h
@@ -36,10 +36,17 @@ namespace itk
 namespace wasm
 {
 
+// dataset index, array index
+using InputArrayStoreKeyType = std::pair<uint32_t, uint32_t>;
+using InputArrayStoreValueType = std::vector<uint8_t>;
+using InputArrayStoreType = std::map<InputArrayStoreKeyType, InputArrayStoreValueType>;
+
 // Function for the Pipeline Input's and Output's to set / get from the memory store
 
 WebAssemblyInterface_EXPORT const std::string & getMemoryStoreInputJSON(uint32_t memoryIndex, uint32_t index);
 
+WebAssemblyInterface_EXPORT const InputArrayStoreType & getMemoryInputArrayStore();
+
 WebAssemblyInterface_EXPORT void setMemoryStoreOutputDataObject(uint32_t memoryIndex, uint32_t index, const WasmDataObject * dataObject);
 
 WebAssemblyInterface_EXPORT void setMemoryStoreOutputArray(uint32_t memoryIndex, uint32_t index, uint32_t subIndex, size_t address, size_t size);

diff --git a/src/itkWasmExports.cxx b/src/itkWasmExports.cxx
@@ -29,9 +29,6 @@ namespace itk
 namespace wasm
 {
 
-// dataset index, array index
-using InputArrayStoreKeyType = std::pair<uint32_t, uint32_t>;
-using InputArrayStoreType = std::map<InputArrayStoreKeyType, std::vector<char>>;
 static InputArrayStoreType inputArrayStore;
 
 // index
@@ -43,6 +40,11 @@ const std::string & getMemoryStoreInputJSON(uint32_t memoryIndex, uint32_t index
   return inputJSONStore[index];
 }
 
+const InputArrayStoreType & getMemoryInputArrayStore()
+{
+  return inputArrayStore;
+}
+
 using OutputWasmDataObjectStoreType = std::map<uint32_t, WasmDataObject::ConstPointer>;
 static OutputWasmDataObjectStoreType outputWasmDataObjectStore;
 
@@ -75,7 +77,7 @@ size_t itk_wasm_input_array_alloc(uint32_t memoryIndex, uint32_t index, uint32_t
   const auto key = std::make_pair(index, subIndex);
   if (inputArrayStore.count(key))
   {
-    inputArrayStore[key] = std::vector<char>(size);
+    inputArrayStore[key] = InputArrayStoreValueType(size);
   }
   else
   {