tahoma2d/toonz/sources/image/ffmpeg/tiio_mp4.h
manongjohn f2da01b0ab Fix ffmpeg loaded levels (#2876)
* Fix reading ffmpeg frame count and rate

* Make ffmpeg loaded levels uneditable

* Ffmpeg extract frames only when loading

* Fix ffmpeg gif input frame rate
2019-11-13 22:07:58 -07:00

93 lines
2.2 KiB
C++

#pragma once
#ifndef TTIO_MP4_INCLUDED
#define TTIO_MP4_INCLUDED
#include "tproperty.h"
#include "tlevel_io.h"
#include "tiio_ffmpeg.h"
#include <QCoreApplication>
//===========================================================
//
// TLevelWriterMp4
//
//===========================================================
class TLevelWriterMp4 : public TLevelWriter {
public:
TLevelWriterMp4(const TFilePath &path, TPropertyGroup *winfo);
~TLevelWriterMp4();
void setFrameRate(double fps);
TImageWriterP getFrameWriter(TFrameId fid) override;
void save(const TImageP &image, int frameIndex);
void saveSoundTrack(TSoundTrack *st);
static TLevelWriter *create(const TFilePath &path, TPropertyGroup *winfo) {
return new TLevelWriterMp4(path, winfo);
}
private:
Ffmpeg *ffmpegWriter;
int m_lx, m_ly;
int m_scale;
int m_vidQuality;
// void *m_buffer;
};
//===========================================================
//
// TLevelReaderMp4
//
//===========================================================
class TLevelReaderMp4 final : public TLevelReader {
public:
TLevelReaderMp4(const TFilePath &path);
~TLevelReaderMp4();
TImageReaderP getFrameReader(TFrameId fid) override;
static TLevelReader *create(const TFilePath &f) {
return new TLevelReaderMp4(f);
}
TLevelP loadInfo() override;
TImageP load(int frameIndex);
TDimension getSize();
// TThread::Mutex m_mutex;
// void *m_decompressedBuffer;
private:
Ffmpeg *ffmpegReader;
bool ffmpegFramesCreated = false;
TDimension m_size;
int m_frameCount, m_lx, m_ly;
};
//===========================================================================
namespace Tiio {
//===========================================================================
class Mp4WriterProperties : public TPropertyGroup {
Q_DECLARE_TR_FUNCTIONS(Mp4WriterProperties)
public:
// TEnumProperty m_pixelSize;
// TBoolProperty m_matte;
TIntProperty m_vidQuality;
TIntProperty m_scale;
Mp4WriterProperties();
void updateTranslation() override;
};
//===========================================================================
// Tiio::Reader *makeMp4Reader();
// Tiio::Writer *makeMp4Writer();
} // namespace
#endif