Skip to content
Snippets Groups Projects

Parallelising spin evolution

Merged Charlie Hoy requested to merge evolve_parallel into master
All threads resolved!
+ 32
1
@@ -53,12 +53,14 @@ class tqdm(_tqdm):
last_len = [0]
def print_status(s):
def print_status(s, time=None):
len_s = len(s)
_message = s + (' ' * max(last_len[0] - len_s, 0))
kwargs["message"] = _message
if logger is not None:
fp_write_log(_message)
if time is not None:
kwargs["asctime"] = time
fp_write('\r' + logger_prefix % kwargs)
last_len[0] = len_s
@@ -86,6 +88,35 @@ class tqdm(_tqdm):
)
return base
def display(self, msg=None, pos=None):
"""
Use `self.sp` to display `msg` in the specified `pos`.
Consider overloading this function when inheriting to use e.g.:
`self.some_frontend(**self.format_dict)` instead of `self.sp`.
Parameters
----------
msg : str, optional. What to display (default: `repr(self)`).
pos : int, optional. Position to `moveto`
(default: `abs(self.pos)`).
"""
if pos is None:
pos = abs(self.pos)
nrows = self.nrows or 20
if pos >= nrows - 1:
if pos >= nrows:
return False
if msg or msg is None: # override at `nrows - 1`
msg = " ... (more hidden) ..."
if pos:
self.moveto(pos)
_msg = self.__repr__() if msg is None else msg
self.sp(_msg, time.strftime("%Y-%m-%d %H:%M:%S"))
if pos:
self.moveto(-pos)
return True
def trange(*args, **kwargs):
"""
Loading