Allow mask value to be passed in

dhis2 · Jul 29, 2024 · c6c5e3f · c6c5e3f
1 parent c61e90f
commit c6c5e3f
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/src/earthengine/ee_worker.js b/src/earthengine/ee_worker.js
@@ -1,7 +1,6 @@
 import { expose } from 'comlink'
 import { getBufferGeometry } from '../utils/buffers.js'
 import ee from './ee_api_js_worker.js' // https://github.com/google/earthengine-api/pull/173
-// import { ee } from '@google/earthengine/build/ee_api_js_debug' // Run "yarn add @google/earthengine"
 import {
     getInfo,
     getScale,
@@ -28,6 +27,8 @@ const DEFAULT_FEATURE_STYLE = {
 }
 const DEFAULT_TILE_SCALE = 1
 
+const DEFAULT_UNMASK_VALUE = 0
+
 class EarthEngineWorker {
     constructor(options = {}) {
         this.options = options
@@ -294,8 +295,12 @@ class EarthEngineWorker {
 
         // Used for "constrained" WorldPop layers
         // We need to unmask the image to get the correct population density
-        if (unmaskAggregation) {
-            image = image.unmask(0)
+        if (unmaskAggregation || typeof unmaskAggregation === 'number') {
+            image = image.unmask(
+                typeof unmaskAggregation === number
+                    ? unmaskAggregation
+                    : DEFAULT_UNMASK_VALUE
+            )
         }
 
         if (collection) {