Mercurial > hg > orthanc-stone

comparison Framework/Loaders/LoaderStateMachine.cpp @ 815:df442f1ba0c6

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
reorganization
author Sebastien Jodogne <s.jodogne@gmail.com>
date Tue, 28 May 2019 21:59:20 +0200
parents
children 47fc7919977d
comparison
equal deleted inserted replaced
814:aead999345e0 815:df442f1ba0c6
1 /**
2 * Stone of Orthanc
3 * Copyright (C) 2012-2016 Sebastien Jodogne, Medical Physics
4 * Department, University Hospital of Liege, Belgium
5 * Copyright (C) 2017-2019 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 Affero General Public License
9 * as published by the Free Software Foundation, either version 3 of
10 * the License, or (at your option) any later version.
11 *
12 * This program is distributed in the hope that it will be useful, but
13 * WITHOUT ANY WARRANTY; without even the implied warranty of
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15 * Affero General Public License for more details.
16 *
17 * You should have received a copy of the GNU Affero General Public License
18 * along with this program. If not, see <http://www.gnu.org/licenses/>.
19 **/
20
21
22 #include "LoaderStateMachine.h"
23
24 #include <Core/OrthancException.h>
25
26 namespace OrthancStone
27 {
28 void LoaderStateMachine::State::Handle(const OrthancRestApiCommand::SuccessMessage& message)
29 {
30 throw Orthanc::OrthancException(Orthanc::ErrorCode_NotImplemented);
31 }
32
33
34 void LoaderStateMachine::State::Handle(const GetOrthancImageCommand::SuccessMessage& message)
35 {
36 throw Orthanc::OrthancException(Orthanc::ErrorCode_NotImplemented);
37 }
38
39
40 void LoaderStateMachine::State::Handle(const GetOrthancWebViewerJpegCommand::SuccessMessage& message)
41 {
42 throw Orthanc::OrthancException(Orthanc::ErrorCode_NotImplemented);
43 }
44
45
46 void LoaderStateMachine::Schedule(OracleCommandWithPayload* command)
47 {
48 std::auto_ptr<OracleCommandWithPayload> protection(command);
49
50 if (command == NULL)
51 {
52 throw Orthanc::OrthancException(Orthanc::ErrorCode_NullPointer);
53 }
54
55 if (!command->HasPayload())
56 {
57 throw Orthanc::OrthancException(Orthanc::ErrorCode_ParameterOutOfRange,
58 "The payload must contain the next state");
59 }
60
61 pendingCommands_.push_back(protection.release());
62 Step();
63 }
64
65
66 void LoaderStateMachine::Start()
67 {
68 if (active_)
69 {
70 throw Orthanc::OrthancException(Orthanc::ErrorCode_BadSequenceOfCalls);
71 }
72
73 active_ = true;
74
75 for (size_t i = 0; i < simultaneousDownloads_; i++)
76 {
77 Step();
78 }
79 }
80
81
82 void LoaderStateMachine::Step()
83 {
84 if (!pendingCommands_.empty() &&
85 activeCommands_ < simultaneousDownloads_)
86 {
87 oracle_.Schedule(*this, pendingCommands_.front());
88 pendingCommands_.pop_front();
89
90 activeCommands_++;
91 }
92 }
93
94
95 void LoaderStateMachine::Clear()
96 {
97 for (PendingCommands::iterator it = pendingCommands_.begin();
98 it != pendingCommands_.end(); ++it)
99 {
100 delete *it;
101 }
102
103 pendingCommands_.clear();
104 }
105
106
107 void LoaderStateMachine::HandleExceptionMessage(const OracleCommandExceptionMessage& message)
108 {
109 LOG(ERROR) << "Error in the state machine, stopping all processing";
110 Clear();
111 }
112
113
114 template <typename T>
115 void LoaderStateMachine::HandleSuccessMessage(const T& message)
116 {
117 assert(activeCommands_ > 0);
118 activeCommands_--;
119
120 try
121 {
122 dynamic_cast<State&>(message.GetOrigin().GetPayload()).Handle(message);
123 Step();
124 }
125 catch (Orthanc::OrthancException& e)
126 {
127 LOG(ERROR) << "Error in the state machine, stopping all processing: " << e.What();
128 Clear();
129 }
130 }
131
132
133 LoaderStateMachine::LoaderStateMachine(IOracle& oracle,
134 IObservable& oracleObservable) :
135 IObserver(oracleObservable.GetBroker()),
136 oracle_(oracle),
137 active_(false),
138 simultaneousDownloads_(4),
139 activeCommands_(0)
140 {
141 oracleObservable.RegisterObserverCallback(
142 new Callable<LoaderStateMachine, OrthancRestApiCommand::SuccessMessage>
143 (*this, &LoaderStateMachine::HandleSuccessMessage));
144
145 oracleObservable.RegisterObserverCallback(
146 new Callable<LoaderStateMachine, GetOrthancImageCommand::SuccessMessage>
147 (*this, &LoaderStateMachine::HandleSuccessMessage));
148
149 oracleObservable.RegisterObserverCallback(
150 new Callable<LoaderStateMachine, GetOrthancWebViewerJpegCommand::SuccessMessage>
151 (*this, &LoaderStateMachine::HandleSuccessMessage));
152
153 oracleObservable.RegisterObserverCallback(
154 new Callable<LoaderStateMachine, OracleCommandExceptionMessage>
155 (*this, &LoaderStateMachine::HandleExceptionMessage));
156 }
157
158
159 void LoaderStateMachine::SetSimultaneousDownloads(unsigned int count)
160 {
161 if (active_)
162 {
163 throw Orthanc::OrthancException(Orthanc::ErrorCode_BadSequenceOfCalls);
164 }
165 else if (count == 0)
166 {
167 throw Orthanc::OrthancException(Orthanc::ErrorCode_ParameterOutOfRange);
168 }
169 else
170 {
171 simultaneousDownloads_ = count;
172 }
173 }
174 }