vtk实战(四十七)——写入tif格式

#include <vtkImageData.h>
#include <vtkInteractorStyleImage.h>
#include <vtkTIFFWriter.h>
#include <vtkSmartPointer.h>
#include <vtkImageCanvasSource2D.h>
#include <vtkRenderWindow.h>
#include <vtkRenderWindowInteractor.h>
#include <vtkRenderer.h>
#include <vtkImageActor.h>
#include <vtkImageMapper3D.h>
#include <vtkTIFFReader.h>

int main()
{
    std::string filename = "output.tif";
    vtkSmartPointer<vtkImageCanvasSource2D> imageSource =
        vtkSmartPointer<vtkImageCanvasSource2D>::New();
    imageSource->SetScalarTypeToUnsignedChar();
    imageSource->SetExtent(0, 9, 0, 9, 0, 0);
    imageSource->SetNumberOfScalarComponents(3);
    imageSource->SetDrawColor(0, 0, 0, 0);
    imageSource->FillBox(0, 9, 0, 9);
    imageSource->SetDrawColor(255, 0, 0, 0);
    imageSource->FillBox(5, 7, 5, 7);
    imageSource->Update();

    vtkSmartPointer<vtkTIFFWriter> tiffWriter =
        vtkSmartPointer<vtkTIFFWriter>::New();
    tiffWriter->SetFileName(filename.c_str());
    tiffWriter->SetInputConnection(imageSource->GetOutputPort());
    tiffWriter->Write();

    vtkSmartPointer<vtkTIFFReader> reader =
        vtkSmartPointer<vtkTIFFReader>::New();
    reader->SetFileName(filename.c_str());
    reader->Update();

    vtkSmartPointer<vtkImageActor> actor =
        vtkSmartPointer<vtkImageActor>::New();
    actor->GetMapper()->SetInputConnection(reader->GetOutputPort());

    vtkSmartPointer<vtkRenderer> renderer =
        vtkSmartPointer<vtkRenderer>::New();
    renderer->AddActor(actor);
    renderer->ResetCamera();

    vtkSmartPointer<vtkRenderWindow> renderWindow =
        vtkSmartPointer<vtkRenderWindow>::New();
    renderWindow->AddRenderer(renderer);

    vtkSmartPointer<vtkRenderWindowInteractor> renderWindowInteractor =
        vtkSmartPointer<vtkRenderWindowInteractor>::New();
    vtkSmartPointer<vtkInteractorStyleImage> style =
        vtkSmartPointer<vtkInteractorStyleImage>::New();
    renderWindowInteractor->SetInteractorStyle(style);
    renderWindowInteractor->SetRenderWindow(renderWindow);
    renderWindowInteractor->Initialize();
    renderWindowInteractor->Start();
    return 0;
}