LHE_Pi/file_reader.c at master · jjaranda13/LHE_Pi · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
#define STB_IMAGE_STATIC
#define STB_IMAGE_IMPLEMENTATION
#define STB_IMAGE_WRITE_IMPLEMENTATION
#include "stb_image/stb_image.h"
#include "stb_image/stb_image_write.h"

#include "include/globals.h"
#include "include/imgUtil.h"

void init_image_loader (int width, int height)
{
    if (DEBUG) printf ("%s:%s:%d:DEBUG: Called with parameter (%d,%d)\n", __FILE__,__func__ ,__LINE__, width, height);

    if (image_loader_initialized == true)
    {
		fprintf(stderr,"%s:%s:%d:WARN: init_image_loader called but it was already initialized. Skkiping init\n", __FILE__,__func__ ,__LINE__, yuv_model);
		return;
	}

	width_orig_Y = width;
	height_orig_Y = height;

	yuv_model = YUV420;
	height_orig_UV = height/2;
	width_orig_UV = width/2;

	/* Enough memory to hold a YUV444. It is needed as in an
	 * intermediate step uses this memory. Finally it will hold a YUV420
	 */
	orig_Y = (unsigned char**)malloc(height_orig_Y * sizeof(unsigned char *));
	orig_U = (unsigned char**)malloc(height_orig_Y * sizeof(unsigned char *));
	orig_V = (unsigned char**)malloc(height_orig_Y * sizeof(unsigned char *));
	for (int i=0;i<height_orig_Y;i++)
	{
		orig_Y[i] = (unsigned char *)malloc(width_orig_Y* sizeof (unsigned char));
		orig_U[i] = (unsigned char *)malloc(width_orig_Y* sizeof (unsigned char));
		orig_V[i] = (unsigned char *)malloc(width_orig_Y* sizeof (unsigned char));
	}

	image_loader_initialized = true;
	if (DEBUG) printf ("%s:%s:%d:DEBUG: Finished -> width_orig_Y=%d height_orig_Y=%d width_orig_UV=%d height_orig_UV=%d  \n", __FILE__,__func__ ,__LINE__, width_orig_Y, height_orig_Y, height_orig_UV, width_orig_UV);

    return;
}

void init_image_loader_file (char filename[])
{
    if (DEBUG) printf ("%s:%s:%d:DEBUG: Called with parameter (%s)\n", __FILE__,__func__ ,__LINE__, filename);

    if (image_loader_initialized == true)
    {
        fprintf(stderr,"%s:%s:%d:WARN: init_image_loader called but it was already initialized. Skkiping init\n", __FILE__,__func__ ,__LINE__, yuv_model);
        return;
    }

    rgb=stbi_load(filename, &width_orig_Y, &height_orig_Y, &rgb_channels, 0);
	if (rgb == NULL)
    {
		fprintf(stderr,"%s:%s:%d:ERR: Specified file %s can not be loaded\n", __FILE__,__func__ ,__LINE__, filename);
    }
    free(rgb);

    init_image_loader (width_orig_Y, height_orig_Y);

    return;
}

int load_image(char filename[])
{
    if (DEBUG) fprintf(stderr,"%s:%s:%d:INFO: Loading frame %s \n", __FILE__,__func__ ,__LINE__, filename);

    if (image_loader_initialized == false)
    {
        init_image_loader_file(filename);
    }

    rgb=stbi_load(filename, &width_orig_Y, &height_orig_Y, &rgb_channels, 0);
    if (rgb == NULL)
    {
        return 1;
    }

    rgb2yuv(rgb, rgb_channels);
    free(rgb);

    yuv444toyuv420(orig_Y, orig_U, orig_V, width_orig_Y, height_orig_Y);

    return 0;
}

void save_image(char filename[], int width, int height, int channels, unsigned char **Y,unsigned char **U, unsigned char **V, int yuvmodel)
{
    int status;
    unsigned char * data;

    if (DEBUG) fprintf(stderr,"%s:%s:%d:INFO: Saving frame %s \n", __FILE__,__func__ ,__LINE__, filename);
    data = malloc(channels*width*height);
    if (channels==1)
    {
        int pix=0;

        for (int line=0; line < height; line++)
        {
            for (int x=0; x<width; x++)
            {
                data[line*width+x]=Y[line][x];
            }
        }
    }
    else if (channels==3)
    {
        yuv2rgb(Y,U,V,3,width,height, data,yuvmodel);
    }

    status = stbi_write_bmp(filename, width, height, channels, data);
    if (status == 0)
    {
        fprintf(stderr,"%s:%s:%d:ERR: Could not write the specified %s file \n", __FILE__,__func__ ,__LINE__, filename);
    }

    free(data);
    if (DEBUG) fprintf(stderr,"%s:%s:%d:INFO: Save frame Ok\n", __FILE__,__func__ ,__LINE__);
}