#include <stdio.h>
#include <sys/types.h>
#include <unistd.h>
#include <sys/ioctl.h>
#include <errno.h>
#include <string.h>
#include <stdlib.h>
#include <fcntl.h>
#include "fb.h"
#include "matroxfb.h"
#include <ncurses.h>

static int help(void) {
	fprintf(stderr, "usage: matroxset [-f fbdev] [-o output] [-m] [value]\n"
	                "\n"
	                "where -f fbdev  is fbdev device (default /dev/fb1)\n"
	                "      -o output is output number to investigate (0=primary, 1=secondary=default)\n"
	                "      -m        says that CRTC->output mapping should be changed/retrieved\n"
			"      -p        print information about blanking\n"
		        "      -l        list controls\n"
		        "      -c        get/set control value\n"
		        "      -e        edit controls (interactively)\n"
	                "      value     if present, value is set, if missing, value is retrieved\n"
			"\n"
			"For output mode, 128 means monitor, 1 = PAL TV, 2 = NTSC TV\n");
	return 98;
}

static int listControls (int fd)
{
	struct matroxfb_queryctrl qctrl;
	qctrl.id = V4L2_CID_BASE;
	
	printf ("*--------+-----+-----+-----+-----+"
		"-----------------------"
		"---------*\n");
	printf ("|Id      |Curr |Def. |Min  |Max  |Name"
		"                            |\n");
	while (0 == ioctl(fd, MATROXFB_TVOQUERYCTRL,
			  &qctrl)) {
		struct matroxfb_control ctrl;
		memset(&ctrl, 0, sizeof(ctrl));
		ctrl.id = qctrl.id;
		ioctl(fd, MATROXFB_G_TVOCTRL, &ctrl);
		printf("+--------+-----+-----+-----+-----+"
		       "-----------------------"
		       "---------+\n");
		printf("|%8X|%5d|%5d|%5d|%5d|%-32.32s|\n",
		       qctrl.id, ctrl.value,
		       qctrl.default_value,
		       qctrl.minimum, qctrl.maximum,
		       qctrl.name);
		++qctrl.id;
	}
	qctrl.id = V4L2_CID_PRIVATE_BASE;
	while (0 == ioctl(fd, MATROXFB_TVOQUERYCTRL,
			  &qctrl)) {
		struct matroxfb_control ctrl;
		memset(&ctrl, 0, sizeof(ctrl));
		ctrl.id = qctrl.id;
		ioctl(fd, MATROXFB_G_TVOCTRL, &ctrl);
		printf("+--------+-----+-----+-----+-----+"
		       "-----------------------"
		       "---------+\n");
		printf("|%8X|%5d|%5d|%5d|%5d|%-32.32s|\n",
		       qctrl.id, ctrl.value,
		       qctrl.default_value,
		       qctrl.minimum, qctrl.maximum,
		       qctrl.name);
		++qctrl.id;
	}
	printf ("*--------+-----+-----+-----+-----+"
		"-----------------------"
		"---------*\n");
	return 0;
}

static int editControls (int fd)
{
  int c = 0;
  int id = V4L2_CID_BASE;
  int err = 0;

  initscr();
  keypad(stdscr, 1);
  nonl();
  cbreak();
  noecho();
  timeout(-10);

  mvprintw (0, 0, "*--------+-----+-----+-----+-----+"
	    "--------------------------------*\n");
  printw ("|Id      |Curr.|Def. |Min  |Max  |Name"
	  "                            |\n");
  printw("+--------+-----+-----+-----+-----+"
	 "-----------------------"
	 "---------+\n");
  mvprintw (4, 0, "*--------+-----+-----+-----+-----+"
	    "--------------------------------*\n");
  printw("a - maximum value\n"
	 "i - minimum value\n"
	 "d - default value\n"
	 "q - quit\n");

  while (!c || c!= 'q') {
    struct matroxfb_control ctrl;
    struct matroxfb_queryctrl qctrl;

    ctrl.id = id;
    err = ioctl(fd, MATROXFB_G_TVOCTRL, &ctrl);
    if (0 != err) goto _exit;

    qctrl.id = id;
    err = ioctl(fd, MATROXFB_TVOQUERYCTRL, &qctrl);
    if (0 != err) goto _exit;

    switch (c) {
    case 'd':
      ctrl.value = qctrl.default_value;
      err = ioctl(fd, MATROXFB_S_TVOCTRL, &ctrl);
      if (0 != err) goto _exit;
      break;
    case 'i':
      ctrl.value = qctrl.minimum;
      err = ioctl(fd, MATROXFB_S_TVOCTRL, &ctrl);
      if (0 != err) goto _exit;
      break;
    case 'a':
      ctrl.value = qctrl.maximum;
      err = ioctl(fd, MATROXFB_S_TVOCTRL, &ctrl);
      if (0 != err) goto _exit;
      break;

    case KEY_UP:
      --id;
      ctrl.id = id;
      if (0 != ioctl(fd, MATROXFB_G_TVOCTRL, &ctrl)) {
        if (id > V4L2_CID_BASE) {
          id = V4L2_CID_BASE + 3;
        } else {
	  ++id;
 	  beep();
	}
      }
      break;
    case KEY_DOWN:
      ++id;
      ctrl.id = id;
      if (0 != ioctl(fd, MATROXFB_G_TVOCTRL, &ctrl)) {
        if (id < V4L2_CID_PRIVATE_BASE) {
	  id = V4L2_CID_PRIVATE_BASE;
	} else {
	  --id;
	  beep();
	}
      }
      break;
    case KEY_LEFT:
      if (ctrl.value > qctrl.minimum) {
	ctrl.value -= qctrl.step;
	if (ctrl.value < qctrl.minimum) ctrl.value = qctrl.minimum;
	err = ioctl(fd, MATROXFB_S_TVOCTRL, &ctrl);
	if (0 != err) goto _exit;
      } else {
	beep ();
      }
      break;
    case KEY_RIGHT:
      if (ctrl.value < qctrl.maximum) {
	ctrl.value += qctrl.step;
	if (ctrl.value > qctrl.maximum) ctrl.value = qctrl.maximum;
	err = ioctl(fd, MATROXFB_S_TVOCTRL, &ctrl);
	if (0 != err) goto _exit;
      } else {
	beep ();
      }
      break;
    }

    ctrl.id = id;
    err = ioctl(fd, MATROXFB_G_TVOCTRL, &ctrl);
    if (0 != err) goto _exit;

    qctrl.id = id;
    err = ioctl(fd, MATROXFB_TVOQUERYCTRL, &qctrl);
    if (0 != err) goto _exit;

    mvprintw(3,0,"|%8X|%5d|%5d|%5d|%5d|%-32.32s|\n",
	     id, ctrl.value,
	     qctrl.default_value, qctrl.minimum, qctrl.maximum,
	     qctrl.name);

    move(3,9);

    wsyncup(stdscr);
    c = getch();
  }

 _exit:
  endwin ();
  return 0;
}

int main(int argc, char* argv[]) {
	char* fb = "/dev/fb1";
	int fd;
	struct matroxioc_output_mode mom;
	struct fb_vblank vbl;
	int rtn = 0;
	int output = MATROXFB_OUTPUT_SECONDARY;
	int o_present = 0;
	int m_present = 0;
	int p_present = 0;
	int l_present = 0;
	int c_present = 0;
	int e_present = 0;
	int ctrlid = 0;
	int act;
	u_int32_t conns;
	
	while ((rtn = getopt(argc, argv, "o:f:mhdplc:e")) != -1) {
		switch (rtn) {
			case 'o':
				output = strtoul(optarg, NULL, 0);
				o_present = 1;
				break;
			case 'l':
				l_present = 1;
				break;
			case 'm':
				m_present = 1;
				break;
			case 'f':
				fb = optarg;
				break;
			case 'p':
				p_present = 1;
				break;
			case 'c':
				ctrlid = strtoul(optarg, NULL, 0);
				c_present = 1;
				break;
			case 'e':
				e_present = 1;
				break;
			case 'h':
				return help();
			default:
				fprintf(stderr, "Bad commandline\n");
				return 99;
		}
	}
	act = 0;
	if (e_present) {
	 	act = 7;
	} else if (l_present) {
	 	act = 6;
	} else if (c_present) {
	 	act = 5;
	} else if (p_present) {
		if (m_present || o_present) {
			fprintf(stderr, "You cannot use -p together with -m or -o\n");
			return 95;
		}
		act = 4;
	} else if (optind >= argc) {
		if (m_present) {
			if (o_present) {
				fprintf(stderr, "You cannot use -m and -o together\n");
				return 96;
			}
			act = 2;
		} else {
			mom.output = output;
			mom.mode = 0;
		}
	} else {
		if (m_present) {
			conns = strtoul(argv[optind], NULL, 0);
			act = 3;
		} else {
			mom.output = output;
			mom.mode = strtoul(argv[optind], NULL, 0);
			act = 1;
		}
	}
	fd = open(fb, O_RDWR);
	if (fd == -1) {
		fprintf(stderr, "Cannot open %s: %s\n", fb, strerror(errno));
		return 122;
	}
	switch (act) {
		case 0:
			rtn = ioctl(fd, MATROXFB_GET_OUTPUT_MODE, &mom);
			if (rtn)
				break;
			printf("Output mode is %u\n", mom.mode);
			break;
		case 1:
			rtn = ioctl(fd, MATROXFB_SET_OUTPUT_MODE, &mom);
			break;
		case 2:
			rtn = ioctl(fd, MATROXFB_GET_OUTPUT_CONNECTION, &conns);
			if (rtn)
				break;
			printf("This framebuffer is connected to outputs %08X\n", conns);
			break;
		case 3:
			rtn = ioctl(fd, MATROXFB_SET_OUTPUT_CONNECTION, &conns);
			break;
		case 4:
#if 0
			{ int i; for (i = 0; i < 1000000; i++) {
			rtn = ioctl(fd, FBIOGET_VBLANK, &vbl);
			if (rtn)
				break;
			}}
#else
			rtn = ioctl(fd, FBIOGET_VBLANK, &vbl);
			if (rtn)
				break;
#endif
			printf("VBlank flags:          %08X\n", vbl.flags);
			printf("  Symbolic: ");
			{
				static const struct { u_int32_t mask; const char* msg; } *ptr, vals[] = {
					{ FB_VBLANK_HAVE_VBLANK, "vblank" },
					{ FB_VBLANK_HAVE_HBLANK, "hblank" },
					{ FB_VBLANK_HAVE_COUNT, "field no." },
					{ FB_VBLANK_HAVE_VCOUNT, "line no." },
					{ FB_VBLANK_HAVE_HCOUNT, "column no." },
					{ FB_VBLANK_VBLANKING, "vblanking" },
					{ FB_VBLANK_HBLANKING, "hblanking" },
					{ 0, NULL }};
				int ap = 0;
				for (ptr = vals; ptr->msg; ptr++) {
					if (vbl.flags & ptr->mask) {
						if (ap) printf("%s", ", ");
						printf("%s", ptr->msg);
						ap = 1;
					}
				}
				if (!ap)
				    printf("%s", "none");
				printf("%s", "\n");
			}
			printf("Field count:       %12u\n", vbl.count);
			printf("Vertical line:     %12u\n", vbl.vcount);
			printf("Horizontal column: %12u\n", vbl.hcount);
			break;
		case 5: /* get/set controls */
			{
				struct matroxfb_control ctrl;
				int oldval = 0;

				ctrl.id = ctrlid;
				rtn = ioctl(fd, MATROXFB_G_TVOCTRL,
					    &ctrl);
				if (0 == rtn) {
					oldval = ctrl.value;
					if (optind < argc) {
					  ctrl.value 
					    = strtoul(argv[optind], NULL, 0);
					  rtn = ioctl(fd, MATROXFB_S_TVOCTRL,
						      &ctrl);
					  if (0 == rtn) {
					    printf("value: %d -> %d\n",
						   oldval, ctrl.value);
					  }
					} else {
					  printf("value: %d\n", oldval);
					}
				}
			}
			break;
		case 6: /* list controls */
			rtn = listControls(fd);
			break;
		case 7: /* edit controls */
			rtn = editControls (fd);
			break;
		default:
			rtn = -1; errno = EINVAL;
			break;
	}
	if (rtn) {
		fprintf(stderr, "ioctl failed: %s\n", strerror(errno));
	}
	close(fd);
	return 0;
}