mitkCollectionStatistic.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 mitkCollectionStatistic_h
#define mitkCollectionStatistic_h
 
#include <MitkDataCollectionExports.h>
 
#include <mitkDataCollection.h>
 
#include <iostream>
 
namespace mitk {
struct MITKDATACOLLECTION_EXPORT StatisticData
{
  unsigned int m_TruePositive;
  unsigned int m_FalsePositive;
  unsigned int m_TrueNegative;
  unsigned int m_FalseNegative;
 
  double m_DICE;
  double m_Jaccard;
  double m_Sensitivity;
  double m_Specificity;
  double m_RMSD;
 
  StatisticData() :
    m_TruePositive(0), m_FalsePositive(0), m_TrueNegative(0), m_FalseNegative(0),
    m_DICE(0), m_Jaccard(0), m_Sensitivity(0), m_Specificity(0), m_RMSD(-1.0)
  {}
};
 
class ValueToIndexMapper
{
public:
  virtual unsigned char operator() (unsigned char value) const = 0;
};
 
class BinaryValueminusOneToIndexMapper : public virtual ValueToIndexMapper
{
public:
  unsigned char operator() (unsigned char value) const override
  {
    return value-1;
  }
};
 
class BinaryValueToIndexMapper : public virtual ValueToIndexMapper
{
public:
  unsigned char operator() (unsigned char value) const override
  {
    return value;
  }
};
 
class MultiClassValueToIndexMapper : public virtual ValueToIndexMapper
{
public:
  unsigned char operator() (unsigned char value) const override
  {
    if (value == 1 || value == 5)
      return 0;
    else
      return 1;
  }
};
 
class ProgressionValueToIndexMapper : public virtual ValueToIndexMapper
{
public:
  unsigned char operator() (unsigned char value) const override
  {
    if (value == 1 || value == 0)
      return 0;
    else
      return 1;
  }
};
 
class MITKDATACOLLECTION_EXPORT CollectionStatistic
{
public:
 
  CollectionStatistic();
  ~CollectionStatistic();
 
  typedef std::vector<StatisticData> DataVector;
  typedef std::vector<DataVector> MultiDataVector;
 
  void SetCollection(DataCollection::Pointer collection);
  DataCollection::Pointer GetCollection();
 
  void SetClassCount (size_t count);
  size_t GetClassCount();
 
  void SetGoldName(std::string name);
  std::string GetGoldName();
 
  void SetTestName(std::string name);
  std::string GetTestName();
 
  void SetMaskName(std::string name) {m_MaskName = name; }
 
  void SetGroundTruthValueToIndexMapper(const ValueToIndexMapper* mapper);
  const ValueToIndexMapper* GetGroundTruthValueToIndexMapper(void) const;
 
  void SetTestValueToIndexMapper(const ValueToIndexMapper* mapper);
  const ValueToIndexMapper* GetTestValueToIndexMapper(void) const;
 
  void Print(std::ostream& out, std::ostream& sout = std::cout, bool withHeader = false, std::string label = "None");
  bool Update();
  int IsInSameVirtualClass(unsigned char gold, unsigned char test);
 
  std::vector<StatisticData> GetStatisticData(unsigned char c) const;
 
  void ComputeRMSD();
 
private:
  size_t m_ClassCount;
  std::string m_GroundTruthName;
  std::string m_TestName;
  std::string m_MaskName;
  DataCollection::Pointer m_Collection;
 
  std::vector<unsigned char> m_ConnectionGold;
  std::vector<unsigned char> m_ConnectionTest;
  std::vector<unsigned char> m_ConnectionClass;
 
  MultiDataVector m_ImageClassStatistic;
  std::vector<std::string> m_ImageNames;
  DataVector m_ImageStatistic;
  StatisticData m_MeanCompleteStatistic;
  StatisticData m_CompleteStatistic;
 
  const ValueToIndexMapper* m_GroundTruthValueToIndexMapper;
  const ValueToIndexMapper* m_TestValueToIndexMapper;
};
}
 
#endif