nmz787/pbm_viewer.py

## pbm_viewer.py
__license__ = """
This library is free software; you can redistribute it and/or
modify it under the terms of the GNU Lesser General Public License as
published by the Free Software Foundation; either version 2.1 of the
License, or (at your option) any later version.

As a special exception, the copyright holders of this library
hereby recind Section 3 of the GNU Lesser General Public License. This
means that you MAY NOT apply the terms of the ordinary GNU General
Public License instead of this License to any given copy of the
Library. This has been done to prevent users of the Library from being
denied access or the ability to use future improvements.

This library is distributed in the hope that it will be useful, but
WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser
General Public License for more details.

You should have received a copy of the GNU Lesser General Public
License along with this library; if not, write to the Free Software
Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
"""


import wx

class BufferedCanvas(wx.Panel):
    """
    Implements a double-buffered, flicker-free canvas widget.

    Standard usage is to subclass this class, and override the
    draw method.  The draw method is passed a device context, which
    should be used to do your drawing.

    Also, you should NOT call dc.BeginDrawing() and dc.EndDrawing() --
    these methods are automatically called for you, although you still
    need to manually clear the device context.

    If you want to force a redraw (for whatever reason), you should
    call the update method.  This is because the draw method is never
    called as a result of an EVT_PAINT event.
    """

    # These are our two buffers.  Just be aware that when the buffers
    # are flipped, the REFERENCES are swapped.  So I wouldn't want to
    # try holding onto explicit references to one or the other ;)
    buffer = None
    backbuffer = None

    def __init__(self,
                 parent,
                 ID=-1,
                 pos=wx.DefaultPosition,
                 size=wx.DefaultSize,
                 style=wx.NO_FULL_REPAINT_ON_RESIZE):
        wx.Panel.__init__(self,parent,ID,pos,size,style)

        # Bind events
        self.Bind(wx.EVT_PAINT, self.onPaint)
        self.Bind(wx.EVT_SIZE, self.onSize)

        # Disable background erasing (flicker-licious)
        def disable_event(*pargs,**kwargs):
            pass # the sauce, please
        self.Bind(wx.EVT_ERASE_BACKGROUND, disable_event)

        # Ensure that the buffers are setup correctly
        self.onSize(None)

    ##
    ## General methods
    ##

    def draw(self,dc):
        """
        Stub: called when the canvas needs to be re-drawn.
        """
        pass


    def flip(self):
        """
        Flips the front and back buffers.
        """
        self.buffer,self.backbuffer = self.backbuffer,self.buffer
        self.Refresh()


    def update(self):
        """
        Causes the canvas to be updated.
        """
        dc = wx.MemoryDC()
        dc.SelectObject(self.backbuffer)
        #dc.BeginDrawing()
        self.draw(dc)
        #dc.EndDrawing()
        self.flip()

    ##
    ## Event handlers
    ##

    def onPaint(self, event):
        # Blit the front buffer to the screen
        dc = wx.BufferedPaintDC(self, self.buffer)


    def onSize(self, event):
        # Here we need to create a new off-screen buffer to hold
        # the in-progress drawings on.
        width,height = self.GetClientSize()
        if width == 0:
            width = 1
        if height == 0:
            height = 1
        self.buffer = wx.Bitmap(width,height)
        self.backbuffer = wx.Bitmap(width,height)

        # Now update the screen
        self.update()


class TestCanvas(BufferedCanvas):

    def __init__(self,parent,controls, ID=-1):
        self.controls = controls
        self.image = None
        BufferedCanvas.__init__(self,parent,ID)


    def draw(self, dc):
        dc.SetBackground(wx.Brush("White"))
        dc.Clear()
        W,H = self.GetSize()
        x = int(self.controls.x_txt.GetValue())
        y = int(self.controls.y_txt.GetValue())
        if self.image:
            y = min(y, self.y_max)
            x = min(x, self.x_max)
            if W>self.x_max:
              x=0
            if H>self.y_max:
              y=0
            dc.SetBrush(wx.BLUE_BRUSH)
            dc.SetPen(wx.Pen('Black', 1))
            for yy in range(min(H, self.y_max)):
                for xx in range(min(W, self.x_max)):
                    try:
                        y_row_offset = (y+yy)*self.x_max
                        pixel_number = x+xx+ y_row_offset
                        byte_number = int(pixel_number/8)
                        r = 8-((pixel_number+1) % 8)
                        if r == 8:
                            r=0
                        pixel = self.image[byte_number]>>r & 1
                        if pixel:
                            dc.DrawPoint(xx, yy)
                    except IndexError:
                        print('x {} xx {} y {} yy {} w {} h {} x_max {} y_max {}'
                              .format(x, xx, y, yy, W, H, self.x_max, self.y_max))
                        raise
        else:
            dc.SetBrush(wx.BLUE_BRUSH)
            dc.SetPen(wx.Pen('Red', 1))
            for xx in range(W):
                dc.DrawPoint(xx, 0)
                dc.DrawPoint(xx, 100)
            # dc.DrawRectangle(x,
            #                  y,
            #                  W,H)


class ControlPanel(wx.Panel):
    def __init__(self, parent):
        wx.Panel.__init__(self, parent=parent)
        self.gs = wx.GridSizer(2,2,0,0)
        x_lbl = wx.StaticText(self, label='X coord')
        self.x_txt = wx.TextCtrl(self, value='0', style = wx.TE_PROCESS_ENTER)
        y_lbl = wx.StaticText(self, label='Y coord')
        self.y_txt = wx.TextCtrl(self, value='0', style = wx.TE_PROCESS_ENTER)
        self.gs.Add(x_lbl, 0, wx.ALL, 1)
        self.gs.Add(self.x_txt, 0, wx.ALL, 1)
        self.gs.Add(y_lbl, 0, wx.ALL, 1)
        self.gs.Add(self.y_txt, 0, wx.ALL, 1)
        self.x_txt.Bind(wx.EVT_TEXT_ENTER, self.textEnter)
        self.y_txt.Bind(wx.EVT_TEXT_ENTER, self.textEnter)
        self.SetSizerAndFit(self.gs)

    def textEnter(self, event):
        print('textEnter')
        self.canvas.update()


class TestFrame(wx.Frame):

    def __init__(self,
                 parent=None,
                 ID=-1,
                 title="CTRL O to open image",
                 pos=wx.DefaultPosition,
                 size=wx.DefaultSize,
                 style=wx.DEFAULT_FRAME_STYLE):
        wx.Frame.__init__(self,parent,ID,title,pos,size,style)
        self.splitter = wx.SplitterWindow(self)
        self.controlPanel = ControlPanel(self.splitter)
        self.canvas = TestCanvas(self.splitter, self.controlPanel)
        self.controlPanel.canvas = self.canvas
        self.splitter.SplitVertically(self.controlPanel, self.canvas)
        self.splitter.SetMinimumPaneSize(30)
        self.Bind(wx.EVT_CLOSE, self.onClose)
        #self.Bind(wx.EVT_KEY_DOWN, self.onKeyPress)
        self.Bind(wx.EVT_MENU, self.onKeyCombo)
        accel_tbl = wx.AcceleratorTable([(wx.ACCEL_CTRL,  ord('O'), self.Id)])
        self.SetAcceleratorTable(accel_tbl)
        #self.image_path = 'Super_Mario_Bros.pbm'
        #self.open_file()

    def onKeyCombo(self, event):
        """"""
        print("You pressed CTRL+O!")
        openFileDialog = wx.FileDialog(self, "Open", "", "",
          "PBM image files (*.pbm)|*.pbm",
          wx.FD_OPEN | wx.FD_FILE_MUST_EXIST)
        openFileDialog.ShowModal()
        self.image_path = openFileDialog.GetPath()
        print('you chose ' + self.image_path)
        openFileDialog.Destroy()
        self.open_file()

    def open_file(self):
        with wx.BusyInfo("Please wait, working..."):
            with open(self.image_path, 'rb') as f:
                self.canvas.image = f.read()
            b = ''
            i=0
            meta = ''#bytearray()
            def get_line():
              nonlocal b, i, meta
              while b!='\n':
                b=self.canvas.image[i]
                b=chr(b)
                i+=1
                meta+=b
            get_line()
            print(meta)
            meta=''
            b=''
            get_line()
            print(meta)
            w,h = map(int, meta.split())
            self.canvas.y_max = h
            self.canvas.x_max = w
            self.canvas.image=self.canvas.image[i:]

    def onKeyPress(self, event):
        keycode = event.GetKeyCode()
        controlDown = event.CmdDown()
        print('{} {}'.format('CTRL' if controlDown else '', keycode))
        # if keycode == wx.WXK_SPACE:
        #     print("you pressed the spacebar!")
        event.Skip()

    def onClose(self,event):
        self.Show(False)
        self.Destroy()


def main():
    app = wx.App()
    frame = TestFrame()
    frame.Show(True)
    app.MainLoop()

if __name__ == '__main__':
    main()
	__license__ = """
	This library is free software; you can redistribute it and/or
	modify it under the terms of the GNU Lesser General Public License as
	published by the Free Software Foundation; either version 2.1 of the
	License, or (at your option) any later version.

	As a special exception, the copyright holders of this library
	hereby recind Section 3 of the GNU Lesser General Public License. This
	means that you MAY NOT apply the terms of the ordinary GNU General
	Public License instead of this License to any given copy of the
	Library. This has been done to prevent users of the Library from being
	denied access or the ability to use future improvements.

	This library is distributed in the hope that it will be useful, but
	WITHOUT ANY WARRANTY; without even the implied warranty of
	MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser
	General Public License for more details.

	You should have received a copy of the GNU Lesser General Public
	License along with this library; if not, write to the Free Software
	Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
	"""


	import wx

	class BufferedCanvas(wx.Panel):
	"""
	Implements a double-buffered, flicker-free canvas widget.

	Standard usage is to subclass this class, and override the
	draw method. The draw method is passed a device context, which
	should be used to do your drawing.

	Also, you should NOT call dc.BeginDrawing() and dc.EndDrawing() --
	these methods are automatically called for you, although you still
	need to manually clear the device context.

	If you want to force a redraw (for whatever reason), you should
	call the update method. This is because the draw method is never
	called as a result of an EVT_PAINT event.
	"""

	# These are our two buffers. Just be aware that when the buffers
	# are flipped, the REFERENCES are swapped. So I wouldn't want to
	# try holding onto explicit references to one or the other ;)
	buffer = None
	backbuffer = None

	def __init__(self,
	parent,
	ID=-1,
	pos=wx.DefaultPosition,
	size=wx.DefaultSize,
	style=wx.NO_FULL_REPAINT_ON_RESIZE):
	wx.Panel.__init__(self,parent,ID,pos,size,style)

	# Bind events
	self.Bind(wx.EVT_PAINT, self.onPaint)
	self.Bind(wx.EVT_SIZE, self.onSize)

	# Disable background erasing (flicker-licious)
	def disable_event(pargs,*kwargs):
	pass # the sauce, please
	self.Bind(wx.EVT_ERASE_BACKGROUND, disable_event)

	# Ensure that the buffers are setup correctly
	self.onSize(None)

	##
	## General methods
	##

	def draw(self,dc):
	"""
	Stub: called when the canvas needs to be re-drawn.
	"""
	pass


	def flip(self):
	"""
	Flips the front and back buffers.
	"""
	self.buffer,self.backbuffer = self.backbuffer,self.buffer
	self.Refresh()


	def update(self):
	"""
	Causes the canvas to be updated.
	"""
	dc = wx.MemoryDC()
	dc.SelectObject(self.backbuffer)
	#dc.BeginDrawing()
	self.draw(dc)
	#dc.EndDrawing()
	self.flip()

	##
	## Event handlers
	##

	def onPaint(self, event):
	# Blit the front buffer to the screen
	dc = wx.BufferedPaintDC(self, self.buffer)


	def onSize(self, event):
	# Here we need to create a new off-screen buffer to hold
	# the in-progress drawings on.
	width,height = self.GetClientSize()
	if width == 0:
	width = 1
	if height == 0:
	height = 1
	self.buffer = wx.Bitmap(width,height)
	self.backbuffer = wx.Bitmap(width,height)

	# Now update the screen
	self.update()


	class TestCanvas(BufferedCanvas):

	def __init__(self,parent,controls, ID=-1):
	self.controls = controls
	self.image = None
	BufferedCanvas.__init__(self,parent,ID)


	def draw(self, dc):
	dc.SetBackground(wx.Brush("White"))
	dc.Clear()
	W,H = self.GetSize()
	x = int(self.controls.x_txt.GetValue())
	y = int(self.controls.y_txt.GetValue())
	if self.image:
	y = min(y, self.y_max)
	x = min(x, self.x_max)
	if W>self.x_max:
	x=0
	if H>self.y_max:
	y=0
	dc.SetBrush(wx.BLUE_BRUSH)
	dc.SetPen(wx.Pen('Black', 1))
	for yy in range(min(H, self.y_max)):
	for xx in range(min(W, self.x_max)):
	try:
	y_row_offset = (y+yy)*self.x_max
	pixel_number = x+xx+ y_row_offset
	byte_number = int(pixel_number/8)
	r = 8-((pixel_number+1) % 8)
	if r == 8:
	r=0
	pixel = self.image[byte_number]>>r & 1
	if pixel:
	dc.DrawPoint(xx, yy)
	except IndexError:
	print('x {} xx {} y {} yy {} w {} h {} x_max {} y_max {}'
	.format(x, xx, y, yy, W, H, self.x_max, self.y_max))
	raise
	else:
	dc.SetBrush(wx.BLUE_BRUSH)
	dc.SetPen(wx.Pen('Red', 1))
	for xx in range(W):
	dc.DrawPoint(xx, 0)
	dc.DrawPoint(xx, 100)
	# dc.DrawRectangle(x,
	# y,
	# W,H)


	class ControlPanel(wx.Panel):
	def __init__(self, parent):
	wx.Panel.__init__(self, parent=parent)
	self.gs = wx.GridSizer(2,2,0,0)
	x_lbl = wx.StaticText(self, label='X coord')
	self.x_txt = wx.TextCtrl(self, value='0', style = wx.TE_PROCESS_ENTER)
	y_lbl = wx.StaticText(self, label='Y coord')
	self.y_txt = wx.TextCtrl(self, value='0', style = wx.TE_PROCESS_ENTER)
	self.gs.Add(x_lbl, 0, wx.ALL, 1)
	self.gs.Add(self.x_txt, 0, wx.ALL, 1)
	self.gs.Add(y_lbl, 0, wx.ALL, 1)
	self.gs.Add(self.y_txt, 0, wx.ALL, 1)
	self.x_txt.Bind(wx.EVT_TEXT_ENTER, self.textEnter)
	self.y_txt.Bind(wx.EVT_TEXT_ENTER, self.textEnter)
	self.SetSizerAndFit(self.gs)

	def textEnter(self, event):
	print('textEnter')
	self.canvas.update()


	class TestFrame(wx.Frame):

	def __init__(self,
	parent=None,
	ID=-1,
	title="CTRL O to open image",
	pos=wx.DefaultPosition,
	size=wx.DefaultSize,
	style=wx.DEFAULT_FRAME_STYLE):
	wx.Frame.__init__(self,parent,ID,title,pos,size,style)
	self.splitter = wx.SplitterWindow(self)
	self.controlPanel = ControlPanel(self.splitter)
	self.canvas = TestCanvas(self.splitter, self.controlPanel)
	self.controlPanel.canvas = self.canvas
	self.splitter.SplitVertically(self.controlPanel, self.canvas)
	self.splitter.SetMinimumPaneSize(30)
	self.Bind(wx.EVT_CLOSE, self.onClose)
	#self.Bind(wx.EVT_KEY_DOWN, self.onKeyPress)
	self.Bind(wx.EVT_MENU, self.onKeyCombo)
	accel_tbl = wx.AcceleratorTable([(wx.ACCEL_CTRL, ord('O'), self.Id)])
	self.SetAcceleratorTable(accel_tbl)
	#self.image_path = 'Super_Mario_Bros.pbm'
	#self.open_file()

	def onKeyCombo(self, event):
	""""""
	print("You pressed CTRL+O!")
	openFileDialog = wx.FileDialog(self, "Open", "", "",
	"PBM image files (.pbm)\|.pbm",
	wx.FD_OPEN \| wx.FD_FILE_MUST_EXIST)
	openFileDialog.ShowModal()
	self.image_path = openFileDialog.GetPath()
	print('you chose ' + self.image_path)
	openFileDialog.Destroy()
	self.open_file()

	def open_file(self):
	with wx.BusyInfo("Please wait, working..."):
	with open(self.image_path, 'rb') as f:
	self.canvas.image = f.read()
	b = ''
	i=0
	meta = ''#bytearray()
	def get_line():
	nonlocal b, i, meta
	while b!='\n':
	b=self.canvas.image[i]
	b=chr(b)
	i+=1
	meta+=b
	get_line()
	print(meta)
	meta=''
	b=''
	get_line()
	print(meta)
	w,h = map(int, meta.split())
	self.canvas.y_max = h
	self.canvas.x_max = w
	self.canvas.image=self.canvas.image[i:]

	def onKeyPress(self, event):
	keycode = event.GetKeyCode()
	controlDown = event.CmdDown()
	print('{} {}'.format('CTRL' if controlDown else '', keycode))
	# if keycode == wx.WXK_SPACE:
	# print("you pressed the spacebar!")
	event.Skip()

	def onClose(self,event):
	self.Show(False)
	self.Destroy()


	def main():
	app = wx.App()
	frame = TestFrame()
	frame.Show(True)
	app.MainLoop()

	if __name__ == '__main__':
	main()