added gcmt accessor from ISC API. Follows the same logic than comcat and bsi readers. Created test for an event search and summary info from a request.

Author: pabloitu

csep/__init__.py

@@ -61,7 +61,9 @@
     '__version__'
 ]
 
-def load_stochastic_event_sets(filename, type='csv', format='native', **kwargs):
+
+def load_stochastic_event_sets(filename, type='csv', format='native',
+                               **kwargs):
     """ General function to load stochastic event sets
 
     This function returns a generator to iterate through a collection of catalogs.
@@ -106,7 +108,8 @@ def load_stochastic_event_sets(filename, type='csv', format='native', **kwargs):
             raise ValueError('format must be either "native" or "csep!')
 
 
-def load_catalog(filename, type='csep-csv', format='native', loader=None, apply_filters=False, **kwargs):
+def load_catalog(filename, type='csep-csv', format='native', loader=None,
+                 apply_filters=False, **kwargs):
     """ General function to load single catalog
 
     See corresponding class documentation for additional parameters.
@@ -122,9 +125,12 @@ def load_catalog(filename, type='csep-csv', format='native', loader=None, apply_
     Returns (:class:`~csep.core.catalogs.AbstractBaseCatalog`)
     """
 
-
-    if type not in ('ucerf3', 'csep-csv', 'zmap', 'jma-csv', 'ingv_horus', 'ingv_emrcmt', 'ndk') and loader is None:
-        raise ValueError("type must be one of the following: ('ucerf3', 'csep-csv', 'zmap', 'jma-csv', 'ndk', 'ingv_horus', 'ingv_emrcmt').")
+    if type not in (
+            'ucerf3', 'csep-csv', 'zmap', 'jma-csv', 'ingv_horus',
+            'ingv_emrcmt',
+            'ndk') and loader is None:
+        raise ValueError(
+            "type must be one of the following: ('ucerf3', 'csep-csv', 'zmap', 'jma-csv', 'ndk', 'ingv_horus', 'ingv_emrcmt').")
 
     # map to correct catalog class, at some point these could be abstracted into configuration file
     # this maps a human readable string to the correct catalog class and the correct loader function
@@ -167,7 +173,8 @@ def load_catalog(filename, type='csep-csv', format='native', loader=None, apply_
         if loader is None:
             loader = class_loader_mapping[type]['loader']
 
-        catalog = catalog_class.load_catalog(filename=filename, loader=loader, **kwargs)
+        catalog = catalog_class.load_catalog(filename=filename, loader=loader,
+                                             **kwargs)
 
     # convert to csep format if needed
     if format == 'native':
@@ -213,12 +220,15 @@ def query_comcat(start_time, end_time, min_magnitude=2.50,
     # Timezone should be in UTC
     t0 = time.time()
     eventlist = readers._query_comcat(start_time=start_time, end_time=end_time,
-                           min_magnitude=min_magnitude,
-                           min_latitude=min_latitude, max_latitude=max_latitude,
-                           min_longitude=min_longitude, max_longitude=max_longitude,
-                           max_depth=max_depth)
+                                      min_magnitude=min_magnitude,
+                                      min_latitude=min_latitude,
+                                      max_latitude=max_latitude,
+                                      min_longitude=min_longitude,
+                                      max_longitude=max_longitude,
+                                      max_depth=max_depth)
     t1 = time.time()
-    comcat = catalogs.CSEPCatalog(data=eventlist, date_accessed=utc_now_datetime(), **kwargs)
+    comcat = catalogs.CSEPCatalog(data=eventlist,
+                                  date_accessed=utc_now_datetime(), **kwargs)
     print("Fetched ComCat catalog in {} seconds.\n".format(t1 - t0))
 
     if apply_filters:
@@ -229,9 +239,13 @@ def query_comcat(start_time, end_time, min_magnitude=2.50,
 
     if verbose:
         print("Downloaded catalog from ComCat with following parameters")
-        print("Start Date: {}\nEnd Date: {}".format(str(comcat.start_time), str(comcat.end_time)))
-        print("Min Latitude: {} and Max Latitude: {}".format(comcat.min_latitude, comcat.max_latitude))
-        print("Min Longitude: {} and Max Longitude: {}".format(comcat.min_longitude, comcat.max_longitude))
+        print("Start Date: {}\nEnd Date: {}".format(str(comcat.start_time),
+                                                    str(comcat.end_time)))
+        print(
+            "Min Latitude: {} and Max Latitude: {}".format(comcat.min_latitude,
+                                                           comcat.max_latitude))
+        print("Min Longitude: {} and Max Longitude: {}".format(
+            comcat.min_longitude, comcat.max_longitude))
         print("Min Magnitude: {}".format(comcat.min_magnitude))
         print(f"Found {comcat.event_count} events in the ComCat catalog.")
 
@@ -266,11 +280,14 @@ def query_bsi(start_time, end_time, min_magnitude=2.50,
     t0 = time.time()
     eventlist = readers._query_bsi(start_time=start_time, end_time=end_time,
                                    min_magnitude=min_magnitude,
-                                   min_latitude=min_latitude, max_latitude=max_latitude,
-                                   min_longitude=min_longitude, max_longitude=max_longitude,
+                                   min_latitude=min_latitude,
+                                   max_latitude=max_latitude,
+                                   min_longitude=min_longitude,
+                                   max_longitude=max_longitude,
                                    max_depth=max_depth)
     t1 = time.time()
-    bsi = catalogs.CSEPCatalog(data=eventlist, date_accessed=utc_now_datetime(), **kwargs)
+    bsi = catalogs.CSEPCatalog(data=eventlist,
+                               date_accessed=utc_now_datetime(), **kwargs)
     print("Fetched BSI catalog in {} seconds.\n".format(t1 - t0))
 
     if apply_filters:
@@ -280,10 +297,15 @@ def query_bsi(start_time, end_time, min_magnitude=2.50,
             bsi = bsi.filter()
 
     if verbose:
-        print("Downloaded catalog from Bollettino Sismico Italiano (BSI) with following parameters")
-        print("Start Date: {}\nEnd Date: {}".format(str(bsi.start_time), str(bsi.end_time)))
-        print("Min Latitude: {} and Max Latitude: {}".format(bsi.min_latitude, bsi.max_latitude))
-        print("Min Longitude: {} and Max Longitude: {}".format(bsi.min_longitude, bsi.max_longitude))
+        print(
+            "Downloaded catalog from Bollettino Sismico Italiano (BSI) with following parameters")
+        print("Start Date: {}\nEnd Date: {}".format(str(bsi.start_time),
+                                                    str(bsi.end_time)))
+        print("Min Latitude: {} and Max Latitude: {}".format(bsi.min_latitude,
+                                                             bsi.max_latitude))
+        print(
+            "Min Longitude: {} and Max Longitude: {}".format(bsi.min_longitude,
+                                                             bsi.max_longitude))
         print("Min Magnitude: {}".format(bsi.min_magnitude))
         print(f"Found {bsi.event_count} events in the BSI catalog.")
 
@@ -338,6 +360,41 @@ def query_gns(start_time, end_time,  min_magnitude=2.950,
 
     return gns
 
+def query_gcmt(start_time, end_time, min_magnitude=5.0, min_depth=None,
+               max_depth=None,
+               catalog_id=None, verbose=True,
+               min_latitude=None, max_latitude=None,
+               min_longitude=None, max_longitude=None):
+    if min_latitude:
+        searchshape = 'RECT'
+    else:
+        searchshape = 'GLOBAL'
+    events, creation_time = readers._query_gcmt(
+        start_year=start_time.year,
+        start_month=start_time.month,
+        start_day=start_time.day,
+        searchshape=searchshape,
+        start_time=start_time.time().isoformat(),
+        end_year=end_time.year,
+        end_month=end_time.month,
+        end_day=end_time.day,
+        end_time=end_time.time().isoformat(),
+        min_mag=min_magnitude,
+        min_dep=min_depth,
+        max_dep=max_depth,
+        left_lon=min_longitude,
+        right_lon=max_longitude,
+        bot_lat=min_latitude,
+        top_lat=max_latitude,
+        verbose=verbose
+    )
+    catalog = catalogs.CSEPCatalog(data=events, name='ISC Bulletin - gCMT',
+                                   catalog_id=catalog_id,
+                                   date_accessed=creation_time)
+    catalog.filter([f'magnitude >= {min_magnitude}'], in_place=True)
+    return catalog
+
+
 def load_evaluation_result(fname):
     """ Load evaluation result stored as json file
 
@@ -361,7 +418,8 @@ def load_evaluation_result(fname):
             evaluation_type = json_dict['type']
         except:
             evaluation_type = 'default'
-    eval_result = evaluation_result_factory[evaluation_type].from_dict(json_dict)
+    eval_result = evaluation_result_factory[evaluation_type].from_dict(
+        json_dict)
     return eval_result
 
 
@@ -404,15 +462,18 @@ class with the region and magnitude members correctly assigned.
 
     # sanity checks
     if not os.path.exists(fname):
-        raise FileNotFoundError(f"Could not locate file {fname}. Unable to load forecast.")
+        raise FileNotFoundError(
+            f"Could not locate file {fname}. Unable to load forecast.")
     # sanity checks
     if loader is not None and not callable(loader):
-        raise AttributeError("Loader must be callable. Unable to load forecast.")
+        raise AttributeError(
+            "Loader must be callable. Unable to load forecast.")
     extension = os.path.splitext(fname)[-1][1:]
     if extension not in forecast_loader_mapping.keys() and loader is None:
-        raise AttributeError("File extension should be in ('dat','xml','h5','bin') if loader not provided.")
+        raise AttributeError(
+            "File extension should be in ('dat','xml','h5','bin') if loader not provided.")
 
-    if extension in ('xml','h5','bin'):
+    if extension in ('xml', 'h5', 'bin'):
         raise NotImplementedError
 
     # assign default loader
@@ -425,7 +486,8 @@ class with the region and magnitude members correctly assigned.
     return forecast
 
 
-def load_catalog_forecast(fname, catalog_loader=None, format='native', type='ascii', **kwargs):
+def load_catalog_forecast(fname, catalog_loader=None, format='native',
+                          type='ascii', **kwargs):
     """ General function to handle loading catalog forecasts.
 
         Currently, just a simple wrapper, but can contain more complex logic in the future.
@@ -444,10 +506,12 @@ def load_catalog_forecast(fname, catalog_loader=None, format='native', type='asc
     """
     # sanity checks
     if not os.path.exists(fname):
-        raise FileNotFoundError(f"Could not locate file {fname}. Unable to load forecast.")
+        raise FileNotFoundError(
+            f"Could not locate file {fname}. Unable to load forecast.")
     # sanity checks
     if catalog_loader is not None and not callable(catalog_loader):
-        raise AttributeError("Loader must be callable. Unable to load forecast.")
+        raise AttributeError(
+            "Loader must be callable. Unable to load forecast.")
     # factory methods for loading different types of catalogs
     catalog_loader_mapping = {
         'ascii': catalogs.CSEPCatalog.load_ascii_catalogs,
@@ -456,17 +520,18 @@ def load_catalog_forecast(fname, catalog_loader=None, format='native', type='asc
     if catalog_loader is None:
         catalog_loader = catalog_loader_mapping[type]
     # try and parse information from filename and send to forecast constructor
-    if format == 'native' and type=='ascii':
+    if format == 'native' and type == 'ascii':
         try:
             basename = str(os.path.basename(fname.rstrip('/')).split('.')[0])
             split_fname = basename.split('_')
             name = split_fname[0]
-            start_time = strptime_to_utc_datetime(split_fname[1], format="%Y-%m-%dT%H-%M-%S-%f")
+            start_time = strptime_to_utc_datetime(split_fname[1],
+                                                  format="%Y-%m-%dT%H-%M-%S-%f")
             # update kwargs
             _ = kwargs.setdefault('name', name)
             _ = kwargs.setdefault('start_time', start_time)
         except:
             pass
     # create observed_catalog forecast
-    return CatalogForecast(filename=fname, loader=catalog_loader, catalog_format=format, catalog_type=type, **kwargs)
-
+    return CatalogForecast(filename=fname, loader=catalog_loader,
+                           catalog_format=format, catalog_type=type, **kwargs)