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