Mercurial > hg > orthanc-stone

comparison Resources/Orthanc/Core/MultiThreading/BagOfTasksProcessor.h @ 200:03afbee0cc7b

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
integration of Orthanc core into Stone
author Sebastien Jodogne <s.jodogne@gmail.com>
date Fri, 23 Mar 2018 11:04:03 +0100
parents
children
comparison
equal deleted inserted replaced
199:dabe9982fca3 200:03afbee0cc7b
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 "BagOfTasks.h"
37 #include "SharedMessageQueue.h"
38
39 #include <stdint.h>
40 #include <map>
41
42 namespace Orthanc
43 {
44 class BagOfTasksProcessor : public boost::noncopyable
45 {
46 private:
47 enum BagStatus
48 {
49 BagStatus_Running,
50 BagStatus_Canceled,
51 BagStatus_Failed
52 };
53
54
55 struct Bag
56 {
57 size_t size_;
58 size_t done_;
59 BagStatus status_;
60
61 Bag() :
62 size_(0),
63 done_(0),
64 status_(BagStatus_Failed)
65 {
66 }
67
68 explicit Bag(size_t size) :
69 size_(size),
70 done_(0),
71 status_(BagStatus_Running)
72 {
73 }
74 };
75
76 class Task;
77
78
79 typedef std::map<uint64_t, Bag> Bags;
80 typedef std::map<uint64_t, bool> ExitStatus;
81
82 SharedMessageQueue queue_;
83
84 boost::mutex mutex_;
85 uint64_t countBags_;
86 Bags bags_;
87 std::vector<boost::thread*> threads_;
88 ExitStatus exitStatus_;
89 bool continue_;
90
91 boost::condition_variable bagFinished_;
92
93 static void Worker(BagOfTasksProcessor* that);
94
95 void Cancel(int64_t bag);
96
97 bool Join(int64_t bag);
98
99 float GetProgress(int64_t bag);
100
101 void SignalProgress(Task& task,
102 Bag& bag);
103
104 public:
105 class Handle : public boost::noncopyable
106 {
107 friend class BagOfTasksProcessor;
108
109 private:
110 BagOfTasksProcessor& that_;
111 uint64_t bag_;
112 bool hasJoined_;
113 bool status_;
114
115 Handle(BagOfTasksProcessor& that,
116 uint64_t bag,
117 bool empty) :
118 that_(that),
119 bag_(bag),
120 hasJoined_(empty)
121 {
122 }
123
124 public:
125 ~Handle()
126 {
127 Join();
128 }
129
130 void Cancel()
131 {
132 that_.Cancel(bag_);
133 }
134
135 bool Join();
136
137 float GetProgress()
138 {
139 return that_.GetProgress(bag_);
140 }
141 };
142
143
144 explicit BagOfTasksProcessor(size_t countThreads);
145
146 ~BagOfTasksProcessor();
147
148 Handle* Submit(BagOfTasks& tasks);
149 };
150 }