WaveletTransform.h

Go to the documentation of this file.

/*
 * The Progressive Graphics File; http://www.libpgf.org
 * 
 * $Date: 2006-05-18 16:03:32 +0200 (Do, 18 Mai 2006) $
 * $Revision: 194 $
 * 
 * This file Copyright (C) 2006 xeraina GmbH, Switzerland
 * 
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU LESSER GENERAL PUBLIC LICENSE
 * as published by the Free Software Foundation; either version 2.1
 * of the License, or (at your option) any later version.
 * 
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 * 
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
 */


#ifndef PGF_WAVELETTRANSFORM_H
#define PGF_WAVELETTRANSFORM_H

#include "PGFtypes.h"
#include "Subband.h"

// Constants
#define FilterWidth         5                   
#define FilterHeight        3                   

#ifdef __PGFROISUPPORT__
class CRoiIndices {
    friend class CWaveletTransform;

    CRoiIndices() 
    : m_nLevels(0)
    , m_indices(0) 
    {}

    ~CRoiIndices() { Destroy(); }

    void Destroy()                              { delete[] m_indices; m_indices = 0; }
    void CreateIndices();
    void ComputeIndices(UINT32 width, UINT32 height, const PGFRect& rect);
    const PGFRect& GetIndices(int level) const  { ASSERT(m_indices); ASSERT(level >= 0 && level < m_nLevels); return m_indices[level]; }
    void SetLevels(int levels)                  { ASSERT(levels > 0); m_nLevels = levels; }
    void ComputeTileIndex(UINT32 width, UINT32 height, UINT32 pos, bool horizontal, bool isMin);

public:
    UINT32 GetNofTiles(int level) const         { ASSERT(level >= 0 && level < m_nLevels); return 1 << (m_nLevels - level - 1); }

private:
    int      m_nLevels;         
    PGFRect *m_indices;         

};
#endif //__PGFROISUPPORT__


class CWaveletTransform {
    friend class CSubband;

public:
    CWaveletTransform(UINT32 width, UINT32 height, int levels, DataT* data = NULL);

    ~CWaveletTransform() { Destroy(); }
    
    OSError ForwardTransform(int level, int quant);

    OSError InverseTransform(int level, UINT32* width, UINT32* height, DataT** data);

    CSubband* GetSubband(int level, Orientation orientation) {
        ASSERT(level >= 0 && level < m_nLevels);
        return &m_subband[level][orientation];
    }
    
#ifdef __PGFROISUPPORT__
    void SetROI(const PGFRect& rect);

    const PGFRect& GetTileIndices(int level) const      { return m_ROIindices.GetIndices(level); }

    UINT32 GetNofTiles(int level) const                 { return m_ROIindices.GetNofTiles(level); }

    const PGFRect& GetROI(int level) const              { return m_subband[level][LL].GetROI(); }

#endif // __PGFROISUPPORT__

private:
    void Destroy() { 
        delete[] m_subband; m_subband = 0; 
    #ifdef __PGFROISUPPORT__
        m_ROIindices.Destroy(); 
    #endif
    }
    void InitSubbands(UINT32 width, UINT32 height, DataT* data);
    void ForwardRow(DataT* buff, UINT32 width);
    void InverseRow(DataT* buff, UINT32 width);
    void LinearToMallat(int destLevel,DataT* loRow, DataT* hiRow, UINT32 width);
    void MallatToLinear(int srcLevel, DataT* loRow, DataT* hiRow, UINT32 width);

#ifdef __PGFROISUPPORT__
    CRoiIndices     m_ROIindices;               
#endif //__PGFROISUPPORT__

    int         m_nLevels;                      
    CSubband    (*m_subband)[NSubbands];        
};

#endif //PGF_WAVELETTRANSFORM_H