From 288ba7d317ec7b10e875da4dcfd1b4d732f2763b Mon Sep 17 00:00:00 2001
From: Thomas Buckley-Houston <tom@tombh.co.uk>
Date: Thu, 4 Jun 2026 17:22:43 +0200
Subject: [PATCH] fix: use --tvs-source-path for neighbourhood viewsheds

Previously it was just using the DEM itself as the source.
---
 crates/total-viewsheds/src/config.rs      | 2 +-
 crates/total-viewsheds/src/pre_process.rs | 6 +++++-
 2 files changed, 6 insertions(+), 2 deletions(-)

diff --git a/crates/total-viewsheds/src/config.rs b/crates/total-viewsheds/src/config.rs
index f567ff1..a249c07 100644
--- a/crates/total-viewsheds/src/config.rs
+++ b/crates/total-viewsheds/src/config.rs
@@ -82,7 +82,7 @@ pub struct Compute {
     )]
     pub process: Vec<Process>,
 
-    /// The input DEM file. Currently only `.hgt` files are supported.
+    /// The input DEM file. Can be of any file type supported by GDAL.
     #[arg(value_name = "Path to the DEM file")]
     pub input: std::path::PathBuf,
 
diff --git a/crates/total-viewsheds/src/pre_process.rs b/crates/total-viewsheds/src/pre_process.rs
index 4a01739..8ecb543 100644
--- a/crates/total-viewsheds/src/pre_process.rs
+++ b/crates/total-viewsheds/src/pre_process.rs
@@ -9,7 +9,11 @@ pub fn create_biggest_tvs_subgrid(config: &crate::config::Compute) -> Result<Has
     let neighbourhood_size = config
         .only_save_biggest_viewsheds
         .context("Must specify --only-save-biggest-viewsheds")?;
-    let dataset = gdal::Dataset::open(&config.input)?;
+    let tvs_source_path = config
+        .tvs_source_path
+        .clone()
+        .context("Must specify --tvs-source-path")?;
+    let dataset = gdal::Dataset::open(&tvs_source_path)?;
     let hashmap = find_biggest_total_surfaces(&dataset, i64::from(neighbourhood_size))?;
 
     Ok(hashmap)