From d7fce56491312df2e2714c5547767a27b9e1ace9 Mon Sep 17 00:00:00 2001
From: monty <velimir.vesselinov@gmail.com>
Date: Tue, 6 Aug 2024 08:19:00 -0600
Subject: [PATCH] plotmap

---
 src/NMFkPostprocess.jl | 45 +++++++++++++++++++++---------------------
 1 file changed, 22 insertions(+), 23 deletions(-)

diff --git a/src/NMFkPostprocess.jl b/src/NMFkPostprocess.jl
index 7f881c0..a6d58b9 100644
--- a/src/NMFkPostprocess.jl
+++ b/src/NMFkPostprocess.jl
@@ -474,19 +474,19 @@ function postprocess(krange::Union{AbstractRange{Int},AbstractVector{Int64},Inte
 			@assert signalmap == sortperm(clustermap)
 			@assert clustermap[signalmap] == clusterlabels
 			dumpcsv = true
-			if length(lon) == length(chnew)
-				if plotmap
-					if isnothing(hover)
-						hover = Hnames
-					end
-					if plotmap_scope == :well
-						NMFk.plot_wells("$(Hcasefilename)-$(k)-map.$(map_format)", lon, lat, chnew; figuredir=figuredir, hover=hover, title="Signals: $k")
-					else
-						NMFk.plotmap(lon, lat, chnew; filename=joinpath(figuredir, "$(Hcasefilename)-$(k)-map.$(map_format)"), title="Signals: $k", scope=string(plotmap_scope), map_dict...)
-					end
+			if plotmap && length(lon) == length(chnew)
+				if isnothing(hover)
+					hover = Hnames
+				end
+				if plotmap_scope == :well
+					NMFk.plot_wells("$(Hcasefilename)-$(k)-map.$(map_format)", lon, lat, chnew; figuredir=figuredir, hover=hover, title="Signals: $k")
+				else
+					NMFk.plotmap(lon, lat, chnew; filename=joinpath(figuredir, "$(Hcasefilename)-$(k)-map.$(map_format)"), title="Signals: $k", scope=string(plotmap_scope), map_dict...)
 				end
 				DelimitedFiles.writedlm("$resultdir/$(Hcasefilename)-$(k).csv", [["Name" "X" "Y" permutedims(clusterlabels) "Signal"]; Hnames lon lat Hm[:,signalmap] chnew], ',')
-			else
+				dumpcsv = false
+			end
+			if dumpcsv
 				DelimitedFiles.writedlm("$resultdir/$(Hcasefilename)-$(k).csv", [["Name" permutedims(clusterlabels) "Signal"]; Hnames Hm[:,signalmap] chnew], ',')
 			end
 			cs = sortperm(chnew)
@@ -580,25 +580,24 @@ function postprocess(krange::Union{AbstractRange{Int},AbstractVector{Int64},Inte
 			# 	snew2[snew .== "S$(i)"] .= "S$(ws[i])"
 			# end
 			dumpcsv = true
-			if length(lon) == length(cwnew)
-				if plotmap
-					if isnothing(hover)
-						hover = Wnames
-					end
-					if plotmap_scope == :well
-						NMFk.plot_wells("$(Hcasefilename)-$(k)-map.$(map_format)", lon, lat, cwnew; figuredir=figuredir, hover=hover, title="Signals: $k")
-					else
-						NMFk.plotmap(lon, lat, cwnew; filename=joinpath(figuredir, "$(Hcasefilename)-$(k)-map.$(map_format)"), title="Signals: $k", scope=string(plotmap_scope), map_dict...)
-					end
+			if plotmap && length(lon) == length(cwnew)
+				if isnothing(hover)
+					hover = Wnames
+				end
+				if plotmap_scope == :well
+					NMFk.plot_wells("$(Hcasefilename)-$(k)-map.$(map_format)", lon, lat, cwnew; figuredir=figuredir, hover=hover, title="Signals: $k")
+				else
+					NMFk.plotmap(lon, lat, cwnew; filename=joinpath(figuredir, "$(Hcasefilename)-$(k)-map.$(map_format)"), title="Signals: $k", scope=string(plotmap_scope), map_dict...)
 				end
 				DelimitedFiles.writedlm("$resultdir/$(Wcasefilename)-$(k).csv", [["Name" "X" "Y" permutedims(clusterlabels) "Signal"]; Wnames lon lat Wm[:,signalmap] cwnew], ',')
 				dumpcsv = false
-			elseif length(lon) != length(chnew)
-				@warn("Lat/Lon data ($(length(lon))) does not match the number of either W matrix rows ($(length(cwnew))) or H matrix columns ($(length(chnew)))!")
 			end
 			if dumpcsv
 				DelimitedFiles.writedlm("$resultdir/$(Wcasefilename)-$(k).csv", [["Name" permutedims(clusterlabels) "Signal"]; Wnames Wm[:,signalmap] cwnew], ',')
 			end
+			if !isnothing(lon) && (lonlength(lon) != length(chnew)) && (lonlength(lon) != length(cwnew))
+				@warn("Lat/Lon data ($(length(lon))) does not match the number of either W matrix rows ($(length(cwnew))) or H matrix columns ($(length(chnew)))!")
+			end
 			cs = sortperm(cwnew)
 			if createplots
 				xticks = ["S$i" for i=1:k]