grmFullLambdaScheduleSegmenter.txx 2.05 KB
Newer Older
Pierre Lassalle's avatar
Pierre Lassalle committed
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17
/*=========================================================================

  Program: Generic Region Merging Library
  Language: C++
  author: Lassalle Pierre
  contact: lassallepierre34@gmail.com



  Copyright (c) Centre National d'Etudes Spatiales. All rights reserved


     This software is distributed WITHOUT ANY WARRANTY; without even
     the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
     PURPOSE.  See the above copyright notices for more information.

=========================================================================*/
Pierre Lassalle's avatar
Pierre Lassalle committed
18 19 20 21
#ifndef GRM_FULL_LAMBDA_SCHEDULE_SEGMENTER_TXX
#define GRM_FULL_LAMBDA_SCHEDULE_SEGMENTER_TXX

#include "grmFullLambdaScheduleSegmenter.h"
Pierre Lassalle's avatar
Pierre Lassalle committed
22 23 24

namespace grm
{
25 26 27 28 29 30 31 32 33 34 35 36 37
template<class TImage>
void
FullLambdaScheduleSegmenter<TImage>::UpdateSpecificAttributes(NodePointerType n, PixelType p)
{

  n->m_Means.reserve(p.GetSize());

  for(std::size_t b = 0; b < p.GetSize(); ++b)
    {
    n->m_Means.push_back(p[b]);
    }

}
Pierre Lassalle's avatar
Pierre Lassalle committed
38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76

	template<class TImage>
	float
	FullLambdaScheduleSegmenter<TImage>::ComputeMergingCost(NodePointerType n1, NodePointerType n2)
	{
		float eucDist = 0.0;
		const float a1 = static_cast<float>(n1->m_Area);
		const float a2 = static_cast<float>(n2->m_Area);
		const float a_sum = a1 + a2;

		for(unsigned int b = 0; b < this->m_NumberOfComponentsPerPixel; b++)
		{
			eucDist += (n1->m_Means[b] - n2->m_Means[b])*(n1->m_Means[b] - n2->m_Means[b]);
		}

		// Retrieve the length of the boundary between n1 and n2
		auto toN2 = GraphOperatorType::FindEdge(n1, n2);

		float cost = (((a1*a2)/a_sum)*eucDist) / (static_cast<float>(toN2->m_Boundary));

		return cost;
	}

	template<class TImage>
	void
	FullLambdaScheduleSegmenter<TImage>::UpdateSpecificAttributes(NodePointerType n1, NodePointerType n2)
	{
		const float a1 = static_cast<float>(n1->m_Area);
		const float a2 = static_cast<float>(n2->m_Area);
		const float a_sum = a1 + a2;

		for(unsigned int b = 0; b < this->m_NumberOfComponentsPerPixel; ++b)
		{
			n1->m_Means[b] = (a1 * n1->m_Means[b] + a2 * n2->m_Means[b]) / a_sum;
		}
	}
} // end of namespace grm

#endif