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
|
#==========================================================================
#
# 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.
#
#==========================================================================*/
# a short program to check the value returned by the GetNameOfClass() methods
from __future__ import print_function
import itk
import sys
itk.auto_progress(2)
# must force the load to return all the names with dir(itk)
itk.force_load()
# itk.ImageToImageFilter
# a list of classes to exclude. Typically, the classes with a custom New()
# method, which return a subclass of the current class
exclude = ["ForwardFFTImageFilter",
"InverseFFTImageFilter",
"OutputWindow",
"FFTComplexToComplexImageFilter",
"ComplexToComplexFFTImageFilter",
"templated_class",
"HalfHermitianToRealInverseFFTImageFilter",
"RealToHalfHermitianForwardFFTImageFilter",
"CustomColormapFunction"]
wrongName = 0
totalName = 0
for t in dir(itk):
if t not in exclude:
T = itk.__dict__[t]
# first case - that's a templated class
if isinstance(T, itk.Vector.__class__) and len(T) > 0:
# use only the first specialization - all of them return the same
# name
i = T.values()[0]
# GetNameOfClass() is a virtual method of the LightObject class,
# so we must instantiate an object with the New() method
if 'New' in dir(i):
I = i.New()
# be sure that the type of the instantiated object is the same
# than the one of the class. It can be different if the class
# is an "abstract" one and don't provide any New() method.
# In that case, the one of the superclass is used.
if 'GetNameOfClass' in dir(I):
# print("Checking", t)
totalName += 1
n = I.GetNameOfClass()
if n != t and itk.class_(I) == i:
msg = "%s: wrong class name: %s" % (t, n)
print(msg, file=sys.stderr)
wrongName += 1
else:
if 'New' in dir(T):
I = T.New()
if 'GetNameOfClass' in dir(I):
# print("Checking", t)
totalName += 1
n = I.GetNameOfClass()
if n != t and itk.class_(I) == T:
msg = "%s: wrong class name: %s" % (t, n)
print(msg, file=sys.stderr)
wrongName += 1
print("%s classes checked." % totalName)
if wrongName:
print(
"%s classes are not providing the correct name." % wrongName,
file=sys.stderr)
sys.exit(1)
|