Dirac - A Video Codec

Created by the British Broadcasting Corporation.


seq_compress.h

Go to the documentation of this file.
00001 /* ***** BEGIN LICENSE BLOCK *****
00002 *
00003 * $Id: seq_compress.h,v 1.33 2008/10/20 04:20:12 asuraparaju Exp $ $Name: Dirac_1_0_2 $
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_encoder/enc_queue.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 EncPicture *CompressNextPicture();
00126 
00128         virtual void SetPicTypeAndRefs( PictureParams& pparams ) = 0;
00129 
00131         const EncPicture *GetPictureEncoded();
00132 
00133         DiracByteStats EndSequence();
00134 
00136 
00141         bool Finished(){return m_all_done;}
00142 
00144         void SignalEOS() { m_eos_signalled = true; }
00145 
00147     int PTSOffset(){return m_delay;}
00148 
00149     protected:
00150 
00152         void SetMotionParameters();
00153 
00155 
00161         virtual int CodedToDisplay(const int pnum) = 0;
00162 
00164         void MakeSequenceReport();
00165 
00167         virtual void CleanBuffers();
00168 
00170     //Purely virtual. The child class will have to define it.
00171         virtual void UpdateCBRModel(EncPicture& my_picture, const PictureByteIO* picture_byteio) = 0;
00172 
00174         void UpdateIntraPicCBRModel( const PictureParams& , const bool is_a_cut );
00175 
00177         bool CanEncode();
00178 
00180         bool m_all_done;
00181 
00183 
00188         bool m_just_finished;
00189 
00191         OLBParams* m_basic_olb_params0;
00192 
00194         OLBParams* m_basic_olb_params1;
00195 
00197         const OLBParams* m_basic_olb_params2;
00198 
00200         OLBParams* m_intra_olbp;
00201 
00203         SourceParams& m_srcparams;
00204 
00206         EncoderParams& m_encparams;
00207 
00209         PicturePredParams& m_predparams;
00210 
00212     int m_L1_sep;
00213 
00215         PictureParams m_pparams;
00216 
00218         StreamPicInput* m_pic_in;
00219 
00221         EncQueue m_enc_pbuffer;
00222 
00223         //state variables for CompressNextPicture
00224 
00226         int m_current_display_pnum;
00227 
00229         int m_current_code_pnum;
00230 
00232         int m_show_pnum;
00233 
00235         int m_last_picture_read;
00236 
00238     int m_gop_start_num;
00239 
00241         int m_delay;
00242 
00244         QualityMonitor m_qmonitor;
00245 
00247         RateController* m_ratecontrol;
00248 
00250         PictureCompressor m_pcoder;
00251 
00253         DiracByteStream& m_dirac_byte_stream;
00254 
00256         bool m_eos_signalled;
00257 
00258     private:
00260 
00264         SequenceCompressor(const SequenceCompressor& cpy);
00265 
00267 
00271         SequenceCompressor& operator=(const SequenceCompressor& rhs);
00272 
00273 
00274     };
00275 
00277 
00283     class FrameSequenceCompressor : public SequenceCompressor
00284     {
00285     public:
00287 
00296         FrameSequenceCompressor(StreamPicInput* pin,
00297                            EncoderParams& encp,
00298                            DiracByteStream& dirac_byte_stream);
00299 
00301 
00304         virtual ~FrameSequenceCompressor(){};
00305 
00307 
00313         virtual bool LoadNextFrame();
00314         
00316         virtual void SetPicTypeAndRefs( PictureParams& pparams );
00317 
00318 protected:
00319         virtual int CodedToDisplay(const int pnum);
00320         virtual void UpdateCBRModel(EncPicture& my_picture, const PictureByteIO* picture_byteio);
00321 
00322     };
00323 
00325 
00331     class FieldSequenceCompressor : public SequenceCompressor
00332     {
00333     public:
00335 
00343         FieldSequenceCompressor(StreamPicInput* pin,
00344                            EncoderParams& encp,
00345                            DiracByteStream& dirac_byte_stream);
00346 
00348 
00351         virtual ~FieldSequenceCompressor();
00352 
00354 
00361         virtual bool LoadNextFrame();
00362 
00363         
00365         virtual void SetPicTypeAndRefs( PictureParams& pparams );
00366 
00367     protected:
00368 
00369         virtual int CodedToDisplay(const int pnum);
00370 
00371         virtual void UpdateCBRModel(EncPicture& my_picture, const PictureByteIO* picture_byteio);
00372     private:
00374 
00378         void PreMotionEstmationFilter (PicArray& comp);
00379 
00380         // Field1 bytes
00381         int m_field1_bytes;
00382         // Field2 bytes
00383         int m_field2_bytes;
00384     };
00385 } // namespace dirac
00386 
00387 #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.