// SPDX-FileCopyrightText: Copyright (c) Ken Martin, Will Schroeder, Bill Lorensen
// SPDX-License-Identifier: BSD-3-Clause

// .NAME
// .SECTION Description
// this program tests the ColorTransferFunction

#include "vtkColorTransferFunction.h"
#include "vtkDebugLeaks.h"

#include <sstream>

int Test(ostream& strm)
{
  int i, j, k;
  vtkColorTransferFunction* ctf1 = vtkColorTransferFunction::New();

  // actual test
  strm << "Test vtkColorTransferFunction Start" << endl;

  ctf1->AddRGBPoint(0.0, 1, 0, 0);
  ctf1->AddHSVPoint(1.0, 1, 1, .6);
  ctf1->AddRGBSegment(2.0, 1, 1, 1, 10.0, 0, 0, 0);
  ctf1->AddHSVSegment(11.0, 1, 1, .6, 15.0, .1, .2, .3);
  strm << *ctf1;

  double rgb[3];
  ctf1->GetColor(.5, rgb);
  strm << "GetColor(.5) = " << rgb[0] << ", " << rgb[1] << ", " << rgb[2] << endl;

  strm << "GetRedValue(.5) = " << ctf1->GetRedValue(.5) << endl;
  strm << "GetGreenValue(.5) = " << ctf1->GetGreenValue(.5) << endl;
  strm << "GetBlueValue(.5) = " << ctf1->GetBlueValue(.5) << endl;

  strm << "MapValue(12) = " << (int)ctf1->MapValue(12)[0] << ", " << (int)ctf1->MapValue(12)[1]
       << ", " << (int)ctf1->MapValue(12)[2] << endl;

  strm << "GetRange = " << ctf1->GetRange()[0] << "," << ctf1->GetRange()[1] << endl;

  double table[3][256];

  ctf1->GetTable(0, 15, 256, &table[0][0]);
  strm << "GetTable(0, 15, 256, &table[0][0])" << endl;
  for (i = 0; i < 256; i++)
  {
    for (j = 0; j < 3; j++)
    {
      strm << table[j][i] << " ";
    }
    strm << endl;
  }

  strm << "BuildFunctionFrom(0, 15, 256, &table[0][0])" << endl;
  vtkColorTransferFunction* ctf2 = vtkColorTransferFunction::New();
  ctf2->BuildFunctionFromTable(0, 15, 256, &table[0][0]);

  ctf2->SetColorSpaceToRGB();
  ctf2->GetTable(0, 15, 512);

  ctf2->SetColorSpaceToHSV();
  ctf2->GetTable(0, 15, 512);

  ctf1->DeepCopy(ctf2);
  strm << "ctf1->DeepCopy(ctf2)" << endl;
  strm << *ctf1;

  ctf1->RemovePoint(10);
  strm << *ctf1;

  ctf1->RemoveAllPoints();
  strm << *ctf1;

  char* cData = new char[128];
  unsigned char* ucData = new unsigned char[128];
  short* sData = new short[128];
  unsigned short* usData = new unsigned short[128];
  int* iData = new int[128];
  unsigned int* uiData = new unsigned int[128];
  long* lData = new long[128];
  unsigned long* ulData = new unsigned long[128];
  float* fData = new float[128];
  double* dData = new double[128];

  for (k = 0; k < 128; k++)
  {
    *(cData + k) = static_cast<char>(static_cast<float>(k) / 255.0);
    *(ucData + k) = static_cast<unsigned char>(static_cast<float>(k) / 255.0);
    *(sData + k) = static_cast<short>(static_cast<float>(k) / 255.0);
    *(usData + k) = static_cast<unsigned short>(static_cast<float>(k) / 255.0);
    *(iData + k) = static_cast<int>(static_cast<float>(k) / 255.0);
    *(uiData + k) = static_cast<unsigned int>(static_cast<float>(k) / 255.0);
    *(lData + k) = static_cast<long>(static_cast<float>(k) / 255.0);
    *(ulData + k) = static_cast<unsigned long>(static_cast<float>(k) / 255.0);
    *(fData + k) = static_cast<float>(static_cast<float>(k) / 255.0);
    *(dData + k) = static_cast<double>(static_cast<float>(k) / 255.0);
  }

  unsigned char* ucResult = new unsigned char[128 * 4];
  for (k = 1; k <= 4; k++)
  {
    ctf2->MapScalarsThroughTable2((void*)cData, ucResult, VTK_CHAR, 128, 1, k);
    ctf2->MapScalarsThroughTable2((void*)ucData, ucResult, VTK_UNSIGNED_CHAR, 128, 1, k);
    ctf2->MapScalarsThroughTable2((void*)sData, ucResult, VTK_SHORT, 128, 1, k);
    ctf2->MapScalarsThroughTable2((void*)usData, ucResult, VTK_UNSIGNED_SHORT, 128, 1, k);
    ctf2->MapScalarsThroughTable2((void*)iData, ucResult, VTK_INT, 128, 1, k);
    ctf2->MapScalarsThroughTable2((void*)uiData, ucResult, VTK_UNSIGNED_INT, 128, 1, k);
    ctf2->MapScalarsThroughTable2((void*)lData, ucResult, VTK_LONG, 128, 1, k);
    ctf2->MapScalarsThroughTable2((void*)ulData, ucResult, VTK_UNSIGNED_LONG, 128, 1, k);
    ctf2->MapScalarsThroughTable2((void*)fData, ucResult, VTK_FLOAT, 128, 1, k);
    ctf2->MapScalarsThroughTable2((void*)dData, ucResult, VTK_DOUBLE, 128, 1, k);
  }

  ctf1->Delete();
  ctf2->Delete();

  delete[] ucResult;
  delete[] cData;
  delete[] ucData;
  delete[] sData;
  delete[] usData;
  delete[] iData;
  delete[] uiData;
  delete[] lData;
  delete[] ulData;
  delete[] fData;
  delete[] dData;

  strm << "Test vtkColorTransferFunction End" << endl;
  return 0;
}

int otherColorTransferFunction(int, char*[])
{
  std::ostringstream vtkmsg_with_warning_C4701;
  return Test(vtkmsg_with_warning_C4701);
}
