X-Git-Url: https://git.ao2.it/libam7xxx.git/blobdiff_plain/b9a31ad87f63016dc9f914f61ab06606a948aa90..4612a1475669091669d76054f341cd4a02b5c2d0:/src/picoproj.c

diff --git a/src/picoproj.c b/src/picoproj.c
index c3f8b10..c3b199c 100644
--- a/src/picoproj.c
+++ b/src/picoproj.c
@@ -35,6 +35,8 @@ static void usage(char *name)
 	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\t\t\t\t2 - YUV - NV12\n");
+	printf("\t-l <log level>\t\tthe verbosity level of libam7xxx output (0-5)\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");
@@ -47,26 +49,44 @@ int main(int argc, char *argv[])
 	int opt;
 
 	char filename[FILENAME_MAX] = {0};
-	int image_fd = -1;
-	am7xxx_device dev;
+	int image_fd;
+	struct stat st;
+	am7xxx_context *ctx;
+	am7xxx_device *dev;
+	int log_level = AM7XXX_LOG_INFO;
 	int format = AM7XXX_IMAGE_FORMAT_JPEG;
 	int width = 800;
 	int height = 480;
-	uint8_t *image = NULL;
-	unsigned int size = 59475;
+	unsigned char *image;
+	unsigned int size;
+	am7xxx_device_info device_info;
 
-	while ((opt = getopt(argc, argv, "f:F:W:H:h")) != -1) {
+	while ((opt = getopt(argc, argv, "f:F:l:W:H:h")) != -1) {
 		switch (opt) {
 		case 'f':
 			strncpy(filename, optarg, FILENAME_MAX);
 			break;
 		case 'F':
 			format = atoi(optarg);
-			if (format != 1) {
+			switch(format) {
+			case AM7XXX_IMAGE_FORMAT_JPEG:
+				fprintf(stdout, "JPEG format\n");
+				break;
+			case AM7XXX_IMAGE_FORMAT_NV12:
+				fprintf(stdout, "NV12 format\n");
+				break;
+			default:
 				fprintf(stderr, "Unsupported format\n");
 				exit(EXIT_FAILURE);
 			}
 			break;
+		case 'l':
+			log_level = atoi(optarg);
+			if (log_level < AM7XXX_LOG_FATAL || log_level > AM7XXX_LOG_TRACE) {
+				fprintf(stderr, "Unsupported log level, falling back to AM7XXX_LOG_ERROR\n");
+				log_level = AM7XXX_LOG_ERROR;
+			}
+			break;
 		case 'W':
 			width = atoi(optarg);
 			if (width < 0) {
@@ -81,43 +101,89 @@ int main(int argc, char *argv[])
 				exit(EXIT_FAILURE);
 			}
 			break;
+		case 'h':
+			usage(argv[0]);
+			exit(EXIT_SUCCESS);
+			break;
 		default: /* '?' */
 			usage(argv[0]);
 			exit(EXIT_FAILURE);
 		}
 	}
 
-	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;
+	if (filename[0] == '\0') {
+		fprintf(stderr, "An image file MUST be specified.\n");
+		exit_code = EXIT_FAILURE;
+		goto out;
+	}
 
-		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;
-		}
+	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;
 
-	dev = am7xxx_init();
-	if (dev == NULL) {
+	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;
+	}
+
+	ret = am7xxx_init(&ctx);
+	if (ret < 0) {
 		perror("am7xxx_init");
 		exit_code = EXIT_FAILURE;
 		goto out_munmap;
 	}
 
+	am7xxx_set_log_level(ctx, log_level);
+
+	ret = am7xxx_open_device(ctx, &dev, 0);
+	if (ret < 0) {
+		perror("am7xxx_open_device");
+		exit_code = EXIT_FAILURE;
+		goto cleanup;
+	}
+
+
+	ret = am7xxx_close_device(dev);
+	if (ret < 0) {
+		perror("am7xxx_close_device");
+		exit_code = EXIT_FAILURE;
+		goto cleanup;
+	}
+
+	ret = am7xxx_open_device(ctx, &dev, 0);
+	if (ret < 0) {
+		perror("am7xxx_open_device");
+		exit_code = EXIT_FAILURE;
+		goto cleanup;
+	}
+
+	ret = am7xxx_get_device_info(dev, &device_info);
+	if (ret < 0) {
+		perror("am7xxx_get_info");
+		exit_code = EXIT_FAILURE;
+		goto cleanup;
+	}
+	printf("Native resolution: %dx%d\n",
+	       device_info.native_width, device_info.native_height);
+
+	ret = am7xxx_set_power_mode(dev, AM7XXX_POWER_LOW);
+	if (ret < 0) {
+		perror("am7xxx_set_power_mode");
+		exit_code = EXIT_FAILURE;
+		goto cleanup;
+	}
+
 	ret = am7xxx_send_image(dev, format, width, height, image, size);
 	if (ret < 0) {
 		perror("am7xxx_send_image");
@@ -128,21 +194,17 @@ int main(int argc, char *argv[])
 	exit_code = EXIT_SUCCESS;
 
 cleanup:
-	am7xxx_shutdown(dev);
+	am7xxx_shutdown(ctx);
 
 out_munmap:
-	if (image != NULL) {
-		ret = munmap(image, size);
-		if (ret < 0)
-			perror("munmap");
-	}
+	ret = munmap(image, size);
+	if (ret < 0)
+		perror("munmap");
 
 out_close_image_fd:
-	if (image_fd >= 0) {
-		ret = close(image_fd);
-		if (ret < 0)
-			perror("close");
-	}
+	ret = close(image_fd);
+	if (ret < 0)
+		perror("close");
 
 out:
 	exit(exit_code);