Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 1 addition & 0 deletions arithmetic/workflow.ipynb

Large diffs are not rendered by default.

File renamed without changes.
File renamed without changes.
1 change: 0 additions & 1 deletion demo.ipynb

This file was deleted.

15 changes: 11 additions & 4 deletions environment.yml
Original file line numberDiff line numberDiff line change
Expand Up@@ -2,9 +2,16 @@ channels:
- conda-forge
dependencies:
- python =3.12
- myqueue
- rich
- pip
- matplotlib
- myqueue =26.3.0
- rich =15.0.0
- texteditor =1.4.1
- pip =25.2
- ase=3.24.0
- matplotlib=3.10.1
- xmlschema=3.4.3
- optimade=1.2.3
- qe=7.5
- qe-tools=2.0.0
- conda_subprocess =0.0.11
- pip:
- git+https://gitlab.com/asm-dtu/perqueue.git
5 changes: 5 additions & 0 deletions nfdi/source/envs/postprocessing.yaml
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,5 @@
channels:
- conda-forge
dependencies:
- paraview=5.9.1=hfc1cbd4_3_egl
- tectonic=0.8.0=ha1fef3e_1
5 changes: 5 additions & 0 deletions nfdi/source/envs/preprocessing.yaml
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,5 @@
channels:
- conda-forge
dependencies:
- gmsh=4.6.0=hd134328_0
- meshio=5.0.5=pyhd8ed1ab_0
5 changes: 5 additions & 0 deletions nfdi/source/envs/processing.yaml
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,5 @@
channels:
- conda-forge
dependencies:
- fenics=2019.1.0=py39hf3d152e_26
- sympy=1.13.3
3 changes: 3 additions & 0 deletions nfdi/source/macros.tex.template
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,3 @@
\newcommand{\domainSize}{${DOMAINSIZE}}
\newcommand{\numDofs}{${NUMDOFS}}
\newcommand{\plotDataPath}{${PLOTDATAPATH}}
22 changes: 22 additions & 0 deletions nfdi/source/paper.tex
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,22 @@
\documentclass[12pt]{article}
\usepackage{graphicx}
\usepackage{pgfplots}

\input{macros}

\title{An exemplary workflow}

\begin{document}
\maketitle

\begin{figure}[h]
\centering%
\begin{tikzpicture}
\begin{axis}[]
\addplot[color=red,smooth,thick] table[col sep=comma,header=true,x index=2,y index=0] {\plotDataPath};
\end{axis}
\end{tikzpicture}
\caption{Solution of the poisson equation over the line $L = \{(x, y)\in\Omega = {(0, \domainSize{})}^2 | x = y\}$ using \numDofs{} DoFs in the finite element model.}%
\end{figure}

\end{document}
108 changes: 108 additions & 0 deletions nfdi/source/poisson.py
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,108 @@
"""
solution of the poisson equation on the unit square
"""

from argparse import ArgumentParser
import dolfin as df


def boundary_expression():
"""Defines the function to be used for the boundary conditions"""
return "1.0 + x[0] * x[0] + 2.0 * x[1] * x[1]"


def solve_poisson(
meshfile: str, degree: int, bc_expression: str = boundary_expression()
):
"""solves the poisson equation

Parameters
----------
meshfile : str
FilePath to the mesh in xdmf format.
degree : int
Degree of the finite element space.

Returns
-------
solution : df.Function
"""
mesh = df.Mesh()
with df.XDMFFile(meshfile) as instream:
instream.read(mesh)
func_space = df.FunctionSpace(mesh, "CG", degree)
boundary_data = df.Expression(bc_expression, degree=2)

def boundary(_, on_boundary):
return on_boundary

boundary_conditions = df.DirichletBC(func_space, boundary_data, boundary)
trial_function = df.TrialFunction(func_space)
test_function = df.TestFunction(func_space)
source = df.Constant(-6.0)
lhs = df.dot(df.grad(trial_function), df.grad(test_function)) * df.dx
rhs = source * test_function * df.dx

solution = df.Function(func_space)
df.solve(lhs == rhs, solution, boundary_conditions)
return solution


def solve_and_write_output(
mesh: str, degree: int, outputfile: str, numdofs=None, return_dofs=False
):
"""solves the poisson equation and writes the solution
and the number of degrees of freedom to the given file

Parameters
----------
meshfile : str
FilePath to the mesh in xdmf format.
degree : int
Degree of the finite element space.
outputfile : str
FilePath to the output file into which the solution is written.
numdofs : optional, str
FilePath to which the number of degrees of freedom is written.
return_dofs : optional, bool
If True, return number of degrees of freedom.
"""
discrete_solution = solve_poisson(mesh, degree)
discrete_solution.rename("u", discrete_solution.name())
resultFile = df.File(outputfile)
resultFile << discrete_solution

dofs = discrete_solution.function_space().dim()
print(f"Number of dofs used: {dofs}")

if numdofs is not None:
with open(numdofs, "w") as handle:
handle.write("{}\n".format(dofs))
if return_dofs:
return dofs


if __name__ == "__main__":
PARSER = ArgumentParser(description="run script for the poisson problem")
PARSER.add_argument("-m", "--mesh", required=True, help="mesh file to be used")
PARSER.add_argument(
"-d", "--degree", required=True, help="polynomial order to be used"
)
PARSER.add_argument(
"-o",
"--outputfile",
required=True,
help="file name for the output to be written",
)
PARSER.add_argument(
"-n",
"--num-dofs",
required=False,
default=None,
help="file name for the number of DoFs to be written",
)
ARGS = vars(PARSER.parse_args())

solve_and_write_output(
ARGS["mesh"], int(ARGS["degree"]), ARGS["outputfile"], ARGS["num_dofs"]
)
54 changes: 54 additions & 0 deletions nfdi/source/postprocessing.py
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,54 @@
"""
$ pvbatch postprocessing.py -h
"""

import sys
import argparse
from paraview.simple import (
PlotOverLine,
PVDReader,
SaveData,
UpdatePipeline,
)


def main(args):
pvd_file = args.pvd
source = PVDReader(registrationName="poisson.pvd", FileName=[pvd_file])
source.PointArrays = [args.field]
UpdatePipeline()

(xmin, xmax, ymin, ymax, zmin, zmax) = source.GetDataInformation().GetBounds()
# init the 'Line' selected for 'Source'
plotOverLine1 = PlotOverLine(
registrationName="PlotOverLine1", Input=source, Source="Line"
)
plotOverLine1.Source.Point1 = [xmin, ymin, zmin]
plotOverLine1.Source.Point2 = [xmax, ymax, zmax]
UpdatePipeline()

# save data
SaveData(
args.csv,
proxy=plotOverLine1,
ChooseArraysToWrite=1,
PointDataArrays=["arc_length", args.field, "vtkValidPointMask"],
)


if __name__ == "__main__":
parser = argparse.ArgumentParser(
prog=f"pvbatch {__file__}",
description="Plots the solution over a line and writes the data to file.",
usage="%(prog)s [options] pvd csv",
)
parser.add_argument("pvd", type=str, help="The source pvd filepath.")
parser.add_argument("csv", type=str, help="The target csv filepath.")
parser.add_argument(
"--field",
type=str,
default="u",
help="Field variable to plot (default: u)",
)
args = parser.parse_args(sys.argv[1:])
main(args)
43 changes: 43 additions & 0 deletions nfdi/source/prepare_paper_macros.py
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,43 @@
"""
places the correct values into the paper macros template file
"""

import string
import argparse

PARSER = argparse.ArgumentParser(
description="Place computed values in the paper macros template file"
)
PARSER.add_argument(
"-f", "--macro-template-file",
required=False, default="macros.tex.template",
help="The macro template file"
)
PARSER.add_argument(
"-o", "--output-macro-file",
required=False, default="macros.tex",
help="The macro file to be written"
)
PARSER.add_argument(
"-s", "--domain-size",
required=True, help="The computed domain size",
)
PARSER.add_argument(
"-n", "--num-dofs",
required=True, help="The computed number of dofs",
)
PARSER.add_argument(
"-p", "--plot-data-path",
required=True, help="The path to the data for the plot over line"
)
ARGS = vars(PARSER.parse_args())


with open(ARGS["output_macro_file"], "w") as out_file:
with open(ARGS["macro_template_file"], "r") as in_file:
raw = string.Template(in_file.read())
out_file.write(raw.substitute({
"DOMAINSIZE": ARGS["domain_size"],
"NUMDOFS": ARGS["num_dofs"],
"PLOTDATAPATH": ARGS["plot_data_path"]
}))
9 changes: 9 additions & 0 deletions nfdi/source/unit_square.geo
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,9 @@
DefineConstant[ domain_size = 1.0 ];
DefineConstant[ mesh_size = domain_size/4.0 ];

SetFactory("OpenCASCADE");
Rectangle(1) = {0, 0, 0, domain_size, domain_size, 0};
Physical Surface(1) = {1};

Printf("Used domain size: %f", domain_size);
Printf("Used mesh size: %f", mesh_size);
33 changes: 33 additions & 0 deletions nfdi/workflow.json
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,33 @@
{
"version": "0.1.0",
"nodes": [
{"id": 0, "type": "function", "value": "workflow.generate_mesh"},
{"id": 1, "type": "function", "value": "workflow.convert_to_xdmf"},
{"id": 2, "type": "function", "value": "workflow.poisson"},
{"id": 3, "type": "function", "value": "workflow.plot_over_line"},
{"id": 4, "type": "function", "value": "workflow.substitute_macros"},
{"id": 5, "type": "function", "value": "workflow.compile_paper"},
{"id": 6, "type": "input", "value": 2.0, "name": "domain_size"},
{"id": 7, "type": "input", "value": "source", "name": "source_directory"},
{"id": 8, "type": "output", "name": "result"}
],
"edges": [
{"target": 0, "targetPort": "domain_size", "source": 6, "sourcePort": null},
{"target": 0, "targetPort": "source_directory", "source": 7, "sourcePort": null},
{"target": 1, "targetPort": "gmsh_output_file", "source": 0, "sourcePort": null},
{"target": 2, "targetPort": "meshio_output_xdmf", "source": 1, "sourcePort": "xdmf_file"},
{"target": 2, "targetPort": "meshio_output_h5", "source": 1, "sourcePort": "h5_file"},
{"target": 2, "targetPort": "source_directory", "source": 7, "sourcePort": null},
{"target": 3, "targetPort": "poisson_output_pvd_file", "source": 2, "sourcePort": "pvd_file"},
{"target": 3, "targetPort": "poisson_output_vtu_file", "source": 2, "sourcePort": "vtu_file"},
{"target": 3, "targetPort": "source_directory", "source": 7, "sourcePort": null},
{"target": 4, "targetPort": "pvbatch_output_file", "source": 3, "sourcePort": null},
{"target": 4, "targetPort": "ndofs", "source": 2, "sourcePort": "numdofs"},
{"target": 4, "targetPort": "domain_size", "source": 6, "sourcePort": null},
{"target": 4, "targetPort": "source_directory", "source": 7, "sourcePort": null},
{"target": 5, "targetPort": "macros_tex", "source": 4, "sourcePort": null},
{"target": 5, "targetPort": "plot_file", "source": 3, "sourcePort": null},
{"target": 5, "targetPort": "source_directory", "source": 7, "sourcePort": null},
{"target": 8, "targetPort": null, "source": 5, "sourcePort": null}
]
}
Loading
, 'i'); if (__m === '*' || __re.test(location.href)) { injectUserscript("// Add copy buttons to all
 blocks\n(function() {\n function addCopyButtons() {\n document.querySelectorAll('pre code').forEach(function(codeBlock) {\n if (codeBlock.parentElement.hasAttribute('data-copy-added')) return;\n codeBlock.parentElement.setAttribute('data-copy-added', 'true');\n \n var btn = document.createElement('button');\n btn.textContent = 'Copy';\n btn.style.cssText = 'position:absolute;top:4px;right:4px;padding:2px 8px;font-size:11px;background:#4ecdc4;border:none;border-radius:4px;color:#1a1a2e;cursor:pointer;opacity:0.7;transition:opacity 0.2s;';\n btn.onmouseover = function() { this.style.opacity = '1'; };\n btn.onmouseout = function() { this.style.opacity = '0.7'; };\n btn.onclick = function() {\n navigator.clipboard.writeText(codeBlock.textContent).then(function() {\n btn.textContent = 'Copied!';\n setTimeout(function() { btn.textContent = 'Copy'; }, 1500);\n });\n };\n codeBlock.parentElement.style.position = 'relative';\n codeBlock.parentElement.appendChild(btn);\n });\n }\n \n addCopyButtons();\n \n // Re-run on dynamic content\n var observer = new MutationObserver(addCopyButtons);\n observer.observe(document.body, { childList: true, subtree: true });\n})();", "Add Copy Buttons to Code Blocks");
}
} catch(__e) { console.warn('[Userscript:Add Copy Buttons to Code Blocks]', __e); }
})();
(function(){
try {
var __m = "github.com";
var __re = new RegExp('^' + "github\\.com" + '
Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 1 addition & 0 deletions arithmetic/workflow.ipynb

Large diffs are not rendered by default.

File renamed without changes.
File renamed without changes.
1 change: 0 additions & 1 deletion demo.ipynb

This file was deleted.

15 changes: 11 additions & 4 deletions environment.yml
Original file line numberDiff line numberDiff line change
Expand Up@@ -2,9 +2,16 @@ channels:
- conda-forge
dependencies:
- python =3.12
- myqueue
- rich
- pip
- matplotlib
- myqueue =26.3.0
- rich =15.0.0
- texteditor =1.4.1
- pip =25.2
- ase=3.24.0
- matplotlib=3.10.1
- xmlschema=3.4.3
- optimade=1.2.3
- qe=7.5
- qe-tools=2.0.0
- conda_subprocess =0.0.11
- pip:
- git+https://gitlab.com/asm-dtu/perqueue.git
5 changes: 5 additions & 0 deletions nfdi/source/envs/postprocessing.yaml
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,5 @@
channels:
- conda-forge
dependencies:
- paraview=5.9.1=hfc1cbd4_3_egl
- tectonic=0.8.0=ha1fef3e_1
5 changes: 5 additions & 0 deletions nfdi/source/envs/preprocessing.yaml
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,5 @@
channels:
- conda-forge
dependencies:
- gmsh=4.6.0=hd134328_0
- meshio=5.0.5=pyhd8ed1ab_0
5 changes: 5 additions & 0 deletions nfdi/source/envs/processing.yaml
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,5 @@
channels:
- conda-forge
dependencies:
- fenics=2019.1.0=py39hf3d152e_26
- sympy=1.13.3
3 changes: 3 additions & 0 deletions nfdi/source/macros.tex.template
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,3 @@
\newcommand{\domainSize}{${DOMAINSIZE}}
\newcommand{\numDofs}{${NUMDOFS}}
\newcommand{\plotDataPath}{${PLOTDATAPATH}}
22 changes: 22 additions & 0 deletions nfdi/source/paper.tex
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,22 @@
\documentclass[12pt]{article}
\usepackage{graphicx}
\usepackage{pgfplots}

\input{macros}

\title{An exemplary workflow}

\begin{document}
\maketitle

\begin{figure}[h]
\centering%
\begin{tikzpicture}
\begin{axis}[]
\addplot[color=red,smooth,thick] table[col sep=comma,header=true,x index=2,y index=0] {\plotDataPath};
\end{axis}
\end{tikzpicture}
\caption{Solution of the poisson equation over the line $L = \{(x, y)\in\Omega = {(0, \domainSize{})}^2 | x = y\}$ using \numDofs{} DoFs in the finite element model.}%
\end{figure}

\end{document}
108 changes: 108 additions & 0 deletions nfdi/source/poisson.py
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,108 @@
"""
solution of the poisson equation on the unit square
"""

from argparse import ArgumentParser
import dolfin as df


def boundary_expression():
"""Defines the function to be used for the boundary conditions"""
return "1.0 + x[0] * x[0] + 2.0 * x[1] * x[1]"


def solve_poisson(
meshfile: str, degree: int, bc_expression: str = boundary_expression()
):
"""solves the poisson equation

Parameters
----------
meshfile : str
FilePath to the mesh in xdmf format.
degree : int
Degree of the finite element space.

Returns
-------
solution : df.Function
"""
mesh = df.Mesh()
with df.XDMFFile(meshfile) as instream:
instream.read(mesh)
func_space = df.FunctionSpace(mesh, "CG", degree)
boundary_data = df.Expression(bc_expression, degree=2)

def boundary(_, on_boundary):
return on_boundary

boundary_conditions = df.DirichletBC(func_space, boundary_data, boundary)
trial_function = df.TrialFunction(func_space)
test_function = df.TestFunction(func_space)
source = df.Constant(-6.0)
lhs = df.dot(df.grad(trial_function), df.grad(test_function)) * df.dx
rhs = source * test_function * df.dx

solution = df.Function(func_space)
df.solve(lhs == rhs, solution, boundary_conditions)
return solution


def solve_and_write_output(
mesh: str, degree: int, outputfile: str, numdofs=None, return_dofs=False
):
"""solves the poisson equation and writes the solution
and the number of degrees of freedom to the given file

Parameters
----------
meshfile : str
FilePath to the mesh in xdmf format.
degree : int
Degree of the finite element space.
outputfile : str
FilePath to the output file into which the solution is written.
numdofs : optional, str
FilePath to which the number of degrees of freedom is written.
return_dofs : optional, bool
If True, return number of degrees of freedom.
"""
discrete_solution = solve_poisson(mesh, degree)
discrete_solution.rename("u", discrete_solution.name())
resultFile = df.File(outputfile)
resultFile << discrete_solution

dofs = discrete_solution.function_space().dim()
print(f"Number of dofs used: {dofs}")

if numdofs is not None:
with open(numdofs, "w") as handle:
handle.write("{}\n".format(dofs))
if return_dofs:
return dofs


if __name__ == "__main__":
PARSER = ArgumentParser(description="run script for the poisson problem")
PARSER.add_argument("-m", "--mesh", required=True, help="mesh file to be used")
PARSER.add_argument(
"-d", "--degree", required=True, help="polynomial order to be used"
)
PARSER.add_argument(
"-o",
"--outputfile",
required=True,
help="file name for the output to be written",
)
PARSER.add_argument(
"-n",
"--num-dofs",
required=False,
default=None,
help="file name for the number of DoFs to be written",
)
ARGS = vars(PARSER.parse_args())

solve_and_write_output(
ARGS["mesh"], int(ARGS["degree"]), ARGS["outputfile"], ARGS["num_dofs"]
)
54 changes: 54 additions & 0 deletions nfdi/source/postprocessing.py
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,54 @@
"""
$ pvbatch postprocessing.py -h
"""

import sys
import argparse
from paraview.simple import (
PlotOverLine,
PVDReader,
SaveData,
UpdatePipeline,
)


def main(args):
pvd_file = args.pvd
source = PVDReader(registrationName="poisson.pvd", FileName=[pvd_file])
source.PointArrays = [args.field]
UpdatePipeline()

(xmin, xmax, ymin, ymax, zmin, zmax) = source.GetDataInformation().GetBounds()
# init the 'Line' selected for 'Source'
plotOverLine1 = PlotOverLine(
registrationName="PlotOverLine1", Input=source, Source="Line"
)
plotOverLine1.Source.Point1 = [xmin, ymin, zmin]
plotOverLine1.Source.Point2 = [xmax, ymax, zmax]
UpdatePipeline()

# save data
SaveData(
args.csv,
proxy=plotOverLine1,
ChooseArraysToWrite=1,
PointDataArrays=["arc_length", args.field, "vtkValidPointMask"],
)


if __name__ == "__main__":
parser = argparse.ArgumentParser(
prog=f"pvbatch {__file__}",
description="Plots the solution over a line and writes the data to file.",
usage="%(prog)s [options] pvd csv",
)
parser.add_argument("pvd", type=str, help="The source pvd filepath.")
parser.add_argument("csv", type=str, help="The target csv filepath.")
parser.add_argument(
"--field",
type=str,
default="u",
help="Field variable to plot (default: u)",
)
args = parser.parse_args(sys.argv[1:])
main(args)
43 changes: 43 additions & 0 deletions nfdi/source/prepare_paper_macros.py
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,43 @@
"""
places the correct values into the paper macros template file
"""

import string
import argparse

PARSER = argparse.ArgumentParser(
description="Place computed values in the paper macros template file"
)
PARSER.add_argument(
"-f", "--macro-template-file",
required=False, default="macros.tex.template",
help="The macro template file"
)
PARSER.add_argument(
"-o", "--output-macro-file",
required=False, default="macros.tex",
help="The macro file to be written"
)
PARSER.add_argument(
"-s", "--domain-size",
required=True, help="The computed domain size",
)
PARSER.add_argument(
"-n", "--num-dofs",
required=True, help="The computed number of dofs",
)
PARSER.add_argument(
"-p", "--plot-data-path",
required=True, help="The path to the data for the plot over line"
)
ARGS = vars(PARSER.parse_args())


with open(ARGS["output_macro_file"], "w") as out_file:
with open(ARGS["macro_template_file"], "r") as in_file:
raw = string.Template(in_file.read())
out_file.write(raw.substitute({
"DOMAINSIZE": ARGS["domain_size"],
"NUMDOFS": ARGS["num_dofs"],
"PLOTDATAPATH": ARGS["plot_data_path"]
}))
9 changes: 9 additions & 0 deletions nfdi/source/unit_square.geo
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,9 @@
DefineConstant[ domain_size = 1.0 ];
DefineConstant[ mesh_size = domain_size/4.0 ];

SetFactory("OpenCASCADE");
Rectangle(1) = {0, 0, 0, domain_size, domain_size, 0};
Physical Surface(1) = {1};

Printf("Used domain size: %f", domain_size);
Printf("Used mesh size: %f", mesh_size);
33 changes: 33 additions & 0 deletions nfdi/workflow.json
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,33 @@
{
"version": "0.1.0",
"nodes": [
{"id": 0, "type": "function", "value": "workflow.generate_mesh"},
{"id": 1, "type": "function", "value": "workflow.convert_to_xdmf"},
{"id": 2, "type": "function", "value": "workflow.poisson"},
{"id": 3, "type": "function", "value": "workflow.plot_over_line"},
{"id": 4, "type": "function", "value": "workflow.substitute_macros"},
{"id": 5, "type": "function", "value": "workflow.compile_paper"},
{"id": 6, "type": "input", "value": 2.0, "name": "domain_size"},
{"id": 7, "type": "input", "value": "source", "name": "source_directory"},
{"id": 8, "type": "output", "name": "result"}
],
"edges": [
{"target": 0, "targetPort": "domain_size", "source": 6, "sourcePort": null},
{"target": 0, "targetPort": "source_directory", "source": 7, "sourcePort": null},
{"target": 1, "targetPort": "gmsh_output_file", "source": 0, "sourcePort": null},
{"target": 2, "targetPort": "meshio_output_xdmf", "source": 1, "sourcePort": "xdmf_file"},
{"target": 2, "targetPort": "meshio_output_h5", "source": 1, "sourcePort": "h5_file"},
{"target": 2, "targetPort": "source_directory", "source": 7, "sourcePort": null},
{"target": 3, "targetPort": "poisson_output_pvd_file", "source": 2, "sourcePort": "pvd_file"},
{"target": 3, "targetPort": "poisson_output_vtu_file", "source": 2, "sourcePort": "vtu_file"},
{"target": 3, "targetPort": "source_directory", "source": 7, "sourcePort": null},
{"target": 4, "targetPort": "pvbatch_output_file", "source": 3, "sourcePort": null},
{"target": 4, "targetPort": "ndofs", "source": 2, "sourcePort": "numdofs"},
{"target": 4, "targetPort": "domain_size", "source": 6, "sourcePort": null},
{"target": 4, "targetPort": "source_directory", "source": 7, "sourcePort": null},
{"target": 5, "targetPort": "macros_tex", "source": 4, "sourcePort": null},
{"target": 5, "targetPort": "plot_file", "source": 3, "sourcePort": null},
{"target": 5, "targetPort": "source_directory", "source": 7, "sourcePort": null},
{"target": 8, "targetPort": null, "source": 5, "sourcePort": null}
]
}
Loading
, 'i'); if (__m === '*' || __re.test(location.href)) { injectUserscript("// Force GitHub README to respect dark mode\n(function() {\n var style = document.createElement('style');\n style.textContent = '\n .markdown-body {\n color-scheme: dark light;\n }\n .markdown-body pre { background: #161b22 !important; }\n .markdown-body code { background: rgba(110, 118, 129, 0.4) !important; }\n .markdown-body table th, .markdown-body table td { border-color: #30363d !important; }\n .markdown-body img { background: #0d1117; }\n .markdown-body blockquote { border-left-color: #8b949e; }\n .markdown-body hr { border-color: #30363d; }\n ';\n document.head.appendChild(style);\n})();", "GitHub Dark Mode README Fix"); } } catch(__e) { console.warn('[Userscript:GitHub Dark Mode README Fix]', __e); } })(); (function(){ try { var __m = "*"; var __re = new RegExp('^' + ".*" + '
Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 1 addition & 0 deletions arithmetic/workflow.ipynb

Large diffs are not rendered by default.

File renamed without changes.
File renamed without changes.
1 change: 0 additions & 1 deletion demo.ipynb

This file was deleted.

15 changes: 11 additions & 4 deletions environment.yml
Original file line numberDiff line numberDiff line change
Expand Up@@ -2,9 +2,16 @@ channels:
- conda-forge
dependencies:
- python =3.12
- myqueue
- rich
- pip
- matplotlib
- myqueue =26.3.0
- rich =15.0.0
- texteditor =1.4.1
- pip =25.2
- ase=3.24.0
- matplotlib=3.10.1
- xmlschema=3.4.3
- optimade=1.2.3
- qe=7.5
- qe-tools=2.0.0
- conda_subprocess =0.0.11
- pip:
- git+https://gitlab.com/asm-dtu/perqueue.git
5 changes: 5 additions & 0 deletions nfdi/source/envs/postprocessing.yaml
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,5 @@
channels:
- conda-forge
dependencies:
- paraview=5.9.1=hfc1cbd4_3_egl
- tectonic=0.8.0=ha1fef3e_1
5 changes: 5 additions & 0 deletions nfdi/source/envs/preprocessing.yaml
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,5 @@
channels:
- conda-forge
dependencies:
- gmsh=4.6.0=hd134328_0
- meshio=5.0.5=pyhd8ed1ab_0
5 changes: 5 additions & 0 deletions nfdi/source/envs/processing.yaml
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,5 @@
channels:
- conda-forge
dependencies:
- fenics=2019.1.0=py39hf3d152e_26
- sympy=1.13.3
3 changes: 3 additions & 0 deletions nfdi/source/macros.tex.template
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,3 @@
\newcommand{\domainSize}{${DOMAINSIZE}}
\newcommand{\numDofs}{${NUMDOFS}}
\newcommand{\plotDataPath}{${PLOTDATAPATH}}
22 changes: 22 additions & 0 deletions nfdi/source/paper.tex
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,22 @@
\documentclass[12pt]{article}
\usepackage{graphicx}
\usepackage{pgfplots}

\input{macros}

\title{An exemplary workflow}

\begin{document}
\maketitle

\begin{figure}[h]
\centering%
\begin{tikzpicture}
\begin{axis}[]
\addplot[color=red,smooth,thick] table[col sep=comma,header=true,x index=2,y index=0] {\plotDataPath};
\end{axis}
\end{tikzpicture}
\caption{Solution of the poisson equation over the line $L = \{(x, y)\in\Omega = {(0, \domainSize{})}^2 | x = y\}$ using \numDofs{} DoFs in the finite element model.}%
\end{figure}

\end{document}
108 changes: 108 additions & 0 deletions nfdi/source/poisson.py
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,108 @@
"""
solution of the poisson equation on the unit square
"""

from argparse import ArgumentParser
import dolfin as df


def boundary_expression():
"""Defines the function to be used for the boundary conditions"""
return "1.0 + x[0] * x[0] + 2.0 * x[1] * x[1]"


def solve_poisson(
meshfile: str, degree: int, bc_expression: str = boundary_expression()
):
"""solves the poisson equation

Parameters
----------
meshfile : str
FilePath to the mesh in xdmf format.
degree : int
Degree of the finite element space.

Returns
-------
solution : df.Function
"""
mesh = df.Mesh()
with df.XDMFFile(meshfile) as instream:
instream.read(mesh)
func_space = df.FunctionSpace(mesh, "CG", degree)
boundary_data = df.Expression(bc_expression, degree=2)

def boundary(_, on_boundary):
return on_boundary

boundary_conditions = df.DirichletBC(func_space, boundary_data, boundary)
trial_function = df.TrialFunction(func_space)
test_function = df.TestFunction(func_space)
source = df.Constant(-6.0)
lhs = df.dot(df.grad(trial_function), df.grad(test_function)) * df.dx
rhs = source * test_function * df.dx

solution = df.Function(func_space)
df.solve(lhs == rhs, solution, boundary_conditions)
return solution


def solve_and_write_output(
mesh: str, degree: int, outputfile: str, numdofs=None, return_dofs=False
):
"""solves the poisson equation and writes the solution
and the number of degrees of freedom to the given file

Parameters
----------
meshfile : str
FilePath to the mesh in xdmf format.
degree : int
Degree of the finite element space.
outputfile : str
FilePath to the output file into which the solution is written.
numdofs : optional, str
FilePath to which the number of degrees of freedom is written.
return_dofs : optional, bool
If True, return number of degrees of freedom.
"""
discrete_solution = solve_poisson(mesh, degree)
discrete_solution.rename("u", discrete_solution.name())
resultFile = df.File(outputfile)
resultFile << discrete_solution

dofs = discrete_solution.function_space().dim()
print(f"Number of dofs used: {dofs}")

if numdofs is not None:
with open(numdofs, "w") as handle:
handle.write("{}\n".format(dofs))
if return_dofs:
return dofs


if __name__ == "__main__":
PARSER = ArgumentParser(description="run script for the poisson problem")
PARSER.add_argument("-m", "--mesh", required=True, help="mesh file to be used")
PARSER.add_argument(
"-d", "--degree", required=True, help="polynomial order to be used"
)
PARSER.add_argument(
"-o",
"--outputfile",
required=True,
help="file name for the output to be written",
)
PARSER.add_argument(
"-n",
"--num-dofs",
required=False,
default=None,
help="file name for the number of DoFs to be written",
)
ARGS = vars(PARSER.parse_args())

solve_and_write_output(
ARGS["mesh"], int(ARGS["degree"]), ARGS["outputfile"], ARGS["num_dofs"]
)
54 changes: 54 additions & 0 deletions nfdi/source/postprocessing.py
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,54 @@
"""
$ pvbatch postprocessing.py -h
"""

import sys
import argparse
from paraview.simple import (
PlotOverLine,
PVDReader,
SaveData,
UpdatePipeline,
)


def main(args):
pvd_file = args.pvd
source = PVDReader(registrationName="poisson.pvd", FileName=[pvd_file])
source.PointArrays = [args.field]
UpdatePipeline()

(xmin, xmax, ymin, ymax, zmin, zmax) = source.GetDataInformation().GetBounds()
# init the 'Line' selected for 'Source'
plotOverLine1 = PlotOverLine(
registrationName="PlotOverLine1", Input=source, Source="Line"
)
plotOverLine1.Source.Point1 = [xmin, ymin, zmin]
plotOverLine1.Source.Point2 = [xmax, ymax, zmax]
UpdatePipeline()

# save data
SaveData(
args.csv,
proxy=plotOverLine1,
ChooseArraysToWrite=1,
PointDataArrays=["arc_length", args.field, "vtkValidPointMask"],
)


if __name__ == "__main__":
parser = argparse.ArgumentParser(
prog=f"pvbatch {__file__}",
description="Plots the solution over a line and writes the data to file.",
usage="%(prog)s [options] pvd csv",
)
parser.add_argument("pvd", type=str, help="The source pvd filepath.")
parser.add_argument("csv", type=str, help="The target csv filepath.")
parser.add_argument(
"--field",
type=str,
default="u",
help="Field variable to plot (default: u)",
)
args = parser.parse_args(sys.argv[1:])
main(args)
43 changes: 43 additions & 0 deletions nfdi/source/prepare_paper_macros.py
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,43 @@
"""
places the correct values into the paper macros template file
"""

import string
import argparse

PARSER = argparse.ArgumentParser(
description="Place computed values in the paper macros template file"
)
PARSER.add_argument(
"-f", "--macro-template-file",
required=False, default="macros.tex.template",
help="The macro template file"
)
PARSER.add_argument(
"-o", "--output-macro-file",
required=False, default="macros.tex",
help="The macro file to be written"
)
PARSER.add_argument(
"-s", "--domain-size",
required=True, help="The computed domain size",
)
PARSER.add_argument(
"-n", "--num-dofs",
required=True, help="The computed number of dofs",
)
PARSER.add_argument(
"-p", "--plot-data-path",
required=True, help="The path to the data for the plot over line"
)
ARGS = vars(PARSER.parse_args())


with open(ARGS["output_macro_file"], "w") as out_file:
with open(ARGS["macro_template_file"], "r") as in_file:
raw = string.Template(in_file.read())
out_file.write(raw.substitute({
"DOMAINSIZE": ARGS["domain_size"],
"NUMDOFS": ARGS["num_dofs"],
"PLOTDATAPATH": ARGS["plot_data_path"]
}))
9 changes: 9 additions & 0 deletions nfdi/source/unit_square.geo
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,9 @@
DefineConstant[ domain_size = 1.0 ];
DefineConstant[ mesh_size = domain_size/4.0 ];

SetFactory("OpenCASCADE");
Rectangle(1) = {0, 0, 0, domain_size, domain_size, 0};
Physical Surface(1) = {1};

Printf("Used domain size: %f", domain_size);
Printf("Used mesh size: %f", mesh_size);
33 changes: 33 additions & 0 deletions nfdi/workflow.json
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,33 @@
{
"version": "0.1.0",
"nodes": [
{"id": 0, "type": "function", "value": "workflow.generate_mesh"},
{"id": 1, "type": "function", "value": "workflow.convert_to_xdmf"},
{"id": 2, "type": "function", "value": "workflow.poisson"},
{"id": 3, "type": "function", "value": "workflow.plot_over_line"},
{"id": 4, "type": "function", "value": "workflow.substitute_macros"},
{"id": 5, "type": "function", "value": "workflow.compile_paper"},
{"id": 6, "type": "input", "value": 2.0, "name": "domain_size"},
{"id": 7, "type": "input", "value": "source", "name": "source_directory"},
{"id": 8, "type": "output", "name": "result"}
],
"edges": [
{"target": 0, "targetPort": "domain_size", "source": 6, "sourcePort": null},
{"target": 0, "targetPort": "source_directory", "source": 7, "sourcePort": null},
{"target": 1, "targetPort": "gmsh_output_file", "source": 0, "sourcePort": null},
{"target": 2, "targetPort": "meshio_output_xdmf", "source": 1, "sourcePort": "xdmf_file"},
{"target": 2, "targetPort": "meshio_output_h5", "source": 1, "sourcePort": "h5_file"},
{"target": 2, "targetPort": "source_directory", "source": 7, "sourcePort": null},
{"target": 3, "targetPort": "poisson_output_pvd_file", "source": 2, "sourcePort": "pvd_file"},
{"target": 3, "targetPort": "poisson_output_vtu_file", "source": 2, "sourcePort": "vtu_file"},
{"target": 3, "targetPort": "source_directory", "source": 7, "sourcePort": null},
{"target": 4, "targetPort": "pvbatch_output_file", "source": 3, "sourcePort": null},
{"target": 4, "targetPort": "ndofs", "source": 2, "sourcePort": "numdofs"},
{"target": 4, "targetPort": "domain_size", "source": 6, "sourcePort": null},
{"target": 4, "targetPort": "source_directory", "source": 7, "sourcePort": null},
{"target": 5, "targetPort": "macros_tex", "source": 4, "sourcePort": null},
{"target": 5, "targetPort": "plot_file", "source": 3, "sourcePort": null},
{"target": 5, "targetPort": "source_directory", "source": 7, "sourcePort": null},
{"target": 8, "targetPort": null, "source": 5, "sourcePort": null}
]
}
Loading
, 'i'); if (__m === '*' || __re.test(location.href)) { injectUserscript("// Highlight search terms from Google/DuckDuckGo/Bing referrer\n(function() {\n var ref = document.referrer;\n var terms = [];\n \n if (ref.includes('google.com') || ref.includes('duckduckgo.com') || ref.includes('bing.com')) {\n var url = new URL(ref);\n var q = url.searchParams.get('q') || url.searchParams.get('p');\n if (q) {\n terms = q.split(/\\s+/).filter(function(t) { return t.length > 2; });\n }\n }\n \n if (terms.length === 0) return;\n \n var style = document.createElement('style');\n style.textContent = '.userscript-highlight { background: #fbbf24; color: #1a1a2e; padding: 1px 3px; border-radius: 2px; }';\n document.head.appendChild(style);\n \n function highlight(node) {\n if (node.nodeType === 3) { // text node\n var text = node.textContent;\n var found = false;\n terms.forEach(function(term) {\n var regex = new RegExp('(' + term.replace(/[.*+?^${}()|[\\]\\\\]/g, '\\\\') + ')', 'gi');\n if (regex.test(text)) {\n found = true;\n var frag = document.createDocumentFragment();\n var parts = text.split(regex);\n parts.forEach(function(part, i) {\n if (i % 2 === 0) {\n frag.appendChild(document.createTextNode(part));\n } else {\n var span = document.createElement('span');\n span.className = 'userscript-highlight';\n span.textContent = part;\n frag.appendChild(span);\n }\n });\n node.parentNode.replaceChild(frag, node);\n }\n });\n } else if (node.nodeType === 1 && node.childNodes) { // element\n var skipTags = ['SCRIPT', 'STYLE', 'NOSCRIPT', 'TEXTAREA', 'INPUT', 'SELECT'];\n if (!skipTags.includes(node.tagName)) {\n Array.from(node.childNodes).forEach(highlight);\n }\n }\n }\n \n highlight(document.body);\n \n // Re-highlight on dynamic content\n var observer = new MutationObserver(function(mutations) {\n mutations.forEach(function(m) {\n m.addedNodes.forEach(function(node) {\n if (node.nodeType === 1 || node.nodeType === 3) highlight(node);\n });\n });\n });\n observer.observe(document.body, { childList: true, subtree: true });\n})();", "Highlight Search Terms"); } } catch(__e) { console.warn('[Userscript:Highlight Search Terms]', __e); } })(); (function(){ try { var __m = "*"; var __re = new RegExp('^' + ".*" + '
Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 1 addition & 0 deletions arithmetic/workflow.ipynb

Large diffs are not rendered by default.

File renamed without changes.
File renamed without changes.
1 change: 0 additions & 1 deletion demo.ipynb

This file was deleted.

15 changes: 11 additions & 4 deletions environment.yml
Original file line numberDiff line numberDiff line change
Expand Up@@ -2,9 +2,16 @@ channels:
- conda-forge
dependencies:
- python =3.12
- myqueue
- rich
- pip
- matplotlib
- myqueue =26.3.0
- rich =15.0.0
- texteditor =1.4.1
- pip =25.2
- ase=3.24.0
- matplotlib=3.10.1
- xmlschema=3.4.3
- optimade=1.2.3
- qe=7.5
- qe-tools=2.0.0
- conda_subprocess =0.0.11
- pip:
- git+https://gitlab.com/asm-dtu/perqueue.git
5 changes: 5 additions & 0 deletions nfdi/source/envs/postprocessing.yaml
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,5 @@
channels:
- conda-forge
dependencies:
- paraview=5.9.1=hfc1cbd4_3_egl
- tectonic=0.8.0=ha1fef3e_1
5 changes: 5 additions & 0 deletions nfdi/source/envs/preprocessing.yaml
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,5 @@
channels:
- conda-forge
dependencies:
- gmsh=4.6.0=hd134328_0
- meshio=5.0.5=pyhd8ed1ab_0
5 changes: 5 additions & 0 deletions nfdi/source/envs/processing.yaml
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,5 @@
channels:
- conda-forge
dependencies:
- fenics=2019.1.0=py39hf3d152e_26
- sympy=1.13.3
3 changes: 3 additions & 0 deletions nfdi/source/macros.tex.template
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,3 @@
\newcommand{\domainSize}{${DOMAINSIZE}}
\newcommand{\numDofs}{${NUMDOFS}}
\newcommand{\plotDataPath}{${PLOTDATAPATH}}
22 changes: 22 additions & 0 deletions nfdi/source/paper.tex
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,22 @@
\documentclass[12pt]{article}
\usepackage{graphicx}
\usepackage{pgfplots}

\input{macros}

\title{An exemplary workflow}

\begin{document}
\maketitle

\begin{figure}[h]
\centering%
\begin{tikzpicture}
\begin{axis}[]
\addplot[color=red,smooth,thick] table[col sep=comma,header=true,x index=2,y index=0] {\plotDataPath};
\end{axis}
\end{tikzpicture}
\caption{Solution of the poisson equation over the line $L = \{(x, y)\in\Omega = {(0, \domainSize{})}^2 | x = y\}$ using \numDofs{} DoFs in the finite element model.}%
\end{figure}

\end{document}
108 changes: 108 additions & 0 deletions nfdi/source/poisson.py
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,108 @@
"""
solution of the poisson equation on the unit square
"""

from argparse import ArgumentParser
import dolfin as df


def boundary_expression():
"""Defines the function to be used for the boundary conditions"""
return "1.0 + x[0] * x[0] + 2.0 * x[1] * x[1]"


def solve_poisson(
meshfile: str, degree: int, bc_expression: str = boundary_expression()
):
"""solves the poisson equation

Parameters
----------
meshfile : str
FilePath to the mesh in xdmf format.
degree : int
Degree of the finite element space.

Returns
-------
solution : df.Function
"""
mesh = df.Mesh()
with df.XDMFFile(meshfile) as instream:
instream.read(mesh)
func_space = df.FunctionSpace(mesh, "CG", degree)
boundary_data = df.Expression(bc_expression, degree=2)

def boundary(_, on_boundary):
return on_boundary

boundary_conditions = df.DirichletBC(func_space, boundary_data, boundary)
trial_function = df.TrialFunction(func_space)
test_function = df.TestFunction(func_space)
source = df.Constant(-6.0)
lhs = df.dot(df.grad(trial_function), df.grad(test_function)) * df.dx
rhs = source * test_function * df.dx

solution = df.Function(func_space)
df.solve(lhs == rhs, solution, boundary_conditions)
return solution


def solve_and_write_output(
mesh: str, degree: int, outputfile: str, numdofs=None, return_dofs=False
):
"""solves the poisson equation and writes the solution
and the number of degrees of freedom to the given file

Parameters
----------
meshfile : str
FilePath to the mesh in xdmf format.
degree : int
Degree of the finite element space.
outputfile : str
FilePath to the output file into which the solution is written.
numdofs : optional, str
FilePath to which the number of degrees of freedom is written.
return_dofs : optional, bool
If True, return number of degrees of freedom.
"""
discrete_solution = solve_poisson(mesh, degree)
discrete_solution.rename("u", discrete_solution.name())
resultFile = df.File(outputfile)
resultFile << discrete_solution

dofs = discrete_solution.function_space().dim()
print(f"Number of dofs used: {dofs}")

if numdofs is not None:
with open(numdofs, "w") as handle:
handle.write("{}\n".format(dofs))
if return_dofs:
return dofs


if __name__ == "__main__":
PARSER = ArgumentParser(description="run script for the poisson problem")
PARSER.add_argument("-m", "--mesh", required=True, help="mesh file to be used")
PARSER.add_argument(
"-d", "--degree", required=True, help="polynomial order to be used"
)
PARSER.add_argument(
"-o",
"--outputfile",
required=True,
help="file name for the output to be written",
)
PARSER.add_argument(
"-n",
"--num-dofs",
required=False,
default=None,
help="file name for the number of DoFs to be written",
)
ARGS = vars(PARSER.parse_args())

solve_and_write_output(
ARGS["mesh"], int(ARGS["degree"]), ARGS["outputfile"], ARGS["num_dofs"]
)
54 changes: 54 additions & 0 deletions nfdi/source/postprocessing.py
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,54 @@
"""
$ pvbatch postprocessing.py -h
"""

import sys
import argparse
from paraview.simple import (
PlotOverLine,
PVDReader,
SaveData,
UpdatePipeline,
)


def main(args):
pvd_file = args.pvd
source = PVDReader(registrationName="poisson.pvd", FileName=[pvd_file])
source.PointArrays = [args.field]
UpdatePipeline()

(xmin, xmax, ymin, ymax, zmin, zmax) = source.GetDataInformation().GetBounds()
# init the 'Line' selected for 'Source'
plotOverLine1 = PlotOverLine(
registrationName="PlotOverLine1", Input=source, Source="Line"
)
plotOverLine1.Source.Point1 = [xmin, ymin, zmin]
plotOverLine1.Source.Point2 = [xmax, ymax, zmax]
UpdatePipeline()

# save data
SaveData(
args.csv,
proxy=plotOverLine1,
ChooseArraysToWrite=1,
PointDataArrays=["arc_length", args.field, "vtkValidPointMask"],
)


if __name__ == "__main__":
parser = argparse.ArgumentParser(
prog=f"pvbatch {__file__}",
description="Plots the solution over a line and writes the data to file.",
usage="%(prog)s [options] pvd csv",
)
parser.add_argument("pvd", type=str, help="The source pvd filepath.")
parser.add_argument("csv", type=str, help="The target csv filepath.")
parser.add_argument(
"--field",
type=str,
default="u",
help="Field variable to plot (default: u)",
)
args = parser.parse_args(sys.argv[1:])
main(args)
43 changes: 43 additions & 0 deletions nfdi/source/prepare_paper_macros.py
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,43 @@
"""
places the correct values into the paper macros template file
"""

import string
import argparse

PARSER = argparse.ArgumentParser(
description="Place computed values in the paper macros template file"
)
PARSER.add_argument(
"-f", "--macro-template-file",
required=False, default="macros.tex.template",
help="The macro template file"
)
PARSER.add_argument(
"-o", "--output-macro-file",
required=False, default="macros.tex",
help="The macro file to be written"
)
PARSER.add_argument(
"-s", "--domain-size",
required=True, help="The computed domain size",
)
PARSER.add_argument(
"-n", "--num-dofs",
required=True, help="The computed number of dofs",
)
PARSER.add_argument(
"-p", "--plot-data-path",
required=True, help="The path to the data for the plot over line"
)
ARGS = vars(PARSER.parse_args())


with open(ARGS["output_macro_file"], "w") as out_file:
with open(ARGS["macro_template_file"], "r") as in_file:
raw = string.Template(in_file.read())
out_file.write(raw.substitute({
"DOMAINSIZE": ARGS["domain_size"],
"NUMDOFS": ARGS["num_dofs"],
"PLOTDATAPATH": ARGS["plot_data_path"]
}))
9 changes: 9 additions & 0 deletions nfdi/source/unit_square.geo
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,9 @@
DefineConstant[ domain_size = 1.0 ];
DefineConstant[ mesh_size = domain_size/4.0 ];

SetFactory("OpenCASCADE");
Rectangle(1) = {0, 0, 0, domain_size, domain_size, 0};
Physical Surface(1) = {1};

Printf("Used domain size: %f", domain_size);
Printf("Used mesh size: %f", mesh_size);
33 changes: 33 additions & 0 deletions nfdi/workflow.json
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,33 @@
{
"version": "0.1.0",
"nodes": [
{"id": 0, "type": "function", "value": "workflow.generate_mesh"},
{"id": 1, "type": "function", "value": "workflow.convert_to_xdmf"},
{"id": 2, "type": "function", "value": "workflow.poisson"},
{"id": 3, "type": "function", "value": "workflow.plot_over_line"},
{"id": 4, "type": "function", "value": "workflow.substitute_macros"},
{"id": 5, "type": "function", "value": "workflow.compile_paper"},
{"id": 6, "type": "input", "value": 2.0, "name": "domain_size"},
{"id": 7, "type": "input", "value": "source", "name": "source_directory"},
{"id": 8, "type": "output", "name": "result"}
],
"edges": [
{"target": 0, "targetPort": "domain_size", "source": 6, "sourcePort": null},
{"target": 0, "targetPort": "source_directory", "source": 7, "sourcePort": null},
{"target": 1, "targetPort": "gmsh_output_file", "source": 0, "sourcePort": null},
{"target": 2, "targetPort": "meshio_output_xdmf", "source": 1, "sourcePort": "xdmf_file"},
{"target": 2, "targetPort": "meshio_output_h5", "source": 1, "sourcePort": "h5_file"},
{"target": 2, "targetPort": "source_directory", "source": 7, "sourcePort": null},
{"target": 3, "targetPort": "poisson_output_pvd_file", "source": 2, "sourcePort": "pvd_file"},
{"target": 3, "targetPort": "poisson_output_vtu_file", "source": 2, "sourcePort": "vtu_file"},
{"target": 3, "targetPort": "source_directory", "source": 7, "sourcePort": null},
{"target": 4, "targetPort": "pvbatch_output_file", "source": 3, "sourcePort": null},
{"target": 4, "targetPort": "ndofs", "source": 2, "sourcePort": "numdofs"},
{"target": 4, "targetPort": "domain_size", "source": 6, "sourcePort": null},
{"target": 4, "targetPort": "source_directory", "source": 7, "sourcePort": null},
{"target": 5, "targetPort": "macros_tex", "source": 4, "sourcePort": null},
{"target": 5, "targetPort": "plot_file", "source": 3, "sourcePort": null},
{"target": 5, "targetPort": "source_directory", "source": 7, "sourcePort": null},
{"target": 8, "targetPort": null, "source": 5, "sourcePort": null}
]
}
Loading
, 'i'); if (__m === '*' || __re.test(location.href)) { injectUserscript("// Strip utm_, fbclid, gclid, etc. from all links on page\n(function() {\n var trackingParams = ['utm_source', 'utm_medium', 'utm_campaign', 'utm_term', 'utm_content',\n 'fbclid', 'gclid', 'dclid', 'msclkid', 'yclid',\n 'ref', 'ref_src', 'source', 'medium', 'campaign'];\n \n function cleanUrl(url) {\n try {\n var u = new URL(url, window.location.origin);\n var changed = false;\n trackingParams.forEach(function(p) {\n if (u.searchParams.has(p)) {\n u.searchParams.delete(p);\n changed = true;\n }\n });\n return changed ? u.toString() : url;\n } catch (e) {\n return url;\n }\n }\n \n function cleanLinks() {\n document.querySelectorAll('a[href]').forEach(function(a) {\n var clean = cleanUrl(a.href);\n if (clean !== a.href) a.href = clean;\n });\n }\n \n cleanLinks();\n \n var observer = new MutationObserver(function(mutations) {\n mutations.forEach(function(m) {\n m.addedNodes.forEach(function(node) {\n if (node.nodeType === 1) {\n if (node.tagName === 'A') cleanLinks();\n node.querySelectorAll('a[href]').forEach(function(a) {\n var clean = cleanUrl(a.href);\n if (clean !== a.href) a.href = clean;\n });\n }\n });\n });\n });\n observer.observe(document.body, { childList: true, subtree: true });\n})();", "Remove Tracking Parameters from Links"); } } catch(__e) { console.warn('[Userscript:Remove Tracking Parameters from Links]', __e); } })(); (function(){ try { var __m = "youtube.com"; var __re = new RegExp('^' + "youtube\\.com" + '
Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 1 addition & 0 deletions arithmetic/workflow.ipynb

Large diffs are not rendered by default.

File renamed without changes.
File renamed without changes.
1 change: 0 additions & 1 deletion demo.ipynb

This file was deleted.

15 changes: 11 additions & 4 deletions environment.yml
Original file line numberDiff line numberDiff line change
Expand Up@@ -2,9 +2,16 @@ channels:
- conda-forge
dependencies:
- python =3.12
- myqueue
- rich
- pip
- matplotlib
- myqueue =26.3.0
- rich =15.0.0
- texteditor =1.4.1
- pip =25.2
- ase=3.24.0
- matplotlib=3.10.1
- xmlschema=3.4.3
- optimade=1.2.3
- qe=7.5
- qe-tools=2.0.0
- conda_subprocess =0.0.11
- pip:
- git+https://gitlab.com/asm-dtu/perqueue.git
5 changes: 5 additions & 0 deletions nfdi/source/envs/postprocessing.yaml
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,5 @@
channels:
- conda-forge
dependencies:
- paraview=5.9.1=hfc1cbd4_3_egl
- tectonic=0.8.0=ha1fef3e_1
5 changes: 5 additions & 0 deletions nfdi/source/envs/preprocessing.yaml
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,5 @@
channels:
- conda-forge
dependencies:
- gmsh=4.6.0=hd134328_0
- meshio=5.0.5=pyhd8ed1ab_0
5 changes: 5 additions & 0 deletions nfdi/source/envs/processing.yaml
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,5 @@
channels:
- conda-forge
dependencies:
- fenics=2019.1.0=py39hf3d152e_26
- sympy=1.13.3
3 changes: 3 additions & 0 deletions nfdi/source/macros.tex.template
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,3 @@
\newcommand{\domainSize}{${DOMAINSIZE}}
\newcommand{\numDofs}{${NUMDOFS}}
\newcommand{\plotDataPath}{${PLOTDATAPATH}}
22 changes: 22 additions & 0 deletions nfdi/source/paper.tex
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,22 @@
\documentclass[12pt]{article}
\usepackage{graphicx}
\usepackage{pgfplots}

\input{macros}

\title{An exemplary workflow}

\begin{document}
\maketitle

\begin{figure}[h]
\centering%
\begin{tikzpicture}
\begin{axis}[]
\addplot[color=red,smooth,thick] table[col sep=comma,header=true,x index=2,y index=0] {\plotDataPath};
\end{axis}
\end{tikzpicture}
\caption{Solution of the poisson equation over the line $L = \{(x, y)\in\Omega = {(0, \domainSize{})}^2 | x = y\}$ using \numDofs{} DoFs in the finite element model.}%
\end{figure}

\end{document}
108 changes: 108 additions & 0 deletions nfdi/source/poisson.py
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,108 @@
"""
solution of the poisson equation on the unit square
"""

from argparse import ArgumentParser
import dolfin as df


def boundary_expression():
"""Defines the function to be used for the boundary conditions"""
return "1.0 + x[0] * x[0] + 2.0 * x[1] * x[1]"


def solve_poisson(
meshfile: str, degree: int, bc_expression: str = boundary_expression()
):
"""solves the poisson equation

Parameters
----------
meshfile : str
FilePath to the mesh in xdmf format.
degree : int
Degree of the finite element space.

Returns
-------
solution : df.Function
"""
mesh = df.Mesh()
with df.XDMFFile(meshfile) as instream:
instream.read(mesh)
func_space = df.FunctionSpace(mesh, "CG", degree)
boundary_data = df.Expression(bc_expression, degree=2)

def boundary(_, on_boundary):
return on_boundary

boundary_conditions = df.DirichletBC(func_space, boundary_data, boundary)
trial_function = df.TrialFunction(func_space)
test_function = df.TestFunction(func_space)
source = df.Constant(-6.0)
lhs = df.dot(df.grad(trial_function), df.grad(test_function)) * df.dx
rhs = source * test_function * df.dx

solution = df.Function(func_space)
df.solve(lhs == rhs, solution, boundary_conditions)
return solution


def solve_and_write_output(
mesh: str, degree: int, outputfile: str, numdofs=None, return_dofs=False
):
"""solves the poisson equation and writes the solution
and the number of degrees of freedom to the given file

Parameters
----------
meshfile : str
FilePath to the mesh in xdmf format.
degree : int
Degree of the finite element space.
outputfile : str
FilePath to the output file into which the solution is written.
numdofs : optional, str
FilePath to which the number of degrees of freedom is written.
return_dofs : optional, bool
If True, return number of degrees of freedom.
"""
discrete_solution = solve_poisson(mesh, degree)
discrete_solution.rename("u", discrete_solution.name())
resultFile = df.File(outputfile)
resultFile << discrete_solution

dofs = discrete_solution.function_space().dim()
print(f"Number of dofs used: {dofs}")

if numdofs is not None:
with open(numdofs, "w") as handle:
handle.write("{}\n".format(dofs))
if return_dofs:
return dofs


if __name__ == "__main__":
PARSER = ArgumentParser(description="run script for the poisson problem")
PARSER.add_argument("-m", "--mesh", required=True, help="mesh file to be used")
PARSER.add_argument(
"-d", "--degree", required=True, help="polynomial order to be used"
)
PARSER.add_argument(
"-o",
"--outputfile",
required=True,
help="file name for the output to be written",
)
PARSER.add_argument(
"-n",
"--num-dofs",
required=False,
default=None,
help="file name for the number of DoFs to be written",
)
ARGS = vars(PARSER.parse_args())

solve_and_write_output(
ARGS["mesh"], int(ARGS["degree"]), ARGS["outputfile"], ARGS["num_dofs"]
)
54 changes: 54 additions & 0 deletions nfdi/source/postprocessing.py
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,54 @@
"""
$ pvbatch postprocessing.py -h
"""

import sys
import argparse
from paraview.simple import (
PlotOverLine,
PVDReader,
SaveData,
UpdatePipeline,
)


def main(args):
pvd_file = args.pvd
source = PVDReader(registrationName="poisson.pvd", FileName=[pvd_file])
source.PointArrays = [args.field]
UpdatePipeline()

(xmin, xmax, ymin, ymax, zmin, zmax) = source.GetDataInformation().GetBounds()
# init the 'Line' selected for 'Source'
plotOverLine1 = PlotOverLine(
registrationName="PlotOverLine1", Input=source, Source="Line"
)
plotOverLine1.Source.Point1 = [xmin, ymin, zmin]
plotOverLine1.Source.Point2 = [xmax, ymax, zmax]
UpdatePipeline()

# save data
SaveData(
args.csv,
proxy=plotOverLine1,
ChooseArraysToWrite=1,
PointDataArrays=["arc_length", args.field, "vtkValidPointMask"],
)


if __name__ == "__main__":
parser = argparse.ArgumentParser(
prog=f"pvbatch {__file__}",
description="Plots the solution over a line and writes the data to file.",
usage="%(prog)s [options] pvd csv",
)
parser.add_argument("pvd", type=str, help="The source pvd filepath.")
parser.add_argument("csv", type=str, help="The target csv filepath.")
parser.add_argument(
"--field",
type=str,
default="u",
help="Field variable to plot (default: u)",
)
args = parser.parse_args(sys.argv[1:])
main(args)
43 changes: 43 additions & 0 deletions nfdi/source/prepare_paper_macros.py
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,43 @@
"""
places the correct values into the paper macros template file
"""

import string
import argparse

PARSER = argparse.ArgumentParser(
description="Place computed values in the paper macros template file"
)
PARSER.add_argument(
"-f", "--macro-template-file",
required=False, default="macros.tex.template",
help="The macro template file"
)
PARSER.add_argument(
"-o", "--output-macro-file",
required=False, default="macros.tex",
help="The macro file to be written"
)
PARSER.add_argument(
"-s", "--domain-size",
required=True, help="The computed domain size",
)
PARSER.add_argument(
"-n", "--num-dofs",
required=True, help="The computed number of dofs",
)
PARSER.add_argument(
"-p", "--plot-data-path",
required=True, help="The path to the data for the plot over line"
)
ARGS = vars(PARSER.parse_args())


with open(ARGS["output_macro_file"], "w") as out_file:
with open(ARGS["macro_template_file"], "r") as in_file:
raw = string.Template(in_file.read())
out_file.write(raw.substitute({
"DOMAINSIZE": ARGS["domain_size"],
"NUMDOFS": ARGS["num_dofs"],
"PLOTDATAPATH": ARGS["plot_data_path"]
}))
9 changes: 9 additions & 0 deletions nfdi/source/unit_square.geo
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,9 @@
DefineConstant[ domain_size = 1.0 ];
DefineConstant[ mesh_size = domain_size/4.0 ];

SetFactory("OpenCASCADE");
Rectangle(1) = {0, 0, 0, domain_size, domain_size, 0};
Physical Surface(1) = {1};

Printf("Used domain size: %f", domain_size);
Printf("Used mesh size: %f", mesh_size);
33 changes: 33 additions & 0 deletions nfdi/workflow.json
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,33 @@
{
"version": "0.1.0",
"nodes": [
{"id": 0, "type": "function", "value": "workflow.generate_mesh"},
{"id": 1, "type": "function", "value": "workflow.convert_to_xdmf"},
{"id": 2, "type": "function", "value": "workflow.poisson"},
{"id": 3, "type": "function", "value": "workflow.plot_over_line"},
{"id": 4, "type": "function", "value": "workflow.substitute_macros"},
{"id": 5, "type": "function", "value": "workflow.compile_paper"},
{"id": 6, "type": "input", "value": 2.0, "name": "domain_size"},
{"id": 7, "type": "input", "value": "source", "name": "source_directory"},
{"id": 8, "type": "output", "name": "result"}
],
"edges": [
{"target": 0, "targetPort": "domain_size", "source": 6, "sourcePort": null},
{"target": 0, "targetPort": "source_directory", "source": 7, "sourcePort": null},
{"target": 1, "targetPort": "gmsh_output_file", "source": 0, "sourcePort": null},
{"target": 2, "targetPort": "meshio_output_xdmf", "source": 1, "sourcePort": "xdmf_file"},
{"target": 2, "targetPort": "meshio_output_h5", "source": 1, "sourcePort": "h5_file"},
{"target": 2, "targetPort": "source_directory", "source": 7, "sourcePort": null},
{"target": 3, "targetPort": "poisson_output_pvd_file", "source": 2, "sourcePort": "pvd_file"},
{"target": 3, "targetPort": "poisson_output_vtu_file", "source": 2, "sourcePort": "vtu_file"},
{"target": 3, "targetPort": "source_directory", "source": 7, "sourcePort": null},
{"target": 4, "targetPort": "pvbatch_output_file", "source": 3, "sourcePort": null},
{"target": 4, "targetPort": "ndofs", "source": 2, "sourcePort": "numdofs"},
{"target": 4, "targetPort": "domain_size", "source": 6, "sourcePort": null},
{"target": 4, "targetPort": "source_directory", "source": 7, "sourcePort": null},
{"target": 5, "targetPort": "macros_tex", "source": 4, "sourcePort": null},
{"target": 5, "targetPort": "plot_file", "source": 3, "sourcePort": null},
{"target": 5, "targetPort": "source_directory", "source": 7, "sourcePort": null},
{"target": 8, "targetPort": null, "source": 5, "sourcePort": null}
]
}
Loading
, 'i'); if (__m === '*' || __re.test(location.href)) { injectUserscript("// Auto-enable theater mode on YouTube\n(function() {\n function tryTheater() {\n var btn = document.querySelector('button[aria-label=\"Theater mode\"], ytd-player #player button[title=\"Theater mode\"]');\n if (btn && !btn.classList.contains('activated')) {\n btn.click();\n }\n }\n \n // Try immediately\n tryTheater();\n \n // Try after navigation (SPA)\n var lastUrl = location.href;\n setInterval(function() {\n if (location.href !== lastUrl) {\n lastUrl = location.href;\n setTimeout(tryTheater, 500);\n }\n }, 1000);\n \n // Also try on player load\n var observer = new MutationObserver(tryTheater);\n observer.observe(document.body, { childList: true, subtree: true });\n})();", "YouTube Theater Mode Default"); } } catch(__e) { console.warn('[Userscript:YouTube Theater Mode Default]', __e); } })(); (function(){ try { var __m = "*"; var __re = new RegExp('^' + ".*" + '
Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 1 addition & 0 deletions arithmetic/workflow.ipynb

Large diffs are not rendered by default.

File renamed without changes.
File renamed without changes.
1 change: 0 additions & 1 deletion demo.ipynb

This file was deleted.

15 changes: 11 additions & 4 deletions environment.yml
Original file line numberDiff line numberDiff line change
Expand Up@@ -2,9 +2,16 @@ channels:
- conda-forge
dependencies:
- python =3.12
- myqueue
- rich
- pip
- matplotlib
- myqueue =26.3.0
- rich =15.0.0
- texteditor =1.4.1
- pip =25.2
- ase=3.24.0
- matplotlib=3.10.1
- xmlschema=3.4.3
- optimade=1.2.3
- qe=7.5
- qe-tools=2.0.0
- conda_subprocess =0.0.11
- pip:
- git+https://gitlab.com/asm-dtu/perqueue.git
5 changes: 5 additions & 0 deletions nfdi/source/envs/postprocessing.yaml
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,5 @@
channels:
- conda-forge
dependencies:
- paraview=5.9.1=hfc1cbd4_3_egl
- tectonic=0.8.0=ha1fef3e_1
5 changes: 5 additions & 0 deletions nfdi/source/envs/preprocessing.yaml
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,5 @@
channels:
- conda-forge
dependencies:
- gmsh=4.6.0=hd134328_0
- meshio=5.0.5=pyhd8ed1ab_0
5 changes: 5 additions & 0 deletions nfdi/source/envs/processing.yaml
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,5 @@
channels:
- conda-forge
dependencies:
- fenics=2019.1.0=py39hf3d152e_26
- sympy=1.13.3
3 changes: 3 additions & 0 deletions nfdi/source/macros.tex.template
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,3 @@
\newcommand{\domainSize}{${DOMAINSIZE}}
\newcommand{\numDofs}{${NUMDOFS}}
\newcommand{\plotDataPath}{${PLOTDATAPATH}}
22 changes: 22 additions & 0 deletions nfdi/source/paper.tex
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,22 @@
\documentclass[12pt]{article}
\usepackage{graphicx}
\usepackage{pgfplots}

\input{macros}

\title{An exemplary workflow}

\begin{document}
\maketitle

\begin{figure}[h]
\centering%
\begin{tikzpicture}
\begin{axis}[]
\addplot[color=red,smooth,thick] table[col sep=comma,header=true,x index=2,y index=0] {\plotDataPath};
\end{axis}
\end{tikzpicture}
\caption{Solution of the poisson equation over the line $L = \{(x, y)\in\Omega = {(0, \domainSize{})}^2 | x = y\}$ using \numDofs{} DoFs in the finite element model.}%
\end{figure}

\end{document}
108 changes: 108 additions & 0 deletions nfdi/source/poisson.py
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,108 @@
"""
solution of the poisson equation on the unit square
"""

from argparse import ArgumentParser
import dolfin as df


def boundary_expression():
"""Defines the function to be used for the boundary conditions"""
return "1.0 + x[0] * x[0] + 2.0 * x[1] * x[1]"


def solve_poisson(
meshfile: str, degree: int, bc_expression: str = boundary_expression()
):
"""solves the poisson equation

Parameters
----------
meshfile : str
FilePath to the mesh in xdmf format.
degree : int
Degree of the finite element space.

Returns
-------
solution : df.Function
"""
mesh = df.Mesh()
with df.XDMFFile(meshfile) as instream:
instream.read(mesh)
func_space = df.FunctionSpace(mesh, "CG", degree)
boundary_data = df.Expression(bc_expression, degree=2)

def boundary(_, on_boundary):
return on_boundary

boundary_conditions = df.DirichletBC(func_space, boundary_data, boundary)
trial_function = df.TrialFunction(func_space)
test_function = df.TestFunction(func_space)
source = df.Constant(-6.0)
lhs = df.dot(df.grad(trial_function), df.grad(test_function)) * df.dx
rhs = source * test_function * df.dx

solution = df.Function(func_space)
df.solve(lhs == rhs, solution, boundary_conditions)
return solution


def solve_and_write_output(
mesh: str, degree: int, outputfile: str, numdofs=None, return_dofs=False
):
"""solves the poisson equation and writes the solution
and the number of degrees of freedom to the given file

Parameters
----------
meshfile : str
FilePath to the mesh in xdmf format.
degree : int
Degree of the finite element space.
outputfile : str
FilePath to the output file into which the solution is written.
numdofs : optional, str
FilePath to which the number of degrees of freedom is written.
return_dofs : optional, bool
If True, return number of degrees of freedom.
"""
discrete_solution = solve_poisson(mesh, degree)
discrete_solution.rename("u", discrete_solution.name())
resultFile = df.File(outputfile)
resultFile << discrete_solution

dofs = discrete_solution.function_space().dim()
print(f"Number of dofs used: {dofs}")

if numdofs is not None:
with open(numdofs, "w") as handle:
handle.write("{}\n".format(dofs))
if return_dofs:
return dofs


if __name__ == "__main__":
PARSER = ArgumentParser(description="run script for the poisson problem")
PARSER.add_argument("-m", "--mesh", required=True, help="mesh file to be used")
PARSER.add_argument(
"-d", "--degree", required=True, help="polynomial order to be used"
)
PARSER.add_argument(
"-o",
"--outputfile",
required=True,
help="file name for the output to be written",
)
PARSER.add_argument(
"-n",
"--num-dofs",
required=False,
default=None,
help="file name for the number of DoFs to be written",
)
ARGS = vars(PARSER.parse_args())

solve_and_write_output(
ARGS["mesh"], int(ARGS["degree"]), ARGS["outputfile"], ARGS["num_dofs"]
)
54 changes: 54 additions & 0 deletions nfdi/source/postprocessing.py
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,54 @@
"""
$ pvbatch postprocessing.py -h
"""

import sys
import argparse
from paraview.simple import (
PlotOverLine,
PVDReader,
SaveData,
UpdatePipeline,
)


def main(args):
pvd_file = args.pvd
source = PVDReader(registrationName="poisson.pvd", FileName=[pvd_file])
source.PointArrays = [args.field]
UpdatePipeline()

(xmin, xmax, ymin, ymax, zmin, zmax) = source.GetDataInformation().GetBounds()
# init the 'Line' selected for 'Source'
plotOverLine1 = PlotOverLine(
registrationName="PlotOverLine1", Input=source, Source="Line"
)
plotOverLine1.Source.Point1 = [xmin, ymin, zmin]
plotOverLine1.Source.Point2 = [xmax, ymax, zmax]
UpdatePipeline()

# save data
SaveData(
args.csv,
proxy=plotOverLine1,
ChooseArraysToWrite=1,
PointDataArrays=["arc_length", args.field, "vtkValidPointMask"],
)


if __name__ == "__main__":
parser = argparse.ArgumentParser(
prog=f"pvbatch {__file__}",
description="Plots the solution over a line and writes the data to file.",
usage="%(prog)s [options] pvd csv",
)
parser.add_argument("pvd", type=str, help="The source pvd filepath.")
parser.add_argument("csv", type=str, help="The target csv filepath.")
parser.add_argument(
"--field",
type=str,
default="u",
help="Field variable to plot (default: u)",
)
args = parser.parse_args(sys.argv[1:])
main(args)
43 changes: 43 additions & 0 deletions nfdi/source/prepare_paper_macros.py
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,43 @@
"""
places the correct values into the paper macros template file
"""

import string
import argparse

PARSER = argparse.ArgumentParser(
description="Place computed values in the paper macros template file"
)
PARSER.add_argument(
"-f", "--macro-template-file",
required=False, default="macros.tex.template",
help="The macro template file"
)
PARSER.add_argument(
"-o", "--output-macro-file",
required=False, default="macros.tex",
help="The macro file to be written"
)
PARSER.add_argument(
"-s", "--domain-size",
required=True, help="The computed domain size",
)
PARSER.add_argument(
"-n", "--num-dofs",
required=True, help="The computed number of dofs",
)
PARSER.add_argument(
"-p", "--plot-data-path",
required=True, help="The path to the data for the plot over line"
)
ARGS = vars(PARSER.parse_args())


with open(ARGS["output_macro_file"], "w") as out_file:
with open(ARGS["macro_template_file"], "r") as in_file:
raw = string.Template(in_file.read())
out_file.write(raw.substitute({
"DOMAINSIZE": ARGS["domain_size"],
"NUMDOFS": ARGS["num_dofs"],
"PLOTDATAPATH": ARGS["plot_data_path"]
}))
9 changes: 9 additions & 0 deletions nfdi/source/unit_square.geo
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,9 @@
DefineConstant[ domain_size = 1.0 ];
DefineConstant[ mesh_size = domain_size/4.0 ];

SetFactory("OpenCASCADE");
Rectangle(1) = {0, 0, 0, domain_size, domain_size, 0};
Physical Surface(1) = {1};

Printf("Used domain size: %f", domain_size);
Printf("Used mesh size: %f", mesh_size);
33 changes: 33 additions & 0 deletions nfdi/workflow.json
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,33 @@
{
"version": "0.1.0",
"nodes": [
{"id": 0, "type": "function", "value": "workflow.generate_mesh"},
{"id": 1, "type": "function", "value": "workflow.convert_to_xdmf"},
{"id": 2, "type": "function", "value": "workflow.poisson"},
{"id": 3, "type": "function", "value": "workflow.plot_over_line"},
{"id": 4, "type": "function", "value": "workflow.substitute_macros"},
{"id": 5, "type": "function", "value": "workflow.compile_paper"},
{"id": 6, "type": "input", "value": 2.0, "name": "domain_size"},
{"id": 7, "type": "input", "value": "source", "name": "source_directory"},
{"id": 8, "type": "output", "name": "result"}
],
"edges": [
{"target": 0, "targetPort": "domain_size", "source": 6, "sourcePort": null},
{"target": 0, "targetPort": "source_directory", "source": 7, "sourcePort": null},
{"target": 1, "targetPort": "gmsh_output_file", "source": 0, "sourcePort": null},
{"target": 2, "targetPort": "meshio_output_xdmf", "source": 1, "sourcePort": "xdmf_file"},
{"target": 2, "targetPort": "meshio_output_h5", "source": 1, "sourcePort": "h5_file"},
{"target": 2, "targetPort": "source_directory", "source": 7, "sourcePort": null},
{"target": 3, "targetPort": "poisson_output_pvd_file", "source": 2, "sourcePort": "pvd_file"},
{"target": 3, "targetPort": "poisson_output_vtu_file", "source": 2, "sourcePort": "vtu_file"},
{"target": 3, "targetPort": "source_directory", "source": 7, "sourcePort": null},
{"target": 4, "targetPort": "pvbatch_output_file", "source": 3, "sourcePort": null},
{"target": 4, "targetPort": "ndofs", "source": 2, "sourcePort": "numdofs"},
{"target": 4, "targetPort": "domain_size", "source": 6, "sourcePort": null},
{"target": 4, "targetPort": "source_directory", "source": 7, "sourcePort": null},
{"target": 5, "targetPort": "macros_tex", "source": 4, "sourcePort": null},
{"target": 5, "targetPort": "plot_file", "source": 3, "sourcePort": null},
{"target": 5, "targetPort": "source_directory", "source": 7, "sourcePort": null},
{"target": 8, "targetPort": null, "source": 5, "sourcePort": null}
]
}
Loading
, 'i'); if (__m === '*' || __re.test(location.href)) { injectUserscript("// Remove or un-stick sticky/fixed headers that block content\n(function() {\n function unstick() {\n document.querySelectorAll('header, nav, [role=\"banner\"], .header, .navbar, .sticky, .fixed-top, [style*=\"position: fixed\"], [style*=\"position:sticky\"]').forEach(function(el) {\n if (el.style.position === 'fixed' || el.style.position === 'sticky' || \n getComputedStyle(el).position === 'fixed' || getComputedStyle(el).position === 'sticky') {\n el.style.position = 'static';\n el.style.top = 'auto';\n el.style.zIndex = 'auto';\n }\n });\n }\n \n unstick();\n \n var observer = new MutationObserver(unstick);\n observer.observe(document.body, { childList: true, subtree: true, attributes: true, attributeFilter: ['style', 'class'] });\n})();", "Kill Sticky Headers"); } } catch(__e) { console.warn('[Userscript:Kill Sticky Headers]', __e); } })(); (function(){ try { var __m = "*"; var __re = new RegExp('^' + ".*" + '
Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 1 addition & 0 deletions arithmetic/workflow.ipynb

Large diffs are not rendered by default.

File renamed without changes.
File renamed without changes.
1 change: 0 additions & 1 deletion demo.ipynb

This file was deleted.

15 changes: 11 additions & 4 deletions environment.yml
Original file line numberDiff line numberDiff line change
Expand Up@@ -2,9 +2,16 @@ channels:
- conda-forge
dependencies:
- python =3.12
- myqueue
- rich
- pip
- matplotlib
- myqueue =26.3.0
- rich =15.0.0
- texteditor =1.4.1
- pip =25.2
- ase=3.24.0
- matplotlib=3.10.1
- xmlschema=3.4.3
- optimade=1.2.3
- qe=7.5
- qe-tools=2.0.0
- conda_subprocess =0.0.11
- pip:
- git+https://gitlab.com/asm-dtu/perqueue.git
5 changes: 5 additions & 0 deletions nfdi/source/envs/postprocessing.yaml
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,5 @@
channels:
- conda-forge
dependencies:
- paraview=5.9.1=hfc1cbd4_3_egl
- tectonic=0.8.0=ha1fef3e_1
5 changes: 5 additions & 0 deletions nfdi/source/envs/preprocessing.yaml
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,5 @@
channels:
- conda-forge
dependencies:
- gmsh=4.6.0=hd134328_0
- meshio=5.0.5=pyhd8ed1ab_0
5 changes: 5 additions & 0 deletions nfdi/source/envs/processing.yaml
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,5 @@
channels:
- conda-forge
dependencies:
- fenics=2019.1.0=py39hf3d152e_26
- sympy=1.13.3
3 changes: 3 additions & 0 deletions nfdi/source/macros.tex.template
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,3 @@
\newcommand{\domainSize}{${DOMAINSIZE}}
\newcommand{\numDofs}{${NUMDOFS}}
\newcommand{\plotDataPath}{${PLOTDATAPATH}}
22 changes: 22 additions & 0 deletions nfdi/source/paper.tex
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,22 @@
\documentclass[12pt]{article}
\usepackage{graphicx}
\usepackage{pgfplots}

\input{macros}

\title{An exemplary workflow}

\begin{document}
\maketitle

\begin{figure}[h]
\centering%
\begin{tikzpicture}
\begin{axis}[]
\addplot[color=red,smooth,thick] table[col sep=comma,header=true,x index=2,y index=0] {\plotDataPath};
\end{axis}
\end{tikzpicture}
\caption{Solution of the poisson equation over the line $L = \{(x, y)\in\Omega = {(0, \domainSize{})}^2 | x = y\}$ using \numDofs{} DoFs in the finite element model.}%
\end{figure}

\end{document}
108 changes: 108 additions & 0 deletions nfdi/source/poisson.py
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,108 @@
"""
solution of the poisson equation on the unit square
"""

from argparse import ArgumentParser
import dolfin as df


def boundary_expression():
"""Defines the function to be used for the boundary conditions"""
return "1.0 + x[0] * x[0] + 2.0 * x[1] * x[1]"


def solve_poisson(
meshfile: str, degree: int, bc_expression: str = boundary_expression()
):
"""solves the poisson equation

Parameters
----------
meshfile : str
FilePath to the mesh in xdmf format.
degree : int
Degree of the finite element space.

Returns
-------
solution : df.Function
"""
mesh = df.Mesh()
with df.XDMFFile(meshfile) as instream:
instream.read(mesh)
func_space = df.FunctionSpace(mesh, "CG", degree)
boundary_data = df.Expression(bc_expression, degree=2)

def boundary(_, on_boundary):
return on_boundary

boundary_conditions = df.DirichletBC(func_space, boundary_data, boundary)
trial_function = df.TrialFunction(func_space)
test_function = df.TestFunction(func_space)
source = df.Constant(-6.0)
lhs = df.dot(df.grad(trial_function), df.grad(test_function)) * df.dx
rhs = source * test_function * df.dx

solution = df.Function(func_space)
df.solve(lhs == rhs, solution, boundary_conditions)
return solution


def solve_and_write_output(
mesh: str, degree: int, outputfile: str, numdofs=None, return_dofs=False
):
"""solves the poisson equation and writes the solution
and the number of degrees of freedom to the given file

Parameters
----------
meshfile : str
FilePath to the mesh in xdmf format.
degree : int
Degree of the finite element space.
outputfile : str
FilePath to the output file into which the solution is written.
numdofs : optional, str
FilePath to which the number of degrees of freedom is written.
return_dofs : optional, bool
If True, return number of degrees of freedom.
"""
discrete_solution = solve_poisson(mesh, degree)
discrete_solution.rename("u", discrete_solution.name())
resultFile = df.File(outputfile)
resultFile << discrete_solution

dofs = discrete_solution.function_space().dim()
print(f"Number of dofs used: {dofs}")

if numdofs is not None:
with open(numdofs, "w") as handle:
handle.write("{}\n".format(dofs))
if return_dofs:
return dofs


if __name__ == "__main__":
PARSER = ArgumentParser(description="run script for the poisson problem")
PARSER.add_argument("-m", "--mesh", required=True, help="mesh file to be used")
PARSER.add_argument(
"-d", "--degree", required=True, help="polynomial order to be used"
)
PARSER.add_argument(
"-o",
"--outputfile",
required=True,
help="file name for the output to be written",
)
PARSER.add_argument(
"-n",
"--num-dofs",
required=False,
default=None,
help="file name for the number of DoFs to be written",
)
ARGS = vars(PARSER.parse_args())

solve_and_write_output(
ARGS["mesh"], int(ARGS["degree"]), ARGS["outputfile"], ARGS["num_dofs"]
)
54 changes: 54 additions & 0 deletions nfdi/source/postprocessing.py
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,54 @@
"""
$ pvbatch postprocessing.py -h
"""

import sys
import argparse
from paraview.simple import (
PlotOverLine,
PVDReader,
SaveData,
UpdatePipeline,
)


def main(args):
pvd_file = args.pvd
source = PVDReader(registrationName="poisson.pvd", FileName=[pvd_file])
source.PointArrays = [args.field]
UpdatePipeline()

(xmin, xmax, ymin, ymax, zmin, zmax) = source.GetDataInformation().GetBounds()
# init the 'Line' selected for 'Source'
plotOverLine1 = PlotOverLine(
registrationName="PlotOverLine1", Input=source, Source="Line"
)
plotOverLine1.Source.Point1 = [xmin, ymin, zmin]
plotOverLine1.Source.Point2 = [xmax, ymax, zmax]
UpdatePipeline()

# save data
SaveData(
args.csv,
proxy=plotOverLine1,
ChooseArraysToWrite=1,
PointDataArrays=["arc_length", args.field, "vtkValidPointMask"],
)


if __name__ == "__main__":
parser = argparse.ArgumentParser(
prog=f"pvbatch {__file__}",
description="Plots the solution over a line and writes the data to file.",
usage="%(prog)s [options] pvd csv",
)
parser.add_argument("pvd", type=str, help="The source pvd filepath.")
parser.add_argument("csv", type=str, help="The target csv filepath.")
parser.add_argument(
"--field",
type=str,
default="u",
help="Field variable to plot (default: u)",
)
args = parser.parse_args(sys.argv[1:])
main(args)
43 changes: 43 additions & 0 deletions nfdi/source/prepare_paper_macros.py
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,43 @@
"""
places the correct values into the paper macros template file
"""

import string
import argparse

PARSER = argparse.ArgumentParser(
description="Place computed values in the paper macros template file"
)
PARSER.add_argument(
"-f", "--macro-template-file",
required=False, default="macros.tex.template",
help="The macro template file"
)
PARSER.add_argument(
"-o", "--output-macro-file",
required=False, default="macros.tex",
help="The macro file to be written"
)
PARSER.add_argument(
"-s", "--domain-size",
required=True, help="The computed domain size",
)
PARSER.add_argument(
"-n", "--num-dofs",
required=True, help="The computed number of dofs",
)
PARSER.add_argument(
"-p", "--plot-data-path",
required=True, help="The path to the data for the plot over line"
)
ARGS = vars(PARSER.parse_args())


with open(ARGS["output_macro_file"], "w") as out_file:
with open(ARGS["macro_template_file"], "r") as in_file:
raw = string.Template(in_file.read())
out_file.write(raw.substitute({
"DOMAINSIZE": ARGS["domain_size"],
"NUMDOFS": ARGS["num_dofs"],
"PLOTDATAPATH": ARGS["plot_data_path"]
}))
9 changes: 9 additions & 0 deletions nfdi/source/unit_square.geo
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,9 @@
DefineConstant[ domain_size = 1.0 ];
DefineConstant[ mesh_size = domain_size/4.0 ];

SetFactory("OpenCASCADE");
Rectangle(1) = {0, 0, 0, domain_size, domain_size, 0};
Physical Surface(1) = {1};

Printf("Used domain size: %f", domain_size);
Printf("Used mesh size: %f", mesh_size);
33 changes: 33 additions & 0 deletions nfdi/workflow.json
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,33 @@
{
"version": "0.1.0",
"nodes": [
{"id": 0, "type": "function", "value": "workflow.generate_mesh"},
{"id": 1, "type": "function", "value": "workflow.convert_to_xdmf"},
{"id": 2, "type": "function", "value": "workflow.poisson"},
{"id": 3, "type": "function", "value": "workflow.plot_over_line"},
{"id": 4, "type": "function", "value": "workflow.substitute_macros"},
{"id": 5, "type": "function", "value": "workflow.compile_paper"},
{"id": 6, "type": "input", "value": 2.0, "name": "domain_size"},
{"id": 7, "type": "input", "value": "source", "name": "source_directory"},
{"id": 8, "type": "output", "name": "result"}
],
"edges": [
{"target": 0, "targetPort": "domain_size", "source": 6, "sourcePort": null},
{"target": 0, "targetPort": "source_directory", "source": 7, "sourcePort": null},
{"target": 1, "targetPort": "gmsh_output_file", "source": 0, "sourcePort": null},
{"target": 2, "targetPort": "meshio_output_xdmf", "source": 1, "sourcePort": "xdmf_file"},
{"target": 2, "targetPort": "meshio_output_h5", "source": 1, "sourcePort": "h5_file"},
{"target": 2, "targetPort": "source_directory", "source": 7, "sourcePort": null},
{"target": 3, "targetPort": "poisson_output_pvd_file", "source": 2, "sourcePort": "pvd_file"},
{"target": 3, "targetPort": "poisson_output_vtu_file", "source": 2, "sourcePort": "vtu_file"},
{"target": 3, "targetPort": "source_directory", "source": 7, "sourcePort": null},
{"target": 4, "targetPort": "pvbatch_output_file", "source": 3, "sourcePort": null},
{"target": 4, "targetPort": "ndofs", "source": 2, "sourcePort": "numdofs"},
{"target": 4, "targetPort": "domain_size", "source": 6, "sourcePort": null},
{"target": 4, "targetPort": "source_directory", "source": 7, "sourcePort": null},
{"target": 5, "targetPort": "macros_tex", "source": 4, "sourcePort": null},
{"target": 5, "targetPort": "plot_file", "source": 3, "sourcePort": null},
{"target": 5, "targetPort": "source_directory", "source": 7, "sourcePort": null},
{"target": 8, "targetPort": null, "source": 5, "sourcePort": null}
]
}
Loading
, 'i'); if (__m === '*' || __re.test(location.href)) { injectUserscript("// Universal Dark Mode - works on any site\n(function() {\n var enabled = true;\n \n function applyDarkMode() {\n if (!enabled) return;\n \n // Create style element if it doesn't exist\n var style = document.getElementById('universal-dark-mode-style');\n if (!style) {\n style = document.createElement('style');\n style.id = 'universal-dark-mode-style';\n document.head.appendChild(style);\n }\n \n // Dark mode CSS - inverts colors but preserves images/video\n style.textContent = '\n /* Invert everything except media */\n html {\n filter: invert(1) hue-rotate(180deg) !important;\n background: #1a1a2e !important;\n }\n \n /* Restore images, videos, iframes, canvas */\n img, video, iframe, canvas, svg, picture, [style*=\"background-image\"] {\n filter: invert(1) hue-rotate(180deg) !important;\n }\n \n /* Preserve specific elements that should not be inverted */\n .no-dark-mode, .no-dark-mode *,\n [data-theme=\"light\"], [data-theme=\"light\"],\n .ace_editor, .ace_editor *,\n .CodeMirror, .CodeMirror *,\n .monaco-editor, .monaco-editor *,\n .markdown-body pre, .markdown-body pre *,\n .highlight, .highlight *,\n pre code, pre code * {\n filter: none !important;\n }\n \n /* Fix common UI elements */\n .modal, .popup, .dropdown-menu, .tooltip, .popover {\n filter: invert(1) hue-rotate(180deg) !important;\n background: #2d2d44 !important;\n border-color: #444 !important;\n }\n \n /* Scrollbars */\n ::-webkit-scrollbar { background: #1a1a2e !important; }\n ::-webkit-scrollbar-thumb { background: #444 !important; }\n ::-webkit-scrollbar-thumb:hover { background: #555 !important; }\n \n /* Selection */\n ::selection { background: #4ecdc4 !important; color: #1a1a2e !important; }\n ::-moz-selection { background: #4ecdc4 !important; color: #1a1a2e !important; }\n ';\n }\n \n function removeDarkMode() {\n var style = document.getElementById('universal-dark-mode-style');\n if (style) style.remove();\n }\n \n // Toggle with Alt+Shift+D\n document.addEventListener('keydown', function(e) {\n if (e.altKey && e.shiftKey && e.key === 'D') {\n e.preventDefault();\n enabled = !enabled;\n if (enabled) {\n applyDarkMode();\n console.log('[Universal Dark Mode] Enabled');\n } else {\n removeDarkMode();\n console.log('[Universal Dark Mode] Disabled');\n }\n }\n });\n \n // Apply on load\n applyDarkMode();\n \n // Re-apply on dynamic content\n var observer = new MutationObserver(function(mutations) {\n if (enabled && !document.getElementById('universal-dark-mode-style')) {\n applyDarkMode();\n }\n });\n observer.observe(document.head, { childList: true });\n \n console.log('[Universal Dark Mode] Loaded - Press Alt+Shift+D to toggle');\n})();", "Universal Dark Mode"); } } catch(__e) { console.warn('[Userscript:Universal Dark Mode]', __e); } })(); })();
Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 1 addition & 0 deletions arithmetic/workflow.ipynb

Large diffs are not rendered by default.

File renamed without changes.
File renamed without changes.
1 change: 0 additions & 1 deletion demo.ipynb

This file was deleted.

15 changes: 11 additions & 4 deletions environment.yml
Original file line numberDiff line numberDiff line change
Expand Up@@ -2,9 +2,16 @@ channels:
- conda-forge
dependencies:
- python =3.12
- myqueue
- rich
- pip
- matplotlib
- myqueue =26.3.0
- rich =15.0.0
- texteditor =1.4.1
- pip =25.2
- ase=3.24.0
- matplotlib=3.10.1
- xmlschema=3.4.3
- optimade=1.2.3
- qe=7.5
- qe-tools=2.0.0
- conda_subprocess =0.0.11
- pip:
- git+https://gitlab.com/asm-dtu/perqueue.git
5 changes: 5 additions & 0 deletions nfdi/source/envs/postprocessing.yaml
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,5 @@
channels:
- conda-forge
dependencies:
- paraview=5.9.1=hfc1cbd4_3_egl
- tectonic=0.8.0=ha1fef3e_1
5 changes: 5 additions & 0 deletions nfdi/source/envs/preprocessing.yaml
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,5 @@
channels:
- conda-forge
dependencies:
- gmsh=4.6.0=hd134328_0
- meshio=5.0.5=pyhd8ed1ab_0
5 changes: 5 additions & 0 deletions nfdi/source/envs/processing.yaml
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,5 @@
channels:
- conda-forge
dependencies:
- fenics=2019.1.0=py39hf3d152e_26
- sympy=1.13.3
3 changes: 3 additions & 0 deletions nfdi/source/macros.tex.template
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,3 @@
\newcommand{\domainSize}{${DOMAINSIZE}}
\newcommand{\numDofs}{${NUMDOFS}}
\newcommand{\plotDataPath}{${PLOTDATAPATH}}
22 changes: 22 additions & 0 deletions nfdi/source/paper.tex
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,22 @@
\documentclass[12pt]{article}
\usepackage{graphicx}
\usepackage{pgfplots}

\input{macros}

\title{An exemplary workflow}

\begin{document}
\maketitle

\begin{figure}[h]
\centering%
\begin{tikzpicture}
\begin{axis}[]
\addplot[color=red,smooth,thick] table[col sep=comma,header=true,x index=2,y index=0] {\plotDataPath};
\end{axis}
\end{tikzpicture}
\caption{Solution of the poisson equation over the line $L = \{(x, y)\in\Omega = {(0, \domainSize{})}^2 | x = y\}$ using \numDofs{} DoFs in the finite element model.}%
\end{figure}

\end{document}
108 changes: 108 additions & 0 deletions nfdi/source/poisson.py
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,108 @@
"""
solution of the poisson equation on the unit square
"""

from argparse import ArgumentParser
import dolfin as df


def boundary_expression():
"""Defines the function to be used for the boundary conditions"""
return "1.0 + x[0] * x[0] + 2.0 * x[1] * x[1]"


def solve_poisson(
meshfile: str, degree: int, bc_expression: str = boundary_expression()
):
"""solves the poisson equation

Parameters
----------
meshfile : str
FilePath to the mesh in xdmf format.
degree : int
Degree of the finite element space.

Returns
-------
solution : df.Function
"""
mesh = df.Mesh()
with df.XDMFFile(meshfile) as instream:
instream.read(mesh)
func_space = df.FunctionSpace(mesh, "CG", degree)
boundary_data = df.Expression(bc_expression, degree=2)

def boundary(_, on_boundary):
return on_boundary

boundary_conditions = df.DirichletBC(func_space, boundary_data, boundary)
trial_function = df.TrialFunction(func_space)
test_function = df.TestFunction(func_space)
source = df.Constant(-6.0)
lhs = df.dot(df.grad(trial_function), df.grad(test_function)) * df.dx
rhs = source * test_function * df.dx

solution = df.Function(func_space)
df.solve(lhs == rhs, solution, boundary_conditions)
return solution


def solve_and_write_output(
mesh: str, degree: int, outputfile: str, numdofs=None, return_dofs=False
):
"""solves the poisson equation and writes the solution
and the number of degrees of freedom to the given file

Parameters
----------
meshfile : str
FilePath to the mesh in xdmf format.
degree : int
Degree of the finite element space.
outputfile : str
FilePath to the output file into which the solution is written.
numdofs : optional, str
FilePath to which the number of degrees of freedom is written.
return_dofs : optional, bool
If True, return number of degrees of freedom.
"""
discrete_solution = solve_poisson(mesh, degree)
discrete_solution.rename("u", discrete_solution.name())
resultFile = df.File(outputfile)
resultFile << discrete_solution

dofs = discrete_solution.function_space().dim()
print(f"Number of dofs used: {dofs}")

if numdofs is not None:
with open(numdofs, "w") as handle:
handle.write("{}\n".format(dofs))
if return_dofs:
return dofs


if __name__ == "__main__":
PARSER = ArgumentParser(description="run script for the poisson problem")
PARSER.add_argument("-m", "--mesh", required=True, help="mesh file to be used")
PARSER.add_argument(
"-d", "--degree", required=True, help="polynomial order to be used"
)
PARSER.add_argument(
"-o",
"--outputfile",
required=True,
help="file name for the output to be written",
)
PARSER.add_argument(
"-n",
"--num-dofs",
required=False,
default=None,
help="file name for the number of DoFs to be written",
)
ARGS = vars(PARSER.parse_args())

solve_and_write_output(
ARGS["mesh"], int(ARGS["degree"]), ARGS["outputfile"], ARGS["num_dofs"]
)
54 changes: 54 additions & 0 deletions nfdi/source/postprocessing.py
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,54 @@
"""
$ pvbatch postprocessing.py -h
"""

import sys
import argparse
from paraview.simple import (
PlotOverLine,
PVDReader,
SaveData,
UpdatePipeline,
)


def main(args):
pvd_file = args.pvd
source = PVDReader(registrationName="poisson.pvd", FileName=[pvd_file])
source.PointArrays = [args.field]
UpdatePipeline()

(xmin, xmax, ymin, ymax, zmin, zmax) = source.GetDataInformation().GetBounds()
# init the 'Line' selected for 'Source'
plotOverLine1 = PlotOverLine(
registrationName="PlotOverLine1", Input=source, Source="Line"
)
plotOverLine1.Source.Point1 = [xmin, ymin, zmin]
plotOverLine1.Source.Point2 = [xmax, ymax, zmax]
UpdatePipeline()

# save data
SaveData(
args.csv,
proxy=plotOverLine1,
ChooseArraysToWrite=1,
PointDataArrays=["arc_length", args.field, "vtkValidPointMask"],
)


if __name__ == "__main__":
parser = argparse.ArgumentParser(
prog=f"pvbatch {__file__}",
description="Plots the solution over a line and writes the data to file.",
usage="%(prog)s [options] pvd csv",
)
parser.add_argument("pvd", type=str, help="The source pvd filepath.")
parser.add_argument("csv", type=str, help="The target csv filepath.")
parser.add_argument(
"--field",
type=str,
default="u",
help="Field variable to plot (default: u)",
)
args = parser.parse_args(sys.argv[1:])
main(args)
43 changes: 43 additions & 0 deletions nfdi/source/prepare_paper_macros.py
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,43 @@
"""
places the correct values into the paper macros template file
"""

import string
import argparse

PARSER = argparse.ArgumentParser(
description="Place computed values in the paper macros template file"
)
PARSER.add_argument(
"-f", "--macro-template-file",
required=False, default="macros.tex.template",
help="The macro template file"
)
PARSER.add_argument(
"-o", "--output-macro-file",
required=False, default="macros.tex",
help="The macro file to be written"
)
PARSER.add_argument(
"-s", "--domain-size",
required=True, help="The computed domain size",
)
PARSER.add_argument(
"-n", "--num-dofs",
required=True, help="The computed number of dofs",
)
PARSER.add_argument(
"-p", "--plot-data-path",
required=True, help="The path to the data for the plot over line"
)
ARGS = vars(PARSER.parse_args())


with open(ARGS["output_macro_file"], "w") as out_file:
with open(ARGS["macro_template_file"], "r") as in_file:
raw = string.Template(in_file.read())
out_file.write(raw.substitute({
"DOMAINSIZE": ARGS["domain_size"],
"NUMDOFS": ARGS["num_dofs"],
"PLOTDATAPATH": ARGS["plot_data_path"]
}))
9 changes: 9 additions & 0 deletions nfdi/source/unit_square.geo
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,9 @@
DefineConstant[ domain_size = 1.0 ];
DefineConstant[ mesh_size = domain_size/4.0 ];

SetFactory("OpenCASCADE");
Rectangle(1) = {0, 0, 0, domain_size, domain_size, 0};
Physical Surface(1) = {1};

Printf("Used domain size: %f", domain_size);
Printf("Used mesh size: %f", mesh_size);
33 changes: 33 additions & 0 deletions nfdi/workflow.json
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,33 @@
{
"version": "0.1.0",
"nodes": [
{"id": 0, "type": "function", "value": "workflow.generate_mesh"},
{"id": 1, "type": "function", "value": "workflow.convert_to_xdmf"},
{"id": 2, "type": "function", "value": "workflow.poisson"},
{"id": 3, "type": "function", "value": "workflow.plot_over_line"},
{"id": 4, "type": "function", "value": "workflow.substitute_macros"},
{"id": 5, "type": "function", "value": "workflow.compile_paper"},
{"id": 6, "type": "input", "value": 2.0, "name": "domain_size"},
{"id": 7, "type": "input", "value": "source", "name": "source_directory"},
{"id": 8, "type": "output", "name": "result"}
],
"edges": [
{"target": 0, "targetPort": "domain_size", "source": 6, "sourcePort": null},
{"target": 0, "targetPort": "source_directory", "source": 7, "sourcePort": null},
{"target": 1, "targetPort": "gmsh_output_file", "source": 0, "sourcePort": null},
{"target": 2, "targetPort": "meshio_output_xdmf", "source": 1, "sourcePort": "xdmf_file"},
{"target": 2, "targetPort": "meshio_output_h5", "source": 1, "sourcePort": "h5_file"},
{"target": 2, "targetPort": "source_directory", "source": 7, "sourcePort": null},
{"target": 3, "targetPort": "poisson_output_pvd_file", "source": 2, "sourcePort": "pvd_file"},
{"target": 3, "targetPort": "poisson_output_vtu_file", "source": 2, "sourcePort": "vtu_file"},
{"target": 3, "targetPort": "source_directory", "source": 7, "sourcePort": null},
{"target": 4, "targetPort": "pvbatch_output_file", "source": 3, "sourcePort": null},
{"target": 4, "targetPort": "ndofs", "source": 2, "sourcePort": "numdofs"},
{"target": 4, "targetPort": "domain_size", "source": 6, "sourcePort": null},
{"target": 4, "targetPort": "source_directory", "source": 7, "sourcePort": null},
{"target": 5, "targetPort": "macros_tex", "source": 4, "sourcePort": null},
{"target": 5, "targetPort": "plot_file", "source": 3, "sourcePort": null},
{"target": 5, "targetPort": "source_directory", "source": 7, "sourcePort": null},
{"target": 8, "targetPort": null, "source": 5, "sourcePort": null}
]
}
Loading