Mercurial > hg > orthanc
comparison OrthancServer/ServerJobs/DicomModalityStoreJob.h @ 2602:c25f1a52acbc jobs
reorganization
author | Sebastien Jodogne <s.jodogne@gmail.com> |
---|---|
date | Fri, 18 May 2018 15:34:11 +0200 |
parents | |
children | 988936118354 |
comparison
equal
deleted
inserted
replaced
2601:5b6c3d77a2a1 | 2602:c25f1a52acbc |
---|---|
1 /** | |
2 * Orthanc - A Lightweight, RESTful DICOM Store | |
3 * Copyright (C) 2012-2016 Sebastien Jodogne, Medical Physics | |
4 * Department, University Hospital of Liege, Belgium | |
5 * Copyright (C) 2017-2018 Osimis S.A., Belgium | |
6 * | |
7 * This program is free software: you can redistribute it and/or | |
8 * modify it under the terms of the GNU General Public License as | |
9 * published by the Free Software Foundation, either version 3 of the | |
10 * License, or (at your option) any later version. | |
11 * | |
12 * In addition, as a special exception, the copyright holders of this | |
13 * program give permission to link the code of its release with the | |
14 * OpenSSL project's "OpenSSL" library (or with modified versions of it | |
15 * that use the same license as the "OpenSSL" library), and distribute | |
16 * the linked executables. You must obey the GNU General Public License | |
17 * in all respects for all of the code used other than "OpenSSL". If you | |
18 * modify file(s) with this exception, you may extend this exception to | |
19 * your version of the file(s), but you are not obligated to do so. If | |
20 * you do not wish to do so, delete this exception statement from your | |
21 * version. If you delete this exception statement from all source files | |
22 * in the program, then also delete it here. | |
23 * | |
24 * This program is distributed in the hope that it will be useful, but | |
25 * WITHOUT ANY WARRANTY; without even the implied warranty of | |
26 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU | |
27 * General Public License for more details. | |
28 * | |
29 * You should have received a copy of the GNU General Public License | |
30 * along with this program. If not, see <http://www.gnu.org/licenses/>. | |
31 **/ | |
32 | |
33 | |
34 #pragma once | |
35 | |
36 #include "../../Core/JobsEngine/SetOfInstancesJob.h" | |
37 #include "../../Core/DicomNetworking/DicomUserConnection.h" | |
38 | |
39 | |
40 namespace Orthanc | |
41 { | |
42 class DicomModalityStoreJob : public SetOfInstancesJob | |
43 { | |
44 private: | |
45 ServerContext& context_; | |
46 std::string localAet_; | |
47 RemoteModalityParameters remote_; | |
48 std::string moveOriginatorAet_; | |
49 uint16_t moveOriginatorId_; | |
50 std::auto_ptr<DicomUserConnection> connection_; | |
51 | |
52 void OpenConnection() | |
53 { | |
54 if (connection_.get() == NULL) | |
55 { | |
56 connection_.reset(new DicomUserConnection); | |
57 connection_->SetLocalApplicationEntityTitle(localAet_); | |
58 connection_->SetRemoteModality(remote_); | |
59 } | |
60 } | |
61 | |
62 protected: | |
63 virtual bool HandleInstance(const std::string& instance) | |
64 { | |
65 OpenConnection(); | |
66 | |
67 LOG(INFO) << "Sending instance " << instance << " to modality \"" | |
68 << remote_.GetApplicationEntityTitle() << "\""; | |
69 | |
70 std::string dicom; | |
71 context_.ReadDicom(dicom, instance); | |
72 | |
73 if (HasMoveOriginator()) | |
74 { | |
75 connection_->Store(dicom, moveOriginatorAet_, moveOriginatorId_); | |
76 } | |
77 else | |
78 { | |
79 connection_->Store(dicom); | |
80 } | |
81 | |
82 //boost::this_thread::sleep(boost::posix_time::milliseconds(500)); | |
83 | |
84 return true; | |
85 } | |
86 | |
87 public: | |
88 DicomModalityStoreJob(ServerContext& context) : | |
89 context_(context), | |
90 localAet_("ORTHANC"), | |
91 moveOriginatorId_(0) // By default, not a C-MOVE | |
92 { | |
93 } | |
94 | |
95 const std::string& GetLocalAet() const | |
96 { | |
97 return localAet_; | |
98 } | |
99 | |
100 void SetLocalAet(const std::string& aet) | |
101 { | |
102 if (IsStarted()) | |
103 { | |
104 throw OrthancException(ErrorCode_BadSequenceOfCalls); | |
105 } | |
106 else | |
107 { | |
108 localAet_ = aet; | |
109 } | |
110 } | |
111 | |
112 const RemoteModalityParameters& GetRemoteModality() const | |
113 { | |
114 return remote_; | |
115 } | |
116 | |
117 void SetRemoteModality(const RemoteModalityParameters& remote) | |
118 { | |
119 if (IsStarted()) | |
120 { | |
121 throw OrthancException(ErrorCode_BadSequenceOfCalls); | |
122 } | |
123 else | |
124 { | |
125 remote_ = remote; | |
126 } | |
127 } | |
128 | |
129 bool HasMoveOriginator() const | |
130 { | |
131 return moveOriginatorId_ != 0; | |
132 } | |
133 | |
134 const std::string& GetMoveOriginatorAet() const | |
135 { | |
136 if (HasMoveOriginator()) | |
137 { | |
138 return moveOriginatorAet_; | |
139 } | |
140 else | |
141 { | |
142 throw OrthancException(ErrorCode_BadSequenceOfCalls); | |
143 } | |
144 } | |
145 | |
146 uint16_t GetMoveOriginatorId() const | |
147 { | |
148 if (HasMoveOriginator()) | |
149 { | |
150 return moveOriginatorId_; | |
151 } | |
152 else | |
153 { | |
154 throw OrthancException(ErrorCode_BadSequenceOfCalls); | |
155 } | |
156 } | |
157 | |
158 void SetMoveOriginator(const std::string& aet, | |
159 int id) | |
160 { | |
161 if (IsStarted()) | |
162 { | |
163 throw OrthancException(ErrorCode_BadSequenceOfCalls); | |
164 } | |
165 else if (id < 0 || | |
166 id >= 65536) | |
167 { | |
168 throw OrthancException(ErrorCode_ParameterOutOfRange); | |
169 } | |
170 else | |
171 { | |
172 moveOriginatorId_ = static_cast<uint16_t>(id); | |
173 moveOriginatorAet_ = aet; | |
174 } | |
175 } | |
176 | |
177 virtual void ReleaseResources() // For pausing jobs | |
178 { | |
179 connection_.reset(NULL); | |
180 } | |
181 | |
182 virtual void GetJobType(std::string& target) | |
183 { | |
184 target = "DicomModalityStore"; | |
185 } | |
186 | |
187 virtual void GetPublicContent(Json::Value& value) | |
188 { | |
189 value["LocalAet"] = localAet_; | |
190 value["RemoteAet"] = remote_.GetApplicationEntityTitle(); | |
191 | |
192 if (HasMoveOriginator()) | |
193 { | |
194 value["MoveOriginatorAET"] = GetMoveOriginatorAet(); | |
195 value["MoveOriginatorID"] = GetMoveOriginatorId(); | |
196 } | |
197 | |
198 value["InstancesCount"] = static_cast<uint32_t>(GetInstances().size()); | |
199 value["FailedInstancesCount"] = static_cast<uint32_t>(GetFailedInstances().size()); | |
200 } | |
201 }; | |
202 } |