Documentation

climetlab/core/metadata.py

@@ -0,0 +1,46 @@
+# (C) Copyright 2020 ECMWF.
+#
+# This software is licensed under the terms of the Apache Licence Version 2.0
+# which can be obtained at http://www.apache.org/licenses/LICENSE-2.0.
+# In applying this licence, ECMWF does not waive the privileges and immunities
+# granted to it by virtue of its status as an intergovernmental organisation
+# nor does it submit to any jurisdiction.
+#
+
+import json
+
+
+class Annotation:
+    def __init__(self, **kwargs):
+        self._kwargs = kwargs
+
+    def get(self, name):
+        return self._kwargs.get(name)
+
+
+def annotate(pd, **kargs):
+
+    kargs["climetlab"] = "annotation"
+
+    for i, a in enumerate(pd._metadata):
+        try:
+            a = json.loads(a)
+            if a["climetlab"] == "annotation":
+                pd._metadata[i] = json.dumps(kargs)
+                return pd
+        except Exception:
+            pass
+
+    pd._metadata.append(json.dumps(kargs))
+    return pd
+
+
+def annotation(pd):
+    for a in pd._metadata:
+        try:
+            a = json.loads(a)
+            if a["climetlab"] == "annotation":
+                return Annotation(**a)
+        except Exception:
+            pass
+    return Annotation()

climetlab/datasets/hurricane_database.py

@@ -1,11 +1,20 @@
+# (C) Copyright 2020 ECMWF.
+#
+# This software is licensed under the terms of the Apache Licence Version 2.0
+# which can be obtained at http://www.apache.org/licenses/LICENSE-2.0.
+# In applying this licence, ECMWF does not waive the privileges and immunities
+# granted to it by virtue of its status as an intergovernmental organisation
+# nor does it submit to any jurisdiction.
+#
+
 # See https://www.aoml.noaa.gov/hrd/hurdat/Data_Storm.html
 
 import pandas as pd
 import numpy as np
 from . import Dataset
 from climetlab.utils import download_and_cache
 from climetlab.utils.datetime import parse_date
-
+from climetlab.core.metadata import annotate
 
 SIGN = {"N": 1, "W": -1, "E": 1, "S": -1}
 
@@ -112,7 +121,7 @@ def __init__(self, bassin="atlantic", url=None):
                         )
                     )
 
-        self.cyclones = pd.DataFrame(p)
+        self.cyclones = annotate(pd.DataFrame(p), style="cyclone-track")
 
     def to_pandas(self):
         return self.cyclones

climetlab/plotting/drivers/magics.py

@@ -22,6 +22,7 @@
 from climetlab.core.caching import temp_file
 from climetlab.core.ipython import SVG, Image
 from climetlab.core.data import get_data_entry
+from climetlab.core.metadata import annotation
 
 LOG = logging.getLogger(__name__)
 
@@ -321,6 +322,12 @@ def plot_pandas(self, frame, latitude: str, longitude: str, variable: str):
         frame[[latitude, longitude, variable]].to_csv(tmp, header=False, index=False)
         self.plot_csv(tmp, variable)
 
+        print(frame._metadata)
+
+        style = annotation(frame).get("style")
+        if style is not None:
+            self.style(style)
+
     def _apply(self, collection, value, action):
 
         if value is None: