Mercurial > hg > orthanc-transfers

comparison Framework/DownloadArea.h @ 0:95226b754d9e

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
initial release
author Sebastien Jodogne <s.jodogne@gmail.com>
date Mon, 17 Sep 2018 11:34:55 +0200
parents
children 4c3437217518
comparison
equal deleted inserted replaced
-1:000000000000 0:95226b754d9e
1 /**
2 * Transfers accelerator plugin for Orthanc
3 * Copyright (C) 2018 Osimis, Belgium
4 *
5 * This program is free software: you can redistribute it and/or
6 * modify it under the terms of the GNU Affero General Public License
7 * as published by the Free Software Foundation, either version 3 of
8 * the License, or (at your option) any later version.
9 *
10 * This program is distributed in the hope that it will be useful, but
11 * WITHOUT ANY WARRANTY; without even the implied warranty of
12 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
13 * Affero General Public License for more details.
14 *
15 * You should have received a copy of the GNU Affero General Public License
16 * along with this program. If not, see <http://www.gnu.org/licenses/>.
17 **/
18
19
20 #pragma once
21
22 #include "TransferScheduler.h"
23
24 #include <Core/TemporaryFile.h>
25
26 namespace OrthancPlugins
27 {
28 class DownloadArea : public boost::noncopyable
29 {
30 private:
31 class Instance : public boost::noncopyable
32 {
33 private:
34 DicomInstanceInfo info_;
35 Orthanc::TemporaryFile file_;
36
37 class Writer;
38
39 public:
40 Instance(const DicomInstanceInfo& info);
41
42 const DicomInstanceInfo& GetInfo() const
43 {
44 return info_;
45 }
46
47 void WriteChunk(size_t offset,
48 const void* data,
49 size_t size);
50
51 void Commit(OrthancPluginContext* context,
52 bool simulate) const;
53 };
54
55
56 typedef std::map<std::string, Instance*> Instances;
57
58 boost::mutex mutex_;
59 Instances instances_;
60 size_t totalSize_;
61
62
63 void Clear();
64
65 Instance& LookupInstance(const std::string& id);
66
67 void WriteUncompressedBucket(const TransferBucket& bucket,
68 const void* data,
69 size_t size);
70
71 void Setup(const std::vector<DicomInstanceInfo>& instances);
72
73 void CommitInternal(OrthancPluginContext* context,
74 bool simulate);
75
76 public:
77 DownloadArea(const TransferScheduler& scheduler);
78
79 DownloadArea(const std::vector<DicomInstanceInfo>& instances)
80 {
81 Setup(instances);
82 }
83
84 ~DownloadArea()
85 {
86 Clear();
87 }
88
89 size_t GetTotalSize() const
90 {
91 return totalSize_;
92 }
93
94 void WriteBucket(const TransferBucket& bucket,
95 const void* data,
96 size_t size,
97 BucketCompression compression);
98
99 void WriteInstance(const std::string& instanceId,
100 const void* data,
101 size_t size);
102
103 void CheckMD5();
104
105 void Commit(OrthancPluginContext* context);
106 };
107 }