make area default to eltype(cellarea) and to NaN fill

mauro3 · May 23, 2024 · 54a4dba · 54a4dba
1 parent af32487
commit 54a4dba
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/WhereTheWaterFlows.jl b/src/WhereTheWaterFlows.jl
@@ -346,8 +346,8 @@ end
 # initialize output cellarea:
 # - as tuple or one array
 # - as tuple, if cellarea is a tuple
-init_area(dir, cellarea) = (fill!(similar(dir,Float64), 0), )
-init_area(dir, cellarea::Tuple) = map(x -> fill!(similar(dir,Float64), 0), cellarea)
+init_area(dir, cellarea) = (fill!(similar(dir, eltype(cellarea)), NaN), )
+init_area(dir, cellarea::Tuple) = map(x -> init_area(dir,x)[1], cellarea)
 
 # modifies c and area
 function _flowrouting_catchments!(area, len, c, dir, cellarea, feedback_fn, color, ij)