1 #include <creaImageIODicomImageReader2.h>
2 #include "gdcmStringFilter.h"
5 #include <creaImageIOSystem.h>
6 #include "boost/filesystem/path.hpp"
8 #include <creaImageIOTreeAttributeDescriptor.h>
9 #include <vtkStringArray.h>
10 #include <creaImageIOGimmick.h>
19 //=====================================================================
20 DicomImageReader::DicomImageReader()
22 mReader = vtkGDCMImageReader::New();
26 //=====================================================================
28 //=====================================================================
29 DicomImageReader::~DicomImageReader()
33 //=====================================================================
35 //=====================================================================
36 bool DicomImageReader::CanRead(const std::string& filename)
39 reader.SetFileName( filename.c_str() );
43 //=====================================================================
45 //=====================================================================
46 vtkImageData* DicomImageReader::ReadImage(const std::string& filename)
51 mReader->SetFileName(filename.c_str());
53 im = vtkImageData::New();
54 im->ShallowCopy(mReader->GetOutput());
58 if (im!=0) im->Delete();
64 //=====================================================================
65 void DicomImageReader::PushBackExtensions(std::vector<std::string>& v)
71 //========================================================================
72 std::string irclean(const std::string& str)
76 if (str == "GDCM::Unfound")
80 if (str[str.size()-1]==' ')
82 return irclean(str.substr(0,str.size()-1));
84 if (str[str.size()-1]==0)
86 return irclean(str.substr(0,str.size()-1));
92 //========================================================================
93 //=====================================================================
96 void DicomImageReader::getAttributes(const std::string filename,
97 std::map <std::string , std::string> &infos, std::vector<std::string> i_attr)
100 reader.SetFileName( filename.c_str() );
103 std::vector<std::string>::iterator it = i_attr.begin();
104 for(;it != i_attr.end(); it++)
108 sscanf((*it).c_str(),"D%04hx_%04hx",&gr,&el);
109 if ( ( gr!=0 ) && ( el!=0 ) )
111 infos[(*it)] = ( GetStringValueFromTag(reader.GetFile().GetDataSet().GetDataElement(gdcm::Tag(gr,el))) );
117 //=====================================================================
118 void DicomImageReader::ReadAttributes(const std::string& filename,
119 std::map<std::string,std::string>& attr)
121 GimmickMessage(2,"Reading attributes from DICOM file '"
122 <<filename<<"'"<<std::endl);
126 reader.SetFileName( filename.c_str() );
129 gdcm::StringFilter sf;
130 sf.SetFile(reader.GetFile());
131 std::map<std::string,std::string>::iterator i;
132 for (i=attr.begin();i!=attr.end();++i)
134 if ( i->first == "D0004_1500" )
136 boost::filesystem::path full_path(filename);
137 std::string f = full_path.leaf();
140 else if ( i->first == "FullFileName" )
142 i->second = filename;
144 else if ( i->first == "FullFileDirectory" )
146 std::string::size_type last_pos = filename.find_last_of("//");
147 //find first separator
148 i->second = filename.substr(0, last_pos);
154 tree::AttributeDescriptor::GetDicomGroupElementFromKey(i->first,gr,el);
155 //if ( ( gr!=0 ) && ( el!=0 ) )
156 const gdcm::Tag tag(gr, el);
157 if( reader.GetFile().GetDataSet().FindDataElement( tag ) )
159 i->second = irclean(sf.ToString(tag));
170 void DicomImageReader::ReadAttributes2(const std::string& filename,
171 std::map<std::string,std::string>& attr)
175 std::map<std::string,std::string>::iterator i;
176 for (i=attr.begin();i!=attr.end();++i)
178 if ( i->first == "D0004_1500" || i->first == "FullFileName" || i->first == "FullFileDirectory" )
187 tree::AttributeDescriptor::GetDicomGroupElementFromKey(i->first,gr,el);
188 mscan.AddTag(gdcm::Tag(gr,el) );
193 bool b = mscan.IsKey(filename.c_str());
196 const gdcm::Scanner::TagToValue &mapping = mscan.GetMapping(filename.c_str());
197 gdcm::Scanner::TagToValue::const_iterator it = mapping.begin();
198 std::map<std::string, std::string>::iterator i;
199 for (i=attr.begin();i!=attr.end();++i, ++it)
201 if ( i->first == "D0004_1500" )
203 boost::filesystem::path full_path(filename);
204 std::string f = full_path.leaf();
207 else if ( i->first == "FullFileName" )
209 i->second = filename;
211 else if ( i->first == "FullFileDirectory" )
213 std::string::size_type last_pos = filename.find_last_of("//");
214 //find first separator
215 i->second = filename.substr(0, last_pos);
219 const char *value = it->second;
220 i->second = irclean(it->second);
226 const std::string DicomImageReader::GetStringValueFromTag(const gdcm::DataElement& de)
228 static std::string buffer;
229 buffer = ""; // cleanup previous call
232 const gdcm::ByteValue *bv = de.GetByteValue();
233 if( bv ) // Can be Type 2
235 buffer = std::string( bv->GetPointer(), bv->GetLength() );
236 // Will be padded with at least one \0
239 // Since return is a const char* the very first \0 will be considered
240 return buffer.c_str();
242 //=====================================================================
244 } // namespace creaImageIO