Added save plot ability

markowkes · Apr 19, 2024 · 8cbd7a2 · 8cbd7a2 · markowkes · Apr 26, 2024
1 parent a1e18e6
commit 8cbd7a2
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 4 deletions.
diff --git a/Project.toml b/Project.toml
@@ -1,7 +1,7 @@
 name = "Biofilm"
 uuid = "210d8ceb-62d9-4e79-bbbd-4c6870635f0a"
 authors = ["Mark Owkes <[email protected]> and contributors"]
-version = "1.0.3"
+version = "1.1.0"
 
 [deps]
 Accessors = "7d9f7c33-5ae7-4f3b-8dc6-eff91059b697"

diff --git a/src/outputs.jl b/src/outputs.jl
@@ -35,12 +35,21 @@ function outputs(integrator)
         printBiofilmValues(t[end],Xt[:,end],St[:,end],Pb,Sb,Lf[end],p)
 
         # Plot results
-        if p.makePlots && (mod(sol.t[end],plotPeriod)≈0.0 || mod(sol.t[end],plotPeriod)≈plotPeriod)
-            plt = biofilm_plot(sol,p)
-            display(plt)
+        if (mod(sol.t[end],plotPeriod)≈0.0 || mod(sol.t[end],plotPeriod)≈plotPeriod)
+            if p.makePlots || p.savePlots
+                plt = biofilm_plot(sol,p)
+            end
+            if p.makePlots
+                display(plt)
+            end
+            if p.savePlots
+                ispath("savePlots") || mkpath("savePlots")
+                savefig(String("savePlots/plot_$(integrator.t).png"))
+            end
         end
 
     end
+    return
 end 
 
 """

diff --git a/src/parameters.jl b/src/parameters.jl
@@ -41,6 +41,7 @@ function checkTypes_setDefs(d; verbose=false)
     d,err = checkType_setDef(err,d,Float64,              :tol                                        )
     d,err = checkType_setDef(err,d,Float64,              :plotPeriod,         default=d[:outPeriod]  )
     d,err = checkType_setDef(err,d,Bool,                 :makePlots,          default=true           )
+    d,err = checkType_setDef(err,d,Bool,                 :savePlots,          default=false          )
     d,err = checkType_setDef(err,d,Float64,              :discontinuityPeriod,default=d.tFinal       )
     d,err = checkType_setDef(err,d,String,               :optionalPlot,       default="growthrate"   )
     d,err = checkType_setDef(err,d,Tuple{Int64, Int64},  :plotSize,           default=(1600,1000)    )