Removed .pvd wrapper creation and reading

这个提交包含在:
Craig Warren
2023-06-14 12:08:50 +01:00
父节点 8c733da526
当前提交 4d85e21a5a
共有 2 个文件被更改,包括 74 次插入126 次删除

查看文件

@@ -39,27 +39,6 @@ from .utilities.utilities import (get_terminal_width,
logger = logging.getLogger(__name__)
def write_vtk_pvd(gvs):
"""Writes Paraview data file (.pvd) - provides pointers to a collection of
data files, i.e. GeometryViews.
Args:
gvs: list of all GeometryViews.
"""
filename = config.get_model_config().output_file_path
pvd = VtkGroup(str(filename))
# Add filenames of all GeometryViews to group
for gv in gvs:
sim_time = 0
pvd.addFile(str(gv.filename) + gv.vtkfiletype.ext, sim_time,
group = "", part = "0")
pvd.save()
logger.info(f'Written wrapper for geometry files: {filename.name}.pvd')
def save_geometry_views(gvs):
"""Creates and saves geometryviews.
@@ -80,10 +59,6 @@ def save_geometry_views(gvs):
pbar.update(gv.nbytes)
pbar.close()
# Write a Paraview data file (.pvd) if there is more than one GeometryView
if len(gvs) > 1:
write_vtk_pvd(gvs)
logger.info('')

查看文件

@@ -17,11 +17,9 @@
# along with gprMax. If not, see <http://www.gnu.org/licenses/>.
import json
import mmap
import os
from xml.etree import ElementTree as ET
from paraview.simple import (AppendDatasets, Box, ColorBy, GetActiveSource,
from paraview.simple import (AppendDatasets, Box, GetActiveSource,
GetActiveView, GetParaViewVersion, Hide,
OpenDataFile, RenameSource, RenderAllViews,
SetActiveSource, Show, Threshold)
@@ -145,34 +143,12 @@ data = GetActiveSource()
Hide(data)
#####################################
# Get filename or list of filenames #
#####################################
# Single .vti or .vtu file
if len(data.FileName) == 1:
files = data.FileName
dirname = os.path.dirname(files[0])
file = data.FileName
dirname = os.path.dirname(file[0])
# Multiple .vti or .vtu files referenced in a .pvd file
else:
files = []
dirname = os.path.dirname(data.FileName)
tree = ET.parse(data.FileName)
root = tree.getroot()
for elem in root:
for subelem in elem.findall('DataSet'):
tmp = os.path.join(dirname, subelem.get('file'))
files.append(tmp)
#################################################################
# Read and display data from file(s), i.e. materials, sources, #
# receivers, and PMLs #
#################################################################
for file in files:
with open(file, 'rb') as f:
# Read and display data from file, i.e. materials, sources, receivers, and PMLs
with open(file, 'rb') as f:
# Comments () embedded in line 3 of file
f.readline()
f.readline()
@@ -184,25 +160,25 @@ for file in files:
print('\ngprMax version: ' + c['gprMax_version'])
print(file)
################
# Display data #
################
pv_view = GetActiveView()
pv_view.AxesGrid.Visibility = 1 # Show Data Axes Grid
pv_data = OpenDataFile(file)
pv_disp = Show(pv_data, pv_view)
pv_src = GetActiveSource()
Hide(pv_src)
src_name = os.path.split(file)
RenameSource(src_name[1])
################
# Display data #
################
pv_view = GetActiveView()
pv_view.AxesGrid.Visibility = 1 # Show Data Axes Grid
pv_data = OpenDataFile(file)
pv_disp = Show(pv_data, pv_view)
pv_src = GetActiveSource()
Hide(pv_src)
src_name = os.path.split(file)
RenameSource(src_name[1])
# Discretisation
dl = c['dx_dy_dz']
# Number of voxels
nl = c['nx_ny_nz']
# Discretisation
dl = c['dx_dy_dz']
# Number of voxels
nl = c['nx_ny_nz']
# Materials
try:
# Materials
try:
for i, mat in enumerate(c['Materials']):
threshold = threshold_filt(pv_src, i, i, ['CELLS', 'Material'])
RenameSource(mat, threshold)
@@ -212,11 +188,11 @@ for file in files:
thresholddisplay = Show(threshold, pv_view)
thresholddisplay.ColorArrayName = ['CELLS', 'Material']
threshold.UpdatePipeline()
except KeyError:
except KeyError:
print('No materials to load')
# Display any sources
try:
# Display any sources
try:
for item in c['Sources']:
pos = item['position']
name = item['name']
@@ -226,11 +202,11 @@ for file in files:
XLength=dl[0], YLength=dl[1], ZLength=dl[2])
RenameSource(name, src)
Show(src)
except KeyError:
except KeyError:
print('No sources to load')
# Display any receivers
try:
# Display any receivers
try:
for item in c['Receivers']:
pos = item['position']
name = item['name']
@@ -240,14 +216,14 @@ for file in files:
XLength=dl[0], YLength=dl[1], ZLength=dl[2])
RenameSource(name, rx)
Show(rx)
except KeyError:
except KeyError:
print('No receivers to load')
# Display any PMLs
try:
# Display any PMLs
try:
pt = c['PMLthickness']
display_pmls(pt, dl, nl)
except KeyError:
except KeyError:
print('No PMLs to load')
@@ -255,6 +231,3 @@ RenderAllViews()
# Reset view to fit data
pv_view.ResetCamera()
# Show color bar/color legend
# thresholdDisplay.SetScalarBarVisibility(renderview, False)