/media/Data/chenxing/myProj/asmlib-opencv/src/modelimage.h

Go to the documentation of this file.

#ifndef SHAPE_H
#define SHAPE_H
#include <string>
#include <vector>
#include "cv.h"
#include "highgui.h"
#include "shapevec.h"
#include "shapeinfo.h"
using std::string;
using std::vector;
using cv::Point_;
using cv::Mat_;
using cv::Mat;
using cv::imread;
using cv::Size;

namespace StatModel {

class ModelImage
{
protected:
    int nMarkPoints;

    string hostImageName;

    Mat imgdata;

    vector< Mat > imgPyramid;
    vector< Mat_<double> > imgPyrGrayDeriv;
    vector< Mat_<uchar> > imgPyrGray;

    bool imgLoaded;

    ShapeInfo *shapeInfo;

public:
    vector< Point_< int > > points;

    ShapeVec shapeVec;

    bool loadTrainImage();

    bool loadTrainImage(const Mat &img);

    void setShapeInfo(ShapeInfo *si){ shapeInfo = si; }


    bool releaseTrainImage();


    Mat_< double > getLocalStruct(int pId, int k, int level, double step);


    void getPointsOnNorm(int pId, int k, int level,
                         vector< Point_< int > > &V,
                         double step, int pOffset=0);

    Mat & getTrainImage(int level=0, bool gray=false);

    Mat & getDerivImage(int level){ return imgPyrGrayDeriv[level]; }

    inline const string &HostImage() const { return hostImageName; }

    void buildFromShapeVec(SimilarityTrans &trans);

    void setHostImage( const char * hostImageFilename ) {
        hostImageName = hostImageFilename;
    }

    inline int NPoints() const { return nMarkPoints; }

    bool readPTS( const char * filename );

    void initPointsByVector( const vector< cv::Point2i > &V);

    Mat show(int level = 0, int pId = -1, bool showInWin = true, int highLight = 0);

    ModelImage();
};

} // Namespace
#endif // SHAPE_H