itkNeuralNetworkFileReader.h

Go to the documentation of this file.

/*=========================================================================

Program:   Insight Segmentation & Registration Toolkit
Module:    $RCSfile: itkNeuralNetworkFileReader.h,v $
Language:  C++
Date:      $Date: 2008-02-07 15:58:06 $
Version:   $Revision: 1.16 $

Copyright (c) Insight Software Consortium. All rights reserved.
See ITKCopyright.txt or http://www.itk.org/HTML/Copyright.htm for details.

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.

=========================================================================*/

#ifndef __itkNeuralNetworkFileReader_h
#define __itkNeuralNetworkFileReader_h

#include "metaTypes.h"
#include "metaUtils.h"

#include "itkMultilayerNeuralNetworkBase.h"

#include "itkBackPropagationLayer.h"

#include "itkCompletelyConnectedWeightSet.h"

#include "itkSumInputFunction.h"
#include "itkProductInputFunction.h"

#include "itkIdentityTransferFunction.h"
#include "itkLogSigmoidTransferFunction.h"
#include "itkSigmoidTransferFunction.h"
#include "itkTanSigmoidTransferFunction.h"
#include "itkSymmetricSigmoidTransferFunction.h"


namespace itk
{

template<class TNetwork>
class NeuralNetworkFileReader : public Object
{
public:

  typedef NeuralNetworkFileReader                       Self;
  typedef Object                                        Superclass;
  typedef SmartPointer<Self>                            Pointer;
  typedef SmartPointer<const Self>                      ConstPointer;

  itkTypeMacro(NeuralNetworkFileReader,Object);

  itkNewMacro(Self);

  typedef typename TNetwork::MeasurementVectorType     MeasurementVectorType;
  typedef typename TNetwork::TargetVectorType          TargetVectorType;
  typedef Statistics::BackPropagationLayer<
    MeasurementVectorType, TargetVectorType>           BackPropagationLayerType;
  typedef typename BackPropagationLayerType::Pointer   BackPropagationLayerPointer;
  typedef typename MeasurementVectorType::ValueType    MeasurementVectorValueType;

  typedef typename TNetwork::LayerInterfaceType        LayerInterfaceType;
  typedef typename LayerInterfaceType::WeightSetType   WeightSetType;
  typedef typename WeightSetType::Pointer              WeightSetPointer;

// typedef typename TNetwork::Pointer                       NetworkPointer;
// typedef typename TNetwork::ConstPointer                  NetworkConstPointer;


//  typedef typename LayerInterfaceType::TransferFunctionType::Pointer      TransferFunctionPointer;
//  typedef typename LayerInterfaceType::TransferFunctionType::ConstPointer TransferFunctionConstPointer;

//  typedef typename LayerInterfaceType::InputFunctionType::Pointer         InputFunctionPointer;
//  typedef typename LayerInterfaceType::InputFunctionType::ConstPointer    InputFunctionConstPointer;

//  typedef typename LayerInterfaceType::ValueType                    ValueType;

  itkSetStringMacro( FileName );

  itkGetStringMacro( FileName );

  void Update(void);

  TNetwork * GetOutput() const;

#ifdef IGNORE
#undef IGNORE
#endif
  typedef enum { IGNORE=0, ASCII=1, BINARY=2 } NetworkWriteWeightsType;
  itkSetEnumMacro(ReadWeightValuesType, NetworkWriteWeightsType);
  itkGetEnumMacro(ReadWeightValuesType, NetworkWriteWeightsType);


protected:
  NeuralNetworkFileReader();
  ~NeuralNetworkFileReader();
  virtual void PrintSelf( std::ostream& os, Indent indent ) const;

private:

  void ClearFields();
  typedef std::vector<MET_FieldRecordType *>      FieldsContainerType;
  typedef std::vector<typename TNetwork::LayerInterfaceType::Pointer >             LayersContainer;
  typedef std::vector<typename TNetwork::LayerInterfaceType::WeightSetInterfaceType::Pointer>  WeightsContainer;

  typedef struct
    {
    std::string name;
    std::string value;
    }                                 LineType;

  typedef std::list< LineType >       LinesContainer;


  typename TNetwork::Pointer          m_Network;
  NetworkWriteWeightsType             m_ReadWeightValuesType;
  LayersContainer                     m_Layers;
  WeightsContainer                    m_Weights;
  LinesContainer                      m_NameValue;
  std::string                         m_FileName;
  FieldsContainerType                 m_Fields;
  std::ifstream                       m_InputFile;
  bool                                m_BinaryDataByteOrderMSB;
};

} // namespace itk

#ifndef ITK_MANUAL_INSTANTIATION
#include "itkNeuralNetworkFileReader.txx"
#endif

#endif