main.c revision 1.1 
1/*	$NetBSD: main.c,v 1.1 2002/08/27 14:12:17 takemura Exp $	*/
2
3/*-
4 * Copyright (c) 2002 TAKEMRUA Shin
5 * All rights reserved.
6 *
7 * Redistribution and use in source and binary forms, with or without
8 * modification, are permitted provided that the following conditions
9 * are met:
10 * 1. Redistributions of source code must retain the above copyright
11 *    notice, this list of conditions and the following disclaimer.
12 * 2. Redistributions in binary form must reproduce the above copyright
13 *    notice, this list of conditions and the following disclaimer in the
14 *    documentation and/or other materials provided with the distribution.
15 * 3. Neither the name of The NetBSD Foundation nor the names of its
16 *    contributors may be used to endorse or promote products derived
17 *    from this software without specific prior written permission.
18 *
19 * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
20 * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
21 * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
22 * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
23 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
24 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
25 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
26 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
27 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
28 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
29 * POSSIBILITY OF SUCH DAMAGE.
30 */
31
32#include <stdio.h>
33#include <strings.h>
34#include <stdlib.h>
35#include <unistd.h>
36#include <err.h>
37#include <errno.h>
38#include <time.h>
39#include <termios.h>
40#include <sys/fcntl.h>
41#include <sys/mman.h>
42#ifdef __STDC__
43#include <stdarg.h>
44#else
45#include <varargs.h>
46#endif
47
48#include "tpctl.h"
49
50#ifndef lint
51#include <sys/cdefs.h>
52__RCSID("$NetBSD: main.c,v 1.1 2002/08/27 14:12:17 takemura Exp $");
53#endif /* not lint */
54
55void load_data(char *data_file, struct tpctl_data *);
56void save_data(char *data_file, struct tpctl_data *);
57int do_calibration(char *, struct tp *, struct wsmouse_calibcoords *);
58void drawcross(struct fb *, int, int, int, fb_pixel_t);
59int check_esc(void *);
60
61int opt_verbose;
62int opt_noupdate;
63int opt_forceupdate;
64
65static void
66usage(void)
67{
68
69	fprintf(stderr, "usage: %s [-D dispdev] [-d dev] [-f file] [-hnuv]\n",
70	    getprogname());
71	exit(EXIT_FAILURE);
72	/* NOTREACHED */
73}
74
75int
76main(int argc, char *argv[])
77{
78	int tpfd, ch;
79	struct tp tp;
80	struct wsmouse_calibcoords *pref;
81	struct tpctl_data data;
82	char *data_file;
83	char *dev_name;
84	char *dispdev_name;
85
86	/* set default values */
87	opt_verbose = 0;
88	opt_noupdate = 0;
89	opt_forceupdate = 0;
90	dev_name = TPCTL_TP_DEVICE;
91	dispdev_name = TPCTL_FB_DEVICE;
92	data_file = TPCTL_DB_FILENAME;
93
94	/* parse command line */
95	while ((ch = getopt(argc, argv, "d:D:f:hnuv")) != -1) {
96		switch (ch) {
97		case 'D':
98			dispdev_name = optarg;
99			break;
100		case 'd':
101			dev_name = optarg;
102			break;
103		case 'f':
104			data_file = optarg;
105			break;
106		case 'h':
107			usage();
108			/* NOTREACHED */
109		case 'n':
110			opt_noupdate = 1;
111			break;
112		case 'u':
113			opt_forceupdate = 1;
114			break;
115		case 'v':
116			opt_verbose = 1;
117			break;
118		default:
119			usage();
120			/* NOTREACHED */
121		}
122	}
123	if (argv[optind] != NULL) {
124		usage();
125		/* NOTREACHED */
126	}
127
128	/* load calibrarion parameters from specified file */
129	load_data(data_file, &data);
130
131	/* open touch panel device and initialize touch panel routines */
132	if ((tpfd = open(dev_name, O_RDWR)) < 0)
133		errx(EXIT_FAILURE, "can't open touch panel");
134	if (tp_init(&tp, tpfd) < 0)
135		errx(EXIT_FAILURE, "can't initialize touch panel");
136
137	/* find out saved parameters for the touch panel */
138	pref = search_data(&data, tp.id);
139	if (opt_forceupdate || pref == NULL) {
140		/* if the parameters wasn't found or '-f' options was
141		   specified, do 'calibrarion' */
142		struct wsmouse_calibcoords coords;
143
144		/* draw cursors and collect samples */
145		if (do_calibration(dispdev_name, &tp, &coords) < 0) {
146			/* ESC key was pressed to abort */
147			exit(EXIT_FAILURE);
148		}
149		/* update parameters with new one */
150		replace_data(&data, tp.id, &coords);
151		pref = search_data(&data, tp.id);
152	} else {
153		/* nothing is updated,
154		   so you don't have to write back the data */
155		opt_noupdate = 1;
156	}
157
158	if (opt_verbose)
159		write_coords(stdout, tp.id, pref);
160
161	/* set calibration parameters into touch panel device */
162	if (tp_setcalibcoords(&tp, pref) < 0)
163		errx(EXIT_FAILURE, "can't set samples");
164
165	/* save calibrarion parameters from specified file */
166	if (!opt_noupdate)
167		save_data(data_file, &data);
168	/* dispose data */
169	free_data(&data);
170
171	exit(EXIT_SUCCESS);
172	/* NOTREACHED */
173}
174
175/*
176 * load calibrarion parameters from specified file
177 *
178 * return:	none (it won't return if some error occurs)
179 */
180void
181load_data(char *data_file, struct tpctl_data *data)
182{
183	int err;
184
185	init_data(data);
186	err = read_data(data_file, data);
187	switch (err) {
188	case ERR_NONE:
189		break;
190	case ERR_NOFILE:
191		fprintf(stderr, "%s: can't open %s\n", getprogname(),
192		    data_file);
193		/* it might be OK... */
194		break;
195	case ERR_IO:
196		fprintf(stderr, "%s: I/O error on %s\n", getprogname(),
197		    data_file);
198		exit(EXIT_FAILURE);
199		break;
200	case ERR_SYNTAX:
201		fprintf(stderr, "%s: format error at %s, line %d\n",
202		    getprogname(), data_file, data->lineno);
203		exit(EXIT_FAILURE);
204		break;
205	case ERR_DUPNAME:
206		fprintf(stderr, "%s: duplicate entry at %s, line %d\n",
207		    getprogname(), data_file, data->lineno);
208		exit(EXIT_FAILURE);
209		break;
210	default:
211		fprintf(stderr, "%s: internal error\n", getprogname());
212		exit(EXIT_FAILURE);
213		break;
214	}
215}
216
217/*
218 * save calibrarion parameters to specified file
219 *
220 * return:	none (it won't return if some error occurs)
221 */
222void
223save_data(char *data_file, struct tpctl_data *data)
224{
225	int err;
226
227	err = write_data(data_file, data);
228	switch (err) {
229	case ERR_NONE:
230		break;
231	case ERR_NOFILE:
232		fprintf(stderr, "%s: can't open %s\n", getprogname(),
233		    data_file);
234		exit(EXIT_FAILURE);
235		break;
236	case ERR_IO:
237		fprintf(stderr, "%s: I/O error on %s\n", getprogname(),
238		    data_file);
239		exit(EXIT_FAILURE);
240		break;
241	default:
242		fprintf(stderr, "%s: internal error\n", getprogname());
243		exit(EXIT_FAILURE);
244		break;
245	}
246}
247
248/*
249 * draw cursors on frame buffer and collect samples in
250 * wamouse_calibcoords structure.
251 *
252 * return:	0	succeeded
253 *		-1	aborted by user (ESC key was pressed)
254 *		(it won't return if some error occurs)
255 */
256int
257do_calibration(char *dev, struct tp *tp, struct wsmouse_calibcoords *coords)
258{
259	int fbfd;
260	struct fb fb;
261	int i, x, y, xm, ym, cursize, err, res;
262
263	/* open frame buffer device and initialize frame buffer routine */
264	if ((fbfd = open(dev, O_RDWR)) < 0)
265		errx(EXIT_FAILURE, "can't open frame buffer");
266	if (fb_init(&fb, fbfd) < 0)
267		errx(EXIT_FAILURE, "can't map frame buffer");
268
269	memset(coords, 0, sizeof(*coords));
270	coords->minx = 0;
271	coords->miny = 0;
272	coords->maxx = fb.conf.hf_width - 1;
273	coords->maxy = fb.conf.hf_height - 1;
274	coords->samplelen = 5;
275
276	cursize = 20;
277	xm = fb.conf.hf_width/10;
278	ym = fb.conf.hf_height/10;
279
280	/* center */
281	coords->samples[0].x = fb.conf.hf_width/2;
282	coords->samples[0].y = fb.conf.hf_height/2;
283
284	/* top left */
285	coords->samples[1].x = xm;
286	coords->samples[1].y = ym;
287
288	/* bottom left */
289	coords->samples[2].x = xm;
290	coords->samples[2].y = fb.conf.hf_height - ym;
291
292	/* bottom right */
293	coords->samples[3].x = fb.conf.hf_width - xm;
294	coords->samples[3].y = fb.conf.hf_height - ym;
295
296	/* top right */
297	coords->samples[4].x = fb.conf.hf_width - xm;
298	coords->samples[4].y = ym;
299
300	tp_setrawmode(tp);
301	err = 0;
302	for (i = 0; i < coords->samplelen; i++) {
303		drawcross(&fb,
304		    coords->samples[i].x,
305		    coords->samples[i].y,
306		    cursize, fb.white);
307		fb_flush(&fb);
308		tp_flush(tp);
309		res = tp_get(tp, &x, &y, check_esc, 0 /* stdin */);
310		if (res < 0) {
311			err = errno;
312			break;
313		}
314		if (0 < res) {
315			fb_dispmode(&fb, WSDISPLAYIO_MODE_EMUL);
316			return (-1); /* aborted by user */
317		}
318		coords->samples[i].rawx = x;
319		coords->samples[i].rawy = y;
320		drawcross(&fb,
321		    coords->samples[i].x,
322		    coords->samples[i].y,
323		    cursize, fb.black);
324		fb_flush(&fb);
325		tp_waitup(tp, 200, check_esc, 0 /* stdin */);
326	}
327
328	fb_dispmode(&fb, WSDISPLAYIO_MODE_EMUL);
329	close(fbfd);
330
331	if (opt_verbose) {
332		printf("%s: %dx%d (%dbytes/line) %dbit offset=0x%lx\n",
333		    getprogname(),
334		    fb.conf.hf_width,
335		    fb.conf.hf_height,
336		    fb.conf.hf_bytes_per_line,
337		    fb.conf.hf_pixel_width,
338		    fb.conf.hf_offset);
339	}
340
341	if (err) {
342		errno = err;
343		errx(EXIT_FAILURE, "can't get samples");
344	}
345
346	return (0);
347}
348
349/*
350 * draw corss cursor on frame buffer
351 *
352 * return:	none
353 */
354void
355drawcross(struct fb *fb, int x, int y, int size, fb_pixel_t pixel)
356{
357	size /= 2;
358
359	fb_drawline(fb, x,     y - size + 1, x,     y - 1,    pixel);
360	fb_drawline(fb, x + 1, y - size + 1, x + 1, y - 1,    pixel);
361	fb_drawline(fb, x,     y + 2,        x,     y + size, pixel);
362	fb_drawline(fb, x + 1, y + 2,        x + 1, y + size, pixel);
363
364	fb_drawline(fb, x - size + 1, y,     x - 1,    y,     pixel);
365	fb_drawline(fb, x - size + 1, y + 1, x - 1,    y + 1, pixel);
366	fb_drawline(fb, x + 2,        y,     x + size, y,     pixel);
367	fb_drawline(fb, x + 2,        y + 1, x + size, y + 1, pixel);
368}
369
370/*
371 * check ESC key
372 *
373 * date:	input file descriptor
374 *
375 * return:	0	nothing has been occured
376 *		1	ESC key was pressed
377 *		-1	error
378 */
379int
380check_esc(void *data)
381{
382	int fd = (int)data;
383	int flg, n, err;
384	char buf[1];
385	struct termios tm, raw;
386
387	if (tcgetattr(fd, &tm) < 0)
388		return (-1);
389	raw = tm;
390	cfmakeraw(&raw);
391	if (tcsetattr(fd, TCSANOW, &raw) < 0)
392		return (-1);
393	if ((flg = fcntl(fd, F_GETFL)) == -1)
394		return (-1);
395	if (fcntl(fd, F_SETFL, flg | O_NONBLOCK) == -1)
396		return (-1);
397	n = read(fd, buf, 1);
398	err = errno;
399	fcntl(fd, F_SETFL, flg);
400	tcsetattr(fd, TCSANOW, &tm);
401	if (n < 0)
402		return (err == EWOULDBLOCK ? 0 : -1);
403	if (n == 0)
404		return (0); /* EOF */
405	if (*buf == 0x1b)
406		return (1); /* ESC */
407
408	return (0);
409}
410