Add EcutTestPlot

MineralsCloud · Nov 2, 2023 · 53b283e · 53b283e
1 parent 9b41a9f
commit 53b283e
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/src/outputs.jl b/src/outputs.jl
@@ -1,6 +1,21 @@
 using DataFrames: groupby
 using QuantumESPRESSO.PWscf
 
+@userplot EcutTestPlot
+@recipe function f(plot::EcutTestPlot)
+    framestyle --> :box
+    legend_foreground_color --> nothing
+    grid --> false
+    seriestype --> :scatter
+    total_energies = last(plot.args)
+    cutoff_energies = length(plot.args) == 2 ? first(plot.args) : eachindex(total_energies)
+    if length(plot.args)
+        xguide --> "cutoff energy"
+    end
+    yguide --> "total energy"
+    return cutoff_energies, total_energies
+end
+
 @userplot EnergyConvergenceIterationPlot
 @recipe function f(plot::EnergyConvergenceIterationPlot)
     framestyle --> :box