From 2067af5b57c92d81abd7a08ebf2c3b11999713ff Mon Sep 17 00:00:00 2001
From: Delphine <46021476+delphinekw@users.noreply.github.com>
Date: Wed, 28 Jan 2026 13:39:47 +0100
Subject: [PATCH] Update rasterize_lidar_cover_class.R

See Issue 21
---
 R/rasterize_lidar_cover_class.R | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/R/rasterize_lidar_cover_class.R b/R/rasterize_lidar_cover_class.R
index e361816..9684faf 100644
--- a/R/rasterize_lidar_cover_class.R
+++ b/R/rasterize_lidar_cover_class.R
@@ -24,7 +24,8 @@
 rasterize_lidar_cover_class <- function(las = NULL,
                                         res = 1,
                                         fill.holes = TRUE,
-                                        filename = NULL) {
+                                        filename = NULL,
+                                        correct.low.veg = TRUE) {
 
   pts <- lidR::readLAS(las, select = "xyc", filter = "-keep_class 2 3 4 5 6 9 17")
   #table(pts$Classification)
@@ -32,7 +33,9 @@ rasterize_lidar_cover_class <- function(las = NULL,
   # reclassify bridges as ground
   pts$Classification[pts$Classification == 17] <- as.integer(2)
   # reclassify low vegetation (<1m) as ground
-  pts$Classification[pts$Classification == 3] <- as.integer(2)
+  if(correct.low.veg){
+    pts$Classification[pts$Classification == 3] <- as.integer(2)
+  }
   # join classes 4 & 5 (vegetation > 1m high)
   pts$Classification[pts$Classification == 5] <- as.integer(4)