summaryrefslogtreecommitdiff
path: root/src/tree_map.h
diff options
context:
space:
mode:
Diffstat (limited to 'src/tree_map.h')
-rw-r--r--src/tree_map.h2
1 files changed, 2 insertions, 0 deletions
diff --git a/src/tree_map.h b/src/tree_map.h
index e614099fc..df9fd441c 100644
--- a/src/tree_map.h
+++ b/src/tree_map.h
@@ -13,6 +13,7 @@
#define TREE_MAP_H
#include "tile_map.h"
+#include "water_map.h"
/**
* List of tree types along all landscape types.
@@ -133,6 +134,7 @@ static inline void SetTreeGroundDensity(TileIndex t, TreeGround g, uint d)
assert(IsTileType(t, MP_TREES)); // XXX incomplete
SB(_m[t].m2, 4, 2, d);
SB(_m[t].m2, 6, 3, g);
+ SetWaterClass(t, g == TREE_GROUND_SHORE ? WATER_CLASS_SEA : WATER_CLASS_INVALID);
}
/**