1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
|
from sys import stderr as _system_error_stream
from enum import IntEnum, unique
from typing import Union
__all__ = [
"auto_not_in_place",
"AutoProgressTypes",
"auto_progress",
"terminal_progress_callback",
"terminal_import_callback",
"simple_import_callback",
"simple_progress_callback",
"force_load",
]
# The following line defines an ascii string used for dynamically refreshing
# the import and progress callbacks on the same terminal line.
# See http://www.termsys.demon.co.uk/vtansi.htm
# \033 is the C-style octal code for an escape character
# [2000D moves the cursor back 2000 columns, this is a brute force way of
# getting to the start of the line.
# [K erases the end of the line
clrLine = "\033[2000D\033[K"
def auto_not_in_place(v: bool = True):
"""Force it to not run in place"""
import itkConfig
itkConfig.NotInPlace = v
@unique
class AutoProgressTypes(IntEnum):
DISABLE = 0 #
TERMINAL = 1
SIMPLE = 2
def auto_progress(
progress_type: Union[bool, AutoProgressTypes] = AutoProgressTypes.TERMINAL
) -> None:
"""Set up auto progress report
progress_type:
1 or True -> auto progress be used in a terminal
2 -> simple auto progress (without special characters)
0 or False -> disable auto progress
"""
import itkConfig
if progress_type is True or progress_type == AutoProgressTypes.TERMINAL:
itkConfig.ImportCallback = terminal_import_callback
itkConfig.ProgressCallback = terminal_progress_callback
elif progress_type == AutoProgressTypes.SIMPLE:
itkConfig.ImportCallback = simple_import_callback
itkConfig.ProgressCallback = simple_progress_callback
elif progress_type is False or progress_type == AutoProgressTypes.DISABLE:
itkConfig.ImportCallback = None
itkConfig.ProgressCallback = None
else:
raise ValueError("Invalid auto progress type: " + repr(progress_type))
def terminal_progress_callback(name: str, p):
"""Display the progress of an object and clean the display once complete
This function can be used with itkConfig.ProgressCallback
"""
print(clrLine + f"{name}: {p:f}", file=_system_error_stream, end="")
if p == 1:
print(clrLine, file=_system_error_stream, end="")
def terminal_import_callback(name: str, p):
"""Display the loading of a module and clean the display once complete
This function can be used with itkConfig.ImportCallback
"""
print(clrLine + f"Loading {name}... ", file=_system_error_stream, end="")
if p == 1:
print(clrLine, file=_system_error_stream, end="")
def simple_import_callback(name: str, p):
"""Print a message when a module is loading
This function can be used with itkConfig.ImportCallback
"""
if p == 0:
print(f"Loading {name}... ", file=_system_error_stream, end="")
elif p == 1:
print("done", file=_system_error_stream)
def simple_progress_callback(name: str, p):
"""Print a message when an object is running
This function can be used with itkConfig.ProgressCallback
"""
if p == 0:
print(f"Running {name}... ", file=_system_error_stream, end="")
elif p == 1:
print("done", file=_system_error_stream)
def force_load() -> None:
"""force itk to load all the submodules"""
import itk
for k in dir(itk):
getattr(itk, k)
|
