X-Git-Url: https://git.creatis.insa-lyon.fr/pubgit/?a=blobdiff_plain;f=common%2FclitkDicomRT_StructureSet.cxx;h=9fa96b39f2a55c0e25f9b0e97bfd64dca2718a8a;hb=19b462067f2b839ebb7e7cc8ba84aa14f4152e8b;hp=8543fb7ddfefac3bbd5392b8a0e3a1f6bbdc18b8;hpb=36fabdb651b4cc6b2d52d8a2e0f3be2b78b1f965;p=clitk.git diff --git a/common/clitkDicomRT_StructureSet.cxx b/common/clitkDicomRT_StructureSet.cxx index 8543fb7..9fa96b3 100644 --- a/common/clitkDicomRT_StructureSet.cxx +++ b/common/clitkDicomRT_StructureSet.cxx @@ -20,10 +20,6 @@ #include "clitkDicomRT_StructureSet.h" #include #include "gdcmFile.h" -#if GDCM_MAJOR_VERSION == 2 -#include "gdcmReader.h" -#include "gdcmAttribute.h" -#endif //-------------------------------------------------------------------- clitk::DicomRT_StructureSet::DicomRT_StructureSet() @@ -35,6 +31,7 @@ clitk::DicomRT_StructureSet::DicomRT_StructureSet() mName = "NoName"; mDate = "NoDate"; mTime = "NoTime"; + mFile = NULL; } //-------------------------------------------------------------------- @@ -103,26 +100,119 @@ const std::string & clitk::DicomRT_StructureSet::GetTime() const //-------------------------------------------------------------------- -const std::vector & clitk::DicomRT_StructureSet::GetListOfROI() const +// const std::vector & clitk::DicomRT_StructureSet::GetListOfROI() const +// { +// return mListOfROI; +// } +//-------------------------------------------------------------------- + + +//-------------------------------------------------------------------- +clitk::DicomRT_ROI* clitk::DicomRT_StructureSet::GetROIFromROINumber(int n) { - return mListOfROI; + if (mROIs.find(n) == mROIs.end()) { + std::cerr << "No ROI number " << n << std::endl; + return NULL; + } + return mROIs[n]; } //-------------------------------------------------------------------- +//-------------------------------------------------------------------- +clitk::DicomRT_ROI* clitk::DicomRT_StructureSet::GetROIFromROIName(const std::string& name) +{ + std::map::iterator it = mMapOfROIName.begin(); + int number = -1; + while (it != mMapOfROIName.end() && number == -1) { + if (it->second == name) + number = it->first; + else + it++; + } + + if (number == -1) { + std::cerr << "No ROI name " << name << std::endl; + return NULL; + } + + return mROIs[number]; +} +//-------------------------------------------------------------------- +/* +// RP: 08/02/2013 +// RegEx version shall be available when C++x11 supports it propely +// +//-------------------------------------------------------------------- +clitk::DicomRT_ROI* clitk::DicomRT_StructureSet::GetROIFromROINameRegEx(const std::string& regEx) +{ + std::map::iterator it = mMapOfROIName.begin(); + int number = -1; + + while (it != mMapOfROIName.end() && number == -1) { + if (std::tr1::regex_match (it->second, std::tr1::regex(regEx))) + number = it->first; + else + it++; + } + if (number == -1) { + std::cerr << "No ROI name " << number << std::endl; + return NULL; + } + + return mROIs[number]; +} +//-------------------------------------------------------------------- +*/ //-------------------------------------------------------------------- -clitk::DicomRT_ROI* clitk::DicomRT_StructureSet::GetROI(int n) +clitk::DicomRT_ROI* clitk::DicomRT_StructureSet::GetROIFromROINameSubstr(const std::string& s) { - if (mMapOfROIIndex.find(n) == mMapOfROIIndex.end()) { - std::cerr << "No ROI number " << n << std::endl; + std::map::iterator it = mMapOfROIName.begin(); + int number = -1; + + while (it != mMapOfROIName.end() && number == -1) { + if (it->second.find(s) != std::string::npos) + number = it->first; + else + it++; + } + + if (number == -1) { + std::cerr << "No ROI name " << s << std::endl; return NULL; } - // DD(mListOfROI[mMapOfROIIndex[n]]->GetName()); - //DD(mListOfROI[mMapOfROIIndex[n]]->GetROINumber()); - return mListOfROI[mMapOfROIIndex[n]]; + + return mROIs[number]; } //-------------------------------------------------------------------- +//-------------------------------------------------------------------- +clitk::DicomRT_StructureSet::ROIMapContainer * +clitk::DicomRT_StructureSet::GetROIsFromROINameSubstr(const std::string& s) +{ + static ROIMapContainer rois; + rois.clear(); + + ROIMapContainer::iterator it = mROIs.begin(); + int number = -1; + + while (it != mROIs.end()) { + if (it->second->GetName().find(s) != std::string::npos) { + number = it->first; + rois[number] = it->second; + } + it++; + } + + if (number == -1) { + std::cerr << "No ROI name " << s << std::endl; + return NULL; + } + + return &rois; + +} +//-------------------------------------------------------------------- //-------------------------------------------------------------------- void clitk::DicomRT_StructureSet::Print(std::ostream & os) const @@ -133,48 +223,139 @@ void clitk::DicomRT_StructureSet::Print(std::ostream & os) const << "Struct Label = " << mLabel << std::endl << "Struct Name = " << mName << std::endl << "Struct Time = " << mTime << std::endl - << "Number of ROI = " << mListOfROI.size() << std::endl; - for(unsigned int i=0; iPrint(os); + << "Number of ROI = " << mROIs.size() << std::endl; + for(ROIConstIteratorType iter = mROIs.begin(); iter != mROIs.end(); iter++) { + iter->second->Print(os); } } //-------------------------------------------------------------------- +#if GDCM_MAJOR_VERSION == 2 //-------------------------------------------------------------------- -void clitk::DicomRT_StructureSet::Read(const std::string & filename) +int clitk::DicomRT_StructureSet::ReadROINumber(const gdcm::Item & item) +{ + // 0x3006,0x0022 = [ROI Number] + const gdcm::DataSet & nestedds = item.GetNestedDataSet(); + gdcm::Attribute<0x3006,0x0022> roinumber; + roinumber.SetFromDataSet( nestedds ); + return roinumber.GetValue(); +} +//-------------------------------------------------------------------- +#endif + +//-------------------------------------------------------------------- +void clitk::DicomRT_StructureSet::Write(const std::string & filename) { - // Open DICOM #if GDCM_MAJOR_VERSION == 2 - gdcm::Reader reader; - reader.SetFileName(filename.c_str()); - reader.Read(); - const gdcm::File & file = reader.GetFile(); - const gdcm::DataSet & ds = file.GetDataSet(); + // Assert that the gdcm file is still open (we can write only if it was readed) + if (mFile == NULL) { + //assert(mFile != NULL); + FATAL("Sorry, I can write DICOM only if it was read first from a file with 'Read' function"); + } - // Check file type - //Verify if the file is a RT-Structure-Set dicom file + // Loop and update each ROI + int i=0; + for(ROIIteratorType iter = mROIs.begin(); iter != mROIs.end(); iter++) { + iter->second->UpdateDicomItem(); + i++; + } + + // Write [ Structure Set ROI Sequence ] = 0x3006,0x0020 + gdcm::DataSet & ds = mFile->GetDataSet(); + gdcm::Tag tssroisq(0x3006,0x0020); + const gdcm::DataElement &ssroisq = ds.GetDataElement( tssroisq ); + gdcm::DataElement de(ssroisq); + de.SetValue(*mROIInfoSequenceOfItems); + ds.Replace(de); + + // Write [ ROI Contour Sequence ] = 0x3006,0x0039 + DD("ici"); + gdcm::Tag troicsq(0x3006,0x0039); + const gdcm::DataElement &roicsq = ds.GetDataElement( troicsq ); + gdcm::DataElement de2(roicsq); + de2.SetValue(*mROIContoursSequenceOfItems); + ds.Replace(de2); + + //DEBUG + gdcm::DataSet & a = mROIContoursSequenceOfItems->GetItem(1).GetNestedDataSet(); + gdcm::Tag tcsq(0x3006,0x0040); + const gdcm::DataElement& csq = a.GetDataElement( tcsq ); + gdcm::SmartPointer sqi2 = csq.GetValueAsSQ(); + gdcm::Item & j = sqi2->GetItem(1); + gdcm::DataSet & b = j.GetNestedDataSet(); + gdcm::Attribute<0x3006,0x0050> at; + gdcm::Tag tcontourdata(0x3006,0x0050); + gdcm::DataElement contourdata = b.GetDataElement( tcontourdata ); + at.SetFromDataElement( contourdata ); + const double* points = at.GetValues(); + DD(points[0]); + + + // Write dicom + gdcm::Writer writer; + //writer.CheckFileMetaInformationOff(); + writer.SetFileName(filename.c_str()); + writer.SetFile(*mFile); + DD("before write"); + writer.Write(); + DD("End write"); +#else + FATAL("Sorry not compatible with GDCM1, use GDCM2"); +#endif +} +//-------------------------------------------------------------------- + + +//-------------------------------------------------------------------- +void clitk::DicomRT_StructureSet::Read(const std::string & filename) +{ + +//Try to avoid to use extern GDCM library + + //check the RS file is available before conversion + gdcm::Reader RTreader; + RTreader.SetFileName( filename.c_str() ); + if( !RTreader.Read() ) + { + std::cout << "Problem reading file: " << filename << std::endl; + return; + } + + const gdcm::DataSet& ds = RTreader.GetFile().GetDataSet(); + gdcm::MediaStorage ms; - ms.SetFromFile(file); - if( ms != gdcm::MediaStorage::RTStructureSetStorage ) - { - std::cerr << "Error. the file " << filename - << " is not a Dicom Struct ? (must have a SOP Class UID [0008|0016] = 1.2.840.10008.5.1.4.1.1.481.3 ==> [RT Structure Set Storage])" - << std::endl; - exit(0); - } + ms.SetFromFile( RTreader.GetFile() ); + + // (3006,0020) SQ (Sequence with explicit length #=4) # 370, 1 StructureSetROISequence + gdcm::Tag tssroisq(0x3006,0x0020); + if( !ds.FindDataElement( tssroisq ) ) + { + std::cout << "Problem locating 0x3006,0x0020 - Is this a valid RT Struct file?" << std::endl; + return; + } + gdcm::Tag troicsq(0x3006,0x0039); + if( !ds.FindDataElement( troicsq ) ) + { + std::cout << "Problem locating 0x3006,0x0039 - Is this a valid RT Struct file?" << std::endl; + return; + } - gdcm::Attribute<0x8,0x60> modality; - modality.SetFromDataSet( ds ); - if( modality.GetValue() != "RTSTRUCT" ) - { - std::cerr << "Error. the file " << filename - << " is not a Dicom Struct ? (must have 0x0008,0x0060 = RTSTRUCT [RT Structure Set Storage])" - << std::endl; - exit(0); - } + const gdcm::DataElement &roicsq = ds.GetDataElement( troicsq ); + gdcm::SmartPointer sqi = roicsq.GetValueAsSQ(); + if( !sqi || !sqi->GetNumberOfItems() ) + { + return; + } + const gdcm::DataElement &ssroisq = ds.GetDataElement( tssroisq ); + gdcm::SmartPointer ssqi = ssroisq.GetValueAsSQ(); + if( !ssqi || !ssqi->GetNumberOfItems() ) + { + return; + } + // Read global info gdcm::Attribute<0x20,0x10> studyid; studyid.SetFromDataSet( ds ); @@ -196,12 +377,18 @@ void clitk::DicomRT_StructureSet::Read(const std::string & filename) mName = atname.GetValue(); mTime = time.GetValue(); + // Temporary store the list of items + std::map mMapOfROIInfo; + std::map mMapOfROIContours; + + //---------------------------------- // Read all ROI Names and number // 0x3006,0x0020 = [ Structure Set ROI Sequence ] - gdcm::Tag tssroisq(0x3006,0x0020); - const gdcm::DataElement &ssroisq = ds.GetDataElement( tssroisq ); - gdcm::SmartPointer roi_seq = ssroisq.GetValueAsSQ(); + //gdcm::Tag tssroisq(0x3006,0x0020); + //const gdcm::DataElement &ssroisq = ds.GetDataElement( tssroisq ); + mROIInfoSequenceOfItems = ssroisq.GetValueAsSQ(); + gdcm::SmartPointer & roi_seq = mROIInfoSequenceOfItems; assert(roi_seq); // TODO error message for(unsigned int ridx = 0; ridx < roi_seq->GetNumberOfItems(); ++ridx) { @@ -210,13 +397,13 @@ void clitk::DicomRT_StructureSet::Read(const std::string & filename) gdcm::Attribute<0x3006,0x26> roiname; roiname.SetFromDataSet( nestedds ); - std::string name = roiname.GetValue(); // 0x3006,0x0026 = [ROI Name] - gdcm::Attribute<0x3006,0x0022> roinumber; - roinumber.SetFromDataSet( nestedds ); - int nb = roinumber.GetValue(); // 0x3006,0x0022 = [ROI Number] - // Change number if needed + std::string name = roiname.GetValue(); // 0x3006,0x0026 = [ROI Name] - //TODO + // 0x3006,0x0022 = [ROI Number] + int nb = ReadROINumber(item); + + // Store the item + mMapOfROIInfo[nb] = &item; // Check if such a number already exist if (mMapOfROIName.find(nb) != mMapOfROIName.end()) { @@ -226,97 +413,81 @@ void clitk::DicomRT_StructureSet::Read(const std::string & filename) // Add in map mMapOfROIName[nb] = name; } - // DD(mMapOfROIName.size()); //---------------------------------- - // Read all ROI + // Read all ROI item // 0x3006,0x0039 = [ ROI Contour Sequence ] - gdcm::Tag troicsq(0x3006,0x0039); - const gdcm::DataElement &roicsq = ds.GetDataElement( troicsq ); + //gdcm::Tag troicsq(0x3006,0x0039); + //const gdcm::DataElement &roicsq = ds.GetDataElement( troicsq ); gdcm::SmartPointer roi_contour_seq = roicsq.GetValueAsSQ(); + mROIContoursSequenceOfItems = roi_contour_seq; assert(roi_contour_seq); // TODO error message - int n=0; - for(unsigned int ridx = 0; ridx < roi_contour_seq->GetNumberOfItems(); ++ridx) - { + for(unsigned int ridx = 0; ridx < roi_contour_seq->GetNumberOfItems(); ++ridx) { gdcm::Item & item = roi_contour_seq->GetItem( ridx + 1); // Item starts at 1 + // ROI number [Referenced ROI Number] + const gdcm::DataSet& nestedds = item.GetNestedDataSet(); + gdcm::Attribute<0x3006,0x0084> referencedroinumber; + referencedroinumber.SetFromDataSet( nestedds ); + int nb = referencedroinumber.GetValue(); + // Store the item + mMapOfROIContours[nb] = &item; + } - DicomRT_ROI::Pointer roi = DicomRT_ROI::New(); - roi->Read(mMapOfROIName, item); - mListOfROI.push_back(roi); - mMapOfROIIndex[roi->GetROINumber()] = n; - n++; - } + //---------------------------------- + // Create the ROIs + for(std::map::iterator i = mMapOfROIInfo.begin(); i != mMapOfROIInfo.end(); i++) { + int nb = i->first;//ReadROINumber(i);//mROIIndex[i]; + // Create the roi + mROIs[nb] = DicomRT_ROI::New(); + mROIs[nb]->Read(mMapOfROIInfo[nb], mMapOfROIContours[nb]); + } + + return; -#else - gdcm::File reader; - reader.SetFileName(filename.c_str()); - reader.SetMaxSizeLoadEntry(16384); // Needed ... - reader.SetLoadMode(gdcm::LD_NOSHADOW); // don't load shadow tags (in order to save memory) - reader.Load(); +} +//-------------------------------------------------------------------- + +//-------------------------------------------------------------------- +bool clitk::DicomRT_StructureSet::IsDicomRTStruct(const std::string & filename) +{ + // Open DICOM +#if GDCM_MAJOR_VERSION == 2 + // Read gdcm file + mReader = new gdcm::Reader; + mReader->SetFileName(filename.c_str()); + mReader->Read(); + mFile = &(mReader->GetFile()); + const gdcm::DataSet & ds = mFile->GetDataSet(); + // Check file type //Verify if the file is a RT-Structure-Set dicom file - if (!gdcm::Util::DicomStringEqual(reader.GetEntryValue(0x0008,0x0016),"1.2.840.10008.5.1.4.1.1.481.3")) { //SOP clas UID - std::cerr << "Error. the file " << filename - << " is not a Dicom Struct ? (must have a SOP Class UID [0008|0016] = 1.2.840.10008.5.1.4.1.1.481.3 ==> [RT Structure Set Storage])" - << std::endl; - exit(0); - } - if (!gdcm::Util::DicomStringEqual(reader.GetEntryValue(0x0008,0x0060),"RTSTRUCT")) { //SOP clas UID - std::cerr << "Error. the file " << filename - << " is not a Dicom Struct ? (must have 0x0008,0x0060 = RTSTRUCT [RT Structure Set Storage])" - << std::endl; - exit(0); - } - - // Read global info - mStudyID = reader.GetValEntry(0x0020,0x0010)->GetValue(); - mStudyTime = reader.GetValEntry(0x008,0x0020)->GetValue(); - mStudyDate = reader.GetValEntry(0x008,0x0030)->GetValue(); - mLabel = reader.GetValEntry(0x3006,0x002)->GetValue(); - if (!reader.GetValEntry(0x3006,0x004)) { - mName = "Anonymous"; - } - else { - mName = reader.GetValEntry(0x3006,0x004)->GetValue(); - } - mTime = reader.GetValEntry(0x3006,0x009)->GetValue(); - - //---------------------------------- - // Read all ROI Names and number - // 0x3006,0x0020 = [ Structure Set ROI Sequence ] - gdcm::SeqEntry * roi_seq=reader.GetSeqEntry(0x3006,0x0020); - assert(roi_seq); // TODO error message - for (gdcm::SQItem* r=roi_seq->GetFirstSQItem(); r!=0; r=roi_seq->GetNextSQItem()) { - std::string name = r->GetEntryValue(0x3006,0x0026); // 0x3006,0x0026 = [ROI Name] - int nb = atoi(r->GetEntryValue(0x3006,0x0022).c_str()); // 0x3006,0x0022 = [ROI Number] - // Change number if needed + gdcm::MediaStorage ms; + ms.SetFromFile(*mFile); + if( ms != gdcm::MediaStorage::RTStructureSetStorage ) return false; - //TODO + gdcm::Attribute<0x8,0x60> modality; + modality.SetFromDataSet( ds ); + if( modality.GetValue() != "RTSTRUCT" ) return false; + + return true; - // Check if such a number already exist - if (mMapOfROIName.find(nb) != mMapOfROIName.end()) { - std::cerr << "WARNING. A Roi already exist with the number " - << nb << ". I replace." << std::endl; - } - // Add in map - mMapOfROIName[nb] = name; - } - // DD(mMapOfROIName.size()); + //---------------------------------------------------------------------------------------- +#else + mFile = new gdcm::File; + mFile->SetFileName(filename.c_str()); + mFile->SetMaxSizeLoadEntry(16384); // Needed ... + mFile->SetLoadMode(gdcm::LD_NOSHADOW); // don't load shadow tags (in order to save memory) + mFile->Load(); + + // Check file type + //Verify if the file is a RT-Structure-Set dicom file + if (!gdcm::Util::DicomStringEqual(mFile->GetEntryValue(0x0008,0x0016),"1.2.840.10008.5.1.4.1.1.481.3")) + return false; + if (!gdcm::Util::DicomStringEqual(mFile->GetEntryValue(0x0008,0x0060),"RTSTRUCT")) + return false; - //---------------------------------- - // Read all ROI - // 0x3006,0x0039 = [ ROI Contour Sequence ] - gdcm::SeqEntry * roi_contour_seq=reader.GetSeqEntry(0x3006,0x0039); - assert(roi_contour_seq); // TODO error message - int n=0; - for (gdcm::SQItem* r=roi_contour_seq->GetFirstSQItem(); r!=0; r=roi_contour_seq->GetNextSQItem()) { - DicomRT_ROI::Pointer roi = DicomRT_ROI::New(); - roi->Read(mMapOfROIName, r); - mListOfROI.push_back(roi); - mMapOfROIIndex[roi->GetROINumber()] = n; - n++; - } + return true; #endif } @@ -326,22 +497,19 @@ void clitk::DicomRT_StructureSet::Read(const std::string & filename) //-------------------------------------------------------------------- int clitk::DicomRT_StructureSet::AddBinaryImageAsNewROI(vvImage * im, std::string n) { - //DD("AddBinaryImageAsNewROI"); // Search max ROI number int max = -1; - for(unsigned int i=0; iGetROINumber() > max) - max = mListOfROI[i]->GetROINumber(); + for(ROIConstIteratorType iter = mROIs.begin(); iter != mROIs.end(); iter++) { + // for(unsigned int i=0; isecond; + if (roi->GetROINumber() > max) + max = roi->GetROINumber(); } - // DD(max); ++max; - //DD(max); // Compute name std::ostringstream oss; oss << vtksys::SystemTools::GetFilenameName(vtksys::SystemTools::GetFilenameWithoutLastExtension(n)); - // << "_roi_" << max << vtksys::SystemTools::GetFilenameLastExtension(n); - //DD(oss.str()); mMapOfROIName[max] = oss.str(); // Set color @@ -353,9 +521,7 @@ int clitk::DicomRT_StructureSet::AddBinaryImageAsNewROI(vvImage * im, std::strin // Create ROI DicomRT_ROI::Pointer roi = DicomRT_ROI::New(); roi->SetFromBinaryImage(im, max, oss.str(), color, n); - mListOfROI.push_back(roi); - mMapOfROIIndex[mListOfROI.size()-1] = max; - //DD(mMapOfROIIndex[mListOfROI.size()-1]); + mROIs[max] = roi; return max; } //--------------------------------------------------------------------