def renderGUI():
"""
Renders the GUI for the script.
"""
global G
# find the selection set and update some selection
# related flags
haveEmpty = False
haveCamera = False
emptyName = ""
G.selection = Object.GetSelected()
G.curempty = None
if G.selection is not None and len(G.selection) > 0:
mso = G.selection[0]
msotype = mso.getType()
if msotype == 'Empty':
haveEmpty = True
G.curempty = mso
emptyName = G.curempty.getName()
elif msotype == 'Camera':
haveCamera = True
emptyHasCoords = G.coordmap.has_key(emptyName)
removeUnknownsFromCoords()
# clear any buttons that need to have set states
G.buttons.add = G.buttons.delete = None
# clear the window
c = COLOR_BACKGROUND
BGL.glClearColor(c[0], c[1], c[2], c[3])
BGL.glClear(BGL.GL_COLOR_BUFFER_BIT)
# paint the image in the background
if G.image is not None:
#drawImage(G.image, G.imgpos, G.iw, G.ih, G.zoom)
Draw.Image(G.image, G.imgpos[0], G.imgpos[1], G.zoom, G.zoom)
# paint 2D vertices in the image
BGL.glPushAttrib(BGL.GL_POINT_BIT | BGL.GL_CURRENT_BIT)
BGL.glPointSize(POINT_SIZE)
x0 = int(G.imgpos[0] / G.zoom)
y0 = int(G.imgpos[1] / G.zoom)
def drawvc(ec):
emptyname, coord = ec
if Object.Get(emptyname) in G.selection:
c = COLOR_VERTSEL
else:
c = COLOR_VERTUNSEL
BGL.glColor4f(c[0], c[1], c[2], c[3])
BGL.glVertex2f(G.zoom*(coord[0]+x0), G.zoom*(coord[1]+y0))
BGL.glBegin(BGL.GL_POINTS)
map(drawvc, G.coordmap.items())
BGL.glEnd()
BGL.glPopAttrib()
# if we're in add mode then draw some extra stuff
if G.mode == MODE_ADD:
xm,ym = map(int, getWMCoords())
xm -= 10
ym += 10
(xw,yw,ww,hw) = getWinRect()
# draw crosshairs
c = COLOR_CROSSHAIRS
BGL.glColor4f(c[0], c[1], c[2], c[3])
verts = [ (xm,0), (xm,hw), (0,ym), (ww,ym) ]
BGL.glBegin(BGL.GL_LINES)
map(lambda x: BGL.glVertex2d(x[0],x[1]), verts)
BGL.glEnd()
#############################################
# UNCOMMENT THIS SECTION FOR A COOL MINIMAP
# EFFECT - NOT VERY USEFUL THOUGH
#
## draw "minimap" background
#c = COLOR_MINIMAP
#BGL.glColor4f(c[0], c[1], c[2], c[3])
#verts = [ (119,10), (221,10), (221,111), (119,111) ]
#BGL.glBegin(BGL.GL_QUADS)
#map(lambda x: BGL.glVertex2i(x[0],x[1]), verts)
#BGL.glEnd()
#
## paint the image into the minimap
#ix,iy = wc2ic((xm,ym))
#ix,iy = map(int, [ix, iy])
#drawImage(G.ibuf, (120,10), G.iw, G.ih, 10.0, (ix-5,iy-5,10,10))
#
# END OF MINIMAP SECTION
#############################################
# paint the current empty name
if haveEmpty:
c = COLOR_TEXT
BGL.glColor4d(c[0], c[1], c[2], c[3])
BGL.glRasterPos2i(220, 10)
Draw.Text(emptyName, 'small')
# paint the normal GUI buttons
G.buttons.quit = Draw.PushButton('Quit', BUTTON_QUIT, 5, 5, 100, 20, 'Exits the script.')
G.buttons.load = Draw.PushButton('Load Image', BUTTON_LOAD, 5, 25, 100, 20, 'Loads an image.')
G.buttons.zoom = Draw.Number('Zoom', BUTTON_ZOOM, 5, 45, 100, 20, G.zoom, ZOOM_MIN, ZOOM_MAX, 'Adjusts image zoom.')
# paint camera-specific stuff
if haveCamera:
G.buttons.fullopt = Draw.Toggle('Full Optimization', BUTTON_FULLOPT, 110, 5, 120, 20, G.fullopt, 'Full or partial optimization.')
G.buttons.coplanar = Draw.Toggle('Coplanar', BUTTON_COPLANAR, 110, 25, 120, 20, G.coplanar, 'Coplanar or non-coplanar target.')
# Origin offset is not currently working in the Tsai module.
# It should be brought back here when it is.
#G.buttons.ofsz = Draw.Number('OfsZ', BUTTON_OFSZ, 110, 50, 100, 20, G.ofsz, OFS_MIN, OFS_MAX, 'Z origin offset.')
#G.buttons.ofsy = Draw.Number('OfsY', BUTTON_OFSY, 110, 70, 100, 20, G.ofsy, OFS_MIN, OFS_MAX, 'Y origin offset.')
#G.buttons.ofsx = Draw.Number('OfsX', BUTTON_OFSX, 110, 90, 100, 20, G.ofsx, OFS_MIN, OFS_MAX, 'X origin offset.')
G.buttons.calibrate = Draw.PushButton('Calibrate', BUTTON_CALIBRATE, 235, 5, 100, 20, 'Calibrates the selected camera.')
# paint empty-specific stuff
elif haveEmpty and (G.mode == MODE_NORMAL):
if emptyHasCoords:
G.buttons.delete = Draw.PushButton('Delete', BUTTON_DELETE, 110, 5, 100, 20, 'Adds an image calibration coordinate.')
else:
G.buttons.add = Draw.PushButton('Add', BUTTON_ADD, 110, 5, 100, 20, 'Removes an image calibration coordinate.')
def renderGUI():
"""
Renders the GUI for the script.
"""
global G
# find the selection set and update some selection
# related flags
haveEmpty = False
haveCamera = False
emptyName = ""
G.selection = Object.GetSelected()
G.curempty = None
if G.selection is not None and len(G.selection) > 0:
mso = G.selection[0]
msotype = mso.getType()
if msotype == 'Empty':
haveEmpty = True
G.curempty = mso
emptyName = G.curempty.getName()
elif msotype == 'Camera':
haveCamera = True
emptyHasCoords = G.coordmap.has_key(emptyName)
removeUnknownsFromCoords()
# clear any buttons that need to have set states
G.buttons.add = G.buttons.delete = None
# clear the window
c = COLOR_BACKGROUND
BGL.glClearColor(c[0], c[1], c[2], c[3])
BGL.glClear(BGL.GL_COLOR_BUFFER_BIT)
# paint the image in the background
if G.image is not None:
#drawImage(G.image, G.imgpos, G.iw, G.ih, G.zoom)
Draw.Image(G.image, G.imgpos[0], G.imgpos[1], G.zoom, G.zoom)
# paint 2D vertices in the image
BGL.glPushAttrib(BGL.GL_POINT_BIT | BGL.GL_CURRENT_BIT)
BGL.glPointSize(POINT_SIZE)
x0 = int(G.imgpos[0] / G.zoom)
y0 = int(G.imgpos[1] / G.zoom)
def drawvc(ec):
emptyname, coord = ec
if Object.Get(emptyname) in G.selection:
c = COLOR_VERTSEL
else:
c = COLOR_VERTUNSEL
BGL.glColor4f(c[0], c[1], c[2], c[3])
BGL.glVertex2f(G.zoom * (coord[0] + x0), G.zoom * (coord[1] + y0))
BGL.glBegin(BGL.GL_POINTS)
map(drawvc, G.coordmap.items())
BGL.glEnd()
BGL.glPopAttrib()
# if we're in add mode then draw some extra stuff
if G.mode == MODE_ADD:
xm, ym = map(int, getWMCoords())
xm -= 10
ym += 10
(xw, yw, ww, hw) = getWinRect()
# draw crosshairs
c = COLOR_CROSSHAIRS
BGL.glColor4f(c[0], c[1], c[2], c[3])
verts = [(xm, 0), (xm, hw), (0, ym), (ww, ym)]
BGL.glBegin(BGL.GL_LINES)
map(lambda x: BGL.glVertex2d(x[0], x[1]), verts)
BGL.glEnd()
#############################################
# UNCOMMENT THIS SECTION FOR A COOL MINIMAP
# EFFECT - NOT VERY USEFUL THOUGH
#
## draw "minimap" background
#c = COLOR_MINIMAP
#BGL.glColor4f(c[0], c[1], c[2], c[3])
#verts = [ (119,10), (221,10), (221,111), (119,111) ]
#BGL.glBegin(BGL.GL_QUADS)
#map(lambda x: BGL.glVertex2i(x[0],x[1]), verts)
#BGL.glEnd()
#
## paint the image into the minimap
#ix,iy = wc2ic((xm,ym))
#ix,iy = map(int, [ix, iy])
#drawImage(G.ibuf, (120,10), G.iw, G.ih, 10.0, (ix-5,iy-5,10,10))
#
# END OF MINIMAP SECTION
#############################################
# paint the current empty name
if haveEmpty:
c = COLOR_TEXT
BGL.glColor4d(c[0], c[1], c[2], c[3])
BGL.glRasterPos2i(220, 10)
Draw.Text(emptyName, 'small')
# paint the normal GUI buttons
G.buttons.quit = Draw.PushButton('Quit', BUTTON_QUIT, 5, 5, 100, 20,
'Exits the script.')
G.buttons.load = Draw.PushButton('Load Image', BUTTON_LOAD, 5, 25, 100, 20,
'Loads an image.')
G.buttons.zoom = Draw.Number('Zoom', BUTTON_ZOOM, 5, 45, 100, 20, G.zoom,
ZOOM_MIN, ZOOM_MAX, 'Adjusts image zoom.')
# paint camera-specific stuff
if haveCamera:
G.buttons.fullopt = Draw.Toggle('Full Optimization', BUTTON_FULLOPT,
110, 5, 120, 20, G.fullopt,
'Full or partial optimization.')
G.buttons.coplanar = Draw.Toggle('Coplanar', BUTTON_COPLANAR, 110, 25,
120, 20, G.coplanar,
'Coplanar or non-coplanar target.')
# Origin offset is not currently working in the Tsai module.
# It should be brought back here when it is.
#G.buttons.ofsz = Draw.Number('OfsZ', BUTTON_OFSZ, 110, 50, 100, 20, G.ofsz, OFS_MIN, OFS_MAX, 'Z origin offset.')
#G.buttons.ofsy = Draw.Number('OfsY', BUTTON_OFSY, 110, 70, 100, 20, G.ofsy, OFS_MIN, OFS_MAX, 'Y origin offset.')
#G.buttons.ofsx = Draw.Number('OfsX', BUTTON_OFSX, 110, 90, 100, 20, G.ofsx, OFS_MIN, OFS_MAX, 'X origin offset.')
G.buttons.calibrate = Draw.PushButton(
'Calibrate', BUTTON_CALIBRATE, 235, 5, 100, 20,
'Calibrates the selected camera.')
# paint empty-specific stuff
elif haveEmpty and (G.mode == MODE_NORMAL):
if emptyHasCoords:
G.buttons.delete = Draw.PushButton(
'Delete', BUTTON_DELETE, 110, 5, 100, 20,
'Adds an image calibration coordinate.')
else:
G.buttons.add = Draw.PushButton(
'Add', BUTTON_ADD, 110, 5, 100, 20,
'Removes an image calibration coordinate.')
