Rainbow

#include <vtkActor.h>
#include <vtkCamera.h>
#include <vtkLookupTable.h>
#include <vtkMultiBlockDataSet.h>
#include <vtkMultiBlockPLOT3DReader.h>
#include <vtkNamedColors.h>
#include <vtkPolyDataMapper.h>
#include <vtkRenderWindow.h>
#include <vtkRenderWindowInteractor.h>
#include <vtkRenderer.h>
#include <vtkSmartPointer.h>
#include <vtkStructuredGrid.h>
#include <vtkStructuredGridGeometryFilter.h>
#include <vtkStructuredGridOutlineFilter.h>

// This example demonstrates the use and manipulation of lookup tables.

// First create a simple pipeline that reads a structured grid
// and then extracts a plane from the grid. The plane will be colored
// differently by using different lookup tables.
//
// Note: The Update method is manually invoked because it causes the
// reader to read; later on we use the output of the reader to set
// a range for the scalar values.
int main(int argc, char* argv[])
{
  if (argc < 3)
  {
    std::cout << "Usage: " << argv[0]
              << " filexyz.bin fileq.bin e.g.: combxyz.bin combq.bin"
              << std::endl;
    return EXIT_FAILURE;
  }
  vtkSmartPointer<vtkNamedColors> colors =
      vtkSmartPointer<vtkNamedColors>::New();

  vtkSmartPointer<vtkMultiBlockPLOT3DReader> pl3d =
      vtkSmartPointer<vtkMultiBlockPLOT3DReader>::New();
  pl3d->SetXYZFileName(argv[1]);
  pl3d->SetQFileName(argv[2]);
  pl3d->SetScalarFunctionNumber(100);
  pl3d->SetVectorFunctionNumber(202);
  pl3d->Update();

  vtkStructuredGrid* pl3dOutput =
      dynamic_cast<vtkStructuredGrid*>(pl3d->GetOutput()->GetBlock(0));

  std::vector<vtkSmartPointer<vtkRenderer>> renderers;

  // Create the 4 renderers
  vtkSmartPointer<vtkRenderer> grayScaleRenderer =
      vtkSmartPointer<vtkRenderer>::New();
  renderers.push_back(grayScaleRenderer);
  vtkSmartPointer<vtkRenderer> rainbowBlueRedRenderer =
      vtkSmartPointer<vtkRenderer>::New();
  renderers.push_back(rainbowBlueRedRenderer);
  vtkSmartPointer<vtkRenderer> rainbowRedBlueRenderer =
      vtkSmartPointer<vtkRenderer>::New();
  renderers.push_back(rainbowRedBlueRenderer);
  vtkSmartPointer<vtkRenderer> largeContrastRenderer =
      vtkSmartPointer<vtkRenderer>::New();
  renderers.push_back(largeContrastRenderer);

  std::vector<vtkSmartPointer<vtkLookupTable>> luts;

  vtkSmartPointer<vtkLookupTable> grayScaleLut =
      vtkSmartPointer<vtkLookupTable>::New();
  grayScaleLut->SetHueRange(0, 0);
  grayScaleLut->SetSaturationRange(0, 0);
  grayScaleLut->SetValueRange(0.2, 1.0);
  grayScaleLut->SetNumberOfColors(256);
  grayScaleLut->SetHueRange(0.0, 0.667);
  grayScaleLut->Build();
  luts.push_back(grayScaleLut);

  vtkSmartPointer<vtkLookupTable> rainbowBlueRedLut =
      vtkSmartPointer<vtkLookupTable>::New();
  rainbowBlueRedLut->SetNumberOfColors(256);
  rainbowBlueRedLut->SetHueRange(0.667, 0.0);
  rainbowBlueRedLut->Build();
  luts.push_back(rainbowBlueRedLut);

  vtkSmartPointer<vtkLookupTable> rainbowRedBlueLut =
      vtkSmartPointer<vtkLookupTable>::New();
  rainbowRedBlueLut->SetNumberOfColors(256);
  rainbowRedBlueLut->SetHueRange(0.0, 0.667);
  rainbowRedBlueLut->Build();
  luts.push_back(rainbowRedBlueLut);

  vtkSmartPointer<vtkLookupTable> highContrastLut =
      vtkSmartPointer<vtkLookupTable>::New();
  highContrastLut->SetNumberOfColors(256);
  highContrastLut->Build();

  for (int l = 0; l < 16; ++l)
  {
    highContrastLut->SetTableValue(l * 16, 1, 0, 0, 1);
    highContrastLut->SetTableValue(l * 16 + 1, 0, 1, 0, 1);
    highContrastLut->SetTableValue(l * 16 + 2, 0, 0, 1, 1);
    highContrastLut->SetTableValue(l * 16 + 3, 0, 0, 0, 1);
  }
  luts.push_back(highContrastLut);

  vtkSmartPointer<vtkRenderWindow> renderWindow =
      vtkSmartPointer<vtkRenderWindow>::New();
  vtkSmartPointer<vtkRenderWindowInteractor> interactor =
      vtkSmartPointer<vtkRenderWindowInteractor>::New();
  interactor->SetRenderWindow(renderWindow);

  int rendererSize = 256;
  unsigned int xGridDimensions = 2;
  unsigned int yGridDimensions = 2;
  renderWindow->SetSize(rendererSize * xGridDimensions,
                        rendererSize * yGridDimensions);
  for (int row = 0; row < static_cast<int>(yGridDimensions); row++)
  {
    for (int col = 0; col < static_cast<int>(xGridDimensions); col++)
    {
      int index = row * xGridDimensions + col;
      // (xmin, ymin, xmax, ymax)
      double viewport[4] = {
          static_cast<double>(col) / xGridDimensions,
          static_cast<double>(yGridDimensions - (row + 1)) / yGridDimensions,
          static_cast<double>(col + 1) / xGridDimensions,
          static_cast<double>(yGridDimensions - row) / yGridDimensions};
      renderers[index]->SetViewport(viewport);
    }
  }

  vtkSmartPointer<vtkStructuredGridGeometryFilter> plane =
      vtkSmartPointer<vtkStructuredGridGeometryFilter>::New();
  plane->SetInputData(pl3dOutput);
  plane->SetExtent(1, 100, 1, 100, 7, 7);

  for (size_t r = 0; r < renderers.size(); ++r)
  {
    vtkSmartPointer<vtkLookupTable> lut = luts[r];
    vtkSmartPointer<vtkPolyDataMapper> planeMapper =
        vtkSmartPointer<vtkPolyDataMapper>::New();
    planeMapper->SetLookupTable(lut);
    planeMapper->SetInputConnection(plane->GetOutputPort());
    planeMapper->SetScalarRange(pl3dOutput->GetScalarRange());

    vtkSmartPointer<vtkActor> planeActor = vtkSmartPointer<vtkActor>::New();
    planeActor->SetMapper(planeMapper);

    // This creates an outline around the data.
    vtkSmartPointer<vtkStructuredGridOutlineFilter> outline =
        vtkSmartPointer<vtkStructuredGridOutlineFilter>::New();
    outline->SetInputData(pl3dOutput);

    vtkSmartPointer<vtkPolyDataMapper> outlineMapper =
        vtkSmartPointer<vtkPolyDataMapper>::New();
    outlineMapper->SetInputConnection(outline->GetOutputPort());

    vtkSmartPointer<vtkActor> outlineActor = vtkSmartPointer<vtkActor>::New();
    outlineActor->SetMapper(outlineMapper);

    // Create the RenderWindow, Renderer and both Actors
    //
    renderers[r]->AddActor(outlineActor);
    renderers[r]->AddActor(planeActor);

    renderers[r]->SetBackground(colors->GetColor3d("SlateGray").GetData());
    renderers[r]->TwoSidedLightingOff();
    renderers[r]->GetActiveCamera()->Azimuth(20);
    renderers[r]->GetActiveCamera()->Elevation(30);
    renderers[r]->ResetCamera();
    if (r > 0)
    {
      renderers[r]->SetActiveCamera(renderers[0]->GetActiveCamera());
    }
    renderWindow->AddRenderer(renderers[r]);
  }

  // vtkSmartPointer<vtkLookupTable> lut =
  //  vtkSmartPointer<vtkLookupTable>::New();

  // Add the actors to the renderer, set the background and size
  //

  renderWindow->SetSize(rendererSize * xGridDimensions,
                        rendererSize * yGridDimensions);

  interactor->Initialize();

  vtkCamera* cam1 = renderers[0]->GetActiveCamera();
  cam1->SetClippingRange(3.95297, 50);
  cam1->SetFocalPoint(8.88908, 0.595038, 29.3342);
  cam1->SetPosition(-12.3332, 31.7479, 41.2387);
  cam1->SetViewUp(0.060772, -0.319905, 0.945498);

  renderWindow->Render();
  interactor->Start();

  return EXIT_SUCCESS;
}

cmake_minimum_required(VERSION 3.3 FATAL_ERROR)

project(Rainbow)

find_package(VTK COMPONENTS 
  vtkvtkCommonColor
  vtkvtkCommonCore
  vtkvtkCommonDataModel
  vtkvtkFiltersCore
  vtkvtkFiltersGeometry
  vtkvtkIOParallel
  vtkvtkInteractionStyle
  vtkvtkRenderingContextOpenGL2
  vtkvtkRenderingCore
  vtkvtkRenderingFreeType
  vtkvtkRenderingGL2PSOpenGL2
  vtkvtkRenderingOpenGL2 QUIET)
if (NOT VTK_FOUND)
  message("Skipping Rainbow: ${VTK_NOT_FOUND_MESSAGE}")
  return ()
endif()
message (STATUS "VTK_VERSION: ${VTK_VERSION}")
if (VTK_VERSION VERSION_LESS "8.90.0")
  # old system
  include(${VTK_USE_FILE})
  add_executable(Rainbow MACOSX_BUNDLE Rainbow.cxx )
  target_link_libraries(Rainbow PRIVATE ${VTK_LIBRARIES})
else ()
  # include all components
  add_executable(Rainbow MACOSX_BUNDLE Rainbow.cxx )
  target_link_libraries(Rainbow PRIVATE ${VTK_LIBRARIES})
  # vtk_module_autoinit is needed
  vtk_module_autoinit(
    TARGETS Rainbow
    MODULES ${VTK_LIBRARIES}
    )
endif ()

cd Rainbow/build

cmake ..

cmake -DVTK_DIR:PATH=/home/me/vtk_build ..

make

./Rainbow