me_utils.h

/* ***** BEGIN LICENSE BLOCK *****
*
* $Id: me_utils.h,v 1.2 2005/01/30 05:11:42 gabest Exp $ $Name:  $
*
* Version: MPL 1.1/GPL 2.0/LGPL 2.1
*
* The contents of this file are subject to the Mozilla Public License
* Version 1.1 (the "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
* http://www.mozilla.org/MPL/
*
* Software distributed under the License is distributed on an "AS IS" basis,
* WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License for
* the specific language governing rights and limitations under the License.
*
* The Original Code is BBC Research and Development code.
*
* The Initial Developer of the Original Code is the British Broadcasting
* Corporation.
* Portions created by the Initial Developer are Copyright (C) 2004.
* All Rights Reserved.
*
* Contributor(s): Thomas Davies (Original Author)
*
* Alternatively, the contents of this file may be used under the terms of
* the GNU General Public License Version 2 (the "GPL"), or the GNU Lesser
* Public License Version 2.1 (the "LGPL"), in which case the provisions of
* the GPL or the LGPL are applicable instead of those above. If you wish to
* allow use of your version of this file only under the terms of the either
* the GPL or LGPL and not to allow others to use your version of this file
* under the MPL, indicate your decision by deleting the provisions above
* and replace them with the notice and other provisions required by the GPL
* or LGPL. If you do not delete the provisions above, a recipient may use
* your version of this file under the terms of any one of the MPL, the GPL
* or the LGPL.
* ***** END LICENSE BLOCK ***** */

#ifndef _ME_UTILS_H_
#define _ME_UTILS_H_

#include <algorithm>
#include <libdirac_common/motion.h>
#include <libdirac_common/common.h>
namespace dirac
{

    //Utilities for motion estimation//
    //-------------------------------//

    class BlockDiffParams
    {

    public:
        BlockDiffParams(){}

        BlockDiffParams( const int x_p , const int y_p , const int x_l , const int y_l):
            m_xp(x_p),
            m_yp(y_p),
            m_xl(x_l),
            m_yl(y_l)
        {}

        //NB: Assume default copy constructor, assignment = and destructor//

        // Sets ...


        
        void SetBlockLimits( const OLBParams& bparams ,
                             const PicArray& pic_data , 
                             const int xbpos , const int ybpos);

        // ... and gets

        const int Xp() const {return m_xp;}

        const int Yp() const {return m_yp;}

        const int Xl() const {return m_xl;}

        const int Yl() const {return m_yl;}

    private: 

        int m_xp;
        int m_yp;
        int m_xl;
        int m_yl;

    };

    //----Different difference classes, so that-----//
    //bounds-checking need only be done as necessary//

    class BlockDiff
    {
    public:
        /*
             Constructor, initialising the reference and picture data
             \param  ref  the reference picture
             \param  pic  the picture being matched
        */
        BlockDiff( const PicArray& ref , const PicArray& pic );

        virtual ~BlockDiff(){}


        virtual float Diff(  const BlockDiffParams& dparams , const MVector& mv )=0;

    protected:

        const PicArray& pic_data;
        const PicArray& ref_data;

    private:
        BlockDiff( const BlockDiff& cpy );            

        BlockDiff& operator=( const BlockDiff& rhs );    
    };

    class SimpleBlockDiff: public BlockDiff
    {
    public:
        /*
             Constructor, initialising the reference and picture data
             \param  ref  the reference picture
             \param  pic  the picture being matched
        */
        SimpleBlockDiff( const PicArray& ref , const PicArray& pic );


        float Diff(  const BlockDiffParams& dparams , const MVector& mv );

    private:
        SimpleBlockDiff(const SimpleBlockDiff& cpy);

        SimpleBlockDiff& operator=(const SimpleBlockDiff& rhs);
    };

    class BChkBlockDiff: public BlockDiff
    {
    public:
        /*
             Constructor, initialising the reference and picture data
             \param  ref  the reference picture
             \param  pic  the picture being matched
        */
        BChkBlockDiff( const PicArray& ref , const PicArray& pic );


        float Diff(  const BlockDiffParams& dparams , const MVector& mv );

    private:
        BChkBlockDiff(const BChkBlockDiff& cpy); 

        BChkBlockDiff& operator=(const BChkBlockDiff& rhs);
    };

    class IntraBlockDiff
    {
    public:
        /*
             Constructor, initialising the picture data
             \param  pic  the picture being matched
        */
        IntraBlockDiff( const PicArray& pic );


        float Diff( const BlockDiffParams& dparams , ValueType& dc_val );

    private:
        IntraBlockDiff(const IntraBlockDiff& cpy);

        IntraBlockDiff& operator=(const IntraBlockDiff& rhs);

        const PicArray& pic_data;
    };

    class BiBlockDiff
    {
    public:
        /*
             Constructor, initialising the references and picture data
             \param  ref1  the first reference picture
             \param  ref2  the second reference picture
             \param  pic  the picture being matched
        */
        BiBlockDiff( const PicArray& ref1 , const PicArray& ref2 , const PicArray& pic);


        virtual float Diff(  const BlockDiffParams& dparams , const MVector& mv1 , const MVector& mv2 )=0;

    protected:
        const PicArray& pic_data;
        const PicArray& ref_data1;
        const PicArray& ref_data2;

    private:
        BiBlockDiff(const BiBlockDiff& cpy);             
                                                                    
        BiBlockDiff& operator=(const BiBlockDiff& rhs);
    };


    class BiSimpleBlockDiff: public BiBlockDiff
    {
    public:

        /*
             Constructor, initialising the references and picture data
             \param  ref1  the first reference picture
             \param  ref2  the second reference picture
             \param  pic  the picture being matched
        */
        BiSimpleBlockDiff( const PicArray& ref1 , const PicArray& ref2 , const PicArray& pic);


        float Diff(  const BlockDiffParams& dparams , const MVector& mv1 , const MVector& mv2 );

    private:
        BiSimpleBlockDiff(const BiSimpleBlockDiff& cpy);

        BiSimpleBlockDiff& operator=(const BiSimpleBlockDiff& rhs);    
                                                                    
    };

    class BiBChkBlockDiff: public BiBlockDiff
    {
    public:
        BiBChkBlockDiff( const PicArray& ref1 , const PicArray& ref2 , const PicArray& pic );


        float Diff(  const BlockDiffParams& dparams , const MVector& mv1 , const MVector& mv2 );

    private:
        BiBChkBlockDiff(const BiBChkBlockDiff& cpy);             

        BiBChkBlockDiff& operator=(const BiBChkBlockDiff& rhs);     
                                                                
    };

    // Classes where the reference is upconverted

    class BlockDiffUp: public BlockDiff
    {
    public:    
        /*
             Constructor, initialising the reference and picture data
             \param  ref  the reference picture
             \param  pic  the picture being matched
        */
        BlockDiffUp( const PicArray& ref , const PicArray& pic);

        virtual ~BlockDiffUp(){}


        virtual float Diff(  const BlockDiffParams& dparams , const MVector& mv )=0;

    protected:
        int InterpLookup[9][4];

    private:
        BlockDiffUp(const BlockDiffUp& cpy); 
                                                        
        BlockDiffUp& operator=(const BlockDiffUp& rhs);
                                                        
    };

    class SimpleBlockDiffUp: public BlockDiffUp
    {
    public:
        /*
             Constructor, initialising the reference and picture data
             \param  ref  the reference picture
             \param  pic  the picture being matched
        */
        SimpleBlockDiffUp( const PicArray& ref , const PicArray& pic );

        ~SimpleBlockDiffUp(){}


        float Diff(  const BlockDiffParams& dparams , const MVector& mv );

    private:
        SimpleBlockDiffUp(const SimpleBlockDiffUp& cpy);

        SimpleBlockDiffUp& operator=(const SimpleBlockDiffUp& rhs);
    };

    class BChkBlockDiffUp: public BlockDiffUp{

    public:
        /*
             Constructor, initialising the reference and picture data
             \param  ref  the reference picture
             \param  pic  the picture being matched
        */
        BChkBlockDiffUp(const PicArray& ref,const PicArray& pic);

        ~BChkBlockDiffUp(){}


        float Diff(  const BlockDiffParams& dparams , const MVector& mv );
    private:
        BChkBlockDiffUp(const BChkBlockDiffUp& cpy);

        BChkBlockDiffUp& operator=(const BChkBlockDiffUp& rhs);    
    };

    class BiBlockDiffUp: public BiBlockDiff
    {
    public:    
        /*
             Constructor, initialising the reference and picture data
             \param  ref1  the first reference picture
             \param  ref2  the second reference picture
             \param  pic  the picture being matched
        */
        BiBlockDiffUp( const PicArray& ref1 , const PicArray& ref2 , const PicArray& pic);

        virtual ~BiBlockDiffUp(){}


        virtual float Diff(  const BlockDiffParams& dparams , const MVector& mv1 , const MVector& mv2 )=0;
    protected:
        int InterpLookup[9][4];

    private:
        BiBlockDiffUp(const BlockDiffUp& cpy); 
                                                        
        BiBlockDiffUp& operator=(const BlockDiffUp& rhs);
                                                        
    };

    class BiSimpleBlockDiffUp: public BiBlockDiffUp
    {
    public:
        /*
             Constructor, initialising the reference and picture data
             \param  ref1  the first reference picture
             \param  ref2  the second reference picture
             \param  pic  the picture being matched
        */
        BiSimpleBlockDiffUp( const PicArray& ref1 , const PicArray& ref2 , const PicArray& pic );


        float Diff(  const BlockDiffParams& dparams , const MVector& mv1 , const MVector& mv2 );
    private:
        BiSimpleBlockDiffUp(const BiSimpleBlockDiffUp& cpy);

        BiSimpleBlockDiffUp& operator=(const BiSimpleBlockDiffUp& rhs);
    };

    class BiBChkBlockDiffUp: public BiBlockDiffUp
    {
    public:
        /*
             Constructor, initialising the reference and picture data
             \param  ref1  the first reference picture
             \param  ref2  the second reference picture
             \param  pic  the picture being matched
        */
        BiBChkBlockDiffUp( const PicArray& ref , const PicArray& ref2 , const PicArray& pic );


        float Diff(  const BlockDiffParams& dparams , const MVector& mv1 , const MVector& mv2 );
    private:
        BiBChkBlockDiffUp(const BiBChkBlockDiffUp& cpy);

        BiBChkBlockDiffUp& operator=(const BiBChkBlockDiffUp& rhs);    
    };

} // namespace dirac
#endif

---