bug fixes + annotations
[IRC.git] / Robust / src / Benchmarks / SSJava / EyeTracking / EyeDetector.java
1 /*
2  * Copyright 2009 (c) Florian Frankenberger (darkblue.de)
3  * 
4  * This file is part of LEA.
5  * 
6  * LEA is free software: you can redistribute it and/or modify it under the
7  * terms of the GNU Lesser General Public License as published by the Free
8  * Software Foundation, either version 3 of the License, or (at your option) any
9  * later version.
10  * 
11  * LEA is distributed in the hope that it will be useful, but WITHOUT ANY
12  * WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR
13  * A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more
14  * details.
15  * 
16  * You should have received a copy of the GNU Lesser General Public License
17  * along with LEA. If not, see <http://www.gnu.org/licenses/>.
18  */
19
20 /**
21  * No description given.
22  * 
23  * @author Florian Frankenberger
24  */
25 @LATTICE("IMG")
26 @METHODDEFAULT("OUT<THIS,THIS<IN,OUT*,THISLOC=THIS,RETURNLOC=OUT")
27 class EyeDetector {
28
29   @LOC("IMG")
30   private int width;
31   @LOC("IMG")
32   private int height;
33   @LOC("IMG")
34   private int[] pixelBuffer;
35   @LOC("IMG")
36   double percent;
37
38   public EyeDetector(Image image, Rectangle2D faceRect) {
39
40     percent = 0.15 * faceRect.getWidth();
41     Rectangle2D adjustedFaceRect =
42         new Rectangle2D(faceRect.getX() + percent, faceRect.getY() + percent, faceRect.getWidth()
43             - percent, faceRect.getHeight() - 2 * percent);
44
45     width = (int) adjustedFaceRect.getWidth() / 2;
46     height = (int) adjustedFaceRect.getHeight() / 2;
47     pixelBuffer = new int[width * height];
48
49     int startX = (int) adjustedFaceRect.getX();
50     int startY = (int) adjustedFaceRect.getY();
51
52     for (int y = 0; y < height; y++) {
53       for (int x = 0; x < width; x++) {
54         pixelBuffer[(y * width) + x] = (int) image.getPixel(x + startX, y + startY);
55       }
56     }
57
58   }
59
60   @LATTICE("OUT,V<THIS,THIS<C,C*,THISLOC=THIS,RETURNLOC=OUT")
61   public Point detectEye() {
62     @LOC("OUT") Point eyePosition = null;
63     @LOC("V") float brightness = 255f;
64     for (@LOC("C") int y = 0; y < height; ++y) {
65       for (@LOC("C") int x = 0; x < width; ++x) {
66         @LOC("V") final int position = y * width + x;
67         @LOC("V") final int[] color =
68             new int[] { (pixelBuffer[position] & 0xFF0000) >> 16,
69                 (pixelBuffer[position] & 0x00FF00) >> 8, pixelBuffer[position] & 0x0000FF };
70         // System.out.println("("+x+","+y+")="+color[0]+" "+color[1]+" "+color[2]);
71         @LOC("V") final float acBrightness = getBrightness(color);
72
73         if (acBrightness < brightness) {
74           eyePosition = new Point(x + (int) percent, y + (int) percent);
75           brightness = acBrightness;
76         }
77       }
78     }
79
80     return eyePosition;
81   }
82
83   private static float getBrightness(@LOC("IN") int[] color) {
84     @LOC("IN") int min = Math.min(Math.min(color[0], color[1]), color[2]);
85     @LOC("IN") int max = Math.max(Math.max(color[0], color[1]), color[2]);
86
87     return 0.5f * (max + min);
88   }
89 }