template/fullcost.hh

Go to the documentation of this file.

/* -*- mia-c++  -*-
 *
 * This file is part of MIA - a toolbox for medical image analysis 
 * Copyright (c) Leipzig, Madrid 1999-2017 Gert Wollny
 *
 * MIA 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 3 of the License, or
 * (at your option) any later version.
 *
 * This program 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 General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with MIA; if not, see <http://www.gnu.org/licenses/>.
 *
 */

#ifndef mia_internal_fullcost_hh
#define mia_internal_fullcost_hh

#include <mia/core/product_base.hh>
#include <mia/core/vector.hh>
#include <mia/core/import_handler.hh>

NS_MIA_BEGIN

template <typename Transform>
class EXPORT_HANDLER TFullCost : public CProductBase {
public: 

        typedef typename Transform::Size Size; 

        typedef TFullCost<Transform> plugin_data; 
        
        typedef TFullCost<Transform> plugin_type;

        static const char *type_descr;
        
        static const char *data_descr;

        typedef std::shared_ptr<TFullCost<Transform> > Pointer; 
        
        TFullCost(double weight);


        double evaluate(const Transform& t, CDoubleVector& gradient) const;

        double cost_value(const Transform& t) const;

        double cost_value() const;

        void reinit();
        
        void set_size(const Size& size); 
        
        bool get_full_size(Size& size) const; 
        
protected: 
        double get_weight() const; 
        
        const Size& get_current_size() const; 
private:
        virtual double do_evaluate(const Transform& t, CDoubleVector& gradient) const = 0;
        virtual double do_value(const Transform& t) const = 0;
        virtual double do_value() const = 0;
        virtual void do_reinit();
        virtual void do_set_size() = 0; 
        virtual bool do_get_full_size(Size& size) const; 
        
        double m_weight;
        Size m_current_size; 

}; 

template <typename Transform>
class EXPORT_HANDLER TFullCostPlugin: public TFactory<TFullCost<Transform> > {
public:
        TFullCostPlugin(const char *name);
private:
        virtual TFullCost<Transform> *do_create() const;
        virtual TFullCost<Transform> *do_create(float weight) const = 0;
        float m_weight;
}; 


NS_MIA_END

#endif