Merge pull request #11 from kthyng/add_plotting

added plotting code and made cmocean optional

Author: kthyng

particle_tracking_manager/__init__.py

@@ -1,9 +1,4 @@
 """Particle Tracking Manager."""
 
-import cmocean
-
 from .models.opendrift.model_opendrift import OpenDriftModel
 from .the_manager import ParticleTrackingManager
-
-
-cmap = cmocean.tools.crop_by_percent(cmocean.cm.amp, 20, which="max", N=None)

particle_tracking_manager/plotting.py

@@ -0,0 +1,27 @@
+"""Some extra plotting code."""
+
+import opendrift
+
+
+def plot_dest(o):
+    """This is copied from an opendrift example."""
+
+    import cmocean
+
+    cmap = cmocean.tools.crop_by_percent(cmocean.cm.amp, 20, which="max", N=None)
+
+    od = opendrift.open_xarray(o.outfile_name)
+    density = od.get_histogram(pixelsize_m=5000).isel(time=-1).isel(origin_marker=0)
+    density = density.where(density > 0)
+    density = density / density.sum() * 100
+    vmax = density.max()
+
+    return o.plot(
+        background=density,
+        clabel="Probability of final location [%]",
+        markersize=0.5,
+        lalpha=0.02,
+        vmin=0,
+        vmax=vmax,
+        cmap=cmap,
+    )