mitkTwoTrackerMatching.h

Go to the documentation of this file.

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

  NifTK: A software platform for medical image computing.

  Copyright (c) University College London (UCL). All rights reserved.

  This software is distributed WITHOUT ANY WARRANTY; without even
  the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
  PURPOSE.

  See LICENSE.txt in the top level directory for details.

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

#ifndef mitkTwoTrackerMatching_h
#define mitkTwoTrackerMatching_h

#include "niftkOpenCVExports.h"
#include <cv.h>
#include <itkObject.h>
#include <itkObjectFactory.h>
#include <mitkCommon.h>
#include "mitkTrackingMatrices.h"
#include "mitkTrackingAndTimeStampsContainer.h"

namespace mitk
{

class NIFTKOPENCV_EXPORT TwoTrackerMatching : public itk::Object
{
public: 
  mitkClassMacroItkParent ( TwoTrackerMatching, itk::Object)
  itkNewMacro (TwoTrackerMatching)

  
  void Initialise (std::string directory1, std::string directory2);

  cv::Mat GetTrackerMatrix ( unsigned int index, long long * TimingError = NULL, unsigned int TrackerIndex = 0 );

  bool IsReady () 
  { 
    return m_Ready;
  } 
  
  int GetNumberOfFrames () 
  {
    return m_FrameNumbers.size();
  }

  void SetLagMilliseconds(unsigned long long Lag, bool LagIsNegative =false);

  void FlipMats1 ();
  void FlipMats2 ();

protected:
  TwoTrackerMatching();
  virtual ~TwoTrackerMatching();

  TwoTrackerMatching(const TwoTrackerMatching&); // Purposefully not implemented.
  TwoTrackerMatching& operator=(const TwoTrackerMatching&); // Purposefully not implemented.
  
  std::vector<unsigned int>             m_FrameNumbers;
  TrackingAndTimeStampsContainer        m_TimeStampsContainer1;
  TrackingAndTimeStampsContainer        m_TimeStampsContainer2;
  bool                                  m_Ready;
  std::string                           m_Directory1;
  std::string                           m_Directory2;

  TrackingMatrices                      m_TrackingMatrices11; // The tracking matrices in directory 1
  TrackingMatrices                      m_TrackingMatrices22; // The tracking matrices in directory 2
  TrackingMatrices                      m_TrackingMatrices12; // The tracking matrices in directory 2 corresponding with the timestamps in directory 1
  TrackingMatrices                      m_TrackingMatrices21; // The tracking matrices in directory 1 corresponding with the timestamps in directory 2

private:
  
  void                                  ConvertMatrices(const TrackingAndTimeStampsContainer& container1, TrackingMatrices& container2);
  void                                  LookupMatrices(const TrackingAndTimeStampsContainer& container1,
                                                       const TrackingAndTimeStampsContainer& container2,
                                                       TrackingMatrices& outputContainer);
  bool                                  CheckTimingErrorStats();
  void                                  CreateLookUps();

  unsigned long long                    m_Lag; // The delay between tracker1 and tracker2
  bool                                  m_LagIsNegative; // Controls the direction of lag

  bool                                  m_FlipMat1; // flip matrices in directory1
  bool                                  m_FlipMat2; // flip matrices in directory2

};


} // namespace


#endif // niftkTwoTrackerMatching_h