public static WaterType ToWaterType(this TerrainDef def) { if (def.IsSea()) { return(WaterType.SeaWater); } else if (def.IsMarsh() || def.IsMud()) { return(WaterType.MudWater); } else if (def.IsRiver() || def.IsLakeOrPond()) { return(WaterType.RawWater); } return(WaterType.NoWater); }