image.py

来自「机器人人3D仿真工具,可以加入到Simbad仿真环境下应用。」· Python 代码 · 共 49 行

# Desc: Generic image panel
# Author: Andrew Howard
# Date: 19 Sep 2004
# CVS: $Id: image.py,v 1.2 2006/02/10 16:34:24 inspectorg Exp $

import sys
from wxPython.wx import *


class gzImagePanel(wxPanel):
    """Generic image panel."""

    def __init__(self, parent, pos, size):

        wxPanel.__init__(self, parent, -1, pos, size)
        EVT_PAINT(self, self.OnImagePaint)

        self.image = wxEmptyImage(size[0], size[1])
        self.image_size = size[0] * size[1] * 3
        self.bitmap = None

        return


    def SetData(self, image):
        """Set the image data; must be an RGB string."""

        self.image.SetData(image[:self.image_size])
        self.bitmap = wxBitmapFromImage(self.image)
        return


    def SaveFile(self, filename, type):
        """Save the image to a file."""

        self.image.SaveFile(filename, type)
        return
    

    def OnImagePaint(self, event):
        """Handle paint events"""

        dc = wxPaintDC(self)
        if self.bitmap:
            dc.DrawBitmap(self.bitmap, 0, 0, False)

        return