spacepaste

Paste details

reply | raw | toggle line wrapping

  1.  
  2. #!/usr/bin/python
  3. import traceback
  4. import sys, math, os
  5. import copy
  6. from PIL import Image, ImageDraw
  7. import cv2
  8. import numpy as np
  10. #import gui_related
  12. import wx
  14. app = wx.App(False)
  16. #print wx.GetDisplaySize()
  19. lines=[]
  20. filename='noyes2.str'
  21. with open(os.path.join(os.path.dirname(__file__), filename), 'rb') as fl:
  22. lines=fl.read()
  24. bmp=[]
  26. print 'A5 at: ', lines.index('\xA5')
  27. pixes=[]
  28. pix=[]
  29. class fib_pattern(object):
  30. def write_as_12_bit(self, lines):
  31. #fl1=open('out.txt1', 'wb')
  32. i=lines.index('\xd7') - 3
  33. firstTime=True
  34. size = (4096, 4096) # size of the image to create
  35. im = Image.new('L', size) # create the image, 'L' for greyscale
  36. self.maxX=0
  37. self.maxY=0
  38. self.minY=4096
  39. self.minX=4096
  40. while True:
  41. try:
  42. b1 = ord(lines[i])
  43. b2 = ord(lines[i+1])
  44. b3 = ord(lines[i+2])
  45. b4 = ord(lines[i+3])
  47. #something to do with how nybble order gets packed by C compilers
  48. nib1 = b2 >> 4
  49. nib2 = b2 & 0xF
  50. x = b1 | (nib2<<8)
  51. y = (b3<<4) | nib1
  53. dwell = b4
  55. #keep some statistics
  56. if x>self.maxX:
  57. self.maxX=x
  58. if y>self.maxY:
  59. self.maxY=y
  60. if x<self.minX:
  61. self.minX=x
  62. if y<self.minY:
  63. self.minY=y
  65. #write the pixel to the image
  66. im.putpixel((x,y), dwell)
  68. #advance by 4 bytes
  69. i+=4
  71. except IndexError:
  72. break
  73. im.save('out_test1.png', 'PNG')
  74. print "maxX %d, maxY %d" % (self.maxX, self.maxY)
  75. print "minX %d, minY %d" % (self.minX, self.minY)
  76. print 'num pixels: %d' % (i/4)
  77. self.numPixels = (i/4)
  79. def write_as_video(self, lines):
  80. #fl1=open('out.txt1', 'wb')
  81. i=lines.index('\xd7') - 3
  82. print 'i', i
  83. firstTime=True
  84. size = (4096, 4096) # size of the image to create
  85. #im = Image.new('L', size) # create the image, 'L' for greyscale
  87. width = (self.maxX - self.minX) + 1
  88. height = (self.maxY - self.minY) + 1
  90. print 'width %d height %d maxX %d minX %d maxY %d minY %d' % (width, height, self.maxX, self.minX, self.maxY, self.minY)
  92. #out = cv2.VideoWriter('output.avi',cv2.cv.CV_FOURCC('D', 'I', 'V', 'X'), 20.0, (width,height))
  93. #fourcc = cv2.cv.CV_FOURCC('m', 'p', '4', 'v') # note the lower case
  94. #fourcc = cv2.cv.CV_FOURCC('M','J','P','G')
  95. #fourcc = cv2.cv.CV_FOURCC('m', 'p', '4', '2')
  96. #fourcc = cv2.cv.CV_FOURCC('i','Y','U', 'V')#'D','I','V','X') #CAP_PROP
  97. #fourcc = cv2.cv.CV_FOURCC(*'XVID')
  98. #print fourcc
  99. #out = cv2.VideoWriter()
  100. #success = out.open('output.avi',0,30,( width, height),True)
  101. m = max(width, height)
  102. #out = cv2.VideoWriter("test.avi", cv2.cv.CV_FOURCC('i','Y','u','v'), #'F', 'M', 'P', '4'),'H', '2', '6', '4'
  103. # 30, (width, height), False) #import cv syntax CreateVideoWriter
  104. #out = cv2.VideoWriter("testmp42.mp4", cv2.cv.CV_FOURCC('M','P','4','2'), #'F', 'M', 'P', '4'),'H', '2', '6', '4' 'D','I','V','3' (works oks)
  105. # 90, (width, height), False) #'X','2','6','4' width, height
  106. #out = cv2.VideoWriter("testmp4.avi", cv2.cv.CV_FOURCC('D','I','V','3'), 30, (width, height), True)
  107. #out = cv2.VideoWriter("test_x264.avi", cv2.cv.CV_FOURCC('X','2','6','4'), 30, (width, height), True)
  108. #print 'video opened? %s' % out.isOpened()
  109. #print success
  110. blank_image = np.zeros((width, height,1), np.uint8)
  111. #blank_image = cv2.cv.CreateImage((m, m), 8, 1)
  112. #cv_image = cv2.cv.CreateImage((height, width), cv2.IPL_DEPTH_8U,1)
  113. cv2.namedWindow('FIB pattern raster sequence')#, cv2.CV_WINDOW_NORMAL|cv2.CV_WINDOW_KEEPRATIO | cv2.CV_GUI_EXPANDED)
  114. lastKey=0
  115. #available_screen_size = gui_related.get_available_screen_size()
  117. topBarHeight, sideBarWidth, widthAvailable, heightAvailable = wx.ClientDisplayRect()
  118. heightAvailable -= topBarHeight # account for the highGui window's title bar
  120. if width > widthAvailable:
  121. scaleWidth=float(widthAvailable)/width
  122. else:
  123. scaleWidth = 1
  124. if height > heightAvailable:
  125. scaleHeight = float(heightAvailable)/height
  126. else:
  127. scaleHeight = 1
  128. scale = min(scaleWidth, scaleHeight)
  129. scaled_frame_size = (int(width*scale), int(height*scale))
  131. while True:
  132. try:
  134. b1 = ord(lines[i])
  135. b2 = ord(lines[i+1])
  136. b3 = ord(lines[i+2])
  137. b4 = ord(lines[i+3])
  139. #something to do with how nybble order gets packed by C compilers
  140. nib1 = b2 >> 4
  141. nib2 = b2 & 0xF
  142. x = b1 | (nib2<<8)
  143. y = (b3<<4) | nib1
  145. dwell = b4
  148. #write the pixel to the image
  149. #im.putpixel((x,y), dwell)
  150. blank_image[(y-self.minY, x-self.minX)] = dwell
  152. #cv2.SetReal2D(blank_image, x-self.minX,y-self.minY, dwell)
  156. #cv2.cv.WriteFrame(out,
  157. #im = cv2.cv.fromarray(blank_image)
  159. #cv2.Set2D(cv_image, x, y, dwell)
  160. #advance by 4 bytes
  161. i+=4
  162. if i%256==0:
  163. pass
  164. #out.write(cv2.cvtColor(blank_image,cv2.COLOR_GRAY2RGB))
  165. #out.write(blank_image)
  166. #cv2.WriteFrame(out, blank_image)
  167. #blank_image = cv.CreateImage((width, height), 8, 1)
  168. #cv2.SetZero(blank_image)
  170. #cv2.imshow('img1',blank_image)
  171. #cv2.waitKey(30) # Waits forever for user to press any key
  172. #print 'frame'
  173. #blank_image = np.zeros((width,height,1), np.uint8)
  175. if (i/4%(self.numPixels/100)==1):
  176. print 'percent done: %d' % (float(i/4) / self.numPixels *100)
  177. if not lastKey == 27:
  178. cv2.imshow('FIB pattern raster sequence',cv2.resize(blank_image, scaled_frame_size) )
  179. lastKey = cv2.waitKey(30) # Waits forever for user to press any key
  180. print i/4, self.numPixels
  182. except IndexError:
  183. print "IndexError, last tuple started at byte: %d " % (i - 4)
  184. #traceback.print_exc()
  185. #out.write(blank_image)
  186. #out.write(cv2.cvtColor(blank_image,cv2.COLOR_GRAY2RGB))
  187. #out.write(blank_image)
  188. #cv2.waitKey(30)
  190. break
  191. #im.save('out_test1.png', 'PNG')
  192. #out.release()
  193. #out=None
  194. print "maxX %d, maxY %d" % (self.maxX, self.maxY)
  195. print "minX %d, minY %d" % (self.minX, self.minY)
  196. print 'num pixels: %d' % (i/4)
  197. while cv2.waitKey(0) != 27:
  198. cv2.imshow('FIB pattern raster sequence',cv2.resize(blank_image, scaled_frame_size) )
  200. cv2.destroyAllWindows()
  202. def write_image(img):
  203. w = max([p[0] for p in img])
  204. h = max([p[1] for p in img])
  206. size = (4096, 4096) # size of the image to create
  207. im = Image.new('L', size) # create the image
  208. draw = ImageDraw.Draw(im) # create a drawing object that is
  209. # used to draw on the new image
  210. red = (255,0,0) # color of our text
  211. table = sum(zip(*img), ())
  212. #draw.point(img)
  213. im.putdata(table)
  215. """
  216. #fff7fb
  217. #ece7f2
  218. #d0d1e6
  219. #a6bddb
  220. #74a9cf
  221. #3690c0
  222. #0570b0
  223. #034e7b
  224. """
  226. del draw # I'm done drawing so I don't need this anymore
  227. im.save('out_test1.png', 'PNG')
  229. pat = fib_pattern()
  230. pat.write_as_12_bit(copy.deepcopy(lines))
  231. pat.write_as_video(lines)
  232. #write_as_byte_w_offset(lines)
  233. #write_image(pixes)
  235. sys.exit(1)
  237. pixels = []
  238. pix=[0,0,0]
  240. """
  241. for i in range(0, len(lines), 2):
  242. #num = int(lines[i].encode('hex'), 16)
  243. num = struct.unpack( "H", lines[i:i+2] )
  244. sys.stdout.write(str(num))
  245. pix[i%3]=num
  246. if i%3 ==2:
  247. print ''
  248. pixels.append((pix[0], pix[1], pix[2]))
  249. else:
  250. sys.stdout.write(', ')
  252. #print c, '\t %s' % lines[i]
  253. """
  254. fl=open(filename, 'rb')
  255. for i in range(0,8):
  256. fl.read(1)
  257. r = BitReader(fl)
  258. i=0
  259. j=0
  260. while True:
  261. b1 = fl.read(1)
  262. b23 = fl.read(1)
  263. b3_latter = fl.read(1)
  264. b2 = ord(b23)>>4 & (int('f',16))
  265. if not i==2:
  266. x = r.readbits(12)
  267. else:
  268. x = r.readbits(8)
  269. if ( r.read == 0 ):
  270. break
  271. pix[i%3]=x
  272. i+=1
  273. if i==3:
  275. pixels.append((pix[0], pix[1], pix[2]))
  276. #print '%s, %s, %s' % (pix[0], pix[1], pix[2])
  277. i=0
  278. j+=1
  279. #if j==50:
  280. # break
  282. h = max([p[0] for p in pixels])
  283. w = max([p[1] for p in pixels])
  284. z = max([p[2] for p in pixels])
  286. print 'maxes of h: %s, w: %s, z: %s' %(h, w, z)
  287. import array
  288. pixmap = [array.array('i',(0 for i in range(0,h+1))) for i in range(0,h+1)]
  290. for pixel in pixels:
  291. pixmap[pixel[0]][pixel[1]] = pixel[2]
  293. pixlist = []
  294. fl1=open('out.raw', 'wb')
  295. for i in range(0, h):
  296. for j in range(0, w):
  297. fl1.write(struct.pack('i', pixmap[i][j]))
  298. fl1.close()
  300. fl.close()
  301.  
Powered by spacepaste. Paste removal.