Skip to content
Snippets Groups Projects
Verified Commit 1f86ba3a authored by Florian Atteneder's avatar Florian Atteneder
Browse files

fix filename assertions

parent 53c52aa2
No related branches found
No related tags found
No related merge requests found
......@@ -160,11 +160,11 @@ end
visz(dir::AbstractString; kwargs...) = visz([dir]; kwargs...)
function visz(dirs::AbstractVector{<:AbstractString};
filename="output.h5", scatter::Bool=true)
filename="output1d.h5", scatter::Bool=true)
length(dirs) == 0 && return
@assert filename in ("output.h5", "substep_output.h5", "interpolate.h5")
@assert filename in ("output1d.h5", "substep_output.h5", "interpolate.h5")
plt_cmd = scatter ? scatterlines! : lines!
......@@ -304,7 +304,7 @@ function visz(dirs::AbstractVector{<:AbstractString};
Ks = [ m.tree.dims[1] for m in meshes ]
out_xs = [ out.x for out in outputs ]
cell_counts = if filename in ("output.h5","substep_output.h5")
cell_counts = if filename in ("output1d.h5","substep_output.h5")
[ [ m.element.Npts for k = 1:K ] for (m,K) in zip(meshes,Ks) ]
else # == "interpolate.h5"
cellidxs = [ dg1d.find_cell_index.(x, m) for (x,m) in zip(out_xs, meshes) ]
......@@ -412,7 +412,7 @@ function visz(dirs::AbstractVector{<:AbstractString};
xdata = plt_xdata[j]
mxdata = xdata[mask]
nx = length(mxdata)
if filename in ("output.h5","substep_output.h5") && length(var) != nx
if filename in ("output1d.h5","substep_output.h5") && length(var) != nx
# this is a cell variable
@views fill!(buf[mask], 1.0)
offset = 1
......@@ -457,7 +457,7 @@ function visz(dirs::AbstractVector{<:AbstractString};
xdata = plt_xdata[j]
mxdata = xdata[mask]
nx = length(mxdata)
if filename in ("output.h5","substep_output.h5") && length(var) != nx
if filename in ("output1d.h5","substep_output.h5") && length(var) != nx
# this is a cell variable
@views fill!(buf[mask], 1.0)
offset = 1
......@@ -660,7 +660,7 @@ end
function cvg(dirs::AbstractVector{<:AbstractString};
filename="interpolate.h5")
@assert filename in ("output.h5", "interpolate.h5")
@assert filename in ("output1d.h5", "interpolate.h5")
@assert length(dirs) >= 2
ms = [ match(r"(\d*)$", d) for d in dirs ]
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment