StraightenPanorama.h

Go to the documentation of this file.

// -*- c-basic-offset: 4 -*-
#ifndef _BASICALGORITHMS_STRAIGHTENPANORAMA_H
#define _BASICALGORITHMS_STRAIGHTENPANORAMA_H

#include <hugin_shared.h>
#include <algorithms/basic/RotatePanorama.h>


namespace HuginBase {


class IMPEX StraightenPanorama : public RotatePanorama
{

    public:
        explicit StraightenPanorama(PanoramaData& panorama)
         : RotatePanorama(panorama, Matrix3())
        {};
        
        virtual ~StraightenPanorama() {};
          
        
    public:
        static Matrix3 calcStraighteningRotation(const PanoramaData& panorama);       
        
    public:
        virtual bool runAlgorithm()
        {
            for(unsigned int i=0;i<o_panorama.getNrOfImages();i++)
            {
                const SrcPanoImage & img=o_panorama.getImage(i);
                //if translation parameters are non-zero, straighten does not work in current form
                //TODO: fix straighten with non-zero translation parameters
                if(img.getX()!=0 || img.getY()!=0 || img.getZ()!=0)
                    return true;
            };
            o_transformMat = calcStraighteningRotation(o_panorama);
            RotatePanorama::runAlgorithm();
            
            return true; // let's hope so.
        }
};


}
#endif //_H