anta-database 0.3.2__tar.gz → 0.3.3__tar.gz

{anta_database-0.3.2 → anta_database-0.3.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: anta_database
-Version: 0.3.2
+Version: 0.3.3
 Summary: SQLite database for the AntArchitecture Community Data
 Author: Antoine Hermant
 Author-email: antoine.hermant@etik.com

{anta_database-0.3.2 → anta_database-0.3.3}/anta_database/compiling/compile.py

@@ -863,6 +863,8 @@ class CompileDatabase:
                             max_depth[i] = depth_slice[max_idx]
                             age_max_depth[i] = ds.IRH_AGE.values[max_idx]
 
+                    age_max_depth = np.abs(age_max_depth)
+                    max_depth = np.abs(max_depth)
                     flight_id = ds.attrs["flight ID"]
                     acq_year = ds.attrs["acquisition year"]
                     project = ds.attrs["project"]
@@ -872,10 +874,10 @@ class CompileDatabase:
                     df = gpd.GeoDataFrame({
                         "geometry": [Point(x, y) for x, y in zip(ds.PSX.values, ds.PSY.values)],
                         "irh_num": N.astype('int16'),
-                        "max_age": age_max_depth.astype('int16'),
+                        "max_age": age_max_depth.astype('int32'),
                         "max_depth": max_depth.astype('int16'),
-                        "flight_id": [flight_id] * len(N),
-                        "doi_data": [doi_data] * len(N),
+                        # "flight_id": [flight_id] * len(N),
+                        # "doi_data": [doi_data] * len(N),
                     }, crs="EPSG:3031")
 
                     all_dfs.append(df)

{anta_database-0.3.2 → anta_database-0.3.3}/anta_database/compiling/reconnect.py

@@ -43,7 +43,7 @@ class Reconnect:
 
 
     def process(self, df_connect: pd.DataFrame, tree, file_path: str):
-        data_dir = self._db.db_dir
+        data_dir = self._db._db_dir
         full_path = os.path.join(data_dir, file_path)
         with h5py.File(full_path, 'a') as f:
             with xr.open_dataset(f, engine='h5netcdf') as ds:

{anta_database-0.3.2 → anta_database-0.3.3}/anta_database/database/database.py

@@ -92,11 +92,13 @@ class Database:
             LEFT JOIN
                 variables v ON dv.variable_id = v.id
             LEFT JOIN
-                dataset_ages da ON d.id = da.dataset_id
+                dataset_ages da ON d.id = da.dataset_id AND v.name = 'IRH_DEPTH'
             LEFT JOIN
                 ages a ON da.age_id = a.id
             LEFT JOIN
-                regions r ON da.region_id = r.id
+                dataset_regions dr ON d.id = dr.dataset_id
+            LEFT JOIN
+                regions r ON dr.region_id = r.id
         '''
         conditions = []
         params = []
@@ -415,21 +417,21 @@ class Database:
         Helper method to build the SQL query and parameters for filtering.
         Returns the query string and parameters list.
         """
-        select_clause = 's.name AS dataset, \
+        select_clause = "s.name AS dataset, \
                         s.citation, \
                         s.DOI_dataset, \
                         s.DOI_publication, \
                         institutes_list.institutes, \
                         projects_list.projects, \
                         d.acq_year, \
-                        a.age, \
-                        a.age_unc, \
+                        CASE WHEN v.name = 'IRH_DEPTH' THEN a.age ELSE NULL END AS age, \
+                        CASE WHEN v.name = 'IRH_DEPTH' THEN a.age_unc ELSE NULL END AS age_unc, \
                         v.name AS var, \
                         d.flight_id, \
                         r.region, \
                         r.IMBIE_basin, \
                         radar_instruments_list.radar_instruments \
-        '
+        "
         query = f'''
             SELECT {select_clause}
             FROM
@@ -474,11 +476,13 @@ class Database:
             LEFT JOIN
                 variables v ON dv.variable_id = v.id
             LEFT JOIN
-                dataset_ages da ON d.id = da.dataset_id
+                dataset_ages da ON d.id = da.dataset_id AND v.name = 'IRH_DEPTH'
             LEFT JOIN
                 ages a ON da.age_id = a.id
             LEFT JOIN
-                regions r ON da.region_id = r.id
+                dataset_regions dr ON d.id = dr.dataset_id
+            LEFT JOIN
+                regions r ON dr.region_id = r.id
         '''
         conditions = []
         params = []
@@ -536,21 +540,21 @@ class Database:
               retain_query: Optional[bool] = True,
               ) -> 'MetadataResult':
 
-        select_clause = 's.name AS dataset, \
+        select_clause = "s.name AS dataset, \
                         s.citation, \
                         s.DOI_dataset, \
                         s.DOI_publication, \
                         institutes_list.institutes, \
                         projects_list.projects, \
                         d.acq_year, \
-                        a.age, \
-                        a.age_unc, \
+                        CASE WHEN v.name = 'IRH_DEPTH' THEN a.age ELSE NULL END AS age, \
+                        CASE WHEN v.name = 'IRH_DEPTH' THEN a.age_unc ELSE NULL END AS age_unc, \
                         v.name AS var, \
                         d.flight_id, \
                         r.region, \
                         r.IMBIE_basin, \
                         radar_instruments_list.radar_instruments \
-        '
+        "
         query, params = self._build_query_and_params(age, var, dataset, institute, project, acquisition_year, flight_id, region, IMBIE_basin, radar_instrument, select_clause)
 
         conn = sqlite3.connect(self._file_db_path)
@@ -686,18 +690,18 @@ class Database:
 
     def get_files(
         self,
-        metadata: Union[None, Dict, 'MetadataResult'] = None,
-        data_dir: Union[None, str] = None,
+        metadata: Optional[Union[None, Dict, 'MetadataResult']] = None,
+        data_dir: Optional[Union[None, str]] = None,
     ):
 
         md = metadata or self._md
         if not md:
-            print('Please provide metadata of the files you want to generate the data from. Exiting...')
+            print('Please provide metadata of the files you want to generate the data from.')
             return
 
         data_dir = data_dir or self._db_dir
         if not data_dir:
-            print('No data directory provided. Exiting...')
+            print('No data directory provided.')
             return
 
         file_paths = self._get_file_paths_from_metadata(metadata=md)

{anta_database-0.3.2 → anta_database-0.3.3}/anta_database/indexing/index_database.py

@@ -126,11 +126,9 @@ class IndexDatabase:
             CREATE TABLE IF NOT EXISTS dataset_ages (
                 dataset_id INTEGER,
                 age_id INTEGER,
-                region_id INTEGER,
                 FOREIGN KEY (dataset_id) REFERENCES datasets (id),
                 FOREIGN KEY (age_id) REFERENCES ages (id),
-                FOREIGN KEY (region_id) REFERENCES regions (id),
-                PRIMARY KEY (dataset_id, age_id, region_id)
+                PRIMARY KEY (dataset_id, age_id)
                 )
             ''')
 
@@ -164,6 +162,16 @@ class IndexDatabase:
                 )
             ''')
 
+        cursor.execute('''
+            CREATE TABLE IF NOT EXISTS dataset_regions (
+                dataset_id INTEGER,
+                region_id INTEGER,
+                FOREIGN KEY (dataset_id) REFERENCES datasets (id),
+                FOREIGN KEY (region_id) REFERENCES regions (id),
+                PRIMARY KEY (dataset_id, region_id)
+                )
+            ''')
+
         for f in tqdm(h5_files, desc="Indexing files"):
             _, file_name = os.path.split(f)
 
@@ -281,7 +289,7 @@ class IndexDatabase:
             # Insert variables
             var_list = ['ICE_THK', 'SURF_ELEV', 'BED_ELEV', 'BASAL_UNIT', 'IRH_NUM', 'IRH_DEPTH']
             for var in var_list:
-                if var in ds_vars or var == 'IRH_DEPTH':
+                if var in ds_vars:
                     cursor.execute('INSERT OR IGNORE INTO variables (name) VALUES (?)', (var,))
                     var_id = cursor.execute('SELECT id FROM variables WHERE name = ?', (var,)).fetchone()[0]
                     cursor.execute('''
@@ -294,13 +302,18 @@ class IndexDatabase:
                 for age, age_unc in age_uncs.iterrows():
                     cursor.execute('INSERT OR IGNORE INTO ages (age, age_unc) VALUES (?, ?)', (str(age), str(age_unc['age_unc'])))
                     age_id = cursor.execute('SELECT id FROM ages WHERE age = ? AND age_unc = ?', (str(age), str(age_unc['age_unc']))).fetchone()[0]
-                    for basin, region in basin_mapping.items():
-                        cursor.execute('INSERT OR IGNORE INTO regions (region, IMBIE_basin) VALUES (?, ?)', (region, basin))
-                        region_id = cursor.execute('SELECT id FROM regions WHERE region = ? AND IMBIE_basin = ?', (region, basin)).fetchone()[0]
-                        cursor.execute('''
-                            INSERT OR IGNORE INTO dataset_ages (dataset_id, age_id, region_id)
-                            VALUES (?, ?, ?)
-                        ''', (dataset_row_id, age_id, region_id))
+                    cursor.execute('''
+                        INSERT OR IGNORE INTO dataset_ages (dataset_id, age_id)
+                        VALUES (?, ?)
+                    ''', (dataset_row_id, age_id))
+
+            for basin, region in basin_mapping.items():
+                cursor.execute('INSERT OR IGNORE INTO regions (region, IMBIE_basin) VALUES (?, ?)', (region, basin))
+                region_id = cursor.execute('SELECT id FROM regions WHERE region = ? AND IMBIE_basin = ?', (region, basin)).fetchone()[0]
+                cursor.execute('''
+                    INSERT OR IGNORE INTO dataset_regions (dataset_id, region_id)
+                    VALUES (?, ?)
+                ''', (dataset_row_id, region_id))
 
         conn.commit()
         conn.close()

{anta_database-0.3.2 → anta_database-0.3.3}/anta_database/plotting/plotting.py

@@ -64,11 +64,12 @@ class Plotting:
             metadata: Union[None, Dict, 'MetadataResult'] = None,
             downscale_factor: Optional[int] = None,
             title: Optional[str] = None,
-            xlim: tuple = (None, None),
-            ylim: tuple = (None, None),
+            xlim: Optional[tuple] = (None, None),
+            ylim: Optional[tuple] = (None, None),
+            vmin: Optional[float] = None,
+            vmax: Optional[float] = None,
             scale_factor: float = 1.0,
             marker_size: Optional[float] = 0.5,
-            latex: bool = False,
             cmap: Optional['LinearSegmentedColormap'] = None,
             grounding_line: Optional[bool] = True,
             basins: Optional[bool] = True,
@@ -86,8 +87,9 @@ class Plotting:
             title=title,
             xlim=xlim,
             ylim=ylim,
+            vmin=vmin,
+            vmax=vmax,
             scale_factor=scale_factor,
-            latex=latex,
             cmap=cmap,
             grounding_line=grounding_line,
             basins=basins,
@@ -100,11 +102,12 @@ class Plotting:
             metadata: Union[None, Dict, 'MetadataResult'] = None,
             downscale_factor: Optional[int] = None,
             title: Optional[str] = None,
-            xlim: tuple = (None, None),
-            ylim: tuple = (None, None),
+            xlim: Optional[tuple] = (None, None),
+            ylim: Optional[tuple] = (None, None),
+            vmin: Optional[float] = None,
+            vmax: Optional[float] = None,
             scale_factor: float = 1.0,
             marker_size: Optional[float] = 0.1,
-            latex: bool = False,
             cmap: Optional['LinearSegmentedColormap'] = None,
             grounding_line: Optional[bool] = True,
             basins: Optional[bool] = True,
@@ -122,8 +125,9 @@ class Plotting:
             title=title,
             xlim=xlim,
             ylim=ylim,
+            vmin=vmin,
+            vmax=vmax,
             scale_factor=scale_factor,
-            latex=latex,
             cmap=cmap,
             grounding_line=grounding_line,
             basins=basins,
@@ -136,11 +140,12 @@ class Plotting:
             metadata: Union[None, Dict, 'MetadataResult'] = None,
             downscale_factor: Optional[int] = None,
             title: Optional[str] = None,
-            xlim: tuple = (None, None),
-            ylim: tuple = (None, None),
+            xlim: Optional[tuple] = (None, None),
+            ylim: Optional[tuple] = (None, None),
+            vmin: Optional[float] = None,
+            vmax: Optional[float] = None,
             scale_factor: float = 1.0,
             marker_size: Optional[float] = 0.1,
-            latex: bool = False,
             cmap: Optional['LinearSegmentedColormap'] = None,
             grounding_line: Optional[bool] = True,
             basins: Optional[bool] = True,
@@ -159,7 +164,8 @@ class Plotting:
             title=title,
             xlim=xlim,
             ylim=ylim,
-            latex=latex,
+            vmin=vmin,
+            vmax=vmax,
             cmap=cmap,
             grounding_line=grounding_line,
             basins=basins,
@@ -172,11 +178,12 @@ class Plotting:
             metadata: Union[None, Dict, 'MetadataResult'] = None,
             downscale_factor: Optional[int] = None,
             title: Optional[str] = None,
-            xlim: tuple = (None, None),
-            ylim: tuple = (None, None),
+            xlim: Optional[tuple] = (None, None),
+            ylim: Optional[tuple] = (None, None),
+            vmin: Optional[float] = None,
+            vmax: Optional[float] = None,
             scale_factor: float = 1.0,
             marker_size: Optional[float] = 0.3,
-            latex: bool = False,
             cmap: Optional['LinearSegmentedColormap'] = None,
             grounding_line: Optional[bool] = True,
             basins: Optional[bool] = True,
@@ -193,9 +200,10 @@ class Plotting:
             title=title,
             xlim=xlim,
             ylim=ylim,
+            vmin=vmin,
+            vmax=vmax,
             scale_factor=scale_factor,
             marker_size=marker_size,
-            latex=latex,
             cmap=cmap,
             grounding_line=grounding_line,
             basins=basins,
@@ -203,16 +211,16 @@ class Plotting:
             save=save,
         )
 
-    def transect(
+    def transect_1D(
             self,
             metadata: Union[None, Dict, 'MetadataResult'] = None,
+            elevation: Optional[bool] = False,
             downscale_factor: Optional[int] = None,
             title: Optional[str] = None,
-            xlim: tuple = (None, None),
-            ylim: tuple = (3500, 0),
+            xlim: Optional[tuple] = (None, None),
+            ylim: Optional[tuple] = (None, None),
             scale_factor: float = 1.0,
             marker_size: Optional[float] = 2,
-            latex: bool = False,
             cmap: Optional['LinearSegmentedColormap'] = None,
             grounding_line: Optional[bool] = True,
             basins: Optional[bool] = True,
@@ -223,7 +231,8 @@ class Plotting:
         Plot the color-coded values of the given variable on Antarcitic map
         """
         self._base_plot(
-            color_by='transect',
+            color_by='transect_1D',
+            elevation=elevation,
             metadata=metadata,
             downscale_factor=downscale_factor,
             title=title,
@@ -231,7 +240,6 @@ class Plotting:
             ylim=ylim,
             scale_factor=scale_factor,
             marker_size=marker_size,
-            latex=latex,
             cmap=cmap,
             grounding_line=grounding_line,
             basins=basins,
@@ -250,22 +258,24 @@ class Plotting:
                 plt.close()
 
     def _base_plot(
-        self,
-        metadata: Union[None, Dict, 'MetadataResult'] = None,
-        downscale_factor: Optional[int] = None,
-        title: Optional[str] = None,
-        xlim: tuple = (None, None),
-        ylim: tuple = (None, None),
-        scale_factor: float = 1.0,
-        marker_size: Optional[float] = 0.1,
-        latex: bool = False,
-        save: Optional[str] = None,
-        color_by: str = 'dataset',  # 'dataset', 'flight_id', 'depth', 'density'
-        cmap: Optional['LinearSegmentedColormap'] = None,
-        grounding_line: Optional[bool] = True,
-        basins: Optional[bool] = True,
-        stations: Optional[bool] = True,
-        ncol: Optional[int] = None,
+            self,
+            metadata: Union[None, Dict, 'MetadataResult'] = None,
+            elevation: Optional[bool] = False,
+            downscale_factor: Optional[int] = None,
+            title: Optional[str] = None,
+            xlim: Optional[tuple] = (None, None),
+            ylim: Optional[tuple] = (None, None),
+            vmin: Optional[float] = None,
+            vmax: Optional[float] = None,
+            scale_factor: float = 1.0,
+            marker_size: Optional[float] = 0.1,
+            save: Optional[str] = None,
+            color_by: str = 'dataset',  # 'dataset', 'flight_id', 'depth', 'density'
+            cmap: Optional['LinearSegmentedColormap'] = None,
+            grounding_line: Optional[bool] = True,
+            basins: Optional[bool] = True,
+            stations: Optional[bool] = True,
+            ncol: Optional[int] = None,
     ) -> None:
         # --- Setup ---
         if metadata is None:
@@ -277,8 +287,6 @@ class Plotting:
 
         total_traces = len(metadata['flight_id'])
 
-        # if latex:
-
         if not self._pre_plot_check(metadata):
             return
 
@@ -292,7 +300,7 @@ class Plotting:
         if basins:
             grounding_line = False
         # --- Plot Grounding Line ---
-        if True and color_by != 'transect': # FIXME
+        if True and color_by != 'transect_1D': # FIXME
             gl = pd.read_pickle(self._gl_path)
             ax.plot(gl.x/1000, gl.y/1000, linewidth=1, color='k')
 
@@ -302,8 +310,6 @@ class Plotting:
         values = None
         label = None
         extend = None
-        vmin = 0
-        vmax = 10
 
         if color_by == 'dataset':
             if not title:
@@ -442,32 +448,32 @@ class Plotting:
                     if cmap == None:
                         cmap = cmaps.bukavu
                     extend = 'both'
-                    vmin = -1000
-                    vmax = 1000
+                    if not vmin: vmin = -1000
+                    if not vmax: vmax = 1000
                     if not title:
                         title = f'AntADatabase Bed Elevation'
                 elif var == 'ICE_THK':
                     if cmap == None:
                         cmap = cmaps.torch_r
                     extend = 'max'
-                    vmin = 0
-                    vmax = 4000
+                    if not vmin: vmin = 0
+                    if not vmax: vmax = 4000
                     if not title:
                         title = f'AntADatabase Ice Thickness'
                 elif var == 'SURF_ELEV':
                     if cmap == None:
                         cmap = cmaps.torch_r
                     extend = 'max'
-                    vmin = 2000
-                    vmax = 4000
+                    if not vmin: vmin = 2000
+                    if not vmax: vmax = 4000
                     if not title:
                         title = f'AntADatabase Surface Elevation'
                 elif var == 'BASAL_UNIT':
                     if cmap == None:
                         cmap = cmaps.torch_r
                     extend = 'both'
-                    vmin = 2000
-                    vmax = 4000
+                    if not vmin: vmin = 2000
+                    if not vmax: vmax = 400
                     if not title:
                         title = f'AntADatabase Basal Unit'
                 scatter = plt.scatter(df['PSX']/1000, df['PSY']/1000, c=df[var], cmap=cmap, s=marker_size, vmin=vmin, vmax=vmax, linewidths=0, rasterized=True)
@@ -478,8 +484,6 @@ class Plotting:
                     if cmap == None:
                         cmap = cmaps.torch_r
                     extend = 'both'
-                    vmin = None
-                    vmax = None
                     age = list(metadata['age'])
                     if not title:
                         title = f'AntADatabase IRH Depth'
@@ -493,8 +497,6 @@ class Plotting:
                     if cmap == None:
                         cmap = cmaps.torch_r
                     extend = 'both'
-                    vmin = None
-                    vmax = None
                     age = list(metadata['age'])
                     if len(age) > 1:
                         print('WARNING: Multiple layers provided: ', age,
@@ -508,7 +510,7 @@ class Plotting:
                 for age in metadata['age']:
                     scatter = plt.scatter(df['PSX']/1000, df['PSY']/1000, c=df[int(age)], cmap=cmap, s=marker_size, vmin=vmin, vmax=vmax, linewidths=0, rasterized=True)
 
-        if color_by == 'transect':
+        if color_by == 'transect_1D':
             flight_id = list(metadata['flight_id'])
             if len(flight_id) > 1:
                 flight_id = flight_id[0]
@@ -519,7 +521,7 @@ class Plotting:
 
             metadata_impl = self._db.query(flight_id=flight_id, dataset=metadata['dataset'], retain_query=False)
             if not title:
-                title = f'Transect {metadata['flight_id'][0]} from {metadata['reference'][0]}'
+                title = f'Transect {metadata_impl['flight_id'][0]} from {metadata_impl['reference'][0]}'
 
             f = self._db._get_file_paths_from_metadata(metadata_impl)[0]
             full_path = os.path.join(self._db._db_dir, f)
@@ -530,15 +532,49 @@ class Plotting:
                 print('Distance not in varibles, cannot plot along transect')
                 return
 
+            if elevation:
+                if 'SURF_ELEV' in ds.variables:
+                    ds['IRH_DEPTH'] = ds.SURF_ELEV - ds.IRH_DEPTH
+
+                elif 'ICE_THK' in ds.variables and 'BED_ELEV' in ds.variables:
+                    ds['SURF_ELEV'] = (ds.ICE_THK + ds.BED_ELEV) - ds.IRH_DEPTH
+                    ds['IRH_DEPTH'] = ds.SURF_ELEV - ds.IRH_DEPTH
+                else:
+                    print('Cannot plot IRH Elevation from the variables in the file, missing either ICE_THK and BED_ELEV or SURF_ELEV')
+                    return
+
             cmap = self._custom_cmap_density()
-            colors = [cmap(i) for i in np.linspace(0.1, 0.9, ds.sizes['IRH_AGE'])]
-            for age, color in zip(ds.IRH_AGE, colors):
+            colors = [cmap(i) for i in np.linspace(0.1, 0.9, len(metadata_impl['age']))]
+            for age, color in zip(list(map(int, metadata['age'])), colors):
+                if age not in ds.IRH_AGE:
+                    print(f'{metadata_impl['flight_id'][0]} does not contain age {age}, skipping')
+                    continue
+
                 ax.scatter(ds.Distance/1000, ds.IRH_DEPTH.sel(IRH_AGE=age),
-                           color=color, s=marker_size, linewidths=0.1)
-                plt.plot([], [], color=color, label=age.values, linewidth=3)
-                # ds.IRH_DEPTH.sel(IRH_AGE=age).where(~np.isnan(ds.IRH_DEPTH.sel(IRH_AGE=age)), drop=True).plot()
-            scatter = ax.scatter(ds.Distance/1000, ds.ICE_THK, color='k', s=marker_size, linewidths=0.1)
-            plt.plot([], [], color='k', label='Bed Depth', linewidth=3)
+                        color=color, s=marker_size, linewidths=0.1)
+                plt.plot([], [], color=color, label=age, linewidth=3)
+
+            if elevation:
+                if 'BED_ELEV' in ds.variables:
+                    scatter = ax.scatter(ds.Distance/1000, ds.BED_ELEV, color='k', s=marker_size, linewidths=0.1)
+                    plt.plot([], [], color='k', label='Bed Elevation', linewidth=3)
+                elif 'ICE_THK' in ds.variables and 'SURF_ELEV' in ds.variables:
+                    ds['BED_ELEV'] = ds.SURF_ELEV - ds.ICE_THK
+                    scatter = ax.scatter(ds.Distance/1000, ds.BED_ELEV, color='k', s=marker_size, linewidths=0.1)
+                    plt.plot([], [], color='k', label='Bed Elevation', linewidth=3)
+                else:
+                    print('Cannot plot Bed Elevation from the variables in the file')
+                ax.scatter(ds.Distance/1000, ds.SURF_ELEV, color='grey', s=marker_size, linewidths=0.1)
+                plt.plot([], [], color='grey', label='Surface Elevation', linewidth=3)
+
+            else:
+                scatter = ax.scatter(ds.Distance/1000, ds.ICE_THK, color='k', s=marker_size, linewidths=0.1)
+                plt.plot([], [], color='k', label='Bed Depth', linewidth=3)
+
+            if elevation:
+                ylim = (ds.BED_ELEV.min() - 200, ds.SURF_ELEV.max()+200) if ylim == (None, None) else ylim
+            else:
+                ylim = (ds.ICE_THK.max() + 200, 0) if ylim == (None, None) else ylim
 
             if ncol == None:
                 if ds.sizes['IRH_AGE'] > 7:
@@ -579,7 +615,7 @@ class Plotting:
 
         ax.set_xlim(*xlim)
         ax.set_ylim(*ylim)
-        if color_by != 'transect':
+        if color_by != 'transect_1D':
             x0, x1 = ax.get_xlim() if xlim == (None, None) else xlim
             y0, y1 = ax.get_ylim() if ylim == (None, None) else ylim
             x_extent = x1 - x0
@@ -607,16 +643,19 @@ class Plotting:
             cbar.ax.xaxis.set_ticks_position('bottom')
             if label:
                 cbar.set_label(label)
-        elif color_by == 'transect':
+        elif color_by == 'transect_1D':
             ax.legend(ncols=2)
-            ax.set_xlabel('Distance [km]')
-            ax.set_ylabel('Depth below surface [m]')
+            ax.set_xlabel('Distance along transect [km]')
+            if elevation:
+                ax.set_ylabel('Elevation above sea level [m]')
+            else:
+                ax.set_ylabel('Depth below surface [m]')
             plt.gcf().set_size_inches(10 * scale_factor, 10 * 2/3)
 
         plt.tight_layout()
 
         # --- Plot IMBIE basins ---
-        if basins and color_by != 'transect':
+        if basins and color_by != 'transect_1D':
             sf_basins = shapefile.Reader(self._imbie_path)
             basin_patches = []
 
@@ -667,7 +706,7 @@ class Plotting:
                         )]
                     )
         # --- Plot ice core sites ---
-        if stations and color_by != 'transect':
+        if stations and color_by != 'transect_1D':
             site_coords = pd.read_pickle(self._site_coords_path)
             for i in site_coords.index:
                 site = site_coords.loc[i]

{anta_database-0.3.2 → anta_database-0.3.3}/anta_database.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: anta_database
-Version: 0.3.2
+Version: 0.3.3
 Summary: SQLite database for the AntArchitecture Community Data
 Author: Antoine Hermant
 Author-email: antoine.hermant@etik.com