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 }