#include "tiio_webm.h"
#include "tsystem.h"
#include "trasterimage.h"
#include "tsound.h"
#include "timageinfo.h"
#include <QStringList>
//===========================================================
//
// TImageWriterWebm
//
//===========================================================
class TImageWriterWebm : public TImageWriter {
public:
int m_frameIndex;
TImageWriterWebm(const TFilePath &path, int frameIndex, TLevelWriterWebm *lwg)
: TImageWriter(path), m_frameIndex(frameIndex), m_lwg(lwg) {
m_lwg->addRef();
}
~TImageWriterWebm() { m_lwg->release(); }
bool is64bitOutputSupported() override { return false; }
void save(const TImageP &img) override { m_lwg->save(img, m_frameIndex); }
private:
TLevelWriterWebm *m_lwg;
};
//===========================================================
//
// TLevelWriterWebm;
//
//===========================================================
TLevelWriterWebm::TLevelWriterWebm(const TFilePath &path, TPropertyGroup *winfo)
: TLevelWriter(path, winfo) {
if (!m_properties) m_properties = new Tiio::WebmWriterProperties();
std::string scale = m_properties->getProperty("Scale")->getValueAsString();
m_scale = QString::fromStdString(scale).toInt();
std::string quality =
m_properties->getProperty("Quality")->getValueAsString();
m_vidQuality = QString::fromStdString(quality).toInt();
ffmpegWriter = new Ffmpeg();
ffmpegWriter->setPath(m_path);
if (TSystem::doesExistFileOrLevel(m_path)) TSystem::deleteFile(m_path);
}
//-----------------------------------------------------------
TLevelWriterWebm::~TLevelWriterWebm() {
// QProcess createWebm;
QStringList preIArgs;
QStringList postIArgs;
int outLx = m_lx;
int outLy = m_ly;
// set scaling
if (m_scale != 0) {
outLx = m_lx * m_scale / 100;
outLy = m_ly * m_scale / 100;
}
// ffmpeg doesn't like resolutions that aren't divisible by 2.
if (outLx % 2 != 0) outLx++;
if (outLy % 2 != 0) outLy++;
// calculate quality (bitrate)
int pixelCount = m_lx * m_ly;
int bitRate = pixelCount / 150; // crude but gets decent values
double quality = m_vidQuality / 100.0;
double tempRate = (double)bitRate * quality;
int finalBitrate = (int)tempRate;
int crf = 51 - (m_vidQuality * 51 / 100);
preIArgs << "-framerate";
preIArgs << QString::number(m_frameRate);
postIArgs << "-c:v";
postIArgs << "libvpx";
postIArgs << "-s";
postIArgs << QString::number(outLx) + "x" + QString::number(outLy);
postIArgs << "-b";
postIArgs << QString::number(finalBitrate) + "k";
postIArgs << "-speed";
postIArgs << "3";
postIArgs << "-quality";
postIArgs << "good";
ffmpegWriter->runFfmpeg(preIArgs, postIArgs, false, false, true);
ffmpegWriter->cleanUpFiles();
}
//-----------------------------------------------------------
TImageWriterP TLevelWriterWebm::getFrameWriter(TFrameId fid) {
// if (IOError != 0)
// throw TImageException(m_path, buildGifExceptionString(IOError));
if (fid.getLetter() != 0) return TImageWriterP(0);
int index = fid.getNumber();
TImageWriterWebm *iwg = new TImageWriterWebm(m_path, index, this);
return TImageWriterP(iwg);
}
//-----------------------------------------------------------
void TLevelWriterWebm::setFrameRate(double fps) {
m_frameRate = fps;
ffmpegWriter->setFrameRate(fps);
}
void TLevelWriterWebm::saveSoundTrack(TSoundTrack *st) {
ffmpegWriter->saveSoundTrack(st);
}
//-----------------------------------------------------------
void TLevelWriterWebm::save(const TImageP &img, int frameIndex) {
TRasterImageP image(img);
m_lx = image->getRaster()->getLx();
m_ly = image->getRaster()->getLy();
ffmpegWriter->createIntermediateImage(img, frameIndex);
}
//===========================================================
//
// TImageReaderWebm
//
//===========================================================
class TImageReaderWebm final : public TImageReader {
public:
int m_frameIndex;
TImageReaderWebm(const TFilePath &path, int index, TLevelReaderWebm *lra)
: TImageReader(path), m_lra(lra), m_frameIndex(index) {
m_lra->addRef();
}
~TImageReaderWebm() { m_lra->release(); }
TImageP load() override { return m_lra->load(m_frameIndex); }
TDimension getSize() const { return m_lra->getSize(); }
TRect getBBox() const { return TRect(); }
private:
TLevelReaderWebm *m_lra;
// not implemented
TImageReaderWebm(const TImageReaderWebm &);
TImageReaderWebm &operator=(const TImageReaderWebm &src);
};
//===========================================================
//
// TLevelReaderWebm
//
//===========================================================
TLevelReaderWebm::TLevelReaderWebm(const TFilePath &path) : TLevelReader(path) {
ffmpegReader = new Ffmpeg();
ffmpegReader->setPath(m_path);
ffmpegReader->disablePrecompute();
ffmpegFileInfo tempInfo = ffmpegReader->getInfo();
double fps = tempInfo.m_frameRate;
m_frameCount = tempInfo.m_frameCount;
m_size = TDimension(tempInfo.m_lx, tempInfo.m_ly);
m_lx = m_size.lx;
m_ly = m_size.ly;
ffmpegReader->getFramesFromMovie();
// set values
m_info = new TImageInfo();
m_info->m_frameRate = fps;
m_info->m_lx = m_lx;
m_info->m_ly = m_ly;
m_info->m_bitsPerSample = 8;
m_info->m_samplePerPixel = 4;
}
//-----------------------------------------------------------
TLevelReaderWebm::~TLevelReaderWebm() {
// ffmpegReader->cleanUpFiles();
}
//-----------------------------------------------------------
TLevelP TLevelReaderWebm::loadInfo() {
if (m_frameCount == -1) return TLevelP();
TLevelP level;
for (int i = 1; i <= m_frameCount; i++) level->setFrame(i, TImageP());
return level;
}
//-----------------------------------------------------------
TImageReaderP TLevelReaderWebm::getFrameReader(TFrameId fid) {
// if (IOError != 0)
// throw TImageException(m_path, buildAVIExceptionString(IOError));
if (fid.getLetter() != 0) return TImageReaderP(0);
int index = fid.getNumber();
TImageReaderWebm *irm = new TImageReaderWebm(m_path, index, this);
return TImageReaderP(irm);
}
//------------------------------------------------------------------------------
TDimension TLevelReaderWebm::getSize() { return m_size; }
//------------------------------------------------
TImageP TLevelReaderWebm::load(int frameIndex) {
return ffmpegReader->getImage(frameIndex);
}
Tiio::WebmWriterProperties::WebmWriterProperties()
: m_vidQuality("Quality", 1, 100, 90), m_scale("Scale", 1, 100, 100) {
bind(m_vidQuality);
bind(m_scale);
}
// Tiio::Reader* Tiio::makeWebmReader(){ return nullptr; }
// Tiio::Writer* Tiio::makeWebmWriter(){ return nullptr; }