research/TCS/pressure/process_digits.py

   1 #! /usr/bin/python
   2
   3 from PIL import Image
   4 import Image, ImageDraw
   5 import sys
   6 import numpy
   7
   8
   9
  10
  11 def PixelGood(pixel, threshold):
  12         #print str(pixel)
  13         return GreyScale(pixel) < threshold;
  14         #return (pixel[1] > (pixel[0] + pixel[2]) and pixel[1] > threshold)
  15
  16 def rect_size(e):
  17         return (e[3] - e[1]) * (e[2] - e[0])
  18
  19 def GreyScale(pixel):
  20         greyscale = 0
  21         for i in range(0, len(pixel)):
  22                 greyscale += pixel[i]
  23         greyscale = greyscale / len(pixel)
  24         return greyscale
  25
  26 def SubDivide(pix, bounds, fill_fraction=0.1, threshold=200, seperation=4, min_area=250):
  27         rect = []
  28
  29         xStart = bounds[2]
  30         xEnd = bounds[0]
  31         yStart = bounds[3]
  32         yEnd = bounds[1]
  33         xLast = bounds[0]
  34         good = 0
  35         for x in range(bounds[0], bounds[2]):
  36                 for y in range(bounds[1], bounds[3]):
  37                         if PixelGood(pix[x, y], threshold):
  38                                 if (x - xLast) > seperation:
  39                                         if ((xEnd - xStart) * (yEnd - yStart) > min_area):
  40                                                 rect.append([xStart, yStart, xEnd, yEnd])
  41
  42                                         xStart = bounds[2]
  43                                         xEnd = bounds[0]
  44                                         yStart = bounds[3]
  45                                         yEnd = bounds[1]
  46                                         good = 0
  47
  48                                 if (x < xStart):
  49                                         xStart = x
  50                                 if (y < yStart):
  51                                         yStart = y
  52                                 if (x > xEnd):
  53                                         xEnd = x
  54                                 if (y > yEnd):
  55                                         yEnd = y
  56                                 xLast = x
  57                                 good += 1
  58
  59         rect.append([xStart, yStart, xEnd, yEnd])
  60
  61         return rect
  62
  63 def GoodFraction(pix, bounds,threshold=200):
  64         good = 0
  65         for x in range(int(bounds[0]), int(bounds[2])):
  66                 for y in range(int(bounds[1]), int(bounds[3])):
  67                         if PixelGood(pix[x, y], threshold):
  68                                 good += 1
  69
  70         return float(good) / float(int(bounds[2] - bounds[0])*int(bounds[3] - bounds[1]))
  71
  72 def PickDigit(pix, r):
  73
  74         # top, mid, bot, ltop, lbot, rtop, rmid
  75         d = {
  76                 "1111011" : 9,
  77                 "1111111" : 8,
  78                 "1000011" : 7,
  79                 #"1010011" : 7,
  80                 "1111101" : 6,
  81                 "1111001" : 5,
  82                 "0101011" : 4,
  83                 "1110011" : 3,
  84                 "1110110" : 2,
  85                 "0000011" : 1,
  86                 "1011111" : 0
  87
  88         }
  89
  90         w = r[2] - r[0]
  91         h = r[3] - r[1]
  92
  93         if (w < 0.4 * h):
  94                 return 1
  95
  96
  97         top = GoodFraction(pix,(r[0]+0.0*w, r[1], r[0]+1.0*w, r[1]+0.15*h)) # Top
  98         mid = GoodFraction(pix,(r[0]+0.0*w, r[1]+0.45*h, r[0]+1.0*w, r[1]+0.55*h)) # Middle
  99         bot = GoodFraction(pix,(r[0]+0.0*w, r[1]+0.9*h, r[0]+1.0*w, r[1]+h)) # Bottom
 100         ltop = GoodFraction(pix,(r[0], r[1]+0.1*h, r[0]+0.4*w, r[1]+0.4*h)) # Left Top
 101         lbot = GoodFraction(pix,(r[0], r[1]+0.6*h, r[0]+0.4*w, r[1]+0.9*h)) #Left Bottom
 102         rtop = GoodFraction(pix,(r[0]+0.6*w, r[1]+0.1*h,r[0]+w, r[1]+0.4*h)) # Right Top
 103         rbot = GoodFraction(pix,(r[0]+0.6*w, r[1]+0.6*h, r[0]+w, r[1]+0.9*h)) # Right Bottom
 104
 105         filled = ""
 106
 107         fraction = 0.4
 108         for f in [top,mid,bot,ltop,lbot,rtop,rbot]:
 109                 if f > fraction:
 110                         filled += "1"
 111                 else:
 112                         filled += "0"
 113
 114
 115
 116
 117                 #Check where the boxes are
 118         """
 119         if (filled in d) == False:
 120                 img = Image.new("RGB", (r[2]-r[0], r[3]-r[1]), "white")
 121                 draw = ImageDraw.Draw(img)
 122                 for x in range(r[0], r[2]):
 123                         for y in range(r[1], r[3]):
 124                                 draw.rectangle((x-r[0], y-r[1], x-r[0], y-r[1]), fill=pix[x, y], outline=None)
 125
 126                 boxes = []
 127                 boxes.append((0, 0, w, 0.15*h)) # Top
 128                 boxes.append((0, 0.45*h, w, 0.55*h)) # Middle
 129                 boxes.append((0, 0.9*h, w, h)) # Bottom
 130                 boxes.append((0, 0, 0.4*w, 0.5*h)) # Left Top
 131                 boxes.append((0, 0.5*h, 0.4*w, h)) # Left Bottom
 132                 boxes.append((0.6*w, 0, w, 0.5*h)) # Right Top
 133                 boxes.append((0.6*w, 0.5*h, w, h)) # Right Bottom
 134                 for box in boxes:
 135                         draw.rectangle(box, fill=None, outline="blue")
 136                 img.show()
 137                 #return 0
 138         """
 139
 140         if filled in d:
 141                 return d[filled]
 142         else:
 143                 return None
 144 def Process(fileName):
 145
 146         results = []
 147         image = Image.open(fileName)
 148         pix = image.load()
 149         rect = SubDivide(pix, (0, 0, image.size[0], image.size[1]))
 150         #print "Found " + str(len(rect)) + " rectangles"
 151         #test = Image.new("RGB", image.size, "white")
 152         #test.paste(image)
 153         #draw1 = ImageDraw.Draw(test)
 154         #for i in range(0, len(rect)):
 155         #       draw1.rectangle(rect[i], fill=None, outline="red")
 156
 157
 158                 #img = Image.new("RGB", (rect[i][2]-rect[i][0], rect[i][3]-rect[i][1]), "white")
 159                 #draw2 = ImageDraw.Draw(img)
 160                 #for x in range(rect[i][0], rect[i][2]):
 161                 #       for y in range(rect[i][1], rect[i][3]):
 162                 #               draw2.rectangle((x-rect[i][0], y-rect[i][1], x-rect[i][0], y-rect[i][1]), fill=pix[x, y], outline=None)
 163                 #img.show()
 164
 165         #test.show()
 166
 167         for r in rect:
 168                 digit = PickDigit(pix, r)
 169                 if digit != None:
 170                         results.append(digit)
 171                 else:
 172                         return None
 173         return results
 174
 175 def PixToList(pix, bounds):
 176         a = []
 177         for x in range(bounds[0], bounds[2]):
 178                 a.append([])
 179
 180         for x in range(bounds[0], bounds[2]):
 181                 for y in range(bounds[1], bounds[3]):
 182
 183                         a[x - bounds[0]].append(GreyScale(pix[x, y]))
 184                         #print(str(x) + "," + str(y), len(fft[x - bounds[0]]))
 185                         #print(str(fft[x - bounds[0]][y - bounds[1]]))
 186         return a
 187
 188
 189 if __name__ == "__main__":
 190
 191         if (len(sys.argv) != 2):
 192                 print("Usage " + str(sys.argv[0]) + " input_image")
 193
 194
 195
 196
 197         #sys.exit(0)
 198         r = Process(sys.argv[1])
 199         if r == None:
 200                 print("?")
 201                 sys.exit(1)
 202
 203         if (len(r) == 4):
 204                 print(str(r[0]) + "." + str(r[1]) + str(r[2]) + "e-"  + str(r[3]))
 205         elif (len(r) == 3):
 206                 print(str(r[0]) + "." + str(r[1]) + "e-" + str(r[2]))
 207         else:
 208                 print("#?")
 209         sys.exit(0)
 210
 211
 212