Check if either kml or kmz files exist.

R-package/R/plot_results.r

@@ -12,6 +12,23 @@
 #'
 #' @export
 plot_results <- function(inPath, color = NULL) {
+    num_files <- length(list.files(
+        inPath,
+        pattern = ".kmz$",
+        all.files = FALSE,
+        include.dirs = TRUE,
+        no.. = TRUE
+    ))
+    
+    if (num_files == 0) {
+        num_files <- length(list.files(
+            inPath,
+            pattern = ".kml$",
+            all.files = FALSE,
+            include.dirs = TRUE,
+            no.. = TRUE
+        ))  
+    }
 
     # Compares the number of .png to that of step_*.tif. If step_*.tif files are missing the
     # user's attention is drawn.
@@ -21,14 +38,8 @@ plot_results <- function(inPath, color = NULL) {
         all.files = FALSE,
         include.dirs = TRUE,
         no.. = TRUE
-    )) != length(list.files(
-        inPath,
-        pattern = ".png$",
-        all.files = FALSE,
-        include.dirs = TRUE,
-        no.. = TRUE
-    ))) {
-        message("Make sure the number of the step_*.tif files fits to the number of the .png files and that
+    )) != num_files) {
+        message("Make sure the number of the step_*.tif files fits to the number of the step_*.km(l|z) files and that
                 there are no other files from previous runs.
                 If you have resumed a run, then the data of the aborted and the continued run needs 
                 to be in the Results directory. You need them for plotting the classification map.")