X-Git-Url: https://git.cinelerra-gg.org/git/?a=blobdiff_plain;f=cinelerra-5.0%2Fcinelerra%2Fffmpeg.h;h=68649d16218c5f03763c00b282e41589b762a1bc;hb=6c0c8bd0e577001d1cc18c6c27d58e62f58a6bff;hp=28b191489c950117049afe3c0ba23ddc2009a7fe;hpb=91efd376233a15f6572e6f68d28a5bee69797e87;p=goodguy%2Fhistory.git

diff --git a/cinelerra-5.0/cinelerra/ffmpeg.h b/cinelerra-5.0/cinelerra/ffmpeg.h
index 28b19148..68649d16 100644
--- a/cinelerra-5.0/cinelerra/ffmpeg.h
+++ b/cinelerra-5.0/cinelerra/ffmpeg.h
@@ -17,6 +17,7 @@
 #include "ffmpeg.inc"
 #include "filebase.inc"
 #include "fileffmpeg.inc"
+#include "indexstate.inc"
 #include "mutex.h"
 #include "thread.h"
 #include "vframe.inc"
@@ -67,7 +68,7 @@ public:
 
 class FFStream {
 public:
-	FFStream(FFMPEG *ffmpeg, AVStream *st, int idx);
+	FFStream(FFMPEG *ffmpeg, AVStream *st, int fidx);
 	~FFStream();
 	static void ff_lock(const char *cp=0);
 	static void ff_unlock();
@@ -77,12 +78,20 @@ public:
 	virtual int encode_activate();
 	virtual int decode_activate();
 	int read_packet();
+	int seek(int64_t no, double rate);
+	int write_packet(FFPacket &pkt);
+	int flush();
 	int decode(AVFrame *frame);
+	void load_markers(IndexMarks &marks, double rate);
 
-	virtual int decode_frame(AVFrame *frame, int &got_frame) = 0;
+	virtual int is_audio() = 0;
+	virtual int is_video() = 0;
+	virtual int decode_frame(AVPacket *pkt, AVFrame *frame, int &got_frame) = 0;
+	virtual int encode_frame(AVPacket *pkt, AVFrame *frame, int &got_frame) = 0;
 	virtual int init_frame(AVFrame *frame) = 0;
 	virtual int create_filter(const char *filter_spec,
 		AVCodecContext *src_ctx, AVCodecContext *sink_ctx) = 0;
+	virtual void load_markers() = 0;
 	int create_filter(const char *filter_spec);
 	int load_filter(AVFrame *frame);
 	int read_filter(AVFrame *frame);
@@ -120,11 +129,13 @@ public:
 	int frm_count;
 	List<FFrame> frms;
 	Mutex *frm_lock;
+	IndexMarks *index_markers;
 
 	int64_t nudge;
-	int idx;
+	int64_t seek_pos, curr_pos;
+	int fidx;
 	int reading, writing;
-	int eof;
+	int seeked, eof;
 
 	int st_eof() { return eof; }
 	void st_eof(int v) { eof = v; }
@@ -132,16 +143,15 @@ public:
 
 class FFAudioStream : public FFStream {
 	float *inp, *outp, *bfr, *lmt;
-	long sz;
+	int64_t hpos, sz;
 	int nch;
 
 	int read(float *fp, long len);
-	void realloc(long sz, int nch, long len);
-	void realloc(long sz, int nch);
-	void reserve(long sz, int nch);
+	void realloc(long nsz, int nch, long len);
+	void realloc(long nsz, int nch);
+	void reserve(long nsz, int nch);
 	long used();
 	long avail();
-	void reset();
 	void iseek(int64_t ofs);
 	float *get_outp(int len);
 	int64_t put_inp(int len);
@@ -149,17 +159,23 @@ class FFAudioStream : public FFStream {
 	int zero(long len);
 	int write(const double *dp, long len, int ch);
 public:
-	FFAudioStream(FFMPEG *ffmpeg, AVStream *strm, int idx);
+	FFAudioStream(FFMPEG *ffmpeg, AVStream *strm, int idx, int fidx);
 	virtual ~FFAudioStream();
+	int is_audio() { return 1; }
+	int is_video() { return 0; }
+	int get_samples(float *&samples, uint8_t **data, int len);
 	int load_history(uint8_t **data, int len);
-	int decode_frame(AVFrame *frame, int &got_frame);
+	int decode_frame(AVPacket *pkt, AVFrame *frame, int &got_frame);
+	int encode_frame(AVPacket *pkt, AVFrame *frame, int &got_frame);
 	int create_filter(const char *filter_spec,
 		AVCodecContext *src_ctx, AVCodecContext *sink_ctx);
+	void load_markers();
 
 	int encode_activate();
 	int nb_samples();
 	int64_t load_buffer(double ** const sp, int len);
 	int in_history(int64_t pos);
+	void reset_history();
 	int read(double *dp, long len, int ch);
 
 	int init_frame(AVFrame *frame);
@@ -167,10 +183,10 @@ public:
 	int audio_seek(int64_t pos);
 	int encode(double **samples, int len);
 
+	int idx;
 	int channel0, channels;
 	int sample_rate;
 	int mbsz, frame_sz;
-	int64_t seek_pos, curr_pos;
 	int64_t length;
 
 	SwrContext *resample_context;
@@ -178,40 +194,57 @@ public:
 	float *aud_bfr;
 };
 
-class FFVideoStream : public FFStream {
+
+class FFVideoConvert {
 public:
-	FFVideoStream(FFMPEG *ffmpeg, AVStream *strm, int idx);
+	struct SwsContext *convert_ctx;
+
+	FFVideoConvert() { convert_ctx = 0; }
+	~FFVideoConvert() { if( convert_ctx ) sws_freeContext(convert_ctx); }
+
+	static PixelFormat color_model_to_pix_fmt(int color_model);
+	static int pix_fmt_to_color_model(PixelFormat pix_fmt);
+
+	int convert_picture_vframe(VFrame *frame,
+		AVPicture *ip, PixelFormat ifmt, int iw, int ih);
+	int convert_cmodel(VFrame *frame_out,
+		AVPicture *ip, PixelFormat ifmt, int iw, int ih);
+	int transfer_cmodel(VFrame *frame_in,  //defaults->metadata
+		AVFrame *ifp, PixelFormat ifmt, int iw, int ih);
+	int convert_vframe_picture(VFrame *frame,
+		AVPicture *op, PixelFormat ofmt, int ow, int oh);
+	int convert_pixfmt(VFrame *frame_in,
+		 AVPicture *op, PixelFormat ofmt, int ow, int oh);
+	int transfer_pixfmt(VFrame *frame_in,  //metadata->defaults
+		 AVFrame *ofp, PixelFormat ofmt, int ow, int oh);
+};
+
+class FFVideoStream : public FFStream, public FFVideoConvert {
+public:
+	FFVideoStream(FFMPEG *ffmpeg, AVStream *strm, int idx, int fidx);
 	virtual ~FFVideoStream();
-	int decode_frame(AVFrame *frame, int &got_frame);
+	int is_audio() { return 0; }
+	int is_video() { return 1; }
+	int decode_frame(AVPacket *pkt, AVFrame *frame, int &got_frame);
+	int encode_frame(AVPacket *pkt, AVFrame *frame, int &got_frame);
 	int create_filter(const char *filter_spec,
 		AVCodecContext *src_ctx, AVCodecContext *sink_ctx);
+	void load_markers();
 
 	int init_frame(AVFrame *picture);
 	int load(VFrame *vframe, int64_t pos);
 	int video_seek(int64_t pos);
 	int encode(VFrame *vframe);
 
+	int idx;
 	double frame_rate;
 	int width, height;
-	int64_t seek_pos, curr_pos;
 	int64_t length;
 	float aspect_ratio;
 
-	struct SwsContext *convert_ctx;
 	uint8_t *pkt_bfr;
 	int pkt_bfr_sz;
-
-	static PixelFormat color_model_to_pix_fmt(int color_model);
-	static int pix_fmt_to_color_model(PixelFormat pix_fmt);
-
-	int convert_picture_vframe(VFrame *frame,
-		AVPicture *ip, PixelFormat ifmt, int iw, int ih);
-	int convert_cmodel(VFrame *frame_out,
-		AVPicture *ip, PixelFormat ifmt, int iw, int ih);
-	int convert_vframe_picture(VFrame *frame,
-		AVPicture *op, PixelFormat ofmt, int ow, int oh);
-	int convert_pixfmt(VFrame *frame_in,
-		 AVPicture *op, PixelFormat ofmt, int ow, int oh);
+	int64_t start_pts;
 };
 
 class FFMPEG : public Thread {
@@ -222,7 +255,7 @@ public:
 
 	int check_sample_rate(AVCodec *codec, int sample_rate);
 	AVRational check_frame_rate(AVCodec *codec, double frame_rate);
-	AVRational to_sample_aspect_ratio(double aspect_ratio);
+	AVRational to_sample_aspect_ratio(Asset *asset);
 	AVRational to_time_base(int sample_rate);
 
 	static void set_option_path(char *path, const char *fmt, ...);
@@ -277,7 +310,7 @@ public:
 		uint16_t st_idx, st_ch;
 		ffidx() { st_idx = st_ch = 0; }
 		ffidx(const ffidx &t) { st_idx = t.st_idx;  st_ch = t.st_ch; }
-		ffidx(uint16_t idx, uint16_t ch) { st_idx = idx; st_ch = ch; }
+		ffidx(uint16_t fidx, uint16_t ch) { st_idx = fidx; st_ch = ch; }
 	};
 
 	ArrayList<ffidx> astrm_index;
@@ -301,6 +334,7 @@ public:
 
 	FFMPEG(FileBase *file_base=0);
 	~FFMPEG();
+	int scan(IndexState *index_state, int64_t *scan_position, int *canceled);
 
 	int ff_audio_stream(int channel) { return astrm_index[channel].st_idx; }
 	int ff_video_stream(int layer) { return vstrm_index[layer].st_idx; }