142 lines
4.4 KiB
C
142 lines
4.4 KiB
C
/*
|
|
* Copyright © 2018-2021, VideoLAN and dav1d authors
|
|
* Copyright © 2018-2021, Two Orioles, LLC
|
|
* All rights reserved.
|
|
*
|
|
* Redistribution and use in source and binary forms, with or without
|
|
* modification, are permitted provided that the following conditions are met:
|
|
*
|
|
* 1. Redistributions of source code must retain the above copyright notice, this
|
|
* list of conditions and the following disclaimer.
|
|
*
|
|
* 2. Redistributions in binary form must reproduce the above copyright notice,
|
|
* this list of conditions and the following disclaimer in the documentation
|
|
* and/or other materials provided with the distribution.
|
|
*
|
|
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
|
|
* ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
|
|
* WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
|
|
* DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR
|
|
* ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
|
|
* (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
|
|
* LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
|
|
* ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
|
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
|
|
* SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
|
*/
|
|
|
|
#include "config.h"
|
|
|
|
#include <errno.h>
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
|
|
#define XXH_INLINE_ALL
|
|
#include "xxhash.h"
|
|
|
|
#include "output/muxer.h"
|
|
|
|
typedef struct MuxerPriv {
|
|
XXH3_state_t* state;
|
|
FILE *f;
|
|
} xxh3Context;
|
|
|
|
static int xxh3_open(xxh3Context *const xxh3, const char *const file,
|
|
const Dav1dPictureParameters *const p,
|
|
const unsigned fps[2])
|
|
{
|
|
xxh3->state = XXH3_createState();
|
|
if (!xxh3->state) return DAV1D_ERR(ENOMEM);
|
|
XXH_errorcode err = XXH3_128bits_reset(xxh3->state);
|
|
if (err != XXH_OK) {
|
|
XXH3_freeState(xxh3->state);
|
|
xxh3->state = NULL;
|
|
return DAV1D_ERR(ENOMEM);
|
|
}
|
|
|
|
if (!strcmp(file, "-")) {
|
|
xxh3->f = stdout;
|
|
} else if (!(xxh3->f = fopen(file, "wb"))) {
|
|
XXH3_freeState(xxh3->state);
|
|
xxh3->state = NULL;
|
|
fprintf(stderr, "Failed to open %s: %s\n", file, strerror(errno));
|
|
return -1;
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
static int xxh3_write(xxh3Context *const xxh3, Dav1dPicture *const p) {
|
|
const int hbd = p->p.bpc > 8;
|
|
const int w = p->p.w, h = p->p.h;
|
|
uint8_t *yptr = p->data[0];
|
|
|
|
for (int y = 0; y < h; y++) {
|
|
XXH3_128bits_update(xxh3->state, yptr, w << hbd);
|
|
yptr += p->stride[0];
|
|
}
|
|
|
|
if (p->p.layout != DAV1D_PIXEL_LAYOUT_I400) {
|
|
const int ss_ver = p->p.layout == DAV1D_PIXEL_LAYOUT_I420;
|
|
const int ss_hor = p->p.layout != DAV1D_PIXEL_LAYOUT_I444;
|
|
const int cw = (w + ss_hor) >> ss_hor;
|
|
const int ch = (h + ss_ver) >> ss_ver;
|
|
for (int pl = 1; pl <= 2; pl++) {
|
|
uint8_t *uvptr = p->data[pl];
|
|
|
|
for (int y = 0; y < ch; y++) {
|
|
XXH3_128bits_update(xxh3->state, uvptr, cw << hbd);
|
|
uvptr += p->stride[1];
|
|
}
|
|
}
|
|
}
|
|
|
|
dav1d_picture_unref(p);
|
|
|
|
return 0;
|
|
}
|
|
|
|
static void xxh3_close(xxh3Context *const xxh3) {
|
|
XXH128_hash_t hash = XXH3_128bits_digest(xxh3->state);
|
|
XXH3_freeState(xxh3->state);
|
|
XXH128_canonical_t c;
|
|
XXH128_canonicalFromHash(&c, hash);
|
|
|
|
for (int i = 0; i < 16; i++)
|
|
fprintf(xxh3->f, "%2.2x", c.digest[i]);
|
|
fprintf(xxh3->f, "\n");
|
|
|
|
if (xxh3->f != stdout)
|
|
fclose(xxh3->f);
|
|
}
|
|
|
|
static int xxh3_verify(xxh3Context *const xxh3, const char * xxh3_str) {
|
|
XXH128_hash_t hash = XXH3_128bits_digest(xxh3->state);
|
|
XXH3_freeState(xxh3->state);
|
|
|
|
if (strlen(xxh3_str) < 32)
|
|
return -1;
|
|
|
|
XXH128_canonical_t c;
|
|
char t[3] = { 0 };
|
|
for (int i = 0; i < 16; i++) {
|
|
char *ignore;
|
|
memcpy(t, xxh3_str, 2);
|
|
xxh3_str += 2;
|
|
c.digest[i] = (unsigned char) strtoul(t, &ignore, 16);
|
|
}
|
|
XXH128_hash_t verify = XXH128_hashFromCanonical(&c);
|
|
|
|
return !XXH128_isEqual(hash, verify);
|
|
}
|
|
|
|
const Muxer xxh3_muxer = {
|
|
.priv_data_size = sizeof(xxh3Context),
|
|
.name = "xxh3",
|
|
.extension = "xxh3",
|
|
.write_header = xxh3_open,
|
|
.write_picture = xxh3_write,
|
|
.write_trailer = xxh3_close,
|
|
.verify = xxh3_verify,
|
|
};
|