JPEG2YUV - TurboJPEG/OSS sample

jpeg2yuv.c

#include <stdio.h>
#include <stdlib.h>
#include <sys/stat.h>
#include "turbojpeg.h"

typedef struct {
    tjhandle tjh;
    FILE *src;
    unsigned char *src_buff;
    unsigned char *dst_buff;
} j2y_hnd_t;


int close(char *msg, j2y_hnd_t *j, int ret)
{
    fprintf(stderr, "%s\n", msg);
    if (j->src_buff) {
        free(j->src_buff);
    }
    if (j->dst_buff) {
        free(j->dst_buff);
    }
    if (j->src) {
        fclose(j->src);
    }
    if (j->tjh && tjDestroy(j->tjh)) {
        fprintf(stderr, "%s\n", tjGetErrorStr());
    }

    return ret;
}

const char *get_yuv_type(int subsample)
{
    const struct {
        int tjsamp;
        char *type;
    } table[] = {
        { TJSAMP_444,  "YUV444"  },
        { TJSAMP_422,  "YUV422"  },
        { TJSAMP_420,  "YUV422"  },
        { TJSAMP_440,  "YUV440"  },
        { TJSAMP_GRAY, "Gray"    },
        { subsample,   "UNKNOWN" }
    };
    int i = 0;
    while (table[i].tjsamp != subsample) i++;
    return table[i].type;
}

int main(int argc, char **argv)
{
    if (argc < 2) {
        return -1;
    }

    struct stat st;
    if (stat(argv[1], &st) != 0) {
        return -1;
    }

    j2y_hnd_t j = {0};
    j.src_buff = calloc(1, st.st_size);
    if (!j.src_buff) {
        return close("src calloc failed", -1);
    }

    j.src = fopen(argv[1], "rb");
    if (!j.src) {
        return close("failed to open src", &j, -1);
    }

    if (fread(j.src_buff, 1, st.st_size, j.src) < st.st_size) {
        return close("failed to read file", &j, -1);
    }

    j.tjh = tjInitDecompress();
    if (!j.tjh) {
        return close(tjGetErrorStr(), &j, -1);
    }

    int width, height, subsample;
    if (tjDecompressHeader2(j.tjh, j.src_buff, st.st_size, &width, &height,
                            &subsample) != 0) {
        return close(tjGetErrorStr(), &j, -1);
    }

    width = ((width + 3) >> 2) << 2;
    if (subsample == TJSAMP_420 || subsample == TJSAMP_440) {
        height += height & 1;
    }

    unsigned long dst_size = tjBufSizeYUV(width, height, subsample);
    j.dst_buff = calloc(1, dst_size);
    if (!j.dst_buff) {
        return close("dst calloc failed\n", &j, -1);
    }

    if (tjDecompressToYUV(j.tjh, j.src_buff, st.st_size, j.dst_buff, 0)) {
        return close(tjGetErrorStr(), &j, -1);
    }

    char dst_name[256];
    sprintf(dst_name, "%s_dst.yuv", argv[1]);
    FILE *dst_yuv = fopen(dst_name, "wb");
    if (!dst_yuv) {
        return close("failed to open dst file\n", &j, -1);
    }

    fwrite(j.dst_buff, 1, dst_size, dst_yuv);
    fclose(dst_yuv);
    fprintf(stderr, "width: %d, height: %d, subsample: %s\n",
            width, height, get_yuv_type(subsample));

    return close("finished", &j, 0);
}