Split am7xxx functions and definitions
[libam7xxx.git] / picoproj.c
index 2629d02..010d4d1 100644 (file)
@@ -1,4 +1,4 @@
-/* picoproj - communication with AM7xxx based USB pico projectors
+/* picoproj - test program for libam7xxx
  *
  * Copyright (C) 2011  Antonio Ospite <ospite@studenti.unina.it>
  *
 
 #include <stdio.h>
 #include <stdlib.h>
-#include <stdint.h>
 #include <string.h>
-#include <endian.h>
-#include <errno.h>
+#include <sys/mman.h>
+#include <sys/types.h>
+#include <sys/stat.h>
+#include <fcntl.h>
+#include <unistd.h>
 
-typedef enum {
-       AM7x01_PACKET_TYPE_INIT    = 0x01,
-       AM7x01_PACKET_TYPE_IMAGE   = 0x02,
-       AM7x01_PACKET_TYPE_POWER   = 0x04,
-       AM7x01_PACKET_TYPE_UNKNOWN = 0x05,
-} am7x01_packet_type;
+#include "am7xxx.h"
 
-typedef enum {
-       AM7x01_IMAGE_FORMAT_JPEG = 1,
-} am7x01_image_format;
-
-typedef enum {
-       AM7x01_POWER_OFF  = 0,
-       AM7x01_POWER_LOW  = 1,
-       AM7x01_POWER_MID  = 2,
-       AM7x01_POWER_HIGH = 3,
-} am7x01_power_mode;
-
-struct image_header {
-       uint32_t format;
-       uint32_t width;
-       uint32_t height;
-       uint32_t image_size;
-};
-
-struct power_header {
-       uint32_t power_low;
-       uint32_t power_mid;
-       uint32_t power_high;
-};
-
-/*
- * Examples of packet headers:
- *
- * Image widget:
- * +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
- * +02|00|00|00|00|10|3e|10|01|00|00|00|20|03|00|00|e0|01|00|00|53|E8|00|00+
- * +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
- *
- * Brightness widget:
- * +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
- * +04|00|00|00|00|0c|ff|ff|00|00|00|00|00|00|00|00|00|00|00|00|00|00|00|00+
- * +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
- */
-
-static uint8_t reference_image_header[] = {
-       0x02, 0x00, 0x00, 0x00,
-       0x00,
-       0x10,
-       0x3e,
-       0x10,
-       0x01, 0x00, 0x00, 0x00,
-       0x20, 0x03, 0x00, 0x00,
-       0xe0, 0x01, 0x00, 0x00,
-       0x53, 0xE8, 0x00, 0x00
-};
-
-struct header {
-       uint32_t packet_type;
-       uint8_t unknown0;
-       uint8_t header_len;
-       uint8_t unknown2;
-       uint8_t unknown3;
-       union {
-               struct image_header image;
-               struct power_header power;
-       } header_data;
-};
-
-
-static void dump_image_header(struct image_header *i)
+static void usage(char *name)
 {
-       if (i == NULL)
-               return;
-
-       printf("Image header:\n");
-       printf("format:      0x%08x (%u)\n", i->format, i->format);
-       printf("width:       0x%08x (%u)\n", i->width, i->width);
-       printf("height:      0x%08x (%u)\n", i->height, i->height);
-       printf("image size:  0x%08x (%u)\n", i->image_size, i->image_size);
+       printf("usage: %s [OPTIONS]\n\n", name);
+       printf("OPTIONS:\n");
+       printf("\t-f <filename>\t\tthe image file to upload\n");
+       printf("\t-F <format>\t\tthe image format to use (default is JPEG).\n");
+       printf("\t\t\t\tSUPPORTED FORMATS:\n");
+       printf("\t\t\t\t\t1 - JPEG\n");
+       printf("\t-W <image width>\tthe width of the image to upload\n");
+       printf("\t-H <image height>\tthe height of the image to upload\n");
+       printf("\t-h \t\t\tthis help message\n");
 }
 
-static void dump_header(struct header *h)
+int main(int argc, char *argv[])
 {
-       if (h == NULL)
-               return;
-
-       printf("packet_type: 0x%08x (%u)\n", h->packet_type, h->packet_type);
-       printf("unknown0:    0x%02hhx (%hhu)\n", h->unknown0, h->unknown0);
-       printf("header_len:  0x%02hhx (%hhu)\n", h->header_len, h->header_len);
-       printf("unknown2:    0x%02hhx (%hhu)\n", h->unknown2, h->unknown2);
-       printf("unknown3:    0x%02hhx (%hhu)\n", h->unknown3, h->unknown3);
-
-       switch(h->packet_type) {
-       case AM7x01_PACKET_TYPE_IMAGE:
-               dump_image_header(&(h->header_data.image));
-               break;
-
-       default:
-               printf("Packet type not supported!\n");
-               break;
+       int ret;
+       int exit_code = EXIT_SUCCESS;
+       int opt;
+
+       char filename[FILENAME_MAX] = {0};
+       int image_fd = -1;
+       am7xxx_device dev;
+       int format = AM7XXX_IMAGE_FORMAT_JPEG;
+       int width = 800;
+       int height = 480;
+       uint8_t *image = NULL;
+       unsigned int size = 59475;
+
+       while ((opt = getopt(argc, argv, "f:F:W:H:h")) != -1) {
+               switch (opt) {
+               case 'f':
+                       strncpy(filename, optarg, FILENAME_MAX);
+                       break;
+               case 'F':
+                       format = atoi(optarg);
+                       if (format != 1) {
+                               fprintf(stderr, "Unsupported format\n");
+                               exit(EXIT_FAILURE);
+                       }
+                       break;
+               case 'W':
+                       width = atoi(optarg);
+                       if (width < 0) {
+                               fprintf(stderr, "Unsupported width\n");
+                               exit(EXIT_FAILURE);
+                       }
+                       break;
+               case 'H':
+                       height = atoi(optarg);
+                       if (height < 0) {
+                               fprintf(stderr, "Unsupported height\n");
+                               exit(EXIT_FAILURE);
+                       }
+                       break;
+               default: /* '?' */
+                       usage(argv[0]);
+                       exit(EXIT_FAILURE);
+               }
        }
 
-       fflush(stdout);
-}
-
-static void dump_buffer(uint8_t *buffer, unsigned int len)
-{
-       unsigned int i;
-
-       if (buffer == NULL || len == 0)
-               return;
-
-       for (i = 0; i < len; i++) {
-               printf("%02hhX%c", buffer[i], (i < len - 1) ? ' ' : '\n');
+       if (filename[0] != '\0') {
+               struct stat st;
+               
+               image_fd = open(filename, O_RDONLY);
+               if (image_fd < 0) {
+                       perror("open");
+                       exit_code = EXIT_FAILURE;
+                       goto out;
+               }
+               if (fstat(image_fd, &st) < 0) {
+                       perror("fstat");
+                       exit_code = EXIT_FAILURE;
+                       goto out_close_image_fd;
+               }
+               size = st.st_size;
+
+               image = mmap(NULL, st.st_size, PROT_READ, MAP_SHARED, image_fd, 0);
+               if (image == NULL) {
+                       perror("mmap");
+                       exit_code = EXIT_FAILURE;
+                       goto out_close_image_fd;
+               }
        }
-       fflush(stdout);
-}
 
-static int send_data(uint8_t *buffer, unsigned int len)
-{
-       dump_buffer(buffer, len);
-       return 0;
-}
-
-static int send_header(struct header *h)
-{
-       union {
-               struct header header;
-               uint8_t buffer[sizeof (struct header)];
-       } data;
-
-       data.header = *h;
-
-       return send_data(data.buffer, sizeof (struct header));
-}
-
-static int send_image(am7x01_image_format format,
-                     unsigned int width,
-                     unsigned int height,
-                     uint8_t *image,
-                     unsigned int size)
-{
-       int ret;
-       struct header h = {
-               .packet_type = htole32(AM7x01_PACKET_TYPE_IMAGE),
-               .unknown0    = 0x00,
-               .header_len  = sizeof(struct image_header),
-               .unknown2    = 0x3e,
-               .unknown3    = 0x10,
-               .header_data = {
-                       .image = {
-                               .format     = htole32(format),
-                               .width      = htole32(width),
-                               .height     = htole32(height),
-                               .image_size = htole32(size),
-                       },
-               },
-       };
-
-       dump_header(&h);
-       printf("\n");
-
-       printf("Dump Buffers\n");
-       dump_buffer(reference_image_header, sizeof(struct header));
-
-       ret = send_header(&h);
-       if (ret < 0)
-               return ret;
+       dev = am7xxx_init();
+       if (dev == NULL) {
+               perror("am7xxx_init");
+               exit_code = EXIT_FAILURE;
+               goto out_munmap;
+       }
 
-       if (image == NULL || size == 0)
-               return 0;
+       ret = am7xxx_send_image(dev, format, width, height, image, size);
+       if (ret < 0) {
+               perror("am7xxx_send_image");
+               exit_code = EXIT_FAILURE;
+               goto cleanup;
+       }
 
-       return send_data(image, size);
-}
+       exit_code = EXIT_SUCCESS;
 
+cleanup:
+       am7xxx_shutdown(dev);
 
-int main(int argc, char *argv[])
-{
-       int ret;
+out_munmap:
+       if (image != NULL) {
+               ret = munmap(image, size);
+               if (ret < 0)
+                       perror("munmap");
+       }
 
-       ret = send_image(AM7x01_IMAGE_FORMAT_JPEG, 800, 480, 59475);
-       if (ret < 0) {
-               perror("send_image");
-               exit(EXIT_FAILURE);
+out_close_image_fd:
+       if (image_fd >= 0) {
+               ret = close(image_fd);
+               if (ret < 0)
+                       perror("close");
        }
-       exit(EXIT_SUCCESS);
+
+out:
+       exit(exit_code);
 }