vtk實戰(四十六)——寫入pnm格式

#include <vtkImageData.h>
#include <vtkPNMWriter.h>
#include <vtkSmartPointer.h>
#include <vtkImageCanvasSource2D.h>
#include <vtkSmartPointer.h>
#include <vtkRenderWindow.h>
#include <vtkRenderWindowInteractor.h>
#include <vtkRenderer.h>
#include <vtkImageActor.h>
#include <vtkImageMapper3D.h>
#include <vtkPNMReader.h>
#include <vtkInteractorStyleImage.h>

int main()
{
    std::string filename = "test.pnm";
    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<vtkPNMWriter> pnmWriter =
        vtkSmartPointer<vtkPNMWriter>::New();
    pnmWriter->SetFileName(filename.c_str());
    pnmWriter->SetInputConnection(imageSource->GetOutputPort());
    pnmWriter->Write();

    vtkSmartPointer<vtkPNMReader> reader =
        vtkSmartPointer<vtkPNMReader>::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;
}