1 #include <creaImageIODicomImageReader2.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 std::map<std::string,std::string>::iterator i;
130 for (i=attr.begin();i!=attr.end();++i)
132 if ( i->first == "D0004_1500" )
134 boost::filesystem::path full_path(filename);
135 std::string f = full_path.leaf();
138 else if ( i->first == "FullFileName" )
140 i->second = filename;
142 else if ( i->first == "FullFileDirectory" )
144 std::string::size_type last_pos = filename.find_last_of("//");
145 //find first separator
146 i->second = filename.substr(0, last_pos);
153 tree::AttributeDescriptor::GetDicomGroupElementFromKey(i->first,gr,el);
154 if ( ( gr!=0 ) && ( el!=0 ) )
157 std::string val = GetStringValueFromTag(reader.GetFile().GetDataSet().GetDataElement(gdcm::Tag(gr,el)));
158 i->second = irclean(val);
165 void DicomImageReader::ReadAttributes2(const std::string& filename,
166 std::map<std::string,std::string>& attr)
170 std::map<std::string,std::string>::iterator i;
171 for (i=attr.begin();i!=attr.end();++i)
173 if ( i->first == "D0004_1500" || i->first == "FullFileName" || i->first == "FullFileDirectory" )
182 tree::AttributeDescriptor::GetDicomGroupElementFromKey(i->first,gr,el);
183 mscan.AddTag(gdcm::Tag(gr,el) );
188 bool b = mscan.IsKey(filename.c_str());
191 const gdcm::Scanner::TagToValue &mapping = mscan.GetMapping(filename.c_str());
192 gdcm::Scanner::TagToValue::const_iterator it = mapping.begin();
193 std::map<std::string, std::string>::iterator i;
194 for (i=attr.begin();i!=attr.end();++i, ++it)
196 if ( i->first == "D0004_1500" )
198 boost::filesystem::path full_path(filename);
199 std::string f = full_path.leaf();
202 else if ( i->first == "FullFileName" )
204 i->second = filename;
206 else if ( i->first == "FullFileDirectory" )
208 std::string::size_type last_pos = filename.find_last_of("//");
209 //find first separator
210 i->second = filename.substr(0, last_pos);
214 const char *value = it->second;
215 i->second = irclean(it->second);
221 const std::string DicomImageReader::GetStringValueFromTag(const gdcm::DataElement& de)
223 static std::string buffer;
224 buffer = ""; // cleanup previous call
227 const gdcm::ByteValue *bv = de.GetByteValue();
228 if( bv ) // Can be Type 2
230 buffer = std::string( bv->GetPointer(), bv->GetLength() );
231 // Will be padded with at least one \0
234 // Since return is a const char* the very first \0 will be considered
235 return buffer.c_str();
237 //=====================================================================
239 } // namespace creaImageIO