forked from Slicer/SlicerGitSVNArchive
-
Notifications
You must be signed in to change notification settings - Fork 2
/
vtkMRMLSliceLogicTest4.cxx
124 lines (100 loc) · 4.13 KB
/
vtkMRMLSliceLogicTest4.cxx
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
117
118
119
120
121
122
123
124
/*==============================================================================
Program: 3D Slicer
Copyright (c) Kitware Inc.
See COPYRIGHT.txt
or http://www.slicer.org/copyright/copyright.txt for details.
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.
This file was originally developed by Jean-Christophe Fillion-Robin, Kitware Inc.
and was partially funded by NIH grant 3P41RR013218-12S1
==============================================================================*/
// MRMLLogic includes
// MRML includes
#include <vtkMRMLColorTableNode.h>
#include <vtkMRMLScene.h>
#include <vtkMRMLScalarVolumeDisplayNode.h>
#include <vtkMRMLScalarVolumeNode.h>
#include <vtkMRMLVolumeArchetypeStorageNode.h>
// VTK includes
#include <vtkColorTransferFunction.h>
#include <vtkImageData.h>
#include <vtkImageReslice.h>
#include <vtkImageMapToColors.h>
#include <vtkImageAppendComponents.h>
#include <vtkSmartPointer.h>
// STD includes
vtkMRMLScalarVolumeNode* loadVolume2(const char* volume, vtkMRMLScene* scene)
{
vtkSmartPointer<vtkMRMLScalarVolumeDisplayNode> displayNode = vtkSmartPointer<vtkMRMLScalarVolumeDisplayNode>::New();
vtkSmartPointer<vtkMRMLScalarVolumeNode> scalarNode = vtkSmartPointer<vtkMRMLScalarVolumeNode>::New();
vtkSmartPointer<vtkMRMLVolumeArchetypeStorageNode> storageNode = vtkSmartPointer<vtkMRMLVolumeArchetypeStorageNode>::New();
displayNode->SetAutoWindowLevel(false);
displayNode->SetInterpolate(false);
storageNode->SetFileName(volume);
if (storageNode->SupportedFileType(volume) == 0)
{
return 0;
}
scalarNode->SetName("foo");
scalarNode->SetScene(scene);
displayNode->SetScene(scene);
//vtkSlicerColorLogic *colorLogic = vtkSlicerColorLogic::New();
//displayNode->SetAndObserveColorNodeID(colorLogic->GetDefaultVolumeColorNodeID());
//colorLogic->Delete();
scene->AddNode(storageNode);
scene->AddNode(displayNode);
scalarNode->SetAndObserveStorageNodeID(storageNode->GetID());
scalarNode->SetAndObserveDisplayNodeID(displayNode->GetID());
scene->AddNode(scalarNode);
storageNode->ReadData(scalarNode);
vtkMRMLColorTableNode* colorNode = vtkMRMLColorTableNode::New();
colorNode->SetTypeToGrey();
scene->AddNode(colorNode);
colorNode->Delete();
displayNode->SetAndObserveColorNodeID(colorNode->GetID());
return scalarNode.GetPointer();
}
int vtkMRMLSliceLogicTest4(int argc, char * argv [] )
{
if( argc < 2 )
{
std::cerr << "Error: missing arguments" << std::endl;
std::cerr << "Usage: " << std::endl;
std::cerr << argv[0] << " input_image " << std::endl;
return EXIT_FAILURE;
}
vtkSmartPointer<vtkMRMLScene> scene = vtkSmartPointer<vtkMRMLScene>::New();
vtkMRMLScalarVolumeNode* scalarNode = loadVolume2(argv[1], scene);
if (scalarNode == 0)
{
std::cerr << "Not a valid volume: " << argv[1] << std::endl;
return EXIT_FAILURE;
}
vtkImageData* img = scalarNode->GetImageData();
vtkSmartPointer<vtkImageReslice> resliceMask = vtkSmartPointer<vtkImageReslice>::New();
resliceMask->SetOutputExtent(0, 400, 0, 500, 0, 0);
resliceMask->SetInput(img);
vtkSmartPointer<vtkImageMapToColors> colors = vtkSmartPointer<vtkImageMapToColors>::New();
colors->SetInput(resliceMask->GetOutput());
vtkSmartPointer<vtkColorTransferFunction> ctf = vtkSmartPointer<vtkColorTransferFunction>::New();
ctf->AddRGBPoint(0, 1., 0., 0.);
colors->SetLookupTable(ctf);
vtkSmartPointer<vtkImageAppendComponents> append = vtkSmartPointer<vtkImageAppendComponents>::New();
append->SetInputConnection(0 ,colors->GetOutput()->GetProducerPort());
//vtkImageBlend* blend = vtkImageBlend::New();
//blend->AddInput(append->GetOutput());
colors->Update();
append->Update();
append->Update();
append->Update();
colors->Update();
//append->Update();
//blend->Update();
//append->Update();
resliceMask->SetOutputExtent(0, 400, 0, 300, 0, 0);
colors->Update();
return EXIT_SUCCESS;
}