feat(Sources/Filters/General/PaintFilter/index.js): Return the promis…

…e generated by workerPromise.ex
Kitware · Sep 18, 2019 · 0c470bc · 0c470bc
1 parent 8d8a543
commit 0c470bc
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/Sources/Filters/General/PaintFilter/index.js b/Sources/Filters/General/PaintFilter/index.js
@@ -50,8 +50,12 @@ function vtkPaintFilter(publicAPI, model) {
   // --------------------------------------------------------------------------
 
   publicAPI.endStroke = () => {
+    let endStrokePromise;
+
     if (workerPromise) {
-      workerPromise.exec('end').then((strokeBuffer) => {
+      endStrokePromise = workerPromise.exec('end');
+
+      endStrokePromise.then((strokeBuffer) => {
         const scalars = model.labelMap.getPointData().getScalars();
         const data = scalars.getData();
 
@@ -113,6 +117,8 @@ function vtkPaintFilter(publicAPI, model) {
         publicAPI.modified();
       });
     }
+
+    return endStrokePromise;
   };
 
   // --------------------------------------------------------------------------