Mercurial > hg > orthanc
view OrthancServer/Search/HierarchicalMatcher.cpp @ 2931:89f2c302fc37
Fix issue #114
author | Sebastien Jodogne <s.jodogne@gmail.com> |
---|---|
date | Tue, 13 Nov 2018 17:50:03 +0100 |
parents | 878b59270859 |
children | 4767d36679ed |
line wrap: on
line source
/** * Orthanc - A Lightweight, RESTful DICOM Store * Copyright (C) 2012-2016 Sebastien Jodogne, Medical Physics * Department, University Hospital of Liege, Belgium * Copyright (C) 2017-2018 Osimis S.A., Belgium * * This program is free software: you can redistribute it and/or * modify it under the terms of the GNU General Public License as * published by the Free Software Foundation, either version 3 of the * License, or (at your option) any later version. * * In addition, as a special exception, the copyright holders of this * program give permission to link the code of its release with the * OpenSSL project's "OpenSSL" library (or with modified versions of it * that use the same license as the "OpenSSL" library), and distribute * the linked executables. You must obey the GNU General Public License * in all respects for all of the code used other than "OpenSSL". If you * modify file(s) with this exception, you may extend this exception to * your version of the file(s), but you are not obligated to do so. If * you do not wish to do so, delete this exception statement from your * version. If you delete this exception statement from all source files * in the program, then also delete it here. * * This program is distributed in the hope that it will be useful, but * WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU * General Public License for more details. * * You should have received a copy of the GNU General Public License * along with this program. If not, see <http://www.gnu.org/licenses/>. **/ #include "../PrecompiledHeadersServer.h" #include "HierarchicalMatcher.h" #include "../../Core/Logging.h" #include "../../Core/OrthancException.h" #include "../../Core/DicomParsing/FromDcmtkBridge.h" #include "../../Core/DicomParsing/ToDcmtkBridge.h" #include "../OrthancInitialization.h" #include <dcmtk/dcmdata/dcfilefo.h> namespace Orthanc { HierarchicalMatcher::HierarchicalMatcher(ParsedDicomFile& query) { Setup(*query.GetDcmtkObject().getDataset(), Configuration::GetGlobalBoolParameter("CaseSensitivePN", false), query.GetEncoding()); } HierarchicalMatcher::~HierarchicalMatcher() { for (Constraints::iterator it = constraints_.begin(); it != constraints_.end(); ++it) { if (it->second != NULL) { delete it->second; } } for (Sequences::iterator it = sequences_.begin(); it != sequences_.end(); ++it) { if (it->second != NULL) { delete it->second; } } } void HierarchicalMatcher::Setup(DcmItem& dataset, bool caseSensitivePN, Encoding encoding) { for (unsigned long i = 0; i < dataset.card(); i++) { DcmElement* element = dataset.getElement(i); if (element == NULL) { throw OrthancException(ErrorCode_InternalError); } DicomTag tag(FromDcmtkBridge::Convert(element->getTag())); if (tag == DICOM_TAG_SPECIFIC_CHARACTER_SET || // Ignore encoding tag.GetElement() == 0x0000) // Ignore all "Group Length" tags { continue; } ValueRepresentation vr = FromDcmtkBridge::LookupValueRepresentation(tag); if (constraints_.find(tag) != constraints_.end() || sequences_.find(tag) != sequences_.end()) { throw OrthancException(ErrorCode_BadRequest); } if (vr == ValueRepresentation_Sequence) { DcmSequenceOfItems& sequence = dynamic_cast<DcmSequenceOfItems&>(*element); if (sequence.card() == 0 || (sequence.card() == 1 && sequence.getItem(0)->card() == 0)) { // Universal matching of a sequence sequences_[tag] = NULL; } else if (sequence.card() == 1) { sequences_[tag] = new HierarchicalMatcher(*sequence.getItem(0), caseSensitivePN, encoding); } else { throw OrthancException(ErrorCode_BadRequest); } } else { std::set<DicomTag> ignoreTagLength; std::auto_ptr<DicomValue> value(FromDcmtkBridge::ConvertLeafElement (*element, DicomToJsonFlags_None, ORTHANC_MAXIMUM_TAG_LENGTH, encoding, ignoreTagLength)); if (value->IsBinary()) { if (!value->GetContent().empty()) { LOG(WARNING) << "This C-Find modality worklist query contains a non-empty tag (" << tag.Format() << ") with UN (unknown) value representation. " << "It will be ignored."; } constraints_[tag] = NULL; } else if (value->IsNull() || value->GetContent().empty()) { // This is an universal matcher constraints_[tag] = NULL; } else { // DICOM specifies that searches must be case sensitive, except // for tags with a PN value representation bool sensitive = true; if (vr == ValueRepresentation_PersonName) { sensitive = caseSensitivePN; } constraints_[tag] = IFindConstraint::ParseDicomConstraint(tag, value->GetContent(), sensitive); } } } } std::string HierarchicalMatcher::Format(const std::string& prefix) const { std::string s; for (Constraints::const_iterator it = constraints_.begin(); it != constraints_.end(); ++it) { s += prefix + it->first.Format() + " "; if (it->second == NULL) { s += "*\n"; } else { s += it->second->Format() + "\n"; } } for (Sequences::const_iterator it = sequences_.begin(); it != sequences_.end(); ++it) { s += prefix + it->first.Format() + " "; if (it->second == NULL) { s += "*\n"; } else { s += "Sequence:\n" + it->second->Format(prefix + " "); } } return s; } bool HierarchicalMatcher::Match(ParsedDicomFile& dicom) const { return MatchInternal(*dicom.GetDcmtkObject().getDataset(), dicom.GetEncoding()); } bool HierarchicalMatcher::MatchInternal(DcmItem& item, Encoding encoding) const { for (Constraints::const_iterator it = constraints_.begin(); it != constraints_.end(); ++it) { if (it->second != NULL) { DcmTagKey tag = ToDcmtkBridge::Convert(it->first); DcmElement* element = NULL; if (!item.findAndGetElement(tag, element).good() || element == NULL) { return false; } std::set<DicomTag> ignoreTagLength; std::auto_ptr<DicomValue> value(FromDcmtkBridge::ConvertLeafElement (*element, DicomToJsonFlags_None, ORTHANC_MAXIMUM_TAG_LENGTH, encoding, ignoreTagLength)); if (value->IsNull() || value->IsBinary() || !it->second->Match(value->GetContent())) { return false; } } } for (Sequences::const_iterator it = sequences_.begin(); it != sequences_.end(); ++it) { if (it->second != NULL) { DcmTagKey tag = ToDcmtkBridge::Convert(it->first); DcmSequenceOfItems* sequence = NULL; if (!item.findAndGetSequence(tag, sequence).good() || sequence == NULL) { continue; } bool match = false; for (unsigned long i = 0; i < sequence->card(); i++) { if (it->second->MatchInternal(*sequence->getItem(i), encoding)) { match = true; break; } } if (!match) { return false; } } } return true; } DcmDataset* HierarchicalMatcher::ExtractInternal(DcmItem& source, Encoding encoding) const { std::auto_ptr<DcmDataset> target(new DcmDataset); for (Constraints::const_iterator it = constraints_.begin(); it != constraints_.end(); ++it) { DcmTagKey tag = ToDcmtkBridge::Convert(it->first); DcmElement* element = NULL; if (source.findAndGetElement(tag, element).good() && element != NULL) { std::auto_ptr<DcmElement> cloned(FromDcmtkBridge::CreateElementForTag(it->first)); cloned->copyFrom(*element); target->insert(cloned.release()); } } for (Sequences::const_iterator it = sequences_.begin(); it != sequences_.end(); ++it) { DcmTagKey tag = ToDcmtkBridge::Convert(it->first); DcmSequenceOfItems* sequence = NULL; if (source.findAndGetSequence(tag, sequence).good() && sequence != NULL) { std::auto_ptr<DcmSequenceOfItems> cloned(new DcmSequenceOfItems(tag)); for (unsigned long i = 0; i < sequence->card(); i++) { if (it->second == NULL) { cloned->append(new DcmItem(*sequence->getItem(i))); } else if (it->second->MatchInternal(*sequence->getItem(i), encoding)) // TODO Might be optimized { // It is necessary to encapsulate the child dataset into a // "DcmItem" object before it can be included in a // sequence. Otherwise, "dciodvfy" reports an error "Bad // tag in sequence - Expecting Item or Sequence Delimiter." std::auto_ptr<DcmDataset> child(it->second->ExtractInternal(*sequence->getItem(i), encoding)); cloned->append(new DcmItem(*child)); } } target->insert(cloned.release()); } } return target.release(); } ParsedDicomFile* HierarchicalMatcher::Extract(ParsedDicomFile& dicom) const { std::auto_ptr<DcmDataset> dataset(ExtractInternal(*dicom.GetDcmtkObject().getDataset(), dicom.GetEncoding())); std::auto_ptr<ParsedDicomFile> result(new ParsedDicomFile(*dataset)); result->SetEncoding(dicom.GetEncoding()); return result.release(); } }