From 109ffe9a777658936a38d0c146579a67c60a0d17 Mon Sep 17 00:00:00 2001
From: xuxiuxi <xuxiuxi@454eff88-639b-444f-9e54-f578c98de674>
Date: 星期四, 11 五月 2017 17:48:48 +0800
Subject: [PATCH] 

---
 RtspFace/FFmpegRTSPServer/FFmpegH264Encoder.h |    9 +++++----
 1 files changed, 5 insertions(+), 4 deletions(-)

diff --git a/RtspFace/FFmpegRTSPServer/FFmpegH264Encoder.h b/RtspFace/FFmpegRTSPServer/FFmpegH264Encoder.h
index 31e345d..d07e113 100644
--- a/RtspFace/FFmpegRTSPServer/FFmpegH264Encoder.h
+++ b/RtspFace/FFmpegRTSPServer/FFmpegH264Encoder.h
@@ -49,18 +49,19 @@
 	{
 	public:
 		FFmpegH264Encoder();
-		~FFmpegH264Encoder();
+		virtual ~FFmpegH264Encoder();
 		
 		virtual void setCallbackFunctionFrameIsReady(std::function<void()> func);
 		
-		void SetupVideo(std::string filename, int Width, int Height, int FPS, int GOB, int BitPerSecond);
+		bool SetupVideo(std::string filename, int Width, int Height, int FPS, int GOB, int BitPerSecond);
 		void CloseVideo();
-		void SetupCodec(const char *filename, int codec_id);
+		bool SetupCodec(const char *filename, int codec_id);
 		void CloseCodec();
 		
 
 		void SendNewFrame(uint8_t * RGBFrame);		
-		void WriteFrame(uint8_t * RGBFrame);
+		bool WriteFrameRGB(uint8_t * RGBFrame);
+		bool WriteFrameYUV420(AVFrame * YUVFrame);
 		virtual char ReleaseFrame();
 
 		void run();	

--
Gitblit v1.8.0