Thiago Franco de Moraes / invesalius3

Commit 011095a80c04584e83c8a6de3e57f6f2460ded59

Authored by Thiago Franco de Moraes 2016-08-03 16:01:34 -0300

Converting polydata_utils to unix format

Showing 1 changed file with 200 additions and 200 deletions Show diff stats

invesalius/data/polydata_utils.py

1	-#--------------------------------------------------------------------------
2	-# Software: InVesalius - Software de Reconstrucao 3D de Imagens Medicas
3	-# Copyright: (C) 2001 Centro de Pesquisas Renato Archer
4	-# Homepage: http://www.softwarepublico.gov.br
5	-# Contact: invesalius@cti.gov.br
6	-# License: GNU - GPL 2 (LICENSE.txt/LICENCA.txt)
7	-#--------------------------------------------------------------------------
8	-# Este programa e software livre; voce pode redistribui-lo e/ou
9	-# modifica-lo sob os termos da Licenca Publica Geral GNU, conforme
10	-# publicada pela Free Software Foundation; de acordo com a versao 2
11	-# da Licenca.
12	-#
13	-# Este programa eh distribuido na expectativa de ser util, mas SEM
14	-# QUALQUER GARANTIA; sem mesmo a garantia implicita de
15	-# COMERCIALIZACAO ou de ADEQUACAO A QUALQUER PROPOSITO EM
16	-# PARTICULAR. Consulte a Licenca Publica Geral GNU para obter mais
17	-# detalhes.
18	-#--------------------------------------------------------------------------
19	-
20	-import sys
21	-
22	-import vtk
23	-import wx
24	-from wx.lib.pubsub import pub as Publisher
25	-
26	-import vtk_utils as vu
27	-
28	-# Update progress value in GUI
29	-UpdateProgress = vu.ShowProgress()
30	-
31	-def ApplyDecimationFilter(polydata, reduction_factor):
32	- """
33	- Reduce number of triangles of the given vtkPolyData, based on
34	- reduction_factor.
35	- """
36	- # Important: vtkQuadricDecimation presented better results than
37	- # vtkDecimatePro
38	- decimation = vtk.vtkQuadricDecimation()
39	- decimation.SetInputData(polydata)
40	- decimation.SetTargetReduction(reduction_factor)
41	- decimation.GetOutput().ReleaseDataFlagOn()
42	- decimation.AddObserver("ProgressEvent", lambda obj, evt:
43	- UpdateProgress(decimation, "Reducing number of triangles..."))
44	- return decimation.GetOutput()
45	-
46	-def ApplySmoothFilter(polydata, iterations, relaxation_factor):
47	- """
48	- Smooth given vtkPolyData surface, based on iteration and relaxation_factor.
49	- """
50	- smoother = vtk.vtkSmoothPolyDataFilter()
51	- smoother.SetInputData(polydata)
52	- smoother.SetNumberOfIterations(iterations)
53	- smoother.SetFeatureAngle(80)
54	- smoother.SetRelaxationFactor(relaxation_factor)
55	- smoother.FeatureEdgeSmoothingOn()
56	- smoother.BoundarySmoothingOn()
57	- smoother.GetOutput().ReleaseDataFlagOn()
58	- smoother.AddObserver("ProgressEvent", lambda obj, evt:
59	- UpdateProgress(smoother, "Smoothing surface..."))
60	-
61	- return smoother.GetOutput()
62	-
63	-
64	-
65	-def FillSurfaceHole(polydata):
66	- """
67	- Fill holes in the given polydata.
68	- """
69	- # Filter used to detect and fill holes. Only fill
70	- print "Filling polydata"
71	- filled_polydata = vtk.vtkFillHolesFilter()
72	- filled_polydata.SetInputData(polydata)
73	- filled_polydata.SetHoleSize(500)
74	- return filled_polydata.GetOutput()
75	-
76	-def CalculateSurfaceVolume(polydata):
77	- """
78	- Calculate the volume from the given polydata
79	- """
80	- # Filter used to calculate volume and area from a polydata
81	- measured_polydata = vtk.vtkMassProperties()
82	- measured_polydata.SetInputData(polydata)
83	- return measured_polydata.GetVolume()
84	-
85	-def CalculateSurfaceArea(polydata):
86	- """
87	- Calculate the volume from the given polydata
88	- """
89	- # Filter used to calculate volume and area from a polydata
90	- measured_polydata = vtk.vtkMassProperties()
91	- measured_polydata.SetInputData(polydata)
92	- return measured_polydata.GetSurfaceArea()
93	-
94	-def Merge(polydata_list):
95	- append = vtk.vtkAppendPolyData()
96	-
97	- for polydata in polydata_list:
98	- triangle = vtk.vtkTriangleFilter()
99	- triangle.SetInputData(polydata)
100	- triangle.Update()
101	- append.AddInputData(triangle.GetOutput())
102	-
103	- append.Update()
104	- clean = vtk.vtkCleanPolyData()
105	- clean.SetInputData(append.GetOutput())
106	- clean.Update()
107	-
108	- return append.GetOutput()
109	-
110	-def Export(polydata, filename, bin=False):
111	- writer = vtk.vtkXMLPolyDataWriter()
112	- print filename, type(filename)
113	- writer.SetFileName(filename.encode('utf-8'))
114	- if bin:
115	- writer.SetDataModeToBinary()
116	- else:
117	- writer.SetDataModeToAscii()
118	- writer.SetInputData(polydata)
119	- writer.Write()
120	-
121	-def Import(filename):
122	- reader = vtk.vtkXMLPolyDataReader()
123	- if isinstance(filename, unicode):
124	- reader.SetFileName(filename.encode(wx.GetDefaultPyEncoding()))
125	- else:
126	- reader.SetFileName(filename)
127	- reader.Update()
128	- return reader.GetOutput()
129	-
130	-def JoinSeedsParts(polydata, point_id_list):
131	- """
132	- The function require vtkPolyData and point id
133	- from vtkPolyData.
134	- """
135	- conn = vtk.vtkPolyDataConnectivityFilter()
136	- conn.SetInputData(polydata)
137	- conn.SetExtractionModeToPointSeededRegions()
138	- UpdateProgress = vu.ShowProgress(1 + len(point_id_list))
139	- pos = 1
140	- for seed in point_id_list:
141	- conn.AddSeed(seed)
142	- UpdateProgress(pos, _("Analysing selected regions..."))
143	- pos += 1
144	-
145	- conn.AddObserver("ProgressEvent", lambda obj, evt:
146	- UpdateProgress(conn, "Getting selected parts"))
147	- conn.Update()
148	-
149	- result = vtk.vtkPolyData()
150	- result.DeepCopy(conn.GetOutput())
151	- return result
152	-
153	-def SelectLargestPart(polydata):
154	- """
155	- """
156	- UpdateProgress = vu.ShowProgress(1)
157	- conn = vtk.vtkPolyDataConnectivityFilter()
158	- conn.SetInputData(polydata)
159	- conn.SetExtractionModeToLargestRegion()
160	- conn.AddObserver("ProgressEvent", lambda obj, evt:
161	- UpdateProgress(conn, "Getting largest part..."))
162	- conn.Update()
163	-
164	- result = vtk.vtkPolyData()
165	- result.DeepCopy(conn.GetOutput())
166	- return result
167	-
168	-def SplitDisconectedParts(polydata):
169	- """
170	- """
171	- conn = vtk.vtkPolyDataConnectivityFilter()
172	- conn.SetInputData(polydata)
173	- conn.SetExtractionModeToAllRegions()
174	- conn.Update()
175	-
176	- nregions = conn.GetNumberOfExtractedRegions()
177	-
178	- conn.SetExtractionModeToSpecifiedRegions()
179	- conn.Update()
180	-
181	- polydata_collection = []
182	-
183	- # Update progress value in GUI
184	- progress = nregions -1
185	- if progress:
186	- UpdateProgress = vu.ShowProgress(progress)
187	-
188	- for region in xrange(nregions):
189	- conn.InitializeSpecifiedRegionList()
190	- conn.AddSpecifiedRegion(region)
191	- conn.Update()
192	-
193	- p = vtk.vtkPolyData()
194	- p.DeepCopy(conn.GetOutput())
195	-
196	- polydata_collection.append(p)
197	- if progress:
198	- UpdateProgress(region, _("Splitting disconnected regions..."))
199	-
200	- return polydata_collection	1	+#--------------------------------------------------------------------------
		2	+# Software: InVesalius - Software de Reconstrucao 3D de Imagens Medicas
		3	+# Copyright: (C) 2001 Centro de Pesquisas Renato Archer
		4	+# Homepage: http://www.softwarepublico.gov.br
		5	+# Contact: invesalius@cti.gov.br
		6	+# License: GNU - GPL 2 (LICENSE.txt/LICENCA.txt)
		7	+#--------------------------------------------------------------------------
		8	+# Este programa e software livre; voce pode redistribui-lo e/ou
		9	+# modifica-lo sob os termos da Licenca Publica Geral GNU, conforme
		10	+# publicada pela Free Software Foundation; de acordo com a versao 2
		11	+# da Licenca.
		12	+#
		13	+# Este programa eh distribuido na expectativa de ser util, mas SEM
		14	+# QUALQUER GARANTIA; sem mesmo a garantia implicita de
		15	+# COMERCIALIZACAO ou de ADEQUACAO A QUALQUER PROPOSITO EM
		16	+# PARTICULAR. Consulte a Licenca Publica Geral GNU para obter mais
		17	+# detalhes.
		18	+#--------------------------------------------------------------------------
		19	+
		20	+import sys
		21	+
		22	+import vtk
		23	+import wx
		24	+from wx.lib.pubsub import pub as Publisher
		25	+
		26	+import vtk_utils as vu
		27	+
		28	+# Update progress value in GUI
		29	+UpdateProgress = vu.ShowProgress()
		30	+
		31	+def ApplyDecimationFilter(polydata, reduction_factor):
		32	+ """
		33	+ Reduce number of triangles of the given vtkPolyData, based on
		34	+ reduction_factor.
		35	+ """
		36	+ # Important: vtkQuadricDecimation presented better results than
		37	+ # vtkDecimatePro
		38	+ decimation = vtk.vtkQuadricDecimation()
		39	+ decimation.SetInputData(polydata)
		40	+ decimation.SetTargetReduction(reduction_factor)
		41	+ decimation.GetOutput().ReleaseDataFlagOn()
		42	+ decimation.AddObserver("ProgressEvent", lambda obj, evt:
		43	+ UpdateProgress(decimation, "Reducing number of triangles..."))
		44	+ return decimation.GetOutput()
		45	+
		46	+def ApplySmoothFilter(polydata, iterations, relaxation_factor):
		47	+ """
		48	+ Smooth given vtkPolyData surface, based on iteration and relaxation_factor.
		49	+ """
		50	+ smoother = vtk.vtkSmoothPolyDataFilter()
		51	+ smoother.SetInputData(polydata)
		52	+ smoother.SetNumberOfIterations(iterations)
		53	+ smoother.SetFeatureAngle(80)
		54	+ smoother.SetRelaxationFactor(relaxation_factor)
		55	+ smoother.FeatureEdgeSmoothingOn()
		56	+ smoother.BoundarySmoothingOn()
		57	+ smoother.GetOutput().ReleaseDataFlagOn()
		58	+ smoother.AddObserver("ProgressEvent", lambda obj, evt:
		59	+ UpdateProgress(smoother, "Smoothing surface..."))
		60	+
		61	+ return smoother.GetOutput()
		62	+
		63	+
		64	+
		65	+def FillSurfaceHole(polydata):
		66	+ """
		67	+ Fill holes in the given polydata.
		68	+ """
		69	+ # Filter used to detect and fill holes. Only fill
		70	+ print "Filling polydata"
		71	+ filled_polydata = vtk.vtkFillHolesFilter()
		72	+ filled_polydata.SetInputData(polydata)
		73	+ filled_polydata.SetHoleSize(500)
		74	+ return filled_polydata.GetOutput()
		75	+
		76	+def CalculateSurfaceVolume(polydata):
		77	+ """
		78	+ Calculate the volume from the given polydata
		79	+ """
		80	+ # Filter used to calculate volume and area from a polydata
		81	+ measured_polydata = vtk.vtkMassProperties()
		82	+ measured_polydata.SetInputData(polydata)
		83	+ return measured_polydata.GetVolume()
		84	+
		85	+def CalculateSurfaceArea(polydata):
		86	+ """
		87	+ Calculate the volume from the given polydata
		88	+ """
		89	+ # Filter used to calculate volume and area from a polydata
		90	+ measured_polydata = vtk.vtkMassProperties()
		91	+ measured_polydata.SetInputData(polydata)
		92	+ return measured_polydata.GetSurfaceArea()
		93	+
		94	+def Merge(polydata_list):
		95	+ append = vtk.vtkAppendPolyData()
		96	+
		97	+ for polydata in polydata_list:
		98	+ triangle = vtk.vtkTriangleFilter()
		99	+ triangle.SetInputData(polydata)
		100	+ triangle.Update()
		101	+ append.AddInputData(triangle.GetOutput())
		102	+
		103	+ append.Update()
		104	+ clean = vtk.vtkCleanPolyData()
		105	+ clean.SetInputData(append.GetOutput())
		106	+ clean.Update()
		107	+
		108	+ return append.GetOutput()
		109	+
		110	+def Export(polydata, filename, bin=False):
		111	+ writer = vtk.vtkXMLPolyDataWriter()
		112	+ print filename, type(filename)
		113	+ writer.SetFileName(filename.encode('utf-8'))
		114	+ if bin:
		115	+ writer.SetDataModeToBinary()
		116	+ else:
		117	+ writer.SetDataModeToAscii()
		118	+ writer.SetInputData(polydata)
		119	+ writer.Write()
		120	+
		121	+def Import(filename):
		122	+ reader = vtk.vtkXMLPolyDataReader()
		123	+ if isinstance(filename, unicode):
		124	+ reader.SetFileName(filename.encode(wx.GetDefaultPyEncoding()))
		125	+ else:
		126	+ reader.SetFileName(filename)
		127	+ reader.Update()
		128	+ return reader.GetOutput()
		129	+
		130	+def JoinSeedsParts(polydata, point_id_list):
		131	+ """
		132	+ The function require vtkPolyData and point id
		133	+ from vtkPolyData.
		134	+ """
		135	+ conn = vtk.vtkPolyDataConnectivityFilter()
		136	+ conn.SetInputData(polydata)
		137	+ conn.SetExtractionModeToPointSeededRegions()
		138	+ UpdateProgress = vu.ShowProgress(1 + len(point_id_list))
		139	+ pos = 1
		140	+ for seed in point_id_list:
		141	+ conn.AddSeed(seed)
		142	+ UpdateProgress(pos, _("Analysing selected regions..."))
		143	+ pos += 1
		144	+
		145	+ conn.AddObserver("ProgressEvent", lambda obj, evt:
		146	+ UpdateProgress(conn, "Getting selected parts"))
		147	+ conn.Update()
		148	+
		149	+ result = vtk.vtkPolyData()
		150	+ result.DeepCopy(conn.GetOutput())
		151	+ return result
		152	+
		153	+def SelectLargestPart(polydata):
		154	+ """
		155	+ """
		156	+ UpdateProgress = vu.ShowProgress(1)
		157	+ conn = vtk.vtkPolyDataConnectivityFilter()
		158	+ conn.SetInputData(polydata)
		159	+ conn.SetExtractionModeToLargestRegion()
		160	+ conn.AddObserver("ProgressEvent", lambda obj, evt:
		161	+ UpdateProgress(conn, "Getting largest part..."))
		162	+ conn.Update()
		163	+
		164	+ result = vtk.vtkPolyData()
		165	+ result.DeepCopy(conn.GetOutput())
		166	+ return result
		167	+
		168	+def SplitDisconectedParts(polydata):
		169	+ """
		170	+ """
		171	+ conn = vtk.vtkPolyDataConnectivityFilter()
		172	+ conn.SetInputData(polydata)
		173	+ conn.SetExtractionModeToAllRegions()
		174	+ conn.Update()
		175	+
		176	+ nregions = conn.GetNumberOfExtractedRegions()
		177	+
		178	+ conn.SetExtractionModeToSpecifiedRegions()
		179	+ conn.Update()
		180	+
		181	+ polydata_collection = []
		182	+
		183	+ # Update progress value in GUI
		184	+ progress = nregions -1
		185	+ if progress:
		186	+ UpdateProgress = vu.ShowProgress(progress)
		187	+
		188	+ for region in xrange(nregions):
		189	+ conn.InitializeSpecifiedRegionList()
		190	+ conn.AddSpecifiedRegion(region)
		191	+ conn.Update()
		192	+
		193	+ p = vtk.vtkPolyData()
		194	+ p.DeepCopy(conn.GetOutput())
		195	+
		196	+ polydata_collection.append(p)
		197	+ if progress:
		198	+ UpdateProgress(region, _("Splitting disconnected regions..."))
		199	+
		200	+ return polydata_collection