Created by the British Broadcasting Corporation.
00001 /* ***** BEGIN LICENSE BLOCK ***** 00002 * 00003 * $Id: seq_compress.h,v 1.24 2008/06/03 09:57:13 tjdwave Exp $ $Name: Dirac_0_10_0 $ 00004 * 00005 * Version: MPL 1.1/GPL 2.0/LGPL 2.1 00006 * 00007 * The contents of this file are subject to the Mozilla Public License 00008 * Version 1.1 (the "License"); you may not use this file except in compliance 00009 * with the License. You may obtain a copy of the License at 00010 * http://www.mozilla.org/MPL/ 00011 * 00012 * Software distributed under the License is distributed on an "AS IS" basis, 00013 * WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License for 00014 * the specific language governing rights and limitations under the License. 00015 * 00016 * The Original Code is BBC Research and Development code. 00017 * 00018 * The Initial Developer of the Original Code is the British Broadcasting 00019 * Corporation. 00020 * Portions created by the Initial Developer are Copyright (C) 2004. 00021 * All Rights Reserved. 00022 * 00023 * Contributor(s): Thomas Davies (Original Author), 00024 * Scott R Ladd, 00025 * Anuradha Suraparaju 00026 * Andrew Kennedy 00027 * 00028 * Alternatively, the contents of this file may be used under the terms of 00029 * the GNU General Public License Version 2 (the "GPL"), or the GNU Lesser 00030 * Public License Version 2.1 (the "LGPL"), in which case the provisions of 00031 * the GPL or the LGPL are applicable instead of those above. If you wish to 00032 * allow use of your version of this file only under the terms of the either 00033 * the GPL or LGPL and not to allow others to use your version of this file 00034 * under the MPL, indicate your decision by deleting the provisions above 00035 * and replace them with the notice and other provisions required by the GPL 00036 * or LGPL. If you do not delete the provisions above, a recipient may use 00037 * your version of this file under the terms of any one of the MPL, the GPL 00038 * or the LGPL. 00039 * ***** END LICENSE BLOCK ***** */ 00040 00041 #ifndef _SEQ_COMPRESS_H_ 00042 #define _SEQ_COMPRESS_H_ 00043 00045 //-------------------------------------// 00046 //Class to manage compressing sequences// 00047 //-------------------------------------// 00049 00050 #include <libdirac_byteio/dirac_byte_stream.h> 00051 #include <libdirac_common/common.h> 00052 #include <libdirac_common/picture_buffer.h> 00053 #include <libdirac_common/pic_io.h> 00054 #include <libdirac_common/dirac_assertions.h> 00055 #include <libdirac_encoder/quality_monitor.h> 00056 #include <libdirac_encoder/picture_compress.h> 00057 #include <libdirac_encoder/rate_control.h> 00058 00059 #include <fstream> 00060 00061 namespace dirac 00062 { 00063 00065 00073 class SequenceCompressor{ 00074 public: 00076 00084 SequenceCompressor(StreamPicInput* pin, 00085 EncoderParams& encp, 00086 DiracByteStream& dirac_byte_stream); 00087 00089 00092 virtual ~SequenceCompressor(); 00093 00095 00103 virtual bool LoadNextFrame() = 0; 00104 00106 00125 const Picture *CompressNextPicture(); 00126 00128 const Picture *GetPictureEncoded(); 00129 00131 const MEData *GetMEData(); 00132 00133 DiracByteStats EndSequence(); 00134 00136 00141 bool Finished(){return m_all_done;} 00142 00144 void SignalEOS() { m_eos_signalled = true; } 00145 00146 protected: 00147 00149 00155 virtual int CodedToDisplay(const int pnum) = 0; 00156 00158 void MakeSequenceReport(); 00159 00161 virtual const Picture& OriginalPicture(int picture_num) 00162 { return m_mebuffer->GetPicture(picture_num); } 00163 00165 virtual void CleanBuffers(); 00166 00168 virtual bool IsNewAccessUnit() = 0; 00169 00171 virtual void RateControlCompress(Picture& my_picture, bool is_a_cut) = 0; 00173 bool CanEncode(); 00174 00176 bool m_all_done; 00177 00179 00184 bool m_just_finished; 00185 00187 SourceParams& m_srcparams; 00188 00190 EncoderParams& m_encparams; 00191 00193 OLBParams m_basic_olb_params0; 00194 OLBParams m_basic_olb_params1; 00195 OLBParams m_basic_olb_params2; 00196 00198 StreamPicInput* m_pic_in; 00199 00201 PictureBuffer* m_fbuffer; 00202 00204 PictureBuffer* m_mebuffer; 00205 00206 //state variables for CompressNextPicture 00207 00209 int m_current_display_pnum; 00210 00212 int m_current_code_pnum; 00213 00215 int m_show_pnum; 00216 00218 int m_last_picture_read; 00219 00221 int m_delay; 00222 00224 QualityMonitor m_qmonitor; 00225 00227 RateController* m_ratecontrol; 00228 00230 PictureCompressor m_pcoder; 00231 00233 DiracByteStream& m_dirac_byte_stream; 00234 00236 bool m_eos_signalled; 00237 00239 int m_prev_gop_start; 00240 00241 private: 00243 00247 SequenceCompressor(const SequenceCompressor& cpy); 00248 00250 00254 SequenceCompressor& operator=(const SequenceCompressor& rhs); 00255 00256 00257 }; 00258 00260 00266 class FrameSequenceCompressor : public SequenceCompressor 00267 { 00268 public: 00270 00279 FrameSequenceCompressor(StreamPicInput* pin, 00280 EncoderParams& encp, 00281 DiracByteStream& dirac_byte_stream); 00282 00284 00287 virtual ~FrameSequenceCompressor(){}; 00288 00290 00296 virtual bool LoadNextFrame(); 00297 protected: 00298 virtual int CodedToDisplay(const int pnum); 00299 virtual bool IsNewAccessUnit(); 00300 virtual void RateControlCompress(Picture& my_picture, bool is_a_cut); 00301 00302 }; 00303 00305 00311 class FieldSequenceCompressor : public SequenceCompressor 00312 { 00313 public: 00315 00323 FieldSequenceCompressor(StreamPicInput* pin, 00324 EncoderParams& encp, 00325 DiracByteStream& dirac_byte_stream); 00326 00328 00331 virtual ~FieldSequenceCompressor(); 00332 00334 00341 virtual bool LoadNextFrame(); 00342 00343 protected: 00344 virtual const Picture& OriginalPicture(int picture_num); 00345 00346 virtual void CleanBuffers(); 00347 00348 virtual int CodedToDisplay(const int pnum); 00349 virtual bool IsNewAccessUnit(); 00350 00351 virtual void RateControlCompress(Picture& my_picture, bool is_a_cut); 00352 private: 00354 00358 void PreMotionEstmationFilter (PicArray& comp); 00359 00361 PictureBuffer* m_origbuffer; 00362 00363 // Field1 bytes 00364 int m_field1_bytes; 00365 // Field2 bytes 00366 int m_field2_bytes; 00367 }; 00368 } // namespace dirac 00369 00370 #endif
© 2004 British Broadcasting Corporation.
Dirac code licensed under the Mozilla Public License (MPL) Version 1.1.
HTML documentation generated by Dimitri van Heesch's
excellent Doxygen tool.