forked from episodes-platform/shared-snippets
		
	feature/ISL-4832-przepisac-catalogeditor-do-pythona #4
@@ -20,7 +20,7 @@ from find_catalog_column import find_catalog_column
 | 
				
			|||||||
 | 
					
 | 
				
			||||||
def sort_by_time(catalog):
 | 
					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:
 | 
					    Parameters:
 | 
				
			||||||
        catalog (np.ndarray): A structured NumPy array representing a seismic catalog in EPISODES Platform format:
 | 
					        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
 | 
					 | 
				
			||||||
		Reference in New Issue
	
	Block a user