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
|
#==========================================================================
#
# Copyright Insight Software Consortium
#
# 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.txt
#
# 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.
#
#==========================================================================*/
"""itk: Top-level container package for ITK wrappers."""
import itkBase
import itkConfig
import itkLazy
import itkTypes
import itkExtras
import os
import sys
thisModule = sys.modules[__name__]
def _GetLazyAttributes(lazyAttributes):
templateNames = [t[0] for t in data['templates']]
isInLibrary = [t[3] for t in data['templates'] if len(t) > 3 ]
attributes = dict([(n, module) for n in templateNames])
attributesInModule = dict([(n, belongs) for n,belongs in zip(templateNames,isInLibrary)])
items = attributes.items()
for k, v in items:
if isInLibrary and attributesInModule[k] is True:
lazyAttributes.setdefault(k,[]).insert(0,v)
else:
lazyAttributes.setdefault(k,[]).append(v)
if itkConfig.LazyLoading:
# If we are loading lazily (on-demand), make a dict mapping the available
# classes/functions/etc. (read from the configuration modules) to the
# modules they are declared in. Then pass that dict to a LazyITKModule
# instance and (later) do some surgery on sys.modules so that the 'itk'
# module becomes that new instance instead of what is executed from this
# file.
lazyAttributes = {}
for module, data in itkBase.module_data.items():
_GetLazyAttributes(lazyAttributes)
if isinstance(thisModule, itkLazy.LazyITKModule):
# Handle reload case where we've already done this once.
# If we made a new module every time, multiple reload()s would fail
# because the identity of sys.modules['itk'] would always be changing.
thisModule.__init__(__name__, lazyAttributes)
del lazyAttributes
else:
thisModule = itkLazy.LazyITKModule(__name__, lazyAttributes)
else:
# We're not lazy-loading. Just load the modules in the order specified in
# the known_modules list for consistency.
for module in itkBase.known_modules:
itkBase.LoadModule(module, thisModule.__dict__)
# Regardless of how it was loaded, fill up the itk module with the ITK types
# and extras.
for k, v in itkTypes.__dict__.items():
if k != 'itkCType' and not k.startswith('_'):
setattr(thisModule, k, v)
for k, v in itkExtras.__dict__.items():
if not k.startswith('_'):
setattr(thisModule, k, v)
# Populate itk.ITKModuleName
for module, data in itkBase.module_data.items():
attributes = {}
_GetLazyAttributes(attributes)
itkModule = itkLazy.LazyITKModule(module, attributes)
setattr(thisModule, module, itkModule)
# Set the __path__ attribute, which is required for this module to be used as a
# package
setattr(thisModule, '__path__', __path__)
if itkConfig.LazyLoading:
# this has to be the last step, else python gets confused about itkTypes
# and itkExtras above. I'm not sure why...
sys.modules[__name__] = thisModule
else:
# do some cleanup
del module, thisModule, itkModule
del itkBase, itkConfig, itkLazy, itkTypes, itkExtras, os, sys
|
