Skip to content

Commit af53201

Browse files
committed
add a few more fields to simplify plotting the largest catchment
1 parent 004c293 commit af53201

File tree

1 file changed

+8
-2
lines changed

1 file changed

+8
-2
lines changed

src/WaterFlow.jl

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -91,7 +91,13 @@ function waterflows(Topo::GeoData, flowdir_fn= WhereTheWaterFlows.d8dir_feature;
9191
feedback_fn=feedback_fn, drain_pits=drain_pits, bnd_as_sink=bnd_as_sink)
9292

9393
catchment_large = prune_catchments(c, minsize; val=0)
94-
95-
Topo_water = addfield(Topo,(;area, slen, dir, nout, nin, c, cellarea_m2, catchment_large))
94+
largest_catchment = catchment_large .== maximum(catchment_large)
95+
largest_area = copy(area)
96+
largest_area[.!largest_catchment] .= NaN
97+
98+
log10_area = log10.(area)
99+
log10_largest_area = log10.(largest_area)
100+
101+
Topo_water = addfield(Topo,(;area, slen, dir, nout, nin, c, cellarea_m2, catchment_large, log10_area, largest_catchment, largest_area, log10_largest_area))
96102
return Topo_water, sinks, pits, bnds
97103
end

0 commit comments

Comments
 (0)