Plot catalogs into a spatial map (#64)

* - catalogs.py

* - Fixed docstring of several plot functions
- Added the option for plot_basemap, plot_catalog and plot_spatial_dataset to include own webservice link for raster image  as argument.

* Modifications in plot_catalog
- Added simple functionality to scale scatter size by magnitudes exponentially, with a exponent base defined by user. Should work in all of the geographic extents.
- Added wrapper for a decent Legend of magnitude size. Has functionality for the ticks wanted, legend location and the spacing between ticks.
- Function was tested on California catalog, Europe catalog and Global catalog is pending still..

* - Finished docstrings of plot_catalog()
- Fixed size_map function for scaling size with magnitude. Now works fine
- Removed plot_catalog from tutorials
- Added extra example in tutorials/plot_customizations,  to get and plot a catalog, and customize it.

Author: pabloitu

csep/core/catalogs.py

@@ -22,7 +22,7 @@
 from csep.utils.log import LoggingMixin
 from csep.utils.readers import csep_ascii
 from csep.utils.file import get_file_extension
-
+from csep.utils.plots import plot_catalog
 
 class AbstractBaseCatalog(LoggingMixin):
     """
@@ -453,6 +453,19 @@ def get_longitudes(self):
         """
         return self.catalog['longitude']
 
+    def get_bbox(self):
+        """
+        Returns bounding box of all events in the catalog
+
+        Returns:
+            (numpy.array): [lon_min, lon_max, lat_min, lat_max]
+        """
+        bbox = numpy.array([numpy.min(self.catalog['longitude']),
+                            numpy.max(self.catalog['longitude']),
+                            numpy.min(self.catalog['latitude']),
+                            numpy.max(self.catalog['latitude'])])
+        return bbox
+
     def get_depths(self):
         """ Returns depths of all events in catalog """
         return self.catalog['depth']
@@ -818,6 +831,29 @@ def p():
         else:
             return bval
 
+    def plot(self, ax=None, show=False, extent=None, set_global=False, plot_args=None):
+        """ Plot catalog according to plate-carree projection
+
+        Args:
+            ax (`matplotlib.pyplot.axes`): Previous axes onto which catalog can be drawn
+            show (bool): if true, show the figure. this call is blocking.
+            extent (list): Force an extent [lon_min, lon_max, lat_min, lat_max]
+            plot_args (optional/dict): dictionary containing plotting arguments for making figures
+
+        Returns:
+            axes: matplotlib.Axes.axes
+        """
+        # no mutable function arguments
+
+
+        plot_args = plot_args or {}
+        plot_args.setdefault('figsize', (10, 10))
+        plot_args.setdefault('title', self.name)
+
+        # this call requires internet connection and basemap
+        ax = plot_catalog(self, ax=ax,show=show, extent=extent,
+                          set_global=set_global, plot_args=plot_args)
+        return ax
 
 class CSEPCatalog(AbstractBaseCatalog):
     """

csep/core/forecasts.py

@@ -407,8 +407,8 @@ def load_ascii(cls, ascii_fname, start_date=None, end_date=None, name=None):
         gds = cls(start_date, end_date, magnitudes=mws, name=name, region=region, data=rates)
         return gds
 
-    def plot(self, show=False, log=True, extent=None, set_global=False, plot_args=None):
-        """ Plot gridded forecast according to plate-carree proejction
+    def plot(self, ax=None, show=False, log=True, extent=None, set_global=False, plot_args=None):
+        """ Plot gridded forecast according to plate-carree projection
 
         Args:
             show (bool): if true, show the figure. this call is blocking.
@@ -434,11 +434,11 @@ def plot(self, show=False, log=True, extent=None, set_global=False, plot_args=No
         if log:
             plot_args.setdefault('clabel', f'log10 M{self.min_magnitude}+ rate per {str(dh)}° x {str(dh)}° per {time}')
             with numpy.errstate(divide='ignore'):
-                ax = plot_spatial_dataset(numpy.log10(self.spatial_counts(cartesian=True)), self.region, show=show,
-                                          extent=extent, set_global=set_global, plot_args=plot_args)
+                ax = plot_spatial_dataset(numpy.log10(self.spatial_counts(cartesian=True)), self.region, ax=ax,
+                                          show=show, extent=extent, set_global=set_global, plot_args=plot_args)
         else:
             plot_args.setdefault('clabel', f'M{self.min_magnitude}+ rate per {str(dh)}° x {str(dh)}° per {time}')
-            ax = plot_spatial_dataset(self.spatial_counts(cartesian=True), self.region, show=show, extent=extent,
+            ax = plot_spatial_dataset(self.spatial_counts(cartesian=True), self.region, ax=ax,show=show, extent=extent,
                                       set_global=set_global, plot_args=plot_args)
         return ax