cylinder.cxx

VTK入门样例(种类齐全)

#include "vtkPolyDataMapper.h"
#include "vtkRenderWindow.h"
#include "vtkCamera.h"
#include "vtkActor.h"
#include "vtkRenderer.h"
#include "vtkPLOT3DReader.h"
#include "vtkPolyDataNormals.h"
#include "vtkRenderWindowInteractor.h"
#include "vtkTextProperty.h"
#include "vtkTexture.h"
#include "vtkFloatArray.h"
#include "vtkContourFilter.h"
#include "vtkCastToConcrete.h"
#include "vtkProgrammableAttributeDataFilter.h"
#include "vtkMath.h"
#include "vtkRendererSource.h"
#include "vtkProgrammableAttributeDataFilter.h"
#include "vtkProperty.h"
#include "vtkSphereSource.h"
#include "vtkPointData.h"
#include "vtkCellData.h"
#include "vtkScalarBarActor.h"

#include "vtkWindowToImageFilter.h"
#include "vtkImageMapper.h"
#include "vtkActor2D.h"
#include "vtkScalarBarActor.h"

int main()
{
	vtkRenderer *ren1= vtkRenderer::New();
	vtkRenderer *ren2= vtkRenderer::New();
    vtkRenderWindow *renWin = vtkRenderWindow::New();
      renWin->AddRenderer( ren1 );
	  renWin->AddRenderer( ren2 );
	vtkRenderWindowInteractor *iren = vtkRenderWindowInteractor::New();
      iren->SetRenderWindow(renWin);

vtkSphereSource *sphere=vtkSphereSource::New();
    sphere-> SetThetaResolution( 40);
    sphere-> SetPhiResolution( 20);
    sphere-> Update();

vtkProgrammableAttributeDataFilter *randomColors=vtkProgrammableAttributeDataFilter::New();
    randomColors-> SetInputConnection (sphere-> GetOutputPort());
    vtkMath *randomColorGenerator=vtkMath::New();
//      int input=randomColors-> GetInput();
//      int output=randomColors-> GetOutput();

    int numCells=((vtkPolyData *)randomColors-> GetInput())-> GetNumberOfCells();

	printf("numCells : %d\n",numCells);

    vtkFloatArray *colors=vtkFloatArray::New();
	  colors->SetNumberOfTuples(numCells);
	int i;
	double f;
	for (i=0;i<numCells;i++)
	{
		f=randomColorGenerator->Random (0, 1);
		colors-> SetValue(i,f);
		printf("i : %d color : %f\n",i,f);
	}

    randomColors->GetOutput()->GetCellData()->CopyScalarsOff();
    randomColors->GetOutput()->GetCellData()->
		PassData(((vtkPolyData *)randomColors-> GetInput())-> GetCellData());
	randomColors->Update();
    randomColors->GetOutput()->GetCellData()->SetScalars(colors);

//    colors-> Delete(); 
    randomColorGenerator-> Delete();

randomColors->Update();

vtkPolyDataMapper *mapper=vtkPolyDataMapper::New();
    mapper ->SetInput(randomColors-> GetPolyDataOutput());
	mapper-> SetScalarRange(randomColors->GetPolyDataOutput()-> GetScalarRange());
//    mapper ->ScalarVisibilityOn();


    vtkActor *sphereActor  = vtkActor::New();
      sphereActor->SetMapper(mapper);


vtkScalarBarActor *scalarBar=vtkScalarBarActor::New();
    scalarBar-> SetLookupTable(mapper-> GetLookupTable());
    scalarBar-> SetTitle ("Color Table");
    scalarBar-> GetPositionCoordinate()->SetCoordinateSystemToNormalizedViewport();
    scalarBar-> GetPositionCoordinate()-> SetValue(0.1, 0.05);  
//    scalarBar-> GetPositionCoordinate()-> 
    scalarBar-> SetOrientationToVertical();
    scalarBar-> SetWidth (0.8);
    scalarBar-> SetHeight (0.9);

	scalarBar-> SetPosition (scalarBar-> GetPosition());
	scalarBar-> SetNumberOfLabels(8);

	  ren1->AddActor(sphereActor);
	  ren2->AddActor2D(scalarBar);
	  ren1->SetBackground( 1, 1, 1 );
	  ren1-> ResetCamera();

ren1-> SetViewport (0, 0, 0.75, 1.0);
ren2-> SetViewport (0.75, 0, 1.0, 1.0);
ren2-> SetBackground (0.3, 0.3, 0.3);

    renWin->SetSize( 600, 600 );
	renWin-> Render();

	iren->Initialize();
    iren->Start(); 

vtkWindowToImageFilter *w2i=vtkWindowToImageFilter::New();
w2i-> SetInput (renWin);
w2i-> SetMagnification( 2);
w2i-> Update();

(w2i-> GetOutput()->NewInstance())->DeepCopy (w2i-> GetOutput());

vtkImageMapper *ia=vtkImageMapper::New();
ia-> SetInput(w2i-> GetOutput()->NewInstance());
scalarBar-> ReleaseGraphicsResources (renWin);
sphereActor-> ReleaseGraphicsResources( renWin);
ia-> SetColorWindow(255);
ia-> SetColorLevel(127.5);

vtkActor2D *ia2=vtkActor2D::New();
ia2-> SetMapper(ia);

renWin-> SetSize( 320, 320);
ren2-> RemoveViewProp (scalarBar);
ren1-> RemoveViewProp (sphereActor);
ren1-> AddActor (ia2);
renWin-> RemoveRenderer (ren2); 
ren1-> SetViewport (0, 0, 1, 1);

renWin-> Render();

	iren->Initialize();
    iren->Start(); 

	ren1->Delete();
    renWin->Delete();
	iren->Delete();
    
	return 0;
}