usElastography.h

/****************************************************************************
 *
 * This file is part of the ustk software.
 * Copyright (C) 2016 - 2017 by Inria. All rights reserved.
 *
 * This software is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License
 * ("GPL") version 2 as published by the Free Software Foundation.
 * See the file LICENSE.txt at the root directory of this source
 * distribution for additional information about the GNU GPL.
 *
 * For using ustk with software that can not be combined with the GNU
 * GPL, please contact Inria about acquiring a ViSP Professional
 * Edition License.
 *
 * This software was developed at:
 * Inria Rennes - Bretagne Atlantique
 * Campus Universitaire de Beaulieu
 * 35042 Rennes Cedex
 * France
 *
 * If you have questions regarding the use of this file, please contact
 * Inria at ustk@inria.fr
 *
 * This file is provided AS IS with NO WARRANTY OF ANY KIND, INCLUDING THE
 * WARRANTY OF DESIGN, MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE.
 *
 * Authors:
 * Pedro Patlan Rosales
 * Marc Pouliquen
 *
 *****************************************************************************/
 
#ifndef __usElastography_h_
#define __usElastography_h_
 
#include <visp3/ustk_core/usConfig.h>
 
#if defined(USTK_HAVE_FFTW)
 
#include <visp3/core/vpImageTools.h>
 
#include <visp3/ustk_core/usImageRF2D.h>
#include <visp3/ustk_elastography/usConvolution2d.h>
#include <visp3/ustk_elastography/usMotionEstimation.h>
#include <visp3/ustk_elastography/usSignalProcessing.h>
 
class VISP_EXPORT usElastography
{
public:
  enum MotionEstimator { OF, BMA_TAYLOR };
  usElastography();
  usElastography(usImageRF2D<short int> &Pre, usImageRF2D<short int> &Post);
  virtual ~usElastography();
 
  double getFPS(void) { return m_FPS; }
  double getSamplingFrequency(void) { return m_samplingFrequency; }
 
  vpImage<unsigned char> run();
 
  void setDecimationFactor(unsigned int decimationFactor) { m_decimationFactor = decimationFactor; }
  void setFPS(double fps);
  void setSamplingFrequency(double samplingFrequency);
  void setLSQpercentage(double per);
  void setMotionEstimator(MotionEstimator t_mest) { m_mEstimatior = t_mest; }
  void setPostCompression(const usImageRF2D<short> &Post);
  void setPreCompression(const usImageRF2D<short> &Pre);
  void setROI(int tx, int ty, int tw, int th);
 
  void updateRF(const usImageRF2D<short int> &image);
  void updateROIPos(int tx, int ty);
 
private:
  usImageRF2D<short int> usImageRF_ROI(const usImageRF2D<short int> &M, uint r, uint c, uint nrows, uint ncols);
 
  vpImage<unsigned char> m_StrainMap;
  usImageRF2D<short int> m_Precomp;
  usImageRF2D<short int> m_Postcomp;
 
  double m_Lsqper;
  bool m_isloadPre;
  bool m_isloadPost;
  double m_FPS;
  double m_samplingFrequency;
  double m_c;
  double m_min_str;
  double m_max_str;
  double m_max_abs;
  int m_h_m;
  int m_w_m;
 
  int m_ix;
  int m_iy;
  int m_rw;
  int m_rh;
  bool m_setROI;
  usImageRF2D<short int> m_PreROI;
  usImageRF2D<short int> m_PostROI;
 
  std::vector<usConvolution2d *> cC;
 
  // Motion estimation
  MotionEstimator m_mEstimatior;
 
#if defined(USTK_HAVE_ARMADILLO) && (ARMA_VERSION_MAJOR > 6) && (ARMA_VERSION_MAJOR > 700)
  usMotionEstimation m_ME;
#endif
 
  vpMatrix U;
  vpMatrix V;
 
  unsigned int m_decimationFactor;
};
 
#endif // FFTW
#endif // __usElastography_h_