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

Go to the documentation of this file.

#ifndef SIMILARITYTRANS_H
#define SIMILARITYTRANS_H

#include "cv.h"
using cv::Mat;

namespace StatModel {

class ShapeVec;

class SimilarityTrans{
public:

    void invTransform(const ShapeVec &src, ShapeVec &dst) const;


    void transform(const ShapeVec &src, ShapeVec &dst) const;

    void setTransformByAlign(const ShapeVec &x, const ShapeVec &xp);

    void warpImage(const Mat &imgSrc, Mat &imgDst) const;

    void warpImgBack(const Mat &imgSrc, Mat &imgDst, bool useDstSize=false) const;

    double getS() const { return sqrt(a*a+b*b); }

    SimilarityTrans():Xt(0), Yt(0), a(1), b(0){}

    SimilarityTrans operator *(const SimilarityTrans & s2) const {
        SimilarityTrans ns;
        ns.a = a*s2.a-b*s2.b;
        ns.b = s2.a*b+s2.b*a;
        ns.Xt = a*s2.Xt - b*s2.Yt + Xt;
        ns.Yt = b*s2.Xt + a*s2.Yt + Yt;
        return ns;
    }

    double Xt;
    double Yt;

    double a;
    double b;
};

} // Namespace
#endif // SIMILARITYTRANS_H