vtkPlusForoughiBoneSurfaceProbability.h

Go to the documentation of this file.

/*=Plus=header=begin======================================================
  Program: Plus
  Copyright (c) Laboratory for Percutaneous Surgery. All rights reserved.
  See License.txt for details.
=========================================================Plus=header=end*/

#ifndef vtkPlusForoughiBoneSurfaceProbability_h
#define vtkPlusForoughiBoneSurfaceProbability_h

#include "vtkPlusImageProcessingExport.h"

#include "vtkSimpleImageToImageFilter.h"
#include "vtkSmartPointer.h"

class vtkPlusImageProcessingExport vtkPlusForoughiBoneSurfaceProbability : public vtkSimpleImageToImageFilter
{
public:
  static vtkPlusForoughiBoneSurfaceProbability* New();
  vtkTypeMacro(vtkPlusForoughiBoneSurfaceProbability, vtkSimpleImageToImageFilter);

  vtkSetMacro(BlurredVSBLoG, int);
  vtkGetMacro(BlurredVSBLoG, int);

  vtkSetMacro(BoneThreshold, double);
  vtkGetMacro(BoneThreshold, double);

  vtkSetMacro(ShadowSigma, double);
  vtkGetMacro(ShadowSigma, double);

  /* Controls the ratio between the shadow map and the reflection number. */
  vtkSetMacro(ShadowVSIntensity, int);
  vtkGetMacro(ShadowVSIntensity, int);

  vtkSetMacro(SmoothingSigma, double);
  vtkGetMacro(SmoothingSigma, double);

  vtkSetMacro(TransducerMargin, int);
  vtkGetMacro(TransducerMargin, int);

protected:
  vtkPlusForoughiBoneSurfaceProbability();
  virtual ~vtkPlusForoughiBoneSurfaceProbability();

  void UpdateKernels();
  void DeleteKernels();

  void Foroughi2007(double* inputBuffer, double* outputBuffer, double smoothingSigma, int transducerMargin, double shadowSigma, double boneThreshold, int blurredVSBLoG, int shadowVSIntensity, int nx, int ny, int nz);
  void Conv2(const double* inputBuffer, const double* kernelBuffer, double* tempBuffer, double* outputBuffer, int nx, int ny, int kx, int ky);
  void ResizeMatrix(const double* inputBuffer, double* outputBuffer, int xClipping, int yClipping, int xInputSize, int yInputSize);
  double GetMaxPixelValue(const double* buffer, int size);
  void Normalize(double* buffer, int size, bool doInverse, double maxValue = 1.0);

  virtual void SimpleExecute(vtkImageData* input, vtkImageData* output);

  int BlurredVSBLoG;
  double BoneThreshold;
  double ShadowSigma;
  int ShadowVSIntensity;
  double SmoothingSigma;
  int TransducerMargin;

  bool KernelUpdateRequested;

  int GaussianKernelSize;
  FrameSizeType FrameSize;

  double* MklGaussianBuffer;
  double* MklLaplacianOfGaussianBuffer;
  double* MklGaussianBufferTemp;
  double* MklLaplacianOfGaussianBufferTemp;
  double* MklReflectionNumberBuffer;
  double* MklShadowValueBuffer;
  double* MklShadowModel;
  double* MklGaussianKernel;
  double* MklLaplacianKernel;

private:
  vtkPlusForoughiBoneSurfaceProbability(const vtkPlusForoughiBoneSurfaceProbability&);  // Not implemented.
  void operator=(const vtkPlusForoughiBoneSurfaceProbability&);  // Not implemented.
};

#endif