From 6b998b5ed066aeece1146fe245b35965319b3cbd Mon Sep 17 00:00:00 2001 From: Mattias Andrée Date: Wed, 10 May 2017 16:59:26 +0200 Subject: Cleaner code MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Signed-off-by: Mattias Andrée --- src/blind-rewrite-head.c | 15 +++++---------- 1 file changed, 5 insertions(+), 10 deletions(-) (limited to 'src/blind-rewrite-head.c') diff --git a/src/blind-rewrite-head.c b/src/blind-rewrite-head.c index 5e25645..894e431 100644 --- a/src/blind-rewrite-head.c +++ b/src/blind-rewrite-head.c @@ -4,11 +4,7 @@ #include #include -#include -#include -#include #include -#include USAGE("[-h] file [(frames | 'auto') [(width | 'same') (height | 'same') [format | 'same']]]") @@ -17,20 +13,19 @@ rewrite(struct stream *stream, int frames_auto) { char head[STREAM_HEAD_MAX]; ssize_t headlen; - size_t frame_size, frame_count, length; + size_t frame_count, length; struct stat st; char *data; - echeck_frame_size(stream->width, stream->height, stream->pixel_size, 0, stream->file); - frame_size = stream->width * stream->height * stream->pixel_size; + echeck_dimensions(stream, WIDTH | HEIGHT, NULL); if (fstat(stream->fd, &st)) eprintf("fstat %s:", stream->file); if (!S_ISREG(st.st_mode)) eprintf("%s: not a regular file\n", stream->file); - frame_count = (size_t)(st.st_size) / frame_size; - if (frame_count * frame_size != (size_t)(st.st_size) - stream->headlen) + frame_count = (size_t)(st.st_size) / stream->frame_size; + if (frame_count * stream->frame_size != (size_t)(st.st_size) - stream->headlen) eprintf("%s: given the select width and height, " "the file has an incomplete frame\n", stream->file); if (frames_auto) @@ -40,7 +35,7 @@ rewrite(struct stream *stream, int frames_auto) SPRINTF_HEAD_ZN(head, stream->frames, stream->width, stream->height, stream->pixfmt, &headlen); - length = stream->frames * frame_size; + length = stream->frames * stream->frame_size; if (length > (size_t)SSIZE_MAX || (size_t)headlen > (size_t)SSIZE_MAX - length) eprintf("%s: video is too long\n", stream->file); -- cgit v1.2.3-70-g09d2