2d241f5b9a
Change-Id: Ib80852d20384688a161b0f63593debbd7cdb3b42 Reviewed-on: https://skia-review.googlesource.com/c/skia/+/241760 Reviewed-by: Mike Reed <reed@google.com> Commit-Queue: Mike Reed <reed@google.com> Auto-Submit: Mike Reed <reed@google.com>
135 lines
4.2 KiB
C++
135 lines
4.2 KiB
C++
/*
|
|
* Copyright 2014 Google Inc.
|
|
*
|
|
* Use of this source code is governed by a BSD-style license that can be
|
|
* found in the LICENSE file.
|
|
*/
|
|
|
|
#include "experimental/ffmpeg/SkVideoEncoder.h"
|
|
#include "include/core/SkCanvas.h"
|
|
#include "include/core/SkStream.h"
|
|
#include "include/core/SkSurface.h"
|
|
#include "include/core/SkTime.h"
|
|
#include "modules/skottie/include/Skottie.h"
|
|
#include "modules/skottie/utils/SkottieUtils.h"
|
|
#include "src/utils/SkOSPath.h"
|
|
#include "tools/flags/CommandLineFlags.h"
|
|
|
|
static DEFINE_string2(input, i, "", "skottie animation to render");
|
|
static DEFINE_string2(output, o, "", "mp4 file to create");
|
|
static DEFINE_string2(assetPath, a, "", "path to assets needed for json file");
|
|
static DEFINE_int_2(fps, f, 25, "fps");
|
|
static DEFINE_bool2(verbose, v, false, "verbose mode");
|
|
static DEFINE_bool2(loop, l, false, "loop mode for profiling");
|
|
static DEFINE_int(set_dst_width, 0, "set destination width (height will be computed)");
|
|
|
|
static void produce_frame(SkSurface* surf, skottie::Animation* anim, double frame_time) {
|
|
anim->seekFrameTime(frame_time);
|
|
surf->getCanvas()->clear(SK_ColorWHITE);
|
|
anim->render(surf->getCanvas());
|
|
}
|
|
|
|
int main(int argc, char** argv) {
|
|
CommandLineFlags::SetUsage("Converts skottie to a mp4");
|
|
CommandLineFlags::Parse(argc, argv);
|
|
|
|
if (FLAGS_input.count() == 0) {
|
|
SkDebugf("-i input_file.json argument required\n");
|
|
return -1;
|
|
}
|
|
|
|
SkString assetPath;
|
|
if (FLAGS_assetPath.count() > 0) {
|
|
assetPath.set(FLAGS_assetPath[0]);
|
|
} else {
|
|
assetPath = SkOSPath::Dirname(FLAGS_input[0]);
|
|
}
|
|
SkDebugf("assetPath %s\n", assetPath.c_str());
|
|
|
|
auto animation = skottie::Animation::Builder()
|
|
.setResourceProvider(skottie_utils::FileResourceProvider::Make(assetPath))
|
|
.makeFromFile(FLAGS_input[0]);
|
|
if (!animation) {
|
|
SkDebugf("failed to load %s\n", FLAGS_input[0]);
|
|
return -1;
|
|
}
|
|
|
|
SkISize dim = animation->size().toRound();
|
|
double duration = animation->duration();
|
|
int fps = FLAGS_fps;
|
|
if (fps < 1) {
|
|
fps = 1;
|
|
} else if (fps > 240) {
|
|
fps = 240;
|
|
}
|
|
|
|
float scale = 1;
|
|
if (FLAGS_set_dst_width > 0) {
|
|
scale = FLAGS_set_dst_width / (float)dim.width();
|
|
dim = { FLAGS_set_dst_width, SkScalarRoundToInt(scale * dim.height()) };
|
|
}
|
|
|
|
const int frames = SkScalarRoundToInt(duration * fps);
|
|
const double frame_duration = 1.0 / fps;
|
|
|
|
if (FLAGS_verbose) {
|
|
SkDebugf("Size %dx%d duration %g, fps %d, frame_duration %g\n",
|
|
dim.width(), dim.height(), duration, fps, frame_duration);
|
|
}
|
|
|
|
SkVideoEncoder encoder;
|
|
|
|
sk_sp<SkSurface> surf, tmp_surf;
|
|
sk_sp<SkData> data;
|
|
|
|
do {
|
|
double loop_start = SkTime::GetSecs();
|
|
|
|
encoder.beginRecording(dim, fps);
|
|
// lazily allocate the surfaces
|
|
if (!surf) {
|
|
surf = SkSurface::MakeRaster(encoder.preferredInfo());
|
|
tmp_surf = surf->makeSurface(surf->width(), surf->height());
|
|
|
|
surf->getCanvas()->scale(scale, scale);
|
|
tmp_surf->getCanvas()->scale(scale, scale);
|
|
}
|
|
|
|
for (int i = 0; i <= frames; ++i) {
|
|
double ts = i * 1.0 / fps;
|
|
if (FLAGS_verbose) {
|
|
SkDebugf("rendering frame %d ts %g\n", i, ts);
|
|
}
|
|
|
|
double normal_time = i * 1.0 / frames;
|
|
double frame_time = normal_time * duration;
|
|
|
|
produce_frame(surf.get(), animation.get(), frame_time);
|
|
|
|
SkPixmap pm;
|
|
SkAssertResult(surf->peekPixels(&pm));
|
|
encoder.addFrame(pm);
|
|
}
|
|
data = encoder.endRecording();
|
|
|
|
if (FLAGS_loop) {
|
|
double loop_dur = SkTime::GetSecs() - loop_start;
|
|
SkDebugf("recording secs %g, frames %d, recording fps %d\n",
|
|
loop_dur, frames, (int)(frames / loop_dur));
|
|
}
|
|
} while (FLAGS_loop);
|
|
|
|
if (FLAGS_output.count() == 0) {
|
|
SkDebugf("missing -o output_file.mp4 argument\n");
|
|
return 0;
|
|
}
|
|
|
|
SkFILEWStream ostream(FLAGS_output[0]);
|
|
if (!ostream.isValid()) {
|
|
SkDebugf("Can't create output file %s\n", FLAGS_output[0]);
|
|
return -1;
|
|
}
|
|
ostream.write(data->data(), data->size());
|
|
return 0;
|
|
}
|