cmtkGeneralLinearModel.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: 2398 $
//
//  $LastChangedDate: 2010-10-05 14:54:37 -0700 (Tue, 05 Oct 2010) $
//
//  $LastChangedBy: torstenrohlfing $
//
*/

#ifndef __cmtkGeneralLinearModel_h_included_
#define __cmtkGeneralLinearModel_h_included_

#include <cmtkconfig.h>

#include <Base/cmtkTypedArray.h>
#include <Base/cmtkMatrix.h>

#include <System/cmtkSmartPtr.h>
#include <System/cmtkThreads.h>

#include <vector>

namespace
cmtk
{


class GeneralLinearModel
{
public:
  typedef GeneralLinearModel Self;

  typedef SmartPointer<Self> SmartPtr;

  GeneralLinearModel( const size_t nParameters, const size_t nData, const double* designMatrix );
  
  ~GeneralLinearModel();

  double GetSingularValue( const size_t n ) const 
  {
    return (*(this->W))[n];
  }

  Matrix2D<double>* GetCorrelationMatrix() const;

  void FitModel( std::vector<TypedArray::SmartPtr>& y, const bool normalizeParameters = true );

  TypedArray::SmartPtr& GetModel( const size_t n )
  {
    return this->Model[n];
  }

  double GetNormFactor( const size_t p )
  {
    // do not normalize constant part
    if ( this->VariableSD[p] > 0 ) 
      return this->VariableSD[p];
    else
      return 1.0;
  }

  TypedArray::SmartPtr& GetTStat( const size_t n )
  {
    return this->TStat[n];
  }

  TypedArray::SmartPtr& GetFStat()
  {
    return this->FStat;
  }

private:
  void InitResults( const size_t nPixels );

  size_t NParameters;

  size_t NData;

  Matrix2D<double> DesignMatrix;

  Matrix2D<double>* U;

  std::vector< Matrix2D<double>* > Up;

  Matrix2D<double>* V;

  std::vector< Matrix2D<double>* > Vp;

  std::vector<double>* W;

  std::vector< std::vector<double>* > Wp;

  std::vector<double> VariableMean;

  std::vector<double> VariableSD;

  std::vector<TypedArray::SmartPtr> Model;

  std::vector<TypedArray::SmartPtr> TStat;

  TypedArray::SmartPtr FStat;
};


} // namespace cmtk

#endif // #ifndef __cmtkGeneralLinearModel_h_included_