spacepaste

Paste details

reply | raw | toggle line wrapping

  1.  
  2. # create a canvas on top of a blank bitmap
  3. # any canvas drawings can now be saved to a standard image file
  4. # tested with Python27 and wxPython28 by vegaseat 05jan2011
  5. import wxversion
  6. wxversion.select('3.0')
  7. import sys
  8. import os
  9. import wx
  10. class MyFrame(wx.Frame):
  11. def __init__(self, parent=None, id=-1, title=None, gerber=None):
  12. wx.Frame.__init__(self, parent, id, title, size=(380,400))
  13. self.statbmp = wx.StaticBitmap(self)
  14. self.gerber = gerber
  15. self.draw_image()
  16. self.save_image()
  18. #self.Destroy()
  20. def draw_image(self):
  22. if self.gerber:
  23. print self.gerber
  24. if os.path.isfile(os.path.abspath(self.gerber)):
  25. import re
  26. with open(self.gerber, 'r') as f:
  27. lines = f.readlines()
  28. _max_dim = None
  29. _min_dim = None
  30. minx = miny = maxy = maxx = None
  31. pts = []
  32. for line in lines:
  33. m = re.match(r"^X(-?\d+)Y(-?\d+)D(\d+)", line)
  34. if m:
  35. x, y, d = m.groups()
  36. x = int(x)
  37. y = int(y)
  38. d = int(d)
  39. if _max_dim is None:
  40. _min_dim = [x, y]
  41. _max_dim = [x, y]
  43. else:
  44. if x<_min_dim[0]:
  45. _min_dim[0] = x
  46. if y<_min_dim[1]:
  47. _min_dim[1] = y
  48. if x>_max_dim[0]:
  49. _max_dim[0] = x
  50. if y>_max_dim[1]:
  51. _max_dim[1] = y
  52. pts.append((x, y, d))
  53. print 'max [{}] min [{}]'.format(_max_dim, _min_dim)
  54. # select the width and height of the blank bitmap
  55. # should fit the client frame
  56. w, h = _max_dim[0] - _min_dim[0], _max_dim[1] - _min_dim[1]
  57. divisor = 10000
  58. w = w/divisor
  59. h = h/divisor
  60. print 'w, h [{}]'.format((w, h))
  61. # create the blank bitmap as a draw background
  62. draw_bmp = wx.EmptyBitmap(w, h)
  63. # create the canvas on top of the draw_bmp
  64. canvas_dc = wx.MemoryDC(draw_bmp)
  65. # fill the canvas white
  66. canvas_dc.SetBrush(wx.Brush('white'))
  67. canvas_dc.Clear()
  68. # draw a bunch of circles ...
  69. # pen colour
  70. canvas_dc.SetPen(wx.Pen('red', 1))
  71. # fill colour
  72. canvas_dc.SetBrush(wx.Brush('orange'))
  73. last = None
  74. for pt in pts:
  75. x = ((pt[0] - _min_dim[0]) / divisor)
  76. y = ((pt[1] - _min_dim[1]) / divisor)
  77. d = pt[2]
  78. if d == 2:
  79. last = (x, y)
  80. elif d == 1:
  81. print 'DrawLine ({}, {}) ({}, {})'.format(last[0], last[1], x, y)
  82. canvas_dc.DrawLine(last[0], last[1], x, y)
  83. elif d == 3:
  84. canvas_dc.DrawCircle(x, y, 10)
  85. else:
  86. # select the width and height of the blank bitmap
  87. # should fit the client frame
  88. w, h = 340, 340
  89. # create the blank bitmap as a draw background
  90. draw_bmp = wx.EmptyBitmap(w, h)
  91. # create the canvas on top of the draw_bmp
  92. canvas_dc = wx.MemoryDC(draw_bmp)
  93. # fill the canvas white
  94. canvas_dc.SetBrush(wx.Brush('white'))
  95. canvas_dc.Clear()
  96. # draw a bunch of circles ...
  97. # pen colour
  98. canvas_dc.SetPen(wx.Pen('red', 1))
  99. # fill colour
  100. canvas_dc.SetBrush(wx.Brush('orange'))
  101. for x in range(10, 180, 10):
  102. y = x
  103. r = x%30
  104. canvas_dc.DrawCircle(x, y, r)
  105. # now put the canvas drawing into a bitmap to display it
  106. # remember the canvas is on top of the draw_bmp
  107. self.statbmp.SetBitmap(draw_bmp)
  108. def save_image(self):
  109. """save the drawing"""
  110. finished_image = self.statbmp.GetBitmap()
  111. #finished_image.SaveFile("mydrawing.png", wx.BITMAP_TYPE_PNG)
  112. finished_image.SaveFile("mydrawing.jpg", wx.BITMAP_TYPE_JPEG)
  113. app = wx.App(0)
  114. MyFrame(title='canvas draw and save', gerber=sys.argv[1]).Show()
  115. app.MainLoop()
  116. # help(wx.PaintDC)
  117.  
Powered by spacepaste. Paste removal.