demo.cpp

VTK是一个可视化工具包

#include "vtkRenderer.h"
#include "vtkRenderWindow.h"
#include "vtkRenderWindowInteractor.h"
#include "vtkVolume16Reader.h"
#include "vtkVolumeReader.h"
#include "vtkImageReader.h"
#include "vtkPolyDataMapper.h"
#include "vtkActor.h"
#include "vtkOutlineFilter.h"
#include "vtkPiecewiseFunction.h"
#include "vtkColorTransferFunction.h"
#include "vtkVolumeProperty.h"
#include "vtkVolumeRayCastCompositeFunction.h"
#include "vtkVolumeRayCastMIPFunction.h"
#include "vtkVolumeRayCastMapper.h"
#include "vtkVolume.h"
#include "vtkVolumeTextureMapper2D.h"



int main (int argc, char **argv)
{
  if (argc < 2)
    {
      cout << "Usage: " << argv[0] << " cell/cell" << endl;
    return 1;
    }

  vtkRenderer *aRenderer = vtkRenderer::New();
  vtkRenderWindow *renWin = vtkRenderWindow::New();
    renWin->AddRenderer(aRenderer);
  vtkRenderWindowInteractor *iren = vtkRenderWindowInteractor::New();
    iren->SetRenderWindow(renWin);

/*  vtkVolume16Reader *v16 = vtkVolume16Reader::New();
     v16->SetDataDimensions(128,128);
     v16->SetDataByteOrderToLittleEndian();
    v16->SetFilePrefix(argv[1]); 
    v16->SetImageRange(140, 180);
    v16->SetDataSpacing(3.2, 3.2, 2.3);
*/
 /*   vtkImageReader *volReader = vtkImageReader::New();
    volReader->SetImageRange (1,5);
    volReader->SetFilePrefix (argv[1]);
    volReader->SetDataSpacing (1,1,1);
    volReader->Update();
 */

	vtkImageReader *reader = vtkImageReader::New();
	reader->SetDataByteOrderToBigEndian();
	reader->SetDataExtent(0,159,0,114,0,64);
	reader->SetFilePrefix(argv[1]);
	reader->SetDataScalarTypeToUnsignedShort();
	reader->SetHeaderSize(0);

 vtkPiecewiseFunction *opacityTFunc=vtkPiecewiseFunction::New();

    opacityTFunc->AddPoint(2000.0, 0.0);
    opacityTFunc->AddPoint(5000.0, 0.5);
    opacityTFunc->AddPoint(10000.0, 0.7);
    opacityTFunc->AddPoint(50000.0, 1.0);
   // opacityTFunc->AddPoint(225, .2);
   // opacityTFunc->AddPoint(430, .9);
   // opacityTFunc->AddPoint(500, .9);
   // opacityTFunc->AddPoint(4095.0, 1.0);

     
  //vtkColorTransferFunction *colorTFunc=vtkColorTransferFunction::New();
   // colorTFunc->AddRGBPoint(0.0, 0.0, 0.5, 0.0);
   // colorTFunc->AddRGBPoint(600.0, 0.5, 1.0, 0.5);
   // colorTFunc->AddRGBPoint(1280.0, 0.2, 0.9, 0.3);
   // colorTFunc->AddRGBPoint(1960.0, 0.27, 0.81, 0.1);
   // colorTFunc->AddRGBPoint(4095.0, 0.5, 0.5, 0.5);


  vtkVolumeProperty *volumeProperty=vtkVolumeProperty::New();
  //  volumeProperty->SetColor(colorTFunc);
    volumeProperty->SetScalarOpacity(opacityTFunc);
    volumeProperty->SetInterpolationTypeToLinear();
    volumeProperty->SetAmbient(0.2);
    volumeProperty->SetDiffuse(0.9);
    volumeProperty->SetSpecular(0.2);
    volumeProperty->SetSpecularPower(10);
    volumeProperty->ShadeOn();
    
//光线合成法
//   vtkVolumeRayCastCompositeFunction 
//      *compositeFunction=vtkVolumeRayCastCompositeFunction::New();
	  
//最大值投射法
   vtkVolumeRayCastMIPFunction *compositeFunction=vtkVolumeRayCastMIPFunction::New();


vtkVolumeRayCastMapper *volumeMapper=vtkVolumeRayCastMapper::New();
    volumeMapper->SetInput(reader->GetOutput());
    volumeMapper->SetVolumeRayCastFunction(compositeFunction);
    volumeMapper->SetNumberOfThreads(3);
//    cerr<GetNumberOfThreads();

//二维纹理映射
/* vtkVolumeTextureMapper2D* volumeMapper=vtkVolumeTextureMapper2D::New();
  volumeMapper->SetInput(reader->GetOutput());
*/
    
  vtkVolume *volume=vtkVolume::New();
    volume->SetMapper(volumeMapper);
    volume->SetProperty(volumeProperty);

  vtkOutlineFilter *outline = vtkOutlineFilter::New();
    outline->SetInput((vtkDataSet *)reader->GetOutput());
  vtkPolyDataMapper *outlineMapper = vtkPolyDataMapper::New();
    outlineMapper->SetInput(outline->GetOutput());
  vtkActor *outlineActor = vtkActor::New();
    outlineActor->SetMapper(outlineMapper);

  aRenderer->AddActor(outlineActor);
  aRenderer->AddVolume(volume);
  
  aRenderer->SetBackground(0,0,0);
  renWin->SetSize(300, 300);

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

  reader->Delete();
  opacityTFunc->Delete();
  //colorTFunc->Delete();
  volumeProperty->Delete();
//  compositeFunction->Delete();
  volumeMapper->Delete();
  volume->Delete();
  outline->Delete();
  outlineMapper->Delete();
  outlineActor->Delete();
  aRenderer->Delete();
  renWin->Delete();
  iren->Delete();

  return 0;
}