feat: ee layer masking

dhis2 · Nov 25, 2023 · 7e29a4e · 7e29a4e
1 parent b47e5fe
commit 7e29a4e
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 7 deletions.
diff --git a/src/earthengine/ee_worker.js b/src/earthengine/ee_worker.js
@@ -123,8 +123,9 @@ class EarthEngineWorker {
             mosaic,
             band,
             bandReducer,
-            mask,
+            maskOperator,
             methods,
+            style,
             cloudScore,
         } = this.options
 
@@ -174,9 +175,9 @@ class EarthEngineWorker {
             }
         }
 
-        // Mask out 0-values
-        if (mask) {
-            eeImage = eeImage.updateMask(eeImage.gt(0))
+        // Only keep pixels above min value
+        if (maskOperator && eeImage[maskOperator]) {
+            eeImage = eeImage.updateMask(eeImage[maskOperator](style?.min || 0))
         }
 
         // Run methods on image
@@ -191,7 +192,7 @@ class EarthEngineWorker {
     getTileUrl() {
         const { datasetId, format, data, filter, style } = this.options
 
-        return new Promise(resolve => {
+        return new Promise((resolve, reject) => {
             switch (format) {
                 case FEATURE_COLLECTION:
                     let dataset = ee.FeatureCollection(datasetId)
@@ -231,7 +232,7 @@ class EarthEngineWorker {
 
                     break
                 default:
-                // TODO: Handle unknown format
+                    reject(new Error('Unknown format'))
             }
         })
     }

diff --git a/src/utils/earthengine.js b/src/utils/earthengine.js
@@ -16,7 +16,7 @@ const workerOptions = [
     'filter',
     'format',
     'legend',
-    'mask',
+    'maskOperator',
     'methods',
     'mosaic',
     'periodReducer',