From 9dbc075369a4cbe0cd6a43ef0c4bf61c16aa3784 Mon Sep 17 00:00:00 2001
From: jasminium <johnmatthewhartley@gmail.com>
Date: Tue, 15 Jun 2021 17:26:37 +0200
Subject: [PATCH] macros for evtk exported geometry views

---
 tools/Paraview macros/etvk.py       | 10 ++--
 tools/Paraview macros/etvk_lines.py | 92 +++++++++++++++++++++++++++++
 2 files changed, 97 insertions(+), 5 deletions(-)
 create mode 100644 tools/Paraview macros/etvk_lines.py

diff --git a/tools/Paraview macros/etvk.py b/tools/Paraview macros/etvk.py
index d7f9458e..16512a9e 100644
--- a/tools/Paraview macros/etvk.py	
+++ b/tools/Paraview macros/etvk.py	
@@ -172,28 +172,28 @@ nl = c['nx_ny_nz']
 try:
     ml = c['Materials']
     load_materials(ml)
-except IndexError:
+except KeyError:
     print('No Materials to load')
 
 # load sources
 try:
     srcs = c['Sources']
     load_src_rx(srcs, dl)
-except IndexError:
-    print('No sources to load}')
+except KeyError:
+    print('No sources to load')
 
 # load rxs
 try:
     rxs = c['Receivers']
     load_src_rx(rxs, dl)
-except IndexError:
+except KeyError:
     print('No receivers to load')
 
 # load pmls
 try:
     pt = c['PMLthickness']
     display_pmls_new(pt, dl, nl)
-except IndexError:
+except KeyError:
     print('No PMLs to load')
 
 RenderAllViews()
diff --git a/tools/Paraview macros/etvk_lines.py b/tools/Paraview macros/etvk_lines.py
new file mode 100644
index 00000000..5ca3ee5f
--- /dev/null
+++ b/tools/Paraview macros/etvk_lines.py	
@@ -0,0 +1,92 @@
+# Copyright (C) 2015-2021: The University of Edinburgh
+#                 Authors: Craig Warren, Antonis Giannopoulos, and John Hartley
+#
+# This file is part of gprMax.
+#
+# gprMax is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# gprMax is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with gprMax.  If not, see <http://www.gnu.org/licenses/>.
+
+import mmap
+import os
+from xml.etree import ElementTree as ET
+import json
+
+
+
+def load_materials(ml):
+    # create a threshold filter for each material type
+    for i, m in enumerate(ml):
+        threshold = Threshold(Input=model)
+        threshold.ThresholdRange = [i, i]
+        print(threshold.Scalars)
+        threshold.Scalars = ['CELLS', 'Material']
+        RenameSource(m, threshold)
+
+        # Show data in view, except for free_space
+        if i != 1:
+            thresholddisplay = Show(threshold, renderview)
+        thresholddisplay.ColorArrayName = ['CELLS', 'Material']
+        threshold.UpdatePipeline()
+
+from paraview.simple import *
+
+# Get whatever source is loaded (should be model)
+model = GetActiveSource()
+
+# Get active view
+renderview = GetActiveView()
+
+# Show Data Axes Grid
+renderview.AxesGrid.Visibility = 1
+
+# Hide display of root data
+Hide(model)
+
+# one file only for now
+
+fp = model.FileName[0]
+
+# materials string
+ms = None
+
+# load json embedded in comment of line 3
+with open(fp, 'rb') as f:
+    f.readline()
+    f.readline()
+    # comments
+    c = f.readline().decode()
+
+# strip comment tags
+c = c[5:-5]
+# model information
+c = json.loads(c)
+
+# discretisation
+dl = c['dx_dy_dz']
+# n voxels
+nl = c['nx_ny_nz']
+
+# load material filters
+try:
+    ml = c['Materials']
+    load_materials(ml)
+except KeyError:
+    print('No Materials to load')
+
+RenderAllViews()
+
+# Reset view to fit data
+renderview.ResetCamera()
+
+# Show color bar/color legend
+# thresholdDisplay.SetScalarBarVisibility(renderview, False)