-
Notifications
You must be signed in to change notification settings - Fork 0
/
thumbnailgenerator.py
139 lines (110 loc) · 4.06 KB
/
thumbnailgenerator.py
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
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
#!/usr/bin/env python
import vtk
from PIL import Image, ImageTk
class KeyPressInteractorStyle(vtk.vtkInteractorStyleTrackballCamera):
def __init__(self,renWin,filename,widget,parent=None):
self.renWin = renWin
self.widget = widget
self.filename = filename
self.parent = vtk.vtkRenderWindowInteractor()
if(parent is not None):
self.parent = parent
self.AddObserver("KeyPressEvent",self.keyPress)
def keyPress(self,obj,event):
key = self.parent.GetKeySym()
if key == "Return":
WriteImage(str(self.filename)+".jpg", self.renWin, rgba=False)
#also update the thumbnail widget and close the renderwindow
image = Image.open("Thumbnails/"+str(self.filename)+".jpg")
image = image.resize((150, 150), Image.ANTIALIAS)
photo = ImageTk.PhotoImage(image)
self.widget.mImage.image = photo#reference keeping
self.widget.mImage.configure(image=photo)
def createAutoThumb(filename,path):
print(filename)
colors = vtk.vtkNamedColors()
# Set the background color.
colors.SetColor("BkgColor", [26, 51, 102, 255])
# create a rendering window and renderer
ren = vtk.vtkRenderer()
renWin = vtk.vtkRenderWindow()
renWin.AddRenderer(ren)
renWin.OffScreenRenderingOn()
# create source
reader = vtk.vtkSTLReader()
reader.SetFileName(path)
# mapper
mapper = vtk.vtkPolyDataMapper()
mapper.SetInputConnection(reader.GetOutputPort())
# actor
actor = vtk.vtkActor()
actor.SetMapper(mapper)
# color the actor
actor.GetProperty().SetColor(colors.GetColor3d("Grey"))
# assign actor to the renderer
ren.AddActor(actor)
ren.SetBackground(colors.GetColor3d("BkgColor"))
renWin.Render()
WriteImage(filename+".jpg", renWin, rgba=False)
def createManualThumb(filename,path,widget):
colors = vtk.vtkNamedColors()
# Set the background color.
colors.SetColor("BkgColor", [26, 51, 102, 255])
# create a rendering window and renderer
ren = vtk.vtkRenderer()
renWin = vtk.vtkRenderWindow()
renWin.AddRenderer(ren)
iren = vtk.vtkRenderWindowInteractor()
iren.SetInteractorStyle(KeyPressInteractorStyle(renWin,filename,widget,parent = iren))
iren.SetRenderWindow(renWin)
# create source
reader = vtk.vtkSTLReader()
reader.SetFileName(path)
# mapper
mapper = vtk.vtkPolyDataMapper()
mapper.SetInputConnection(reader.GetOutputPort())
# actor
actor = vtk.vtkActor()
actor.SetMapper(mapper)
# color the actor
actor.GetProperty().SetColor(colors.GetColor3d("Grey"))
# assign actor to the renderer
ren.AddActor(actor)
ren.SetBackground(colors.GetColor3d("BkgColor"))
iren.Initialize()
renWin.Render()
iren.Start()
def WriteImage(fileName, renWin, rgba=True):
"""
Write the render window view to an image file.
Image types supported are:
BMP, JPEG, PNM, PNG, PostScript, TIFF.
The default parameters are used for all writers, change as needed.
:param fileName: The file name, if no extension then PNG is assumed.
:param renWin: The render window.
:param rgba: Used to set the buffer type.
:return:
"""
import os
#check für /Thumbnails folder
if os.path.exists("Thumbnails") == True:
pass
else:
os.mkdir("Thumbnails")
if fileName:
writer = vtk.vtkJPEGWriter()
windowto_image_filter = vtk.vtkWindowToImageFilter()
windowto_image_filter.SetInput(renWin)
windowto_image_filter.SetScale(1) # image quality
if rgba:
windowto_image_filter.SetInputBufferTypeToRGBA()
else:
windowto_image_filter.SetInputBufferTypeToRGB()
# Read from the front buffer.
windowto_image_filter.ReadFrontBufferOff()
windowto_image_filter.Update()
writer.SetFileName("Thumbnails/"+fileName)
writer.SetInputConnection(windowto_image_filter.GetOutputPort())
writer.Write()
else:
raise RuntimeError('Need a filename.')