GDCM
2.2.0
|
/*========================================================================= Program: GDCM (Grassroots DICOM). A DICOM library Copyright (c) 2006-2011 Mathieu Malaterre All rights reserved. See Copyright.txt or http://gdcm.sourceforge.net/Copyright.html for details. This software is distributed WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the above copyright notice for more information. =========================================================================*/ #include "vtkGDCMPolyDataWriter.h" #include "vtkGDCMPolyDataReader.h" #include "vtkPolyData.h" #include "vtkPolyDataReader.h" #include "vtkMedicalImageProperties.h" #include "vtkRTStructSetProperties.h" #include "vtkStringArray.h" #include "vtkAppendPolyData.h" #include "vtkPolyDataWriter.h" #include "vtkPolyDataMapper.h" #include "vtkPolyDataMapper2D.h" #include "vtkActor2D.h" #include "vtkRenderWindowInteractor.h" #include "vtkMedicalImageProperties.h" #include "vtkRenderWindow.h" #include "vtkRenderer.h" #include "vtkCamera.h" #include "vtkProperty.h" #include "vtkProperty2D.h" #include "vtkImageData.h" #include "gdcmDirectoryHelper.h" using namespace gdcm; /* * Full application which ... RTSTUCT */ int main(int argc, char *argv[]) { if( argc < 2 ) { std::cerr << argv[0] << " directory-with-rtstruct-and-ct-images\n"; return 1; } std::string theDirName(argv[1]); Directory::FilenamesType theRTSeries = DirectoryHelper::GetRTStructSeriesUIDs(theDirName); if (theRTSeries.empty()) { std::cerr << "No RTStructs found for the test, ending." << std::endl; return 1; } Directory::FilenamesType theRTNames = DirectoryHelper::GetFilenamesFromSeriesUIDs(theDirName, theRTSeries[0]); vtkGDCMPolyDataReader * reader = vtkGDCMPolyDataReader::New(); reader->SetFileName( theRTNames[0].c_str() ); reader->Update(); //std::cout << reader->GetMedicalImageProperties()->GetStudyDate() << std::endl; vtkGDCMPolyDataWriter * writer = vtkGDCMPolyDataWriter::New(); int numMasks = reader->GetNumberOfOutputPorts() + 1;//add a blank one in writer->SetNumberOfInputPorts( numMasks ); writer->SetFileName( std::string(theDirName + "/" + "GDCMTestRTStruct." + theRTSeries[0] + ".dcm").c_str()); writer->SetMedicalImageProperties( reader->GetMedicalImageProperties() ); //this line is cheating, we won't have the same stuff, and may not have a struct //to start with. //have to go back to the original data to reconstruct the RTStructureSetProperties //writer->SetRTStructSetProperties( reader->GetRTStructSetProperties() ); //writer->Write(); //loop through the outputs in order to write them out as if they had been created and appended vtkStringArray* roiNames = vtkStringArray::New(); vtkStringArray* roiAlgorithms = vtkStringArray::New(); vtkStringArray* roiTypes = vtkStringArray::New(); roiNames->SetNumberOfValues(numMasks); roiAlgorithms->SetNumberOfValues(numMasks); roiTypes->SetNumberOfValues(numMasks); vtkAppendPolyData* append = vtkAppendPolyData::New(); for (int i = 0; i < reader->GetNumberOfOutputPorts(); ++i) { writer->SetInput(i, reader->GetOutput(i)); append->AddInput(reader->GetOutput(i)); std::string theString = reader->GetRTStructSetProperties()->GetStructureSetROIName(i); roiNames->InsertValue(i, theString); theString = reader->GetRTStructSetProperties()->GetStructureSetROIGenerationAlgorithm(i); roiAlgorithms->InsertValue(i, theString); theString = reader->GetRTStructSetProperties()->GetStructureSetRTROIInterpretedType(i); roiTypes->InsertValue(i, theString); } //ok, now we'll add a blank organ //the blank organ is to test to ensure that blank organs work; there have been crash reports vtkPolyData* blank = vtkPolyData::New(); writer->SetInput(numMasks-1, blank); roiNames->InsertValue(numMasks-1, "blank"); roiAlgorithms->InsertValue(numMasks-1, "blank"); roiTypes->InsertValue(numMasks-1, "ORGAN"); vtkRTStructSetProperties* theProperties = vtkRTStructSetProperties::New(); writer->SetRTStructSetProperties(theProperties); writer->InitializeRTStructSet(theDirName, reader->GetRTStructSetProperties()->GetStructureSetLabel(), reader->GetRTStructSetProperties()->GetStructureSetName(), roiNames, roiAlgorithms, roiTypes); writer->SetRTStructSetProperties(theProperties); writer->Write(); // print reader output: reader->Print( std::cout ); // print first output: reader->GetOutput()->Print( std::cout ); // Now we'll look at it. vtkPolyDataMapper *cubeMapper = vtkPolyDataMapper::New(); cubeMapper->SetInput( append->GetOutput()); cubeMapper->SetScalarRange(0,7); vtkActor *cubeActor = vtkActor::New(); cubeActor->SetMapper(cubeMapper); vtkProperty * property = cubeActor->GetProperty(); property->SetRepresentationToWireframe(); vtkRenderer *renderer = vtkRenderer::New(); vtkRenderWindow *renWin = vtkRenderWindow::New(); renWin->AddRenderer(renderer); vtkRenderWindowInteractor *iren = vtkRenderWindowInteractor::New(); iren->SetRenderWindow(renWin); renderer->AddActor(cubeActor); renderer->ResetCamera(); renderer->SetBackground(1,1,1); renWin->SetSize(300,300); renWin->Render(); iren->Start(); reader->Delete(); append->Delete(); cubeMapper->Delete(); cubeActor->Delete(); renderer->Delete(); renWin->Delete(); iren->Delete(); roiNames->Delete(); roiTypes->Delete(); theProperties->Delete(); roiAlgorithms->Delete(); blank->Delete(); writer->Delete(); return 0; }