static void index_rebuild_fallback_finish(FallbackIndexBuilderContext *context, int stop)
{
struct anim *anim = context->anim;
char fname[FILE_MAX];
char fname_tmp[FILE_MAX];
int i;
for (i = 0; i < IMB_PROXY_MAX_SLOT; i++) {
if (context->proxy_sizes_in_use & proxy_sizes[i]) {
AVI_close_compress(context->proxy_ctx[i]);
MEM_freeN(context->proxy_ctx[i]);
get_proxy_filename(anim, proxy_sizes[i], fname_tmp, TRUE);
get_proxy_filename(anim, proxy_sizes[i], fname, FALSE);
if (stop) {
unlink(fname_tmp);
}
else {
unlink(fname);
rename(fname_tmp, fname);
}
}
}
}
static void end_avi(void *context_v)
{
AviMovie *avi = context_v;
if (avi == NULL) return;
AVI_close_compress(avi);
}
static void end_avi(void)
{
if (avi == NULL) return;
AVI_close_compress(avi);
MEM_freeN(avi);
avi = NULL;
}
static void index_rebuild_fallback(struct anim * anim,
IMB_Timecode_Type UNUSED(tcs_in_use),
IMB_Proxy_Size proxy_sizes_in_use,
int quality,
short *stop, short *do_update,
float *progress)
{
int cnt = IMB_anim_get_duration(anim, IMB_TC_NONE);
int i, pos;
AviMovie * proxy_ctx[IMB_PROXY_MAX_SLOT];
char fname[FILE_MAXDIR+FILE_MAXFILE];
char fname_tmp[FILE_MAXDIR+FILE_MAXFILE];
memset(proxy_ctx, 0, sizeof(proxy_ctx));
/* since timecode indices only work with ffmpeg right now,
don't know a sensible fallback here...
so no proxies, no game to play...
*/
if (proxy_sizes_in_use == IMB_PROXY_NONE) {
return;
}
for (i = 0; i < IMB_PROXY_MAX_SLOT; i++) {
if (proxy_sizes_in_use & proxy_sizes[i]) {
char fname[FILE_MAXDIR+FILE_MAXFILE];
get_proxy_filename(anim, proxy_sizes[i], fname, TRUE);
BLI_make_existing_file(fname);
proxy_ctx[i] = alloc_proxy_output_avi(
anim, fname,
anim->x * proxy_fac[i],
anim->y * proxy_fac[i],
quality);
}
}
for (pos = 0; pos < cnt; pos++) {
struct ImBuf * ibuf = IMB_anim_absolute(
anim, pos, IMB_TC_NONE, IMB_PROXY_NONE);
int next_progress = (int) ((double) pos / (double) cnt);
if (*progress != next_progress) {
*progress = next_progress;
*do_update = 1;
}
if (*stop) {
break;
}
IMB_flipy(ibuf);
for (i = 0; i < IMB_PROXY_MAX_SLOT; i++) {
if (proxy_sizes_in_use & proxy_sizes[i]) {
int x = anim->x * proxy_fac[i];
int y = anim->y * proxy_fac[i];
struct ImBuf * s_ibuf = IMB_scalefastImBuf(
ibuf, x, y);
IMB_convert_rgba_to_abgr(s_ibuf);
AVI_write_frame (proxy_ctx[i], pos,
AVI_FORMAT_RGB32,
s_ibuf->rect, x * y * 4);
/* note that libavi free's the buffer... */
s_ibuf->rect = 0;
IMB_freeImBuf(s_ibuf);
}
}
}
for (i = 0; i < IMB_PROXY_MAX_SLOT; i++) {
if (proxy_sizes_in_use & proxy_sizes[i]) {
AVI_close_compress (proxy_ctx[i]);
MEM_freeN (proxy_ctx[i]);
get_proxy_filename(anim, proxy_sizes[i],
fname_tmp, TRUE);
get_proxy_filename(anim, proxy_sizes[i],
fname, FALSE);
if (*stop) {
unlink(fname_tmp);
} else {
rename(fname_tmp, fname);
}
}
}
}
int main(int argc, char *argv[]) {
int opt_shutter_angle = 0;
int opt_mjpeg_quality = 100;
int nf, i, nc, nr;
int tfs, fps;
vc *pos_i, *pos_h, *pos_y;
es_ctx *es;
rs_ctx *rs;
opterr = 0;
while ((i = getopt(argc, argv, "r:q:")) != -1) {
switch (i) {
case 'r':
opt_shutter_angle = atoi(optarg);
break;
case 'q':
opt_mjpeg_quality = atoi(optarg);
break;
default:
print_help(argv);
}
}
if (argc < optind + 2)
print_help(argv);
if (AVI_open_movie(argv[optind], &mv_in) != AVI_ERROR_NONE) {
printf("error: can't read from %s\n", argv[optind]);
return EXIT_FAILURE;
}
if (mv_in.header->Streams < 1 || mv_in.streams[0].sh.Type != AVIST_VIDEO) {
printf("error: video stream not found on %s\n", argv[optind]);
return EXIT_FAILURE;
}
if (AVI_open_compress(argv[optind + 1], &mv_out, 1, AVI_FORMAT_MJPEG) != AVI_ERROR_NONE) {
printf("error: can't write to %s\n", argv[optind + 1]);
return EXIT_FAILURE;
}
printf("status: setup\n");
prepare_lanc_kernels();
nc = mv_in.header->Width;
nr = mv_in.header->Height;
tfs = mv_in.header->TotalFrames;
fps = 1000000 / mv_in.header->MicroSecPerFrame;
pos_i = (vc *)malloc(tfs * sizeof(vc));
pos_h = (vc *)malloc(tfs * sizeof(vc));
pos_y = (vc *)malloc(nr * sizeof(vc));
AVI_set_compress_option(&mv_out, AVI_OPTION_TYPE_MAIN, 0, AVI_OPTION_WIDTH, &nc);
AVI_set_compress_option(&mv_out, AVI_OPTION_TYPE_MAIN, 0, AVI_OPTION_HEIGHT, &nr);
AVI_set_compress_option(&mv_out, AVI_OPTION_TYPE_MAIN, 0, AVI_OPTION_FRAMERATE, &fps);
AVI_set_compress_option(&mv_out, AVI_OPTION_TYPE_MAIN, 0, AVI_OPTION_QUALITY, &opt_mjpeg_quality);
es = es_init(nc, nr);
rs = rs_init(nc, nr);
printf("status: estimating\n");
for (nf = 0; nf < tfs; nf ++) {
unsigned char *fr = (unsigned char *)AVI_read_frame(&mv_in, AVI_FORMAT_RGB24, nf, 0);
pos_i[nf] = vc_add(
nf > 0 ? pos_i[nf - 1] : vc_set(0.0, 0.0),
es_estimate(es, fr)
);
free(fr);
if ((nf + 1) % 10 == 0) {
printf(".");
fflush(stdout);
}
}
printf("\nstatus: filtering\n");
hipass(pos_i, pos_h, tfs, fps / 2);
printf("status: resampling\n");
for (nf = 0; nf < tfs; nf ++) {
unsigned char *fr = (unsigned char *)AVI_read_frame(&mv_in, AVI_FORMAT_RGB24, nf, 0);
for (i = 0; i < nr; i ++) {
pos_y[i] = interp(
pos_h, tfs,
nf + (i - nr / 2.0) * opt_shutter_angle / (nr * 360.0)
);
}
rs_resample(rs, fr, pos_y);
AVI_write_frame(&mv_out, nf, AVI_FORMAT_RGB24, fr, nc * nr * 3 * sizeof(unsigned char));
if ((nf + 1) % 10 == 0) {
printf(".");
fflush(stdout);
}
}
printf("\nstatus: closing\n");
es_free(es);
rs_free(rs);
free_lanc_kernels();
AVI_close(&mv_in);
AVI_close_compress(&mv_out);
return EXIT_SUCCESS;
}
