138 lines
4.6 KiB
C
138 lines
4.6 KiB
C
|
// Copyright (c) 2009, Karl Blomster
|
||
|
// All rights reserved.
|
||
|
//
|
||
|
// Redistribution and use in source and binary forms, with or without
|
||
|
// modification, are permitted provided that the following conditions are met:
|
||
|
//
|
||
|
// * Redistributions of source code must retain the above copyright notice,
|
||
|
// this list of conditions and the following disclaimer.
|
||
|
// * Redistributions in binary form must reproduce the above copyright notice,
|
||
|
// this list of conditions and the following disclaimer in the documentation
|
||
|
// and/or other materials provided with the distribution.
|
||
|
// * Neither the name of the Aegisub Group nor the names of its contributors
|
||
|
// may be used to endorse or promote products derived from this software
|
||
|
// without specific prior written permission.
|
||
|
//
|
||
|
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
|
||
|
// AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
||
|
// IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
|
||
|
// ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
|
||
|
// LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
|
||
|
// CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
|
||
|
// SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
|
||
|
// INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
|
||
|
// CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
|
||
|
// ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
|
||
|
// POSSIBILITY OF SUCH DAMAGE.
|
||
|
//
|
||
|
// -----------------------------------------------------------------------------
|
||
|
//
|
||
|
// AEGISUB
|
||
|
//
|
||
|
// Website: http://aegisub.cellosoft.com
|
||
|
// Contact: mailto:zeratul@cellosoft.com
|
||
|
//
|
||
|
|
||
|
|
||
|
#pragma once
|
||
|
|
||
|
#include "include/aegisub/video_provider.h"
|
||
|
#include <wx/wxprec.h>
|
||
|
#include <stdio.h>
|
||
|
#include <vector>
|
||
|
|
||
|
// ffmpeg uses 80, so I'm p sure this isn't too small
|
||
|
#define YUV4MPEG_HEADER_MAXLEN 128
|
||
|
|
||
|
|
||
|
class YUV4MPEGVideoProvider : public VideoProvider {
|
||
|
private:
|
||
|
enum Y4M_PixelFormat {
|
||
|
Y4M_PIXFMT_NONE = -1,
|
||
|
Y4M_PIXFMT_420JPEG,
|
||
|
Y4M_PIXFMT_420MPEG2,
|
||
|
Y4M_PIXFMT_420PALDV,
|
||
|
Y4M_PIXFMT_411,
|
||
|
Y4M_PIXFMT_422,
|
||
|
Y4M_PIXFMT_444,
|
||
|
Y4M_PIXFMT_444ALPHA,
|
||
|
Y4M_PIXFMT_MONO,
|
||
|
};
|
||
|
|
||
|
enum Y4M_InterlacingMode {
|
||
|
Y4M_ILACE_NOTSET = -1, // not to be confused with Y4M_ILACE_UNKNOWN
|
||
|
Y4M_ILACE_PROGRESSIVE,
|
||
|
Y4M_ILACE_TFF,
|
||
|
Y4M_ILACE_BFF,
|
||
|
Y4M_ILACE_MIXED,
|
||
|
Y4M_ILACE_UNKNOWN,
|
||
|
};
|
||
|
|
||
|
// this is currently unused :(
|
||
|
enum Y4M_FrameFlags {
|
||
|
Y4M_FFLAG_NOTSET = -1,
|
||
|
Y4M_FFLAG_NONE = 0x0000,
|
||
|
// repeat field/frame flags
|
||
|
Y4M_FFLAG_R_TFF = 0x0001, // TFF
|
||
|
Y4M_FFLAG_R_TFF_R = 0x0002, // TFF and repeat
|
||
|
Y4M_FFLAG_R_BFF = 0x0004, // BFF
|
||
|
Y4M_FFLAG_R_BFF_R = 0x0008, // BFF and repeat
|
||
|
Y4M_FFLAG_R_P = 0x0010, // progressive
|
||
|
Y4M_FFLAG_R_P_R = 0x0020, // progressive and repeat once
|
||
|
Y4M_FFLAG_R_P_RR = 0x0040, // progressive and repeat twice
|
||
|
// temporal sampling flags
|
||
|
Y4M_FFLAG_T_P = 0x0080, // progressive (fields sampled at the same time)
|
||
|
Y4M_FFLAG_T_I = 0x0100, // interlaced (fields sampled at different times)
|
||
|
// spatial sampling flags
|
||
|
Y4M_FFLAG_C_P = 0x0200, // progressive (whole frame subsampled)
|
||
|
Y4M_FFLAG_C_I = 0x0400, // interlaced (fields subsampled independently)
|
||
|
Y4M_FFLAG_C_UNKNOWN = 0x0800, // unknown (only allowed for non-4:2:0 sampling)
|
||
|
};
|
||
|
|
||
|
FILE *sf; // source file
|
||
|
bool inited;
|
||
|
int w, h; // width/height
|
||
|
int num_frames; // length of file in frames
|
||
|
int frame_sz; // size of each frame in bytes
|
||
|
Y4M_PixelFormat pixfmt; // colorspace/pixel format
|
||
|
Y4M_InterlacingMode imode; // interlacing mode
|
||
|
struct {
|
||
|
int num;
|
||
|
int den;
|
||
|
} fps_rat; // framerate
|
||
|
|
||
|
std::vector<int64_t> seek_table; // the position in the file of each frame, in bytes
|
||
|
int cur_fn; // current frame number
|
||
|
|
||
|
wxString errmsg;
|
||
|
|
||
|
void LoadVideo(const Aegisub::String filename);
|
||
|
void Close();
|
||
|
|
||
|
void CheckFileFormat();
|
||
|
void ParseFileHeader(const std::vector<wxString>& tags);
|
||
|
Y4M_FrameFlags ParseFrameHeader(const std::vector<wxString>& tags);
|
||
|
std::vector<wxString> ReadHeader(int64_t startpos, bool reset_pos=false);
|
||
|
int IndexFile();
|
||
|
|
||
|
public:
|
||
|
YUV4MPEGVideoProvider(Aegisub::String filename, double fps);
|
||
|
~YUV4MPEGVideoProvider();
|
||
|
|
||
|
const AegiVideoFrame GetFrame(int n, int formatType);
|
||
|
int GetPosition();
|
||
|
int GetFrameCount();
|
||
|
|
||
|
int GetWidth();
|
||
|
int GetHeight();
|
||
|
double GetFPS();
|
||
|
bool AreKeyFramesLoaded() { return false; }
|
||
|
wxArrayInt GetKeyFrames() { return wxArrayInt(); };
|
||
|
bool IsVFR() { return false; };
|
||
|
FrameRate GetTrueFrameRate() { return FrameRate(); };
|
||
|
Aegisub::String GetDecoderName() { return L"YUV4MPEG"; }
|
||
|
bool IsNativelyByFrames() { return true; }
|
||
|
int GetDesiredCacheSize() { return 8; }
|
||
|
};
|
||
|
|