QmitkIGTTrackingDataEvaluationView.h

Go to the documentation of this file.

/*============================================================================
 
The Medical Imaging Interaction Toolkit (MITK)
 
Copyright (c) German Cancer Research Center (DKFZ)
All rights reserved.
 
Use of this source code is governed by a 3-clause BSD license that can be
found in the LICENSE file.
 
============================================================================*/
 
#ifndef QmitkIGTTrackingDataEvaluationView_h
#define QmitkIGTTrackingDataEvaluationView_h
 
#include <berryISelectionListener.h>
 
#include <QmitkAbstractView.h>
 
#include "ui_QmitkIGTTrackingDataEvaluationViewControls.h"
#include "mitkHummelProtocolEvaluation.h"
 
#include <mitkNavigationDataEvaluationFilter.h>
#include "mitkNavigationDataCSVSequentialPlayer.h"
 
class QmitkIGTTrackingDataEvaluationView : public QmitkAbstractView
{
  Q_OBJECT
 
  public:
 
    static const std::string VIEW_ID;
 
    QmitkIGTTrackingDataEvaluationView();
    ~QmitkIGTTrackingDataEvaluationView() override;
 
    void CreateQtPartControl(QWidget *parent) override;
    void SetFocus() override {}
 
  protected slots:
 
    void OnLoadFileList();
    void OnAddToCurrentList();
    void OnEvaluateData();
    void OnEvaluateDataAll();
    void OnGeneratePointSet();
    void OnGeneratePointSetsOfSinglePositions();
    void OnGenerateRotationLines();
    void OnGenerateGroundTruthPointSet();
    void OnConvertCSVtoXMLFile();
    void OnCSVtoXMLLoadInputList();
    void OnCSVtoXMLLoadOutputList();
    void OnPerfomGridMatching();
    void OnComputeRotation();
 
    void OnOrientationCalculation_CalcRef();
    void OnOrientationCalculation_CalcOrientandWriteToFile();
 
 
  protected:
 
    Ui::QmitkIGTTrackingDataEvaluationViewControls* m_Controls;
 
    std::vector<std::string> m_FilenameVector;
 
    void MessageBox(std::string s);
 
    std::fstream m_CurrentWriteFile;
    void WriteHeader();
    void WriteDataSet(mitk::NavigationDataEvaluationFilter::Pointer evaluationFilter, std::string dataSetName);
 
    //members for orientation calculation
    mitk::Point3D m_RefPoint1;
    mitk::Point3D m_RefPoint2;
    mitk::Point3D m_RefPoint3;
 
    double m_scalingfactor; //scaling factor for visualization, 1 by default
 
    //angle differences: separated file
    std::fstream m_CurrentAngleDifferencesWriteFile;
    void CalculateDifferenceAngles();
    void WriteDifferenceAnglesHeader();
    void WriteDifferenceAnglesDataSet(std::string pos1, std::string pos2, int idx1, int idx2, double angle);
 
    void writeToFile(std::string filename, std::vector<mitk::HummelProtocolEvaluation::HummelProtocolDistanceError> values);
 
    //different help methods to read a csv logging file
    std::vector<mitk::NavigationData::Pointer> GetNavigationDatasFromFile(std::string filename);
    std::vector<std::string> GetFileContentLineByLine(std::string filename);
    mitk::NavigationData::Pointer GetNavigationDataOutOfOneLine(std::string line);
 
    //help method to sonstruct the NavigationDataCSVSequentialPlayer filled with all the options from the UI
    mitk::NavigationDataCSVSequentialPlayer::Pointer ConstructNewNavigationDataPlayer();
 
    //CSV to XML members
    std::vector<std::string> m_CSVtoXMLInputFilenameVector;
    std::vector<std::string> m_CSVtoXMLOutputFilenameVector;
 
    //returns the number of converted lines
    int ConvertOneFile(std::string inputFilename, std::string outputFilename);
 
    double GetAngleBetweenTwoQuaterions(mitk::Quaternion a, mitk::Quaternion b);
 
    mitk::Quaternion GetSLERPAverage(mitk::NavigationDataEvaluationFilter::Pointer);
 
    mitk::PointSet::Pointer m_PointSetMeanPositions;
 
    std::vector<mitk::Quaternion> GetMeanOrientationsOfAllData(std::vector<mitk::NavigationDataEvaluationFilter::Pointer> allData, bool useSLERP = false);
 
    std::vector<mitk::NavigationDataEvaluationFilter::Pointer> GetAllDataFromUIList();
};
 
 
 
#endif