// -*- c-basic-offset: 4 -*-
/** @file CalculateOptimalScale.h
 *
 *  @author Pablo d'Angelo <pablo.dangelo@web.de>
 *
 *  $Id$
 *
 * !! from Panorama.h 1947 
 *
 *  This is free software; you can redistribute it and/or
 *  modify it under the terms of the GNU General Public
 *  License as published by the Free Software Foundation; either
 *  version 2 of the License, or (at your option) any later version.
 *
 *  This software is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 *  Lesser General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public
 *  License along with this software. If not, see
 *  <http://www.gnu.org/licenses/>.
 *
 */


#ifndef _BASICALGORITHMS_CALCULATEOPTIMALSCALE_H
#define _BASICALGORITHMS_CALCULATEOPTIMALSCALE_H

#include <hugin_shared.h>
#include <algorithms/PanoramaAlgorithm.h>

#include <panodata/PanoramaData.h>



namespace HuginBase {


class IMPEX CalculateOptimalScale : public PanoramaAlgorithm
{

    public:
        ///
        explicit CalculateOptimalScale(PanoramaData& panorama)
            : PanoramaAlgorithm(panorama), o_optimalScale(0)
        {}
        
        ///
        virtual ~CalculateOptimalScale()
        {}
        
        
    public:
        ///
        virtual bool modifiesPanoramaData() const
            { return false; }
            
        ///
        virtual bool runAlgorithm()
        {
            o_optimalScale =  calcOptimalScale(o_panorama);
            return true; // let's hope so.
        }
        
    public:
        ///
        static double calcOptimalScale(PanoramaData& panorama);
        
        /** function to calculate the scaling factor so that the distances
         * in the input image and panorama image are similar at the panorama center
         */
        static double calcOptimalPanoScale(const SrcPanoImage & src,
                                           const PanoramaOptions & dest);
        
        ///
        virtual double getResultOptimalScale()
        {
            // [TODO] if(!hasRunSuccessfully()) DEBUG;
            return o_optimalScale;
        }
        
        ///
        virtual unsigned getResultOptimalWidth()
        {
            // [TODO] if(!hasRunSuccessfully()) DEBUG;
            return hugin_utils::roundi(getResultOptimalScale() * o_panorama.getOptions().getWidth());
        }
        
    protected:
        double o_optimalScale;
        
};


class SetWidthOptimal : public CalculateOptimalScale
{
    public:
        ///
        explicit SetWidthOptimal(PanoramaData& panorama)
         : CalculateOptimalScale(panorama)
        {}

        ///
        virtual ~SetWidthOptimal()
        {}
    
    
    public:
        ///
        virtual bool modifiesPanoramaData() const
            { return true; }
        
        ///
        virtual bool runAlgorithm()
        {
            bool success = CalculateOptimalScale::runAlgorithm();
            if(success)
            {
                PanoramaOptions opts = o_panorama.getOptions();
                opts.setWidth(getResultOptimalWidth());
                o_panorama.setOptions(opts);
            }
            return success;
        }
};


} //namespace
#endif
