PyramidGL.h

Go to the documentation of this file.

//  File:       PyramidGL.h
//  Author:     Changchang Wu
//  Description : interface for the PyramdGL
//      class PyramidNaive and PyramidPacked are derived from PyramidGL
//
//  Copyright (c) 2007 University of North Carolina at Chapel Hill
//  All Rights Reserved
//
//  Permission to use, copy, modify and distribute this software and its
//  documentation for educational, research and non-profit purposes, without
//  fee, and without a written agreement is hereby granted, provided that the
//  above copyright notice and the following paragraph appear in all copies.
//  
//  The University of North Carolina at Chapel Hill make no representations
//  about the suitability of this software for any purpose. It is provided
//  'as is' without express or implied warranty. 
//
//  Please send BUG REPORTS to ccwu@cs.unc.edu
//
 
 
 
#ifndef _PYRAMID_GL_H
#define _PYRAMID_GL_H
 
class GLTexImage;
class SiftParam;
class ProgramGPU;
class ShaderMan;
class GlobalUtil;
class SiftPyramid;
 
class PyramidGL:public SiftPyramid
{
protected:
    GLTexImage* _histoPyramidTex;
    GLTexImage* _featureTex;
    GLTexImage* _descriptorTex;
    GLTexImage* _orientationTex;
public:
    void InitializeContext();
    void SetLevelFeatureNum(int idx, int num);
    void GetTextureStorageSize(int num, int &fw, int& fh);
    void GetAlignedStorageSize(int num, int align, int &fw, int &fh);
    static void InterlaceDescriptorF2(int w, int h, float* buf, float* pd, int step);
    static void NormalizeDescriptor(int num, float*pd);
    virtual void DownloadKeypoints();
    virtual int ResizeFeatureStorage();
    virtual void DestroyPerLevelData();
    virtual void DestroySharedData();
    virtual void GetFeatureDescriptors();
    virtual void GenerateFeatureListTex();
    virtual void ReshapeFeatureListCPU();
    virtual void GenerateFeatureDisplayVBO();
    virtual void CleanUpAfterSIFT();
    virtual GLTexImage* GetBaseLevel(int octave, int dataName = DATA_GAUSSIAN)=0;
public:
    PyramidGL(SiftParam&sp);
    virtual ~PyramidGL();
};
 
class PyramidNaive:public PyramidGL, public ShaderMan  
{
protected:
    GLTexImage * _texPyramid;
    GLTexImage * _auxPyramid;
public:
    void DestroyPyramidData();
    void GetSimplifiedOrientation();
    void GenerateFeatureListCPU();
    virtual void GetFeatureOrientations();
    virtual void GenerateFeatureList();
    void DetectKeypointsEX();
    void ComputeGradient();
    GLTexImage* GetLevelTexture(int octave, int level);
    GLTexImage* GetBaseLevel(int octave, int dataName = DATA_GAUSSIAN);
    GLTexImage* GetLevelTexture(int octave, int level, int dataName);
    void BuildPyramid(GLTexInput * input);
    void InitPyramid(int w, int h, int ds);
    void FitPyramid(int w, int h);
    void ResizePyramid(int w, int h);
    void FitHistogramPyramid();
    PyramidNaive(SiftParam & sp);
    ~PyramidNaive();
private:
    void GenerateFeatureList(int i, int j);
};
 
 
class PyramidPacked:public PyramidGL, public ShaderMan
{
    GLTexPacked * _allPyramid;
public:
    PyramidPacked(SiftParam& sp);
    ~PyramidPacked();
    void DestroyPyramidData();
    void DetectKeypointsEX();
    void ComputeGradient();
    void BuildPyramid(GLTexInput * input);
    void InitPyramid(int w, int h, int ds);
    void FitPyramid(int w, int h);
    void ResizePyramid(int w, int h);
    void FitHistogramPyramid();
    void GenerateFeatureListCPU();
    void GenerateFeatureList();
    void GetSimplifiedOrientation();
    void GetFeatureOrientations();
    GLTexImage* GetBaseLevel(int octave, int dataName = DATA_GAUSSIAN);
    GLTexImage* GetLevelTexture(int octave, int level);
    GLTexImage* GetLevelTexture(int octave, int level, int dataName);
    virtual int  IsUsingRectDescription(){return _existing_keypoints & SIFT_RECT_DESCRIPTION; }
private:
    void GenerateFeatureList(int i, int j);
};
 
#endif