Calculates internal and external energy of the new particle configuration proposal. More...

#include <mitkEnergyComputer.h>

Inheritance diagram for EnergyComputer:

Collaboration diagram for EnergyComputer:

Public Types
typedef itk::Image< float, 3 >	ItkFloatImageType

typedef itk::Statistics::MersenneTwisterRandomVariateGenerator	ItkRandGenType

Public Member Functions
	EnergyComputer (ItkFloatImageType mask, ParticleGrid particleGrid, SphereInterpolator interpolator, ItkRandGenType randGen)

void	SetParameters (float particleWeight, float particleWidth, float connectionPotential, float curvThres, float inexBalance, float particlePotential)

void	DrawRandomPosition (vnl_vector_fixed< float, 3 > &R)

virtual float	ComputeExternalEnergy (vnl_vector_fixed< float, 3 > &R, vnl_vector_fixed< float, 3 > &N, Particle *dp)=0

virtual float	ComputeInternalEnergyConnection (Particle *p1, int ep1)=0

virtual float	ComputeInternalEnergyConnection (Particle p1, int ep1, Particle p2, int ep2)=0

virtual float	ComputeInternalEnergy (Particle *dp)=0

int	GetNumActiveVoxels ()

Protected Member Functions
float	SpatProb (vnl_vector_fixed< float, 3 > pos)

float	EvaluateOdf (vnl_vector_fixed< float, 3 > &pos, vnl_vector_fixed< float, 3 > dir)

float	mbesseli0 (float x)

float	mexp (float x)

Protected Attributes
vnl_matrix_fixed< float, 3, 3 >	m_RotationMatrix

SphereInterpolator *	m_SphereInterpolator

ParticleGrid *	m_ParticleGrid

ItkRandGenType *	m_RandGen

ItkFloatImageType *	m_Mask

vnl_vector_fixed< int, 3 >	m_Size

vnl_vector_fixed< float, 3 >	m_Spacing

std::vector< float >	m_CumulatedSpatialProbability

std::vector< int >	m_ActiveIndices

bool	m_UseTrilinearInterpolation

int	m_NumActiveVoxels

float	m_ConnectionPotential

float	m_ParticleChemicalPotential

float	gamma_s

float	gamma_reg_s

float	m_ParticleWeight

float	m_ExtStrength

float	m_IntStrength

float	m_ParticleLength

float	m_SquaredParticleLength

float	m_CurvatureThreshold

Detailed Description

Calculates internal and external energy of the new particle configuration proposal.

Definition at line 31 of file mitkEnergyComputer.h.

Member Typedef Documentation

typedef itk::Image<float, 3> EnergyComputer::ItkFloatImageType

Definition at line 38 of file mitkEnergyComputer.h.

typedef itk::Statistics::MersenneTwisterRandomVariateGenerator EnergyComputer::ItkRandGenType

Definition at line 39 of file mitkEnergyComputer.h.

Constructor & Destructor Documentation

EnergyComputer::EnergyComputer	(	ItkFloatImageType *	mask,
		ParticleGrid *	particleGrid,
		SphereInterpolator *	interpolator,
		ItkRandGenType *	randGen
	)

Definition at line 23 of file mitkEnergyComputer.cpp.

References m_ActiveIndices, m_CumulatedSpatialProbability, m_Mask, m_NumActiveVoxels, m_ParticleGrid, mitk::ParticleGrid::m_ParticleLength, m_ParticleLength, m_RandGen, m_RotationMatrix, m_Size, m_Spacing, m_SphereInterpolator, m_SquaredParticleLength, m_UseTrilinearInterpolation, SphereInterpolator::nverts, and QBALL_ODFSIZE.

Member Function Documentation

virtual float EnergyComputer::ComputeExternalEnergy	(	vnl_vector_fixed< float, 3 > &	R,
		vnl_vector_fixed< float, 3 > &	N,
		Particle *	dp
	)

pure virtual

Implemented in GibbsEnergyComputer.

Referenced by mitk::MetropolisHastingsSampler::MakeProposal().

virtual float EnergyComputer::ComputeInternalEnergy ( Particle * dp )

pure virtual

Implemented in GibbsEnergyComputer.

Referenced by mitk::MetropolisHastingsSampler::MakeProposal().

virtual float EnergyComputer::ComputeInternalEnergyConnection	(	Particle *	p1,
		int	ep1
	)

pure virtual

Implemented in GibbsEnergyComputer.

Referenced by mitk::MetropolisHastingsSampler::ComputeEndPointProposalDistribution(), mitk::MetropolisHastingsSampler::MakeTrackProposal(), and mitk::MetropolisHastingsSampler::RemoveAndSaveTrack().

virtual float EnergyComputer::ComputeInternalEnergyConnection	(	Particle *	p1,
		int	ep1,
		Particle *	p2,
		int	ep2
	)

pure virtual

Implemented in GibbsEnergyComputer.

void EnergyComputer::DrawRandomPosition ( vnl_vector_fixed< float, 3 > & R )

Definition at line 109 of file mitkEnergyComputer.cpp.

References m_ActiveIndices, m_CumulatedSpatialProbability, m_NumActiveVoxels, m_RandGen, m_Size, and m_Spacing.

Referenced by mitk::MetropolisHastingsSampler::MakeProposal().

float EnergyComputer::EvaluateOdf	(	vnl_vector_fixed< float, 3 > &	pos,
		vnl_vector_fixed< float, 3 >	dir
	)

protected

int EnergyComputer::GetNumActiveVoxels ( )

Definition at line 169 of file mitkEnergyComputer.cpp.

References m_NumActiveVoxels.

float EnergyComputer::mbesseli0 ( float x )

protected

Definition at line 149 of file mitkEnergyComputer.cpp.

Referenced by GibbsEnergyComputer::ComputeExternalEnergy().

float EnergyComputer::mexp ( float x )

protected

Definition at line 163 of file mitkEnergyComputer.cpp.

Referenced by GibbsEnergyComputer::ComputeExternalEnergy().

void EnergyComputer::SetParameters	(	float	particleWeight,
		float	particleWidth,
		float	connectionPotential,
		float	curvThres,
		float	inexBalance,
		float	particlePotential
	)