Mercurial > hg > orthanc-transfers
diff Framework/DownloadArea.h @ 0:95226b754d9e
initial release
author | Sebastien Jodogne <s.jodogne@gmail.com> |
---|---|
date | Mon, 17 Sep 2018 11:34:55 +0200 |
parents | |
children | 4c3437217518 |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/Framework/DownloadArea.h Mon Sep 17 11:34:55 2018 +0200 @@ -0,0 +1,107 @@ +/** + * Transfers accelerator plugin for Orthanc + * Copyright (C) 2018 Osimis, Belgium + * + * This program is free software: you can redistribute it and/or + * modify it under the terms of the GNU Affero General Public License + * as published by the Free Software Foundation, either version 3 of + * the License, or (at your option) any later version. + * + * 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 + * Affero General Public License for more details. + * + * You should have received a copy of the GNU Affero General Public License + * along with this program. If not, see <http://www.gnu.org/licenses/>. + **/ + + +#pragma once + +#include "TransferScheduler.h" + +#include <Core/TemporaryFile.h> + +namespace OrthancPlugins +{ + class DownloadArea : public boost::noncopyable + { + private: + class Instance : public boost::noncopyable + { + private: + DicomInstanceInfo info_; + Orthanc::TemporaryFile file_; + + class Writer; + + public: + Instance(const DicomInstanceInfo& info); + + const DicomInstanceInfo& GetInfo() const + { + return info_; + } + + void WriteChunk(size_t offset, + const void* data, + size_t size); + + void Commit(OrthancPluginContext* context, + bool simulate) const; + }; + + + typedef std::map<std::string, Instance*> Instances; + + boost::mutex mutex_; + Instances instances_; + size_t totalSize_; + + + void Clear(); + + Instance& LookupInstance(const std::string& id); + + void WriteUncompressedBucket(const TransferBucket& bucket, + const void* data, + size_t size); + + void Setup(const std::vector<DicomInstanceInfo>& instances); + + void CommitInternal(OrthancPluginContext* context, + bool simulate); + + public: + DownloadArea(const TransferScheduler& scheduler); + + DownloadArea(const std::vector<DicomInstanceInfo>& instances) + { + Setup(instances); + } + + ~DownloadArea() + { + Clear(); + } + + size_t GetTotalSize() const + { + return totalSize_; + } + + void WriteBucket(const TransferBucket& bucket, + const void* data, + size_t size, + BucketCompression compression); + + void WriteInstance(const std::string& instanceId, + const void* data, + size_t size); + + void CheckMD5(); + + void Commit(OrthancPluginContext* context); + }; +}