mitkLabeledImageLookupTable.cpp

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.

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

#include "mitkLabeledImageLookupTable.h"
#include <cstdlib>
#include <vtkLookupTable.h>

mitk::LabeledImageLookupTable::LabeledImageLookupTable() : m_LevelWindow(128, 256)
{
  if (m_LookupTable == nullptr)
  {
    itkWarningMacro(
      "LookupTable is nullptr, it should have been initialized by the default constructor of mitk::LookupTable");
    m_LookupTable = vtkLookupTable::New();
  }
  m_LookupTable->SetNumberOfTableValues(256);

  // set the background to black and fully transparent
  m_LookupTable->SetTableValue(0, 0.0, 0.0, 0.0, 0.0);

  // initialize the remaining 255 colors with random values and
  // an alpha value of 1.0
  double r, g, b;

  //
  // Initialize the random number generator with an arbitrary seed.
  // This way, the generated colors are random, but always the same...
  std::srand(2);
  for (vtkIdType index = 1; index < 256; ++index)
  {
    GenerateRandomColor(r, g, b);
    m_LookupTable->SetTableValue(index, r, g, b);
  }

  // initialize the default level/window settings,
  // which can be accessed via GetLevelWindow();
  m_LevelWindow.SetRangeMinMax(0, 255);
  m_LevelWindow.SetWindowBounds(0, 255);
  m_LevelWindow.SetFixed(true);
}

mitk::LabeledImageLookupTable::LabeledImageLookupTable(const mitk::LabeledImageLookupTable &other)
  : LookupTable(other), m_LevelWindow(other.m_LevelWindow)
{
}

mitk::LabeledImageLookupTable::~LabeledImageLookupTable()
{
}

mitk::LabeledImageLookupTable &mitk::LabeledImageLookupTable::operator=(const mitk::LookupTable &other)
{
  LookupTable::operator=(other);
  if (const auto *lut = dynamic_cast<const LabeledImageLookupTable *>(&other))
  {
    this->m_LevelWindow = lut->m_LevelWindow;
  }
  return *this;
}

void mitk::LabeledImageLookupTable::SetColorForLabel(const mitk::LabeledImageLookupTable::LabelType &label,
                                                     const double &r,
                                                     const double &g,
                                                     const double &b,
                                                     const double a)
{
  if (m_LookupTable == nullptr)
  {
    itkWarningMacro("LookupTable is nullptr, but it should have been initialized by the constructor");
    return;
  }
  m_LookupTable->SetTableValue(label, r, g, b, a);
}

double *mitk::LabeledImageLookupTable::GetColorForLabel(const mitk::LabeledImageLookupTable::LabelType &label)
{
  if (m_LookupTable == nullptr)
  {
    itkWarningMacro("LookupTable is nullptr, but it should have been initialized by the constructor");
    return nullptr;
  }
  return m_LookupTable->GetTableValue(label);
}

void mitk::LabeledImageLookupTable::GenerateRandomColor(double &r, double &g, double &b)
{
  r = GenerateRandomNumber();
  g = GenerateRandomNumber();
  b = GenerateRandomNumber();
}

double mitk::LabeledImageLookupTable::GenerateRandomNumber()
{
  return (((double)(std::rand())) / ((double)(RAND_MAX)));
}

itk::LightObject::Pointer mitk::LabeledImageLookupTable::InternalClone() const
{
  itk::LightObject::Pointer result(new Self(*this));
  result->UnRegister();
  return result;
}