X-Git-Url: https://git.ao2.it/libam7xxx.git/blobdiff_plain/e73c33d0a4c2f4695f58a19b2a0a06af3b0353bd..9da43e7f4d027d916e6801c57a2763c45f66fb4c:/examples/am7xxx-play.c?ds=sidebyside diff --git a/examples/am7xxx-play.c b/examples/am7xxx-play.c index 531c156..38812a4 100644 --- a/examples/am7xxx-play.c +++ b/examples/am7xxx-play.c @@ -17,6 +17,12 @@ * along with this program. If not, see . */ +/** + * @example examples/am7xxx-play.c + * am7xxx-play uses libavdevice, libavformat, libavcodec and libswscale to + * decode the input, encode it to jpeg and display it with libam7xxx. + */ + #include #include #include @@ -29,6 +35,13 @@ #include +/* On some systems ENOTSUP is not defined, fallback to its value on + * linux which is equal to EOPNOTSUPP which is 95 + */ +#ifndef ENOTSUP +#define ENOTSUP 95 +#endif + static unsigned int run = 1; struct video_input_ctx { @@ -84,7 +97,7 @@ static int video_input_init(struct video_input_ctx *input_ctx, /* get information on the input stream (e.g. format, bitrate, framerate) */ ret = avformat_find_stream_info(input_format_ctx, NULL); if (ret < 0) { - fprintf(stderr, "cannot get information on the stream"); + fprintf(stderr, "cannot get information on the stream\n"); goto cleanup; } @@ -115,7 +128,7 @@ static int video_input_init(struct video_input_ctx *input_ctx, /* find the decoder for the video stream */ input_codec = avcodec_find_decoder(input_codec_ctx->codec_id); if (input_codec == NULL) { - fprintf(stderr, "input_codec is NULL!"); + fprintf(stderr, "input_codec is NULL!\n"); ret = -ENOTSUP; goto cleanup; } @@ -163,7 +176,7 @@ static int video_output_init(struct video_output_ctx *output_ctx, int ret; if (input_ctx == NULL) { - fprintf(stderr, "input_ctx must not be NULL!"); + fprintf(stderr, "input_ctx must not be NULL!\n"); ret = -EINVAL; goto out; } @@ -171,7 +184,7 @@ static int video_output_init(struct video_output_ctx *output_ctx, /* create the encoder context */ output_codec_ctx = avcodec_alloc_context3(NULL); if (output_codec_ctx == NULL) { - fprintf(stderr, "cannot allocate output codec context!"); + fprintf(stderr, "cannot allocate output codec context!\n"); ret = -ENOMEM; goto out; } @@ -193,8 +206,10 @@ static int video_output_init(struct video_output_ctx *output_ctx, output_codec_ctx->bit_rate = (input_ctx->codec_ctx)->bit_rate; output_codec_ctx->width = new_output_width; output_codec_ctx->height = new_output_height; - output_codec_ctx->time_base.num = (input_ctx->codec_ctx)->time_base.num; - output_codec_ctx->time_base.den = (input_ctx->codec_ctx)->time_base.den; + output_codec_ctx->time_base.num = + (input_ctx->format_ctx)->streams[input_ctx->video_stream_index]->time_base.num; + output_codec_ctx->time_base.den = + (input_ctx->format_ctx)->streams[input_ctx->video_stream_index]->time_base.den; /* When the raw format is requested we don't actually need to setup * and open a decoder @@ -270,9 +285,12 @@ static int am7xxx_play(const char *input_format_string, int out_buf_size; uint8_t *out_buf; int out_picture_size; + uint8_t *out_picture; struct SwsContext *sw_scale_ctx; - AVPacket packet; + AVPacket in_packet; + AVPacket out_packet; int got_picture; + int got_packet; int ret = 0; ret = video_input_init(&input_ctx, input_format_string, input_path, input_options); @@ -290,7 +308,8 @@ static int am7xxx_play(const char *input_format_string, /* allocate an input frame */ picture_raw = avcodec_alloc_frame(); if (picture_raw == NULL) { - fprintf(stderr, "cannot allocate the raw picture frame!"); + fprintf(stderr, "cannot allocate the raw picture frame!\n"); + ret = -ENOMEM; goto cleanup_output; } @@ -298,6 +317,7 @@ static int am7xxx_play(const char *input_format_string, picture_scaled = avcodec_alloc_frame(); if (picture_scaled == NULL) { fprintf(stderr, "cannot allocate the scaled picture!\n"); + ret = -ENOMEM; goto cleanup_picture_raw; } @@ -308,6 +328,7 @@ static int am7xxx_play(const char *input_format_string, out_buf = av_malloc(out_buf_size * sizeof(uint8_t)); if (out_buf == NULL) { fprintf(stderr, "cannot allocate output data buffer!\n"); + ret = -ENOMEM; goto cleanup_picture_scaled; } @@ -329,12 +350,13 @@ static int am7xxx_play(const char *input_format_string, NULL, NULL, NULL); if (sw_scale_ctx == NULL) { fprintf(stderr, "cannot set up the rescaling context!\n"); + ret = -EINVAL; goto cleanup_out_buf; } while (run) { /* read packet */ - ret = av_read_frame(input_ctx.format_ctx, &packet); + ret = av_read_frame(input_ctx.format_ctx, &in_packet); if (ret < 0) { if (ret == (int)AVERROR_EOF || input_ctx.format_ctx->pb->eof_reached) ret = 0; @@ -344,7 +366,7 @@ static int am7xxx_play(const char *input_format_string, goto end_while; } - if (packet.stream_index != input_ctx.video_stream_index) { + if (in_packet.stream_index != input_ctx.video_stream_index) { /* that is more or less a "continue", but there is * still the packet to free */ goto end_while; @@ -352,7 +374,7 @@ static int am7xxx_play(const char *input_format_string, /* decode */ got_picture = 0; - ret = avcodec_decode_video2(input_ctx.codec_ctx, picture_raw, &got_picture, &packet); + ret = avcodec_decode_video2(input_ctx.codec_ctx, picture_raw, &got_picture, &in_packet); if (ret < 0) { fprintf(stderr, "cannot decode video\n"); run = 0; @@ -371,20 +393,26 @@ static int am7xxx_play(const char *input_format_string, picture_scaled->linesize); if (output_ctx.raw_output) { + out_picture = out_buf; out_picture_size = out_buf_size; } else { picture_scaled->quality = (output_ctx.codec_ctx)->global_quality; - /* TODO: switch to avcodec_encode_video2() eventually */ - out_picture_size = avcodec_encode_video(output_ctx.codec_ctx, - out_buf, - out_buf_size, - picture_scaled); - if (out_picture_size < 0) { + av_init_packet(&out_packet); + out_packet.data = NULL; + out_packet.size = 0; + got_packet = 0; + ret = avcodec_encode_video2(output_ctx.codec_ctx, + &out_packet, + picture_scaled, + &got_packet); + if (ret < 0 || !got_packet) { fprintf(stderr, "cannot encode video\n"); - ret = out_picture_size; run = 0; goto end_while; } + + out_picture = out_packet.data; + out_picture_size = out_packet.size; } #ifdef DEBUG @@ -395,7 +423,7 @@ static int am7xxx_play(const char *input_format_string, else snprintf(filename, NAME_MAX, "out.raw"); file = fopen(filename, "wb"); - fwrite(out_buf, 1, out_picture_size, file); + fwrite(out_picture, 1, out_picture_size, file); fclose(file); #endif @@ -403,7 +431,7 @@ static int am7xxx_play(const char *input_format_string, image_format, (output_ctx.codec_ctx)->width, (output_ctx.codec_ctx)->height, - out_buf, + out_picture, out_picture_size); if (ret < 0) { perror("am7xxx_send_image"); @@ -412,7 +440,9 @@ static int am7xxx_play(const char *input_format_string, } } end_while: - av_free_packet(&packet); + if (!output_ctx.raw_output) + av_free_packet(&out_packet); + av_free_packet(&in_packet); } sws_freeContext(sw_scale_ctx); @@ -520,6 +550,7 @@ static void unset_run(int signo) run = 0; } +#ifdef HAVE_SIGACTION static int set_signal_handler(void (*signal_handler)(int)) { struct sigaction new_action; @@ -547,25 +578,40 @@ static int set_signal_handler(void (*signal_handler)(int)) out: return ret; } +#else +static int set_signal_handler(void (*signal_handler)(int)) +{ + (void)signal_handler; + fprintf(stderr, "set_signal_handler() not implemented, sigaction not available\n"); + return 0; +} +#endif + static void usage(char *name) { printf("usage: %s [OPTIONS]\n\n", name); printf("OPTIONS:\n"); + printf("\t-d \t\tthe device index (default is 0)\n"); printf("\t-f \tthe input device format\n"); printf("\t-i \t\tthe input path\n"); printf("\t-o \t\ta comma separated list of input format options\n"); printf("\t\t\t\tEXAMPLE:\n"); printf("\t\t\t\t\t-o draw_mouse=1,framerate=100,video_size=800x480\n"); - printf("\t-s \t\tthe rescaling method (see swscale.h)\n"); + printf("\t-s \tthe rescaling method (see swscale.h)\n"); printf("\t-u \t\t\tupscale the image if smaller than the display dimensions\n"); printf("\t-F \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 - NV12\n"); printf("\t-q \t\tquality of jpeg sent to the device, between 1 and 100\n"); - printf("\t-p \tpower level of device, between %x (off) and %x (maximum)\n", AM7XXX_POWER_OFF, AM7XXX_POWER_TURBO); - printf("\t\t\t\tWARNING: Level 2 and greater require the master AND\n\t\t\t\t\t the slave connector to be plugged in.\n"); + printf("\t-l \t\tthe verbosity level of libam7xxx output (0-5)\n"); + printf("\t-p \t\tthe power mode of device, between %d (off) and %d (turbo)\n", + AM7XXX_POWER_OFF, AM7XXX_POWER_TURBO); + printf("\t\t\t\tWARNING: Level 2 and greater require the master AND\n"); + printf("\t\t\t\t the slave connector to be plugged in.\n"); + printf("\t-z \t\tthe display zoom mode, between %d (original) and %d (test)\n", + AM7XXX_ZOOM_ORIGINAL, AM7XXX_ZOOM_TEST); printf("\t-h \t\t\tthis help message\n"); printf("\n\nEXAMPLES OF USE:\n"); printf("\t%s -f x11grab -i :0.0 -o video_size=800x480\n", name); @@ -587,13 +633,24 @@ int main(int argc, char *argv[]) unsigned int rescale_method = SWS_BICUBIC; unsigned int upscale = 0; unsigned int quality = 95; + int log_level = AM7XXX_LOG_INFO; + int device_index = 0; am7xxx_power_mode power_mode = AM7XXX_POWER_LOW; + am7xxx_zoom_mode zoom = AM7XXX_ZOOM_ORIGINAL; int format = AM7XXX_IMAGE_FORMAT_JPEG; am7xxx_context *ctx; am7xxx_device *dev; - while ((opt = getopt(argc, argv, "f:i:o:s:uF:q:hp:")) != -1) { + while ((opt = getopt(argc, argv, "d:f:i:o:s:uF:q:l:p:z:h")) != -1) { switch (opt) { + case 'd': + device_index = atoi(optarg); + if (device_index < 0) { + fprintf(stderr, "Unsupported device index\n"); + ret = -EINVAL; + goto out; + } + break; case 'f': input_format_string = strdup(optarg); break; @@ -601,6 +658,7 @@ int main(int argc, char *argv[]) input_path = strdup(optarg); break; case 'o': +#ifdef HAVE_STRTOK_R /* * parse suboptions, the expected format is something * like: @@ -617,10 +675,13 @@ int main(int argc, char *argv[]) av_dict_set(&options, subopt_name, subopt_value, 0); } free(subopts_saved); +#else + fprintf(stderr, "Option '-o' not implemented\n"); +#endif break; case 's': rescale_method = atoi(optarg); - switch(format) { + switch(rescale_method) { case SWS_FAST_BILINEAR: case SWS_BILINEAR: case SWS_BICUBIC: @@ -636,7 +697,7 @@ int main(int argc, char *argv[]) default: fprintf(stderr, "Unsupported rescale method\n"); ret = -EINVAL; - goto out;; + goto out; } break; case 'u': @@ -654,7 +715,7 @@ int main(int argc, char *argv[]) default: fprintf(stderr, "Unsupported format\n"); ret = -EINVAL; - goto out;; + goto out; } break; case 'q': @@ -662,13 +723,15 @@ int main(int argc, char *argv[]) if (quality < 1 || quality > 100) { fprintf(stderr, "Invalid quality value, must be between 1 and 100\n"); ret = -EINVAL; - goto out;; + goto out; } break; - case 'h': - usage(argv[0]); - ret = 0; - goto out; + 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 'p': power_mode = atoi(optarg); @@ -678,14 +741,34 @@ int main(int argc, char *argv[]) case AM7XXX_POWER_MIDDLE: case AM7XXX_POWER_HIGH: case AM7XXX_POWER_TURBO: - fprintf(stdout, "Power mode: %x\n", power_mode); + fprintf(stdout, "Power mode: %d\n", power_mode); break; default: - fprintf(stderr, "Invalid power mode value, must be between %x and %x\n", AM7XXX_POWER_OFF, AM7XXX_POWER_TURBO); + fprintf(stderr, "Invalid power mode value, must be between %d and %d\n", + AM7XXX_POWER_OFF, AM7XXX_POWER_TURBO); ret = -EINVAL; goto out; } break; + case 'z': + zoom = atoi(optarg); + switch(zoom) { + case AM7XXX_ZOOM_ORIGINAL: + case AM7XXX_ZOOM_H: + case AM7XXX_ZOOM_H_V: + case AM7XXX_ZOOM_TEST: + fprintf(stdout, "Zoom: %d\n", zoom); + break; + default: + fprintf(stderr, "Invalid zoom mode value, must be between %d and %d\n", + AM7XXX_ZOOM_ORIGINAL, AM7XXX_ZOOM_TEST); + exit(EXIT_FAILURE); + } + break; + case 'h': + usage(argv[0]); + ret = 0; + goto out; default: /* '?' */ usage(argv[0]); ret = -EINVAL; @@ -732,18 +815,30 @@ int main(int argc, char *argv[]) goto out; } - ret = am7xxx_open_device(ctx, &dev, 0); + am7xxx_set_log_level(ctx, log_level); + + ret = am7xxx_open_device(ctx, &dev, device_index); if (ret < 0) { perror("am7xxx_open_device"); goto cleanup; } + ret = am7xxx_set_zoom_mode(dev, zoom); + if (ret < 0) { + perror("am7xxx_set_zoom_mode"); + goto cleanup; + } + ret = am7xxx_set_power_mode(dev, power_mode); if (ret < 0) { perror("am7xxx_set_power_mode"); goto cleanup; } + /* When setting AM7XXX_ZOOM_TEST don't display the actual image */ + if (zoom == AM7XXX_ZOOM_TEST) + goto cleanup; + ret = am7xxx_play(input_format_string, &options, input_path,