X-Git-Url: https://git.creatis.insa-lyon.fr/pubgit/?a=blobdiff_plain;f=vtk%2FvtkGdcmWriter.cxx;h=4d0168780397b00aba09bb2ca9e0d774ef05adb8;hb=8f38e5061bcb7bebaea78024d8d743ab973e2dab;hp=7d6a697c13e01f790b352569262a970581a3f72b;hpb=f2806c140be3061a45b62c4d3f3ef8870cb8882a;p=gdcm.git diff --git a/vtk/vtkGdcmWriter.cxx b/vtk/vtkGdcmWriter.cxx index 7d6a697c..4d016878 100644 --- a/vtk/vtkGdcmWriter.cxx +++ b/vtk/vtkGdcmWriter.cxx @@ -3,8 +3,8 @@ Program: gdcm Module: $RCSfile: vtkGdcmWriter.cxx,v $ Language: C++ - Date: $Date: 2004/12/10 08:34:08 $ - Version: $Revision: 1.7 $ + Date: $Date: 2005/01/28 10:07:35 $ + Version: $Revision: 1.16 $ Copyright (c) CREATIS (Centre de Recherche et d'Applications en Traitement de l'Image). All rights reserved. See Doc/License.txt or @@ -16,9 +16,10 @@ =========================================================================*/ -#include "gdcmHeader.h" #include "gdcmFile.h" +#include "gdcmFileHelper.h" #include "gdcmDebug.h" +#include "gdcmUtil.h" #include "vtkGdcmWriter.h" #include @@ -26,7 +27,7 @@ #include #include -vtkCxxRevisionMacro(vtkGdcmWriter, "$Revision: 1.7 $"); +vtkCxxRevisionMacro(vtkGdcmWriter, "$Revision: 1.16 $"); vtkStandardNewMacro(vtkGdcmWriter); //----------------------------------------------------------------------------- @@ -35,6 +36,12 @@ vtkGdcmWriter::vtkGdcmWriter() { this->LookupTable = NULL; this->FileDimensionality = 3; + this->WriteType = VTK_GDCM_WRITE_TYPE_EXPLICIT_VR; + + UIDPrefix = ""; + StudyInstanceUID = ""; + SeriesInstanceUID = ""; + FrameOfReferenceInstanceUID = ""; } vtkGdcmWriter::~vtkGdcmWriter() @@ -46,10 +53,53 @@ vtkGdcmWriter::~vtkGdcmWriter() void vtkGdcmWriter::PrintSelf(ostream& os, vtkIndent indent) { this->Superclass::PrintSelf(os,indent); + + os << indent << "Write type : " << this->GetWriteTypeAsString(); } //----------------------------------------------------------------------------- // Public +const char *vtkGdcmWriter::GetWriteTypeAsString() +{ + switch(WriteType) + { + case VTK_GDCM_WRITE_TYPE_EXPLICIT_VR : + return "Explicit VR"; + case VTK_GDCM_WRITE_TYPE_IMPLICIT_VR : + return "Implicit VR"; + case VTK_GDCM_WRITE_TYPE_ACR : + return "ACR"; + case VTK_GDCM_WRITE_TYPE_ACR_LIBIDO : + return "ACR Libido"; + default : + return "Unknow type"; + } +} + +void vtkGdcmWriter::SetUIDPrefix(const char *prefix) +{ + UIDPrefix = prefix; +} + +const char *vtkGdcmWriter::GetUIDPrefix() +{ + return UIDPrefix.c_str(); +} + +void vtkGdcmWriter::NewStudyInstanceUID() +{ + StudyInstanceUID = ""; +} + +void vtkGdcmWriter::NewSeriesInstanceUID() +{ + SeriesInstanceUID = ""; +} + +void vtkGdcmWriter::NewFrameOfReferenceInstanceUID() +{ + FrameOfReferenceInstanceUID = ""; +} //----------------------------------------------------------------------------- // Protected @@ -70,22 +120,29 @@ size_t ReverseData(vtkImageData *image,unsigned char **data) size_t planeSize = dim[1] * lineSize; size_t size = dim[2] * planeSize; - *data = new unsigned char[size]; - - image->GetIncrements(inc); - unsigned char *src = (unsigned char *)image->GetScalarPointerForExtent(extent); - unsigned char *dst = *data + planeSize - lineSize; - for (int plane = extent[4]; plane <= extent[5]; plane++) + if( size>0 ) { - for (int line = extent[2]; line <= extent[3]; line++) + *data = new unsigned char[size]; + + image->GetIncrements(inc); + unsigned char *src = (unsigned char *)image->GetScalarPointerForExtent(extent); + unsigned char *dst = *data + planeSize - lineSize; + for (int plane = extent[4]; plane <= extent[5]; plane++) { - // Copy one line at proper destination: - memcpy((void*)dst, (void*)src, lineSize); + for (int line = extent[2]; line <= extent[3]; line++) + { + // Copy one line at proper destination: + memcpy((void*)dst, (void*)src, lineSize); - src += inc[1]*image->GetScalarSize(); - dst -= lineSize; + src += inc[1] * image->GetScalarSize(); + dst -= lineSize; + } + dst += 2 * planeSize; } - dst += 2 * planeSize; + } + else + { + *data = NULL; } return size; @@ -94,7 +151,7 @@ size_t ReverseData(vtkImageData *image,unsigned char **data) /** * Set the datas informations in the file */ -void SetImageInformation(gdcm::File *file,vtkImageData *image) +void SetImageInformation(gdcm::FileHelper *file,vtkImageData *image) { std::ostringstream str; @@ -106,29 +163,29 @@ void SetImageInformation(gdcm::File *file,vtkImageData *image) str.str(""); str << dim[0]; - file->ReplaceOrCreateByNumber(str.str(),0x0028,0x0011); // Columns + file->InsertValEntry(str.str(),0x0028,0x0011); // Columns str.str(""); str << dim[1]; - file->ReplaceOrCreateByNumber(str.str(),0x0028,0x0010); // Rows + file->InsertValEntry(str.str(),0x0028,0x0010); // Rows if(dim[2]>1) { str.str(""); str << dim[2]; - //file->ReplaceOrCreateByNumber(str.str(),0x0028,0x0012); // Planes - file->ReplaceOrCreateByNumber(str.str(),0x0028,0x0008); // Number of Frames + //file->Insert(str.str(),0x0028,0x0012); // Planes + file->InsertValEntry(str.str(),0x0028,0x0008); // Number of Frames } // Pixel type str.str(""); str << image->GetScalarSize()*8; - file->ReplaceOrCreateByNumber(str.str(),0x0028,0x0100); // Bits Allocated - file->ReplaceOrCreateByNumber(str.str(),0x0028,0x0101); // Bits Stored + file->InsertValEntry(str.str(),0x0028,0x0100); // Bits Allocated + file->InsertValEntry(str.str(),0x0028,0x0101); // Bits Stored str.str(""); str << image->GetScalarSize()*8-1; - file->ReplaceOrCreateByNumber(str.str(),0x0028,0x0102); // High Bit + file->InsertValEntry(str.str(),0x0028,0x0102); // High Bit // Pixel Repr // FIXME : what do we do when the ScalarType is @@ -145,44 +202,44 @@ void SetImageInformation(gdcm::File *file,vtkImageData *image) { str << "1"; // Signed } - file->ReplaceOrCreateByNumber(str.str(),0x0028,0x0103); // Pixel Representation + file->InsertValEntry(str.str(),0x0028,0x0103); // Pixel Representation // Samples per pixel str.str(""); str << image->GetNumberOfScalarComponents(); - file->ReplaceOrCreateByNumber(str.str(),0x0028,0x0002); // Samples per Pixel + file->InsertValEntry(str.str(),0x0028,0x0002); // Samples per Pixel // Spacing double *sp = image->GetSpacing(); str.str(""); str << sp[0] << "\\" << sp[1]; - file->ReplaceOrCreateByNumber(str.str(),0x0028,0x0030); // Pixel Spacing + file->InsertValEntry(str.str(),0x0028,0x0030); // Pixel Spacing str.str(""); str << sp[2]; - file->ReplaceOrCreateByNumber(str.str(),0x0018,0x0088); // Spacing Between Slices + file->InsertValEntry(str.str(),0x0018,0x0088); // Spacing Between Slices // Origin double *org = image->GetOrigin(); str.str(""); str << org[0] << "\\" << org[1] << "\\" << org[2]; - file->ReplaceOrCreateByNumber(str.str(),0x0020,0x0032); // Image Position Patient + file->InsertValEntry(str.str(),0x0020,0x0032); // Image Position Patient // Window / Level double *rng=image->GetScalarRange(); str.str(""); str << rng[1]-rng[0]; - file->ReplaceOrCreateByNumber(str.str(),0x0028,0x1051); // Window Width + file->InsertValEntry(str.str(),0x0028,0x1051); // Window Width str.str(""); str << (rng[1]+rng[0])/2.0; - file->ReplaceOrCreateByNumber(str.str(),0x0028,0x1050); // Window Center + file->InsertValEntry(str.str(),0x0028,0x1050); // Window Center // Pixels unsigned char *data; size_t size = ReverseData(image,&data); - file->SetImageData(data,size); + file->SetUserData(data,size); } /** @@ -286,19 +343,56 @@ void vtkGdcmWriter::RecursiveWrite(int axis, vtkImageData *cache, void vtkGdcmWriter::WriteDcmFile(char *fileName,vtkImageData *image) { // From here, the write of the file begins - gdcm::File *dcmFile = new gdcm::File(); + gdcm::FileHelper *dcmFile = new gdcm::FileHelper(); + + // Set the image UID + if( StudyInstanceUID.empty() ) + StudyInstanceUID = gdcm::Util::CreateUniqueUID( UIDPrefix ); + if( SeriesInstanceUID.empty() ) + SeriesInstanceUID = gdcm::Util::CreateUniqueUID( UIDPrefix ); + if( FrameOfReferenceInstanceUID.empty() ) + FrameOfReferenceInstanceUID = gdcm::Util::CreateUniqueUID( UIDPrefix ); + std::string uid = gdcm::Util::CreateUniqueUID( UIDPrefix ); + + dcmFile->InsertValEntry(uid,0x0008,0x0018); //[SOP Instance UID] + dcmFile->InsertValEntry(uid,0x0002,0x0003); //[Media Stored SOP Instance UID] + dcmFile->InsertValEntry(StudyInstanceUID,0x0020,0x000d); //[Study Instance UID] + dcmFile->InsertValEntry(SeriesInstanceUID,0x0020,0x000e); //[Series Instance UID] + dcmFile->InsertValEntry(FrameOfReferenceInstanceUID,0x0020, 0x0052); //[Frame of Reference UID] // Set the image informations SetImageInformation(dcmFile,image); // Write the image + switch(this->WriteType) + { + case VTK_GDCM_WRITE_TYPE_EXPLICIT_VR : + dcmFile->SetWriteTypeToDcmExplVR(); + break; + case VTK_GDCM_WRITE_TYPE_IMPLICIT_VR : + dcmFile->SetWriteTypeToDcmImplVR(); + break; + case VTK_GDCM_WRITE_TYPE_ACR : + dcmFile->SetWriteTypeToAcr(); + break; + case VTK_GDCM_WRITE_TYPE_ACR_LIBIDO : + dcmFile->SetWriteTypeToAcrLibido(); + break; + default : + dcmFile->SetWriteTypeToDcmExplVR(); + } + if(!dcmFile->Write(fileName)) { vtkErrorMacro( << "File " << this->FileName << "couldn't be written by " << " the gdcm library"); - std::cerr << "not written \n"; } + // Clean up + if( dcmFile->GetUserData() && dcmFile->GetUserDataSize()>0 ) + { + delete[] dcmFile->GetUserData(); + } delete dcmFile; }