Newer
Older
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
#include "gtest/gtest.h"
#include "itkTransformComponent1.h"
#include "itkTransformComponent1Factory.h"
#include "itkMetricComponent1.h"
#include "itkMetricComponent1Factory.h"
namespace {
class ComponentFactoryTest : public ::testing::Test {
public:
typedef std::list< itk::LightObject::Pointer > RegisteredObjectsContainerType;
typedef itk::ComponentBase ComponentType;
typedef std::map<std::string, std::string> CriteriaType;
typedef std::pair<std::string, std::string> CriteriumType;
typedef itk::ComponentFactory::Pointer NodePointer;
virtual void SetUp()
{
}
// each node can hold multiple components (or none). Its the overlord's task to make it one per node.
NodePointer Node1;
NodePointer Node2;
};
TEST_F(ComponentFactoryTest, EmptyObjectFactoryBase)
{
// The Component factory is inspired on the ImageIO factory.
// In ITK Components (transformsIO and imageIO) can be preregistered: CMake\UseITK.cmake sets up the IOFactory_Register_Manager
// In Elastix (by Denis) GPU filters are preregistered using (simpleITK's) Typelists and Cmake
// In this test we manually register 2 dummy modules: itkTransformComponent1 and itkMetricComponent1.
RegisteredObjectsContainerType registeredComponents;
// "When CMake is not used to register the Component classes, there are"
EXPECT_NO_THROW(registeredComponents = itk::ObjectFactoryBase::CreateAllInstance("itkComponentBase"));
// " 0 Component objects available to the Overlord."
EXPECT_EQ(registeredComponents.size(), 0);
}
TEST_F(ComponentFactoryTest, FilledObjectFactoryBase)
{
// In this test we manually register 2 dummy modules: itkTransformComponent1 and itkMetricComponent1.
RegisteredObjectsContainerType registeredComponents;
EXPECT_NO_THROW(itk::TransformComponent1Factory::RegisterOneFactory());
EXPECT_NO_THROW(itk::MetricComponent1Factory::RegisterOneFactory());
// After registering the TransformComponent1 and MetricComponent1object, there are
EXPECT_NO_THROW(registeredComponents = itk::ObjectFactoryBase::CreateAllInstance("itkComponentBase"));
// " 2 Component objects available to the Overlord."
EXPECT_EQ(registeredComponents.size(), 2);
}
TEST_F(ComponentFactoryTest, SetEmptyCriteria)
{
CriteriaType emptyCriteria;
ASSERT_NO_THROW(Node1 = itk::ComponentFactory::New());
EXPECT_NO_THROW(Node1->SetCriteria(emptyCriteria));
ComponentType::Pointer Node1Component;
EXPECT_NO_THROW(Node1Component = Node1->GetComponent());
//Too few criteria means no Component could be selected."
EXPECT_TRUE(Node1Component.IsNull());
}
TEST_F(ComponentFactoryTest, SetSufficientCriteria)
{
CriteriaType criteria2;
criteria2["ComponentInput"] = "Transform";
//criteria1.insert(CriteriumType("ComponentInput", "Metric"));
ASSERT_NO_THROW(Node2 = itk::ComponentFactory::New());
Node2->SetCriteria(criteria2);
ComponentType::Pointer Node2Component;
EXPECT_NO_THROW(Node2Component = Node2->GetComponent());
//Sufficient criteria means one Component was selected."
EXPECT_FALSE(Node2Component.IsNull());
//Based on the criteria MetricComponent1 should be selected
EXPECT_STREQ(Node2Component->GetNameOfClass(), "MetricComponent1");
}
TEST_F(ComponentFactoryTest, AddCriteria)
{
CriteriaType emptyCriteria;
CriteriaType criteria1;
//criteria1.insert(CriteriumType("ComponentOutput","Metric"));
criteria1["ComponentOutput"] = "Transform";
//criteria1.insert(CriteriumType("ComponentInput", "Metric"));
Node1 = itk::ComponentFactory::New();
Node1->SetCriteria(emptyCriteria);
EXPECT_NO_THROW(Node1->AddCriteria(criteria1));
ComponentType::Pointer Node1Component;
EXPECT_NO_THROW(Node1Component = Node1->GetComponent());
//Sufficient criteria means one Component was selected."
EXPECT_FALSE(Node1Component.IsNull());
//Based on the criteria TransformComponent1 should be selected
EXPECT_STREQ(Node1Component->GetNameOfClass(), "TransformComponent1");
}
} // namespace elx