cmtkActiveShapeModel.h

Go to the documentation of this file.

/*
//
//  Copyright 1997-2009 Torsten Rohlfing
//
//  Copyright 2004-2010 SRI International
//
//  This file is part of the Computational Morphometry Toolkit.
//
//  http://www.nitrc.org/projects/cmtk/
//
//  The Computational Morphometry Toolkit 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.
//
//  The Computational Morphometry Toolkit 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 the Computational Morphometry Toolkit.  If not, see
//  <http://www.gnu.org/licenses/>.
//
//  $Revision: 2752 $
//
//  $LastChangedDate: 2011-01-17 11:33:31 -0800 (Mon, 17 Jan 2011) $
//
//  $LastChangedBy: torstenrohlfing $
//
*/

#ifndef __cmtkActiveShapeModel_h_included_
#define __cmtkActiveShapeModel_h_included_

#include <cmtkconfig.h>

#include <Base/cmtkTypes.h>
#include <Base/cmtkDirectionSet.h>

#include <System/cmtkSmartPtr.h>

namespace
cmtk
{


class ActiveShapeModel
{
public:
  typedef SmartPointer<ActiveShapeModel> SmartPtr;

  unsigned int NumberOfPoints;

  unsigned int GetNumberOfPoints() const { return NumberOfPoints; }

  CoordinateVector::SmartPtr Mean;

  unsigned int NumberOfModes;

  unsigned int GetNumberOfModes() const { return NumberOfModes; }

  DirectionSet::SmartPtr Modes;

  CoordinateVector::SmartPtr ModeVariances;

  ActiveShapeModel() : Mean( NULL ), Modes( NULL ) {}

  ActiveShapeModel( CoordinateVector::SmartPtr& mean, DirectionSet::SmartPtr& modes, CoordinateVector::SmartPtr& modeVariances );

  ActiveShapeModel
  ( const Types::Coordinate *const* trainingSet, 
    const unsigned int numberOfSamples,
    const unsigned int numberOfPoints, 
    const unsigned int numberOfModes ) :
    Mean( NULL ), 
    Modes( NULL )
  {
    this->Construct( trainingSet, numberOfSamples, numberOfPoints, numberOfModes ); 
  }

  float Construct( const Types::Coordinate *const* trainingSet, const unsigned int numberOfSamples, const unsigned int numberOfPoints, const unsigned int numberOfModes );

  Types::Coordinate* Generate( Types::Coordinate *const instance, const Types::Coordinate* modeWeights ) const;

  float Decompose( const CoordinateVector* input, Types::Coordinate *const weights = NULL ) const;

 protected:
  void Allocate( const unsigned int numberOfPoints, const unsigned int numberOfModes );
};


} // namespace cmtk

#endif // #ifndef __cmtkActiveShapeModel_h_included_