Modified paraview output

src/fenicsxconcrete/finite_element_problem/base_material.py

@@ -3,6 +3,7 @@
 from abc import ABC, abstractmethod
 from pathlib import Path, PosixPath
 
+import dolfinx as df
 import jsonschema
 import pint
 
@@ -65,6 +66,10 @@ def __init__(
 
  self.residual = None # initialize residual
 
+ # set up xdmf file with mesh info
+ with df.io.XDMFFile(self.mesh.comm, self.pv_output_file, "w") as f:
+ f.write_mesh(self.mesh)
+
  # setup the material object to access the function
  self.setup()
 

src/fenicsxconcrete/finite_element_problem/linear_elasticity_grf.py

@@ -23,8 +23,7 @@ def __init__(
  ) -> None:
  super().__init__(experiment, parameters, pv_name, pv_path)
  # TODO There should be more elegant ways of doing this
- self.pv_egrf_file = Path(pv_path) / (pv_name + "egrf.xdmf")
- self.pv_nugrf_file = Path(pv_path) / (pv_name + "nugrf.xdmf")
+ self.pv_file = Path(pv_path) / (pv_name + ".xdmf")
 
  def setup(self) -> None:
  self.field_function_space = df.fem.FunctionSpace(self.experiment.mesh, ("CG", 1))
@@ -125,19 +124,23 @@ def get_lames_constants(
  return lame1, lame2
 
  # TODO move this to sensor definition!?!?!
- def pv_plot(self, t: int = 0) -> None:
- # TODO add possibility for multiple time steps???
- # Displacement Plot
-
- # "Displacement.xdmf"
- # pv_output_file
- # TODO Look into how to unify in one file
- with df.io.XDMFFile(self.mesh.comm, self.pv_output_file, "w") as xdmf:
- xdmf.write_mesh(self.mesh)
- xdmf.write_function(self.displacement)
- with df.io.XDMFFile(self.mesh.comm, self.pv_egrf_file, "w") as xdmf:
- xdmf.write_mesh(self.mesh)
- xdmf.write_function(self.E_randomfield.field)
- with df.io.XDMFFile(self.mesh.comm, self.pv_nugrf_file, "w") as xdmf:
- xdmf.write_mesh(self.mesh)
- xdmf.write_function(self.nu_randomfield.field)
+ def pv_plot(self, t: pint.Quantity | float = 1) -> None:
+ """creates paraview output at given time step
+
+ Args:
+ t: time point of output (default = 1)
+ """
+ print("create pv plot for t", t)
+ try:
+ _t = t.magnitude
+ except:
+ _t = t
+
+ self.displacement.name = "Displacement"
+ self.E_randomfield.field.name = "E_field"
+ self.nu_randomfield.field.name = "nu_field"
+
+ with df.io.XDMFFile(self.mesh.comm, self.pv_output_file, "a") as f:
+ f.write_function(self.displacement, _t)
+ f.write_function(self.E_randomfield.field, _t)
+ f.write_function(self.nu_randomfield.field, _t)