From 4b038e8d01da1ae427a1d5d37df6b0aa378e5527 Mon Sep 17 00:00:00 2001 From: rstiskalek Date: Wed, 18 Oct 2023 16:30:45 +0100 Subject: [PATCH] Add printing --- csiborgtools/__init__.py | 2 +- csiborgtools/utils.py | 7 +++++++ 2 files changed, 8 insertions(+), 1 deletion(-) diff --git a/csiborgtools/__init__.py b/csiborgtools/__init__.py index e382c56..b791fa6 100644 --- a/csiborgtools/__init__.py +++ b/csiborgtools/__init__.py @@ -16,7 +16,7 @@ from csiborgtools import clustering, field, match, read, summary from .utils import (center_of_mass, delta2ncells, number_counts, # noqa periodic_distance, periodic_distance_two_points, # noqa - binned_statistic, cosine_similarity) # noqa + binned_statistic, cosine_similarity, fprint) # noqa # Arguments to csiborgtools.read.Paths. diff --git a/csiborgtools/utils.py b/csiborgtools/utils.py index 388c4f6..4740bd9 100644 --- a/csiborgtools/utils.py +++ b/csiborgtools/utils.py @@ -15,6 +15,7 @@ """Collection of stand-off utility functions used in the scripts.""" import numpy from numba import jit +from datetime import datetime ############################################################################### # Positions # @@ -262,3 +263,9 @@ def binned_statistic(x, y, left_edges, bin_width, statistic): if numpy.any(mask): out[i] = statistic(y[mask]) return out + + +def fprint(msg, verbose=True): + """Print and flush a message with a timestamp.""" + if verbose: + print(f"{datetime.now()}: {msg}", flush=True)