NIREPDiceCoefficient.cpp

Go to the documentation of this file.

#include "NIREPDiceCoefficient.h"
#include "gecObjectMap.h"


TextTable NIREPDiceCoefficient::ComputeStatistic(const std::vector<std::string>& arg, DisplayDescription * displayDescription)
{
  const gec::ObjectMap* objmap1 =
    dynamic_cast<const gec::ObjectMap *>(this->evaluator->GetSpatialData(arg[0], displayDescription));
  const gec::ObjectMap* objmap2 =
    dynamic_cast<const gec::ObjectMap *>(this->evaluator->GetSpatialData(arg[1], displayDescription));

  int* Intersection = new int[objmap1->GetObjectEntryTableSize()];
  int* X = new int[objmap1->GetObjectEntryTableSize()];
  int* Y = new int[objmap1->GetObjectEntryTableSize()];

  for(int i=0; i<objmap1->GetObjectEntryTableSize(); i++) {
    Intersection[i] = 0;
    X[i] = 0;
    Y[i] = 0;
  }

  const int* dim = objmap1->GetVTKData()->GetDimensions();
  const int size = dim[0] * dim[1] * dim[2];

  // For fast indexing
  const unsigned char* objptr1 =
    static_cast<const unsigned char *>(objmap1->GetVTKData()->GetScalarPointer());
  const unsigned char* objptr2 =
    static_cast<const unsigned char *>(objmap2->GetVTKData()->GetScalarPointer());

  for(int i=0; i<size; i++) {
    int obj1 = static_cast<int>(*objptr1++);
    int obj2 = static_cast<int>(*objptr2++);
    X[obj1]++;
    Y[obj2]++;
    if(obj1 == obj2) {
      Intersection[obj1]++;
    }
  }

  TextTable result;
  result.Title = "Dice Coefficient of " + arg[0] + " and " + arg[1];
  result.ColumnHeading.push_back("Name");
  result.ColumnHeading.push_back("Dice Coefficient");
  std::vector<float> column;
  for(int i=0; i<objmap1->GetObjectEntryTableSize(); i++) {
    column.push_back((2 * static_cast<float>(Intersection[i]) )/(X[i] + Y[i]));
  }
  result.Value.push_back(column);

  delete [] Intersection;
  delete [] X;
  delete [] Y;

  return result;
}