mv_codec.h
上传用户:xjjlds
上传日期:2015-12-05
资源大小:22823k
文件大小:7k
源码类别:

多媒体编程

开发平台:

Visual C++

  1. /* ***** BEGIN LICENSE BLOCK *****
  2. *
  3. * $Id: mv_codec.h,v 1.2 2005/01/30 05:11:40 gabest Exp $ $Name:  $
  4. *
  5. * Version: MPL 1.1/GPL 2.0/LGPL 2.1
  6. *
  7. * The contents of this file are subject to the Mozilla Public License
  8. * Version 1.1 (the "License"); you may not use this file except in compliance
  9. * with the License. You may obtain a copy of the License at
  10. * http://www.mozilla.org/MPL/
  11. *
  12. * Software distributed under the License is distributed on an "AS IS" basis,
  13. * WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License for
  14. * the specific language governing rights and limitations under the License.
  15. *
  16. * The Original Code is BBC Research and Development code.
  17. *
  18. * The Initial Developer of the Original Code is the British Broadcasting
  19. * Corporation.
  20. * Portions created by the Initial Developer are Copyright (C) 2004.
  21. * All Rights Reserved.
  22. *
  23. * Contributor(s): Thomas Davies (Original Author), Scott R Ladd
  24. *
  25. * Alternatively, the contents of this file may be used under the terms of
  26. * the GNU General Public License Version 2 (the "GPL"), or the GNU Lesser
  27. * Public License Version 2.1 (the "LGPL"), in which case the provisions of
  28. * the GPL or the LGPL are applicable instead of those above. If you wish to
  29. * allow use of your version of this file only under the terms of the either
  30. * the GPL or LGPL and not to allow others to use your version of this file
  31. * under the MPL, indicate your decision by deleting the provisions above
  32. * and replace them with the notice and other provisions required by the GPL
  33. * or LGPL. If you do not delete the provisions above, a recipient may use
  34. * your version of this file under the terms of any one of the MPL, the GPL
  35. * or the LGPL.
  36. * ***** END LICENSE BLOCK ***** */
  37. #ifndef _MV_CODEC_H_
  38. #define _MV_CODEC_H_
  39. /////////////////////////////////////////////////
  40. //Class to do motion vector coding and decoding//
  41. //------using adaptive arithmetic coding-------//
  42. /////////////////////////////////////////////////
  43. #include <libdirac_common/arith_codec.h>
  44. #include <libdirac_common/common.h>
  45. #include <libdirac_common/motion.h>
  46. #include <libdirac_common/wavelet_utils.h>
  47. #include <vector>
  48. namespace dirac
  49. {
  50.     //! Codes and decodes all the Motion Vector data
  51.     /*!
  52.         Derived from the ArithCodec class, this codes and decodes all the 
  53.         motion vector data.
  54.      */
  55.     class MvDataCodec: public ArithCodec<MvData>
  56.     {
  57.     public:
  58.         //! Constructor for encoding
  59.             /*!
  60.             Creates a MvDataCodec object to encode MV data, based on parameters
  61.             param    bits_out    the output for the encoded bits
  62.             param    number_of_contexts   the contexts used in the encoding process
  63.             param     cf            the chroma format
  64.          */    
  65.         MvDataCodec(BasicOutputManager* bits_out,
  66.                     size_t number_of_contexts,
  67.                     const ChromaFormat & cf);
  68.         //! Constructor for decoding
  69.             /*!
  70.             Creates a MvDataCodec object to encode MV data, based on parameters
  71.             param    bits_in        the input for the encoded bits
  72.             param    number_of_contexts  the contexts used in the encoding process
  73.             param     cf            the chroma format
  74.          */        
  75.         MvDataCodec(BitInputManager* bits_in,
  76.                     size_t number_of_contexts,
  77.                     const ChromaFormat & cf); 
  78.         //! Initialises the contexts    
  79.         void InitContexts();
  80.         
  81.     private:
  82.         int MB_count;
  83.         const ChromaFormat & m_cformat;
  84.         int b_xp, b_yp;            //position of current block
  85.         int mb_xp, mb_yp;        //position of current MB
  86.         int mb_tlb_x, mb_tlb_y;    //position of top-left block of current MB
  87.         // functions    
  88.         MvDataCodec(const MvDataCodec& cpy);            //private, bodyless copy constructor: class should not be copied
  89.         MvDataCodec& operator=(const MvDataCodec& rhs); //private, bodyless copy operator=: class should not be assigned
  90.         // coding functions    
  91.         void CodeMBSplit(const MvData& in_data);    //code the MB splitting mode
  92.         void CodeMBCom(const MvData& in_data);    //code the MB common ref mode
  93.         void CodePredmode(const MvData& in_data);    //code the block prediction mode
  94.         void CodeMv1(const MvData& in_data);        //code the first motion vector
  95.         void CodeMv2(const MvData& in_data);        //code the second motion vector
  96.         void CodeDC(const MvData& in_data);        //code the dc value of intra blocks
  97.         // decoding functions
  98.         void DecodeMBSplit( MvData& out_data);    //decode the MB splitting mode
  99.         void DecodeMBCom( MvData& out_data);//decode the MB common ref mode
  100.         void DecodePredmode(MvData& out_data);//decode the block prediction mode
  101.         void DecodeMv1( MvData& out_data);    //decode the first motion vector
  102.         void DecodeMv2( MvData& out_data);    //decode the second motion vector
  103.         void DecodeDC( MvData& out_data);    //decode the dc value of intra blocks    
  104.         void DoWorkCode( MvData& in_data );
  105.         void DoWorkDecode(MvData& out_data, int num_bits);
  106.         // Context stuff    
  107.         void Update( const bool symbol , const int context_num );
  108.         void Resize(const int context_num);
  109.         void ResetAll();
  110.         int ChooseContext(const MvData& data, const int BinNumber) const;
  111.         int ChooseContext(const MvData& data) const;
  112.         int ChooseSignContext(const MvData& data) const;
  113.         int ChooseMBSContext(const MvData& data, const int BinNumber) const;
  114.         int ChooseMBCContext(const MvData& data) const;
  115.         int ChoosePredContext(const MvData& data, const int BinNumber) const;
  116.         int ChooseREF1xContext(const MvData& data, const int BinNumber) const;
  117.         int ChooseREF1xSignContext(const MvData& data) const;
  118.         int ChooseREF1yContext(const MvData& data, const int BinNumber) const;
  119.         int ChooseREF1ySignContext(const MvData& data) const;
  120.         int ChooseREF2xContext(const MvData& data, const int BinNumber) const;
  121.         int ChooseREF2xSignContext(const MvData& data) const;
  122.         int ChooseREF2yContext(const MvData& data, const int BinNumber) const;
  123.         int ChooseREF2ySignContext(const MvData& data) const;
  124.         int ChooseYDCContext(const MvData& data, const int BinNumber) const;
  125.         int ChooseUDCContext(const MvData& data, const int BinNumber) const;
  126.         int ChooseVDCContext(const MvData& data, const int BinNumber) const;
  127.         int ChooseYDCSignContext(const MvData& data) const;
  128.         int ChooseUDCSignContext(const MvData& data) const;
  129.         int ChooseVDCSignContext(const MvData& data) const;
  130.         //prediction stuff
  131.         unsigned int MBSplitPrediction(const TwoDArray<int>& mbdata) const;
  132.         bool MBCBModePrediction(const TwoDArray<bool>& mbdata) const;
  133.         unsigned int BlockModePrediction(const TwoDArray<PredMode>& preddata) const;
  134.         MVector Mv1Prediction(const MvArray& mvarray,const TwoDArray<PredMode>& preddata) const;
  135.         MVector Mv2Prediction(const MvArray& mvarray,const TwoDArray<PredMode>& preddata) const;
  136.         ValueType DCPrediction(const TwoDArray<ValueType>& dcdata,const TwoDArray<PredMode>& preddata) const;
  137.     };
  138. } // namespace dirac
  139. #endif