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

Go to the documentation of this file.

#ifndef SHAPEVEC_H
#define SHAPEVEC_H

#include "similaritytrans.h"

#include "cv.h"
#include "highgui.h"
#include <vector>
using std::vector;
using cv::Point_;
using cv::Mat_;
using cv::Mat;
using cv::imread;
using cv::mean;
typedef unsigned int uint;

namespace StatModel {

class ShapeVec : public Mat_< double > {
  public:
    ShapeVec(const Mat_< double > &a):Mat_< double >(a){}

    ShapeVec & operator =(const Mat_< double > &a) {
        Mat_< double >::operator=(a);
        return *this;
    }

    ShapeVec(){}

    void alignTo(const ShapeVec & ref);

    void zeroGravity();

    void scaleToOne();

    void doTranslate(double vX, double vY);
    void doScale(double r);

    double getXMean() const {return mean(rowRange(0, rows / 2))[0];}
    double getYMean() const {return mean(rowRange(rows / 2, rows))[0];}

    double X(int i) const { return (*this)(i, 0); }
    double & X(int i) { return (*this)(i, 0); }
    double Y(int i) const { return (*this)(i + (rows >> 1), 0); }
    double & Y(int i) { return (*this)(i+(rows>>1), 0); }

    cv::Rect_<double> getBoundRect();

    void restoreToPointList(
      vector< Point_< int > > &v,
      const SimilarityTrans &st
    );

    void fromPointList(const vector< cv::Point2i > &v);

    int nPoints() const { return (rows >> 1); }

    SimilarityTrans getShapeTransformFitingSize(
        const cv::Size &rect,
        double scaleRatio=0.9, double xOffset=0, double yOffset=0);
};
} // Namespace
#endif // SHAPEVEC_H