Add options for more plots

opendrift_leeway_webgui/leeway/forms.py

@@ -35,6 +35,10 @@ class Meta:
             "start_time",
             "duration",
             "radius",
+            "send_trajectories",
+            "send_heatmap",
+            "send_animation_wind",
+            "send_animation_sea",
         ]
         help_texts = {
             "duration": "Length of simulation in hours.",

opendrift_leeway_webgui/leeway/migrations/0009_leewaysimulation_send_animation_sea_and_more.py

@@ -0,0 +1,42 @@
+# Generated by Django 4.2.7 on 2023-12-03 14:59
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+    dependencies = [("leeway", "0008_leewaysimulation_traceback")]
+
+    operations = [
+        migrations.AddField(
+            model_name="leewaysimulation",
+            name="send_animation_sea",
+            field=models.BooleanField(
+                default=False,
+                help_text="Attach Gif to e-mail with animation of particles with sea currents.",
+            ),
+        ),
+        migrations.AddField(
+            model_name="leewaysimulation",
+            name="send_animation_wind",
+            field=models.BooleanField(
+                default=False,
+                help_text="Attach Gif to e-mail with animation of particles and wind fields.",
+            ),
+        ),
+        migrations.AddField(
+            model_name="leewaysimulation",
+            name="send_heatmap",
+            field=models.BooleanField(
+                default=False,
+                help_text="Attach PNG to e-mail with heat map of final particle locations.",
+            ),
+        ),
+        migrations.AddField(
+            model_name="leewaysimulation",
+            name="send_trajectories",
+            field=models.BooleanField(
+                default=True,
+                help_text="Attach PNG to e-mail with particle trajectories.",
+            ),
+        ),
+    ]

opendrift_leeway_webgui/leeway/models.py

@@ -35,6 +35,18 @@ class LeewaySimulation(models.Model):
     simulation_started = models.DateTimeField(null=True)
     simulation_finished = models.DateTimeField(null=True)
     radius = models.IntegerField(default=1000)
+    send_trajectories = models.BooleanField(
+        default=True,
+        help_text="Attach PNG to e-mail with particle trajectories.")
+    send_heatmap = models.BooleanField(
+        default=False,
+        help_text="Attach PNG to e-mail with heat map of final particle locations.")
+    send_animation_wind = models.BooleanField(
+        default=False,
+        help_text="Attach Gif to e-mail with animation of particles and wind fields.")
+    send_animation_sea = models.BooleanField(
+        default=False,
+        help_text="Attach Gif to e-mail with animation of particles with sea currents.")
     img = models.FileField(
         null=True, storage=simulation_storage, verbose_name=_("Image file")
     )

opendrift_leeway_webgui/simulation.py

@@ -107,9 +107,18 @@ def main():
     simulation.run(
         duration=timedelta(hours=args.duration), time_step=600, outfile=f"{outfile}.nc"
     )
+
+
     simulation.plot(
         fast=True, legend=True, filename=f"{outfile}.png", linecolor="age_seconds"
     )
+
+    simulation.plot(
+        background=simulation.get_density_array(pixelsize_m=3000),
+        clabel='Probability of origin [%]', fast=True, markersize=.5, lalpha=.02,
+        outfile=f"{outfile}-density.png"
+    )
+
     print(f"Success: {outfile}.png written.")