forked from episodes-platform/shared-snippets
ISL-4832 Removed standalone sort by time script and added information to sort by time script that it requires find catalog column script
This commit is contained in:
parent
9770709a81
commit
d9a962e3bb
@ -20,7 +20,7 @@ from find_catalog_column import find_catalog_column
|
||||
|
||||
def sort_by_time(catalog):
|
||||
"""
|
||||
Sorts the given catalog based on the 'Time' column.
|
||||
Sorts the given catalog based on the 'Time' column. Requires find_catalog_column script to work.
|
||||
|
||||
Parameters:
|
||||
catalog (np.ndarray): A structured NumPy array representing a seismic catalog in EPISODES Platform format:
|
||||
|
@ -1,50 +0,0 @@
|
||||
# -----------------
|
||||
# Copyright © 2024 ACK Cyfronet AGH, Poland.
|
||||
#
|
||||
# Licensed under the Apache License, Version 2.0 (the "License");
|
||||
# you may not use this file except in compliance with the License.
|
||||
# You may obtain a copy of the License at
|
||||
#
|
||||
# http://www.apache.org/licenses/LICENSE-2.0
|
||||
#
|
||||
# Unless required by applicable law or agreed to in writing, software
|
||||
# distributed under the License is distributed on an "AS IS" BASIS,
|
||||
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
# See the License for the specific language governing permissions and
|
||||
# limitations under the License.
|
||||
#
|
||||
# This work was partially funded by DT-GEO Project.
|
||||
# -----------------
|
||||
import numpy as np
|
||||
|
||||
def sort_by_time(catalog):
|
||||
"""
|
||||
Sorts the given catalog based on the 'Time' column.
|
||||
|
||||
Parameters:
|
||||
catalog (np.ndarray): A structured NumPy array representing a seismic catalog in EPISODES Platform format:
|
||||
https://docs.cyfronet.pl/display/ISDOC/Seismic+catalog
|
||||
|
||||
Returns:
|
||||
np.ndarray: The sorted catalog, where rows in 'val' fields are ordered by ascending 'Time' values.
|
||||
|
||||
Raises:
|
||||
ValueError: If the 'Time' column is not found in the catalog.
|
||||
"""
|
||||
|
||||
time_col_index = None
|
||||
for idx, col in enumerate(catalog):
|
||||
if col['field'] == "Time":
|
||||
time_col_index = idx
|
||||
|
||||
if time_col_index is None:
|
||||
raise ValueError("No 'Time' field found in the catalog.")
|
||||
|
||||
time_values = catalog[time_col_index]['val'].flatten()
|
||||
sorted_indexes = np.argsort(time_values)
|
||||
|
||||
sorted_catalog = catalog.copy()
|
||||
for field in sorted_catalog:
|
||||
field['val'] = field['val'][sorted_indexes]
|
||||
|
||||
return sorted_catalog
|
Loading…
Reference in New Issue
Block a user