102 lines
2.2 KiB
Perl
Executable file
102 lines
2.2 KiB
Perl
Executable file
#!/usr/bin/env perl
|
|
|
|
use strict;
|
|
use warnings;
|
|
|
|
my $filename = shift;
|
|
my $arrayname = shift;
|
|
my $track = shift;
|
|
my @timestamps;
|
|
my @hashes;
|
|
my @durations;
|
|
my @widths;
|
|
my @heights;
|
|
my @pixfmts;
|
|
my @keyframes;
|
|
my $ticks = 1;
|
|
my $frames = 0;
|
|
|
|
if (!defined($filename) || !defined($arrayname) || !defined($track)) {
|
|
print("Usage: gendata [filename] [arrayname] [track] > output.cpp\n");
|
|
exit(0);
|
|
}
|
|
|
|
my $version = "Unknown";
|
|
|
|
for (`ffmpeg -version 2>&1`) {
|
|
chomp;
|
|
s/^(ffmpeg version .+) Copyright.+/$1/;
|
|
$version = $_;
|
|
last;
|
|
}
|
|
|
|
print("// Autogenerated from $filename by gendata with $version\n\n");
|
|
print("#ifndef _FFMS_TEST_".uc($arrayname)."_H\n");
|
|
print("#define _FFMS_TEST_".uc($arrayname)."_H\n\n");
|
|
print("#include \"data.h\"\n\n");
|
|
|
|
for (`ffmpeg -i $filename -map $track:0 -f framehash - 2>/dev/null`) {
|
|
if (/^#/) {
|
|
if (!/^#tb/) {
|
|
next;
|
|
}
|
|
chomp;
|
|
s/^#tb \d: //;
|
|
my @tb = split('/');
|
|
$ticks = int($tb[0]);
|
|
next;
|
|
}
|
|
chomp;
|
|
s/\s//g;
|
|
my @fields = split(',');
|
|
push(@hashes, $fields[5]);
|
|
$frames++;
|
|
}
|
|
|
|
for (`ffprobe -show_frames -select_streams $track $filename 2>/dev/null`) {
|
|
chomp;
|
|
|
|
if (/^width/) {
|
|
my @w = split('=');
|
|
push(@widths, int($w[1]));
|
|
} elsif (/^height/) {
|
|
my @h = split('=');
|
|
push(@heights, int($h[1]));
|
|
} elsif (/^pix_fmt/) {
|
|
my @p = split('=');
|
|
push(@pixfmts, $p[1]);
|
|
} elsif (/^key_frame/) {
|
|
my @k = split('=');
|
|
push(@keyframes, (int($k[1]) == 1 ? "true" : "false"));
|
|
} elsif (/^pkt_pts=/) {
|
|
my @t = split('=');
|
|
push(@timestamps, int($t[1]));
|
|
} elsif (/^pkt_duration=/) {
|
|
my @d = split('=');
|
|
push(@durations, int($d[1]));
|
|
} else {
|
|
next;
|
|
}
|
|
}
|
|
|
|
print("const TestFrameData $arrayname\[$frames] = {\n");
|
|
|
|
for (my $i = 0; $i < $frames; $i++) {
|
|
print(" { $timestamps[$i], $durations[$i], $widths[$i], $heights[$i], \"$pixfmts[$i]\", $keyframes[$i], {");
|
|
|
|
my $hash = $hashes[$i];
|
|
$hash =~ s/(..)/ 0x\U$1,/g;
|
|
$hash =~ s/,$//;
|
|
print($hash);
|
|
|
|
print(" } }");
|
|
|
|
if ($i != $frames - 1) {
|
|
print(",");
|
|
}
|
|
|
|
print("\n");
|
|
}
|
|
|
|
print("};\n\n");
|
|
print("#endif // _FFMS_TEST_".uc($arrayname)."_H\n");
|