|
| 1 | +/* |
| 2 | + * Copyright (C) 2012 The Android Open Source Project |
| 3 | + * |
| 4 | + * Licensed under the Apache License, Version 2.0 (the "License"); |
| 5 | + * you may not use this file except in compliance with the License. |
| 6 | + * You may obtain a copy of the License at |
| 7 | + * |
| 8 | + * http://www.apache.org/licenses/LICENSE-2.0 |
| 9 | + * |
| 10 | + * Unless required by applicable law or agreed to in writing, software |
| 11 | + * distributed under the License is distributed on an "AS IS" BASIS, |
| 12 | + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| 13 | + * See the License for the specific language governing permissions and |
| 14 | + * limitations under the License. |
| 15 | + */ |
| 16 | + |
| 17 | +#include <stdint.h> |
| 18 | +#include <sys/types.h> |
| 19 | + |
| 20 | +#include <fcntl.h> |
| 21 | +#include <sys/ioctl.h> |
| 22 | +#include <linux/fb.h> |
| 23 | +#include <linux/input.h> |
| 24 | +#include <errno.h> |
| 25 | +#include <string.h> |
| 26 | +#include <stdio.h> |
| 27 | +#include <cutils/memory.h> |
| 28 | +#include <asm-generic/mman.h> |
| 29 | +#include <sys/mman.h> |
| 30 | +#include <utils/threads.h> |
| 31 | +#include <unistd.h> |
| 32 | +#include <math.h> |
| 33 | + |
| 34 | +using namespace android; |
| 35 | + |
| 36 | +#ifndef FBIO_WAITFORVSYNC |
| 37 | +#define FBIO_WAITFORVSYNC _IOW('F', 0x20, __u32) |
| 38 | +#endif |
| 39 | + |
| 40 | +struct Buffer { |
| 41 | + size_t w; |
| 42 | + size_t h; |
| 43 | + size_t s; |
| 44 | + union { |
| 45 | + void* addr; |
| 46 | + uint32_t* pixels; |
| 47 | + }; |
| 48 | +}; |
| 49 | + |
| 50 | +void clearBuffer(Buffer* buf, uint32_t pixel) { |
| 51 | + android_memset32(buf->pixels, pixel, buf->s * buf->h * 4); |
| 52 | +} |
| 53 | + |
| 54 | +void drawTwoPixels(Buffer* buf, uint32_t pixel, ssize_t x, ssize_t y, size_t w) { |
| 55 | + if (y>0 && y<ssize_t(buf->h)) { |
| 56 | + uint32_t* bits = buf->pixels + y * buf->s; |
| 57 | + if (x>=0 && x<buf->w) { |
| 58 | + bits[x] = pixel; |
| 59 | + } |
| 60 | + ssize_t W(w); |
| 61 | + if ((x+W)>=0 && (x+W)<buf->w) { |
| 62 | + bits[x+W] = pixel; |
| 63 | + } |
| 64 | + } |
| 65 | +} |
| 66 | + |
| 67 | +void drawHLine(Buffer* buf, uint32_t pixel, ssize_t x, ssize_t y, size_t w) { |
| 68 | + if (y>0 && y<ssize_t(buf->h)) { |
| 69 | + ssize_t W(w); |
| 70 | + if (x<0) { |
| 71 | + W += x; |
| 72 | + x = 0; |
| 73 | + } |
| 74 | + if (x+w > buf->w) { |
| 75 | + W = buf->w - x; |
| 76 | + } |
| 77 | + if (W>0) { |
| 78 | + uint32_t* bits = buf->pixels + y * buf->s + x; |
| 79 | + android_memset32(bits, pixel, W*4); |
| 80 | + } |
| 81 | + } |
| 82 | +} |
| 83 | + |
| 84 | +void drawRect(Buffer* buf, uint32_t pixel, ssize_t x, ssize_t y, size_t w, size_t h) { |
| 85 | + ssize_t W(w), H(h); |
| 86 | + if (x<0) { |
| 87 | + w += x; |
| 88 | + x = 0; |
| 89 | + } |
| 90 | + if (y<0) { |
| 91 | + h += y; |
| 92 | + y = 0; |
| 93 | + } |
| 94 | + if (x+w > buf->w) W = buf->w - x; |
| 95 | + if (y+h > buf->h) H = buf->h - y; |
| 96 | + if (W>0 && H>0) { |
| 97 | + uint32_t* bits = buf->pixels + y * buf->s + x; |
| 98 | + for (ssize_t i=0 ; i<H ; i++) { |
| 99 | + android_memset32(bits, pixel, W*4); |
| 100 | + bits += buf->s; |
| 101 | + } |
| 102 | + } |
| 103 | +} |
| 104 | + |
| 105 | +void drawCircle(Buffer* buf, uint32_t pixel, |
| 106 | + size_t x0, size_t y0, size_t radius, bool filled = false) { |
| 107 | + ssize_t f = 1 - radius; |
| 108 | + ssize_t ddF_x = 1; |
| 109 | + ssize_t ddF_y = -2 * radius; |
| 110 | + ssize_t x = 0; |
| 111 | + ssize_t y = radius; |
| 112 | + if (filled) { |
| 113 | + drawHLine(buf, pixel, x0-radius, y0, 2*radius); |
| 114 | + } else { |
| 115 | + drawTwoPixels(buf, pixel, x0-radius, y0, 2*radius); |
| 116 | + } |
| 117 | + while (x < y) { |
| 118 | + if (f >= 0) { |
| 119 | + y--; |
| 120 | + ddF_y += 2; |
| 121 | + f += ddF_y; |
| 122 | + } |
| 123 | + x++; |
| 124 | + ddF_x += 2; |
| 125 | + f += ddF_x; |
| 126 | + if (filled) { |
| 127 | + drawHLine(buf, pixel, x0-x, y0+y, 2*x); |
| 128 | + drawHLine(buf, pixel, x0-x, y0-y, 2*x); |
| 129 | + drawHLine(buf, pixel, x0-y, y0+x, 2*y); |
| 130 | + drawHLine(buf, pixel, x0-y, y0-x, 2*y); |
| 131 | + } else { |
| 132 | + drawTwoPixels(buf, pixel, x0-x, y0+y, 2*x); |
| 133 | + drawTwoPixels(buf, pixel, x0-x, y0-y, 2*x); |
| 134 | + drawTwoPixels(buf, pixel, x0-y, y0+x, 2*y); |
| 135 | + drawTwoPixels(buf, pixel, x0-y, y0-x, 2*y); |
| 136 | + } |
| 137 | + } |
| 138 | +} |
| 139 | + |
| 140 | +class TouchEvents { |
| 141 | + class EventThread : public Thread { |
| 142 | + int fd; |
| 143 | + |
| 144 | + virtual bool threadLoop() { |
| 145 | + input_event event; |
| 146 | + int first_down = 0; |
| 147 | + do { |
| 148 | + read(fd, &event, sizeof(event)); |
| 149 | + if (event.type == EV_ABS) { |
| 150 | + if (event.code == ABS_MT_TRACKING_ID) { |
| 151 | + down = event.value == -1 ? 0 : 1; |
| 152 | + first_down = down; |
| 153 | + } |
| 154 | + if (event.code == ABS_MT_POSITION_X) { |
| 155 | + x = event.value; |
| 156 | + } |
| 157 | + if (event.code == ABS_MT_POSITION_Y) { |
| 158 | + y = event.value; |
| 159 | + } |
| 160 | + } |
| 161 | + } while (event.type == EV_SYN); |
| 162 | + return true; |
| 163 | + } |
| 164 | + |
| 165 | + public: |
| 166 | + int x, y, down; |
| 167 | + EventThread() : Thread(false), |
| 168 | + x(0), y(0), down(0) |
| 169 | + { |
| 170 | + fd = open("/dev/input/event1", O_RDONLY); |
| 171 | + } |
| 172 | +}; |
| 173 | + sp<EventThread> thread; |
| 174 | + |
| 175 | +public: |
| 176 | + TouchEvents() { |
| 177 | + thread = new EventThread(); |
| 178 | + thread->run("EventThread", PRIORITY_URGENT_DISPLAY); |
| 179 | + } |
| 180 | + |
| 181 | + int getMostRecentPosition(int* x, int* y) { |
| 182 | + *x = thread->x; |
| 183 | + *y = thread->y; |
| 184 | + return thread->down; |
| 185 | + } |
| 186 | +}; |
| 187 | + |
| 188 | + |
| 189 | +struct Queue { |
| 190 | + struct position { |
| 191 | + int x, y; |
| 192 | + }; |
| 193 | + int index; |
| 194 | + position q[16]; |
| 195 | + Queue() : index(0) { } |
| 196 | + void push(int x, int y) { |
| 197 | + index++; |
| 198 | + index &= 0xF; |
| 199 | + q[index].x = x; |
| 200 | + q[index].y = y; |
| 201 | + } |
| 202 | + void get(int lag, int* x, int* y) { |
| 203 | + const int i = (index - lag) & 0xF; |
| 204 | + *x = q[i].x; |
| 205 | + *y = q[i].y; |
| 206 | + } |
| 207 | +}; |
| 208 | + |
| 209 | +extern char *optarg; |
| 210 | +extern int optind; |
| 211 | +extern int optopt; |
| 212 | +extern int opterr; |
| 213 | +extern int optreset; |
| 214 | + |
| 215 | +void usage(const char* name) { |
| 216 | + printf("\nusage: %s [-h] [-l lag]\n", name); |
| 217 | +} |
| 218 | + |
| 219 | +int main(int argc, char** argv) { |
| 220 | + fb_var_screeninfo vi; |
| 221 | + fb_fix_screeninfo fi; |
| 222 | + |
| 223 | + int lag = 0; |
| 224 | + int fd = open("/dev/graphics/fb0", O_RDWR); |
| 225 | + ioctl(fd, FBIOGET_VSCREENINFO, &vi); |
| 226 | + ioctl(fd, FBIOGET_FSCREENINFO, &fi); |
| 227 | + void* bits = mmap(0, fi.smem_len, PROT_READ | PROT_WRITE, MAP_SHARED, fd, 0); |
| 228 | + Buffer framebuffer; |
| 229 | + framebuffer.w = vi.xres; |
| 230 | + framebuffer.h = vi.yres; |
| 231 | + framebuffer.s = fi.line_length / (vi.bits_per_pixel >> 3); |
| 232 | + framebuffer.addr = bits; |
| 233 | + |
| 234 | + int ch; |
| 235 | + while ((ch = getopt(argc, argv, "hl:")) != -1) { |
| 236 | + switch (ch) { |
| 237 | + case 'l': |
| 238 | + lag = atoi(optarg); |
| 239 | + break; |
| 240 | + case 'h': |
| 241 | + default: |
| 242 | + usage(argv[0]); |
| 243 | + exit(0); |
| 244 | + } |
| 245 | + } |
| 246 | + argc -= optind; |
| 247 | + argv += optind; |
| 248 | + |
| 249 | + |
| 250 | + TouchEvents touch; |
| 251 | + Queue queue; |
| 252 | + |
| 253 | + |
| 254 | + int x=0, y=0, down=0; |
| 255 | + int lag_x=0, lag_y=0; |
| 256 | + |
| 257 | + clearBuffer(&framebuffer, 0); |
| 258 | + while (true) { |
| 259 | + uint32_t crt = 0; |
| 260 | + int err = ioctl(fd, FBIO_WAITFORVSYNC, &crt); |
| 261 | + |
| 262 | + // draw beam marker |
| 263 | + drawRect(&framebuffer, 0x400000, framebuffer.w-2, 0, 2, framebuffer.h); |
| 264 | + // erase screen |
| 265 | + if (lag) { |
| 266 | + drawCircle(&framebuffer, 0, lag_x, lag_y, 100); |
| 267 | + drawHLine(&framebuffer, 0, 0, lag_y, 32); |
| 268 | + } |
| 269 | + drawCircle(&framebuffer, 0, x, y, 100, true); |
| 270 | + drawHLine(&framebuffer, 0, 0, y, 32); |
| 271 | + |
| 272 | + // draw a line at y=1000 |
| 273 | + drawHLine(&framebuffer, 0x808080, 0, 1000, framebuffer.w); |
| 274 | + |
| 275 | + // get touch events |
| 276 | + touch.getMostRecentPosition(&x, &y); |
| 277 | + queue.push(x, y); |
| 278 | + queue.get(lag, &lag_x, &lag_y); |
| 279 | + |
| 280 | + if (lag) { |
| 281 | + drawCircle(&framebuffer, 0x00FF00, lag_x, lag_y, 100); |
| 282 | + drawHLine(&framebuffer, 0x00FF00, 0, lag_y, 32); |
| 283 | + } |
| 284 | + |
| 285 | + drawCircle(&framebuffer, 0xFFFFFF, x, y, 100, true); |
| 286 | + drawHLine(&framebuffer, 0xFFFFFF, 0, y, 32); |
| 287 | + |
| 288 | + // draw end of frame beam marker |
| 289 | + drawRect(&framebuffer, 0x004000, framebuffer.w-2, 0, 2, framebuffer.h); |
| 290 | + } |
| 291 | + |
| 292 | + close(fd); |
| 293 | + return 0; |
| 294 | +} |
0 commit comments