hashirama/uvg266
1
0
Fork 0
mirror of https://github.com/ultravideo/uvg266.git synced 2024-11-27 19:24:06 +00:00
Code Issues Actions Projects Releases Wiki Activity

Merge pull request #10 from VFR-maniac/fix

Browse source 
Avoid reading one extra frame at the end of the input file.
This commit is contained in:
Marko Viitanen 2014-02-04 07:35:47 -08:00
parent a4cd709fc7 de2c4ab78e
commit 2438386f4b
3 changed files with 29 additions and 18 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
src/encmain.c
View file

@ -184,12 +184,16 @@ int main(int argc, char *argv[])
encoder->in.cur_pic->pred_v = MALLOC(pixel, (cfg->width * cfg->height) >> 2);
// Start coding cycle while data on input and not on the last frame
while(!feof(input) && (!cfg->frames || encoder->frame < cfg->frames)) {
while(!cfg->frames || encoder->frame < cfg->frames) {
int32_t diff;
double temp_psnr[3];
// Read one frame from the input
read_one_frame(input, encoder);
if (!read_one_frame(input, encoder)) {
if (!feof(input))
printf("Failed to read a frame %d\n", encoder->frame);
break;
}
// The actual coding happens here, after this function we have a coded frame
encode_one_frame(encoder);

37
src/encoder.c
View file

@ -447,7 +447,7 @@ void fill_after_frame(FILE *file, unsigned height, unsigned array_width,
}
}
void read_and_fill_frame_data(FILE *file, unsigned width, unsigned height,
int read_and_fill_frame_data(FILE *file, unsigned width, unsigned height,
unsigned array_width, pixel *data)
{
pixel* p = data;
@ -457,7 +457,8 @@ void read_and_fill_frame_data(FILE *file, unsigned width, unsigned height,
while (p < end) {
// Read the beginning of the line from input.
fread(p, sizeof(unsigned char), width, file);
if (width != fread(p, sizeof(unsigned char), width, file))
return 0;
// Fill the rest with the last pixel value.
fill_char = p[width - 1];
@ -468,9 +469,10 @@ void read_and_fill_frame_data(FILE *file, unsigned width, unsigned height,
p += array_width;
}
return 1;
}
void read_one_frame(FILE* file, encoder_control* encoder)
int read_one_frame(FILE* file, encoder_control* encoder)
{
encoder_input* in = &encoder->in;
unsigned width = in->real_width;
@ -480,20 +482,24 @@ void read_one_frame(FILE* file, encoder_control* encoder)
if (width != array_width) {
// In the case of frames not being aligned on 8 bit borders, bits need to be copied to fill them in.
read_and_fill_frame_data(file, width, height, array_width,
in->cur_pic->y_data);
read_and_fill_frame_data(file, width >> 1, height >> 1, array_width >> 1,
in->cur_pic->u_data);
read_and_fill_frame_data(file, width >> 1, height >> 1, array_width >> 1,
in->cur_pic->v_data);
if (!read_and_fill_frame_data(file, width, height, array_width,
in->cur_pic->y_data) ||
!read_and_fill_frame_data(file, width >> 1, height >> 1, array_width >> 1,
in->cur_pic->u_data) ||
!read_and_fill_frame_data(file, width >> 1, height >> 1, array_width >> 1,
in->cur_pic->v_data))
return 0;
} else {
// Otherwise the data can be read directly to the array.
fread(in->cur_pic->y_data, sizeof(unsigned char),
width * height, file);
fread(in->cur_pic->u_data, sizeof(unsigned char),
(width >> 1) * (height >> 1), file);
fread(in->cur_pic->v_data, sizeof(unsigned char),
(width >> 1) * (height >> 1), file);
unsigned y_size = width * height;
unsigned uv_size = (width >> 1) * (height >> 1);
if (y_size != fread(in->cur_pic->y_data, sizeof(unsigned char),
y_size, file) ||
uv_size != fread(in->cur_pic->u_data, sizeof(unsigned char),
uv_size, file) ||
uv_size != fread(in->cur_pic->v_data, sizeof(unsigned char),
uv_size, file))
return 0;
}
if (height != array_height) {
@ -504,6 +510,7 @@ void read_one_frame(FILE* file, encoder_control* encoder)
fill_after_frame(file, height >> 1, array_width >> 1, array_height >> 1,
in->cur_pic->v_data);
}
return 1;
}
/**

2
src/encoder.h
View file

@ -86,7 +86,7 @@ encoder_control *init_encoder_control(config *cfg);
void init_encoder_input(encoder_input *input, FILE* inputfile,
int32_t width, int32_t height);
void encode_one_frame(encoder_control *encoder);
void read_one_frame(FILE *file, encoder_control *encoder);
int read_one_frame(FILE *file, encoder_control *encoder);
void encode_seq_parameter_set(encoder_control *encoder);
void encode_pic_parameter_set(encoder_control *encoder);