Mercurial > hg > orthanc-stone

annotate RenderingPlugin/Sources/Plugin.cpp @ 1887:aa4ed1cf4e8d

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
refactoring using DataAugmentationParameters
author Sebastien Jodogne <s.jodogne@gmail.com>
date Mon, 17 Jan 2022 12:38:04 +0100
parents ca89fec8c48f
children 9bdce2c91620
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
1877
a2955abe4c2e skeleton for the RenderingPlugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
1 /**
a2955abe4c2e skeleton for the RenderingPlugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
2 * Stone of Orthanc
a2955abe4c2e skeleton for the RenderingPlugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
3 * Copyright (C) 2012-2016 Sebastien Jodogne, Medical Physics
a2955abe4c2e skeleton for the RenderingPlugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
4 * Department, University Hospital of Liege, Belgium
a2955abe4c2e skeleton for the RenderingPlugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
5 * Copyright (C) 2017-2022 Osimis S.A., Belgium
a2955abe4c2e skeleton for the RenderingPlugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
6 * Copyright (C) 2021-2022 Sebastien Jodogne, ICTEAM UCLouvain, Belgium
a2955abe4c2e skeleton for the RenderingPlugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
7 *
a2955abe4c2e skeleton for the RenderingPlugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
8 * This program is free software: you can redistribute it and/or
a2955abe4c2e skeleton for the RenderingPlugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
9 * modify it under the terms of the GNU General Public License as
a2955abe4c2e skeleton for the RenderingPlugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
10 * published by the Free Software Foundation, either version 3 of the
a2955abe4c2e skeleton for the RenderingPlugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
11 * License, or (at your option) any later version.
a2955abe4c2e skeleton for the RenderingPlugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
12 *
a2955abe4c2e skeleton for the RenderingPlugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
13 * This program is distributed in the hope that it will be useful, but
a2955abe4c2e skeleton for the RenderingPlugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
14 * WITHOUT ANY WARRANTY; without even the implied warranty of
a2955abe4c2e skeleton for the RenderingPlugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
15 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
a2955abe4c2e skeleton for the RenderingPlugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
16 * General Public License for more details.
a2955abe4c2e skeleton for the RenderingPlugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
17 *
a2955abe4c2e skeleton for the RenderingPlugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
18 * You should have received a copy of the GNU General Public License
a2955abe4c2e skeleton for the RenderingPlugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
19 * along with this program. If not, see <http://www.gnu.org/licenses/>.
a2955abe4c2e skeleton for the RenderingPlugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
20 **/
a2955abe4c2e skeleton for the RenderingPlugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
21
a2955abe4c2e skeleton for the RenderingPlugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
22
1885
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
23 #include "OrthancPluginConnection.h"
1877
a2955abe4c2e skeleton for the RenderingPlugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
24 #include "../Resources/Orthanc/Plugins/OrthancPluginCppWrapper.h"
a2955abe4c2e skeleton for the RenderingPlugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
25
1882
4e80b8afd0da numpy in rendering plugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1877
diff changeset
26 #include "../../OrthancStone/Sources/Toolbox/AffineTransform2D.h"
1883
fba6e550e0ee apply rescale slope/intercept in rendering plugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1882
diff changeset
27 #include "../../OrthancStone/Sources/Toolbox/DicomInstanceParameters.h"
1885
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
28 #include "../../OrthancStone/Sources/Toolbox/DicomStructureSet.h"
1882
4e80b8afd0da numpy in rendering plugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1877
diff changeset
29
1877
a2955abe4c2e skeleton for the RenderingPlugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
30 #include <EmbeddedResources.h>
a2955abe4c2e skeleton for the RenderingPlugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
31
1882
4e80b8afd0da numpy in rendering plugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1877
diff changeset
32 #include <Images/Image.h>
4e80b8afd0da numpy in rendering plugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1877
diff changeset
33 #include <Images/ImageProcessing.h>
4e80b8afd0da numpy in rendering plugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1877
diff changeset
34 #include <Images/NumpyWriter.h>
1877
a2955abe4c2e skeleton for the RenderingPlugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
35 #include <Logging.h>
1884
0c6923982fdd rendering options from GET args
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1883
diff changeset
36 #include <SerializationToolbox.h>
1885
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
37 #include <Toolbox.h>
1877
a2955abe4c2e skeleton for the RenderingPlugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
38
1882
4e80b8afd0da numpy in rendering plugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1877
diff changeset
39 #include <boost/math/constants/constants.hpp>
4e80b8afd0da numpy in rendering plugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1877
diff changeset
40
4e80b8afd0da numpy in rendering plugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1877
diff changeset
41
1885
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
42 class DicomStructureCache : public boost::noncopyable
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
43 {
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
44 private:
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
45 boost::mutex mutex_;
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
46 std::string instanceId_;
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
47 std::unique_ptr<OrthancStone::DicomStructureSet> rtstruct_;
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
48
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
49 DicomStructureCache() // Singleton design pattern
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
50 {
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
51 }
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
52
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
53 public:
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
54 void Invalidate(const std::string& instanceId)
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
55 {
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
56 boost::mutex::scoped_lock lock(mutex_);
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
57
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
58 if (instanceId_ == instanceId)
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
59 {
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
60 rtstruct_.reset(NULL);
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
61 }
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
62 }
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
63
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
64 static DicomStructureCache& GetSingleton()
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
65 {
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
66 static DicomStructureCache instance;
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
67 return instance;
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
68 }
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
69
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
70 class Accessor : public boost::noncopyable
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
71 {
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
72 private:
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
73 boost::mutex::scoped_lock lock_;
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
74 std::string instanceId_;
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
75 const OrthancStone::DicomStructureSet* rtstruct_;
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
76
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
77 public:
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
78 Accessor(DicomStructureCache& that,
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
79 const std::string& instanceId) :
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
80 lock_(that.mutex_),
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
81 instanceId_(instanceId),
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
82 rtstruct_(NULL)
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
83 {
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
84 if (that.instanceId_ == instanceId &&
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
85 that.rtstruct_.get() != NULL)
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
86 {
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
87 rtstruct_ = that.rtstruct_.get();
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
88 }
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
89 else
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
90 {
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
91 try
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
92 {
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
93 OrthancStone::OrthancPluginConnection connection;
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
94 OrthancStone::FullOrthancDataset dataset(connection, "/instances/" + instanceId + "/tags?ignore-length=3006-0050");
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
95 that.rtstruct_.reset(new OrthancStone::DicomStructureSet(dataset));
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
96 that.instanceId_ = instanceId;
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
97 rtstruct_ = that.rtstruct_.get();
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
98 }
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
99 catch (Orthanc::OrthancException&)
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
100 {
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
101 }
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
102 }
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
103 }
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
104
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
105 const std::string& GetInstanceId() const
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
106 {
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
107 return instanceId_;
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
108 }
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
109
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
110 bool IsValid() const
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
111 {
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
112 return rtstruct_ != NULL;
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
113 }
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
114
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
115 const OrthancStone::DicomStructureSet& GetRtStruct() const
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
116 {
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
117 if (IsValid())
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
118 {
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
119 return *rtstruct_;
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
120 }
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
121 else
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
122 {
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
123 throw Orthanc::OrthancException(Orthanc::ErrorCode_BadSequenceOfCalls);
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
124 }
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
125 }
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
126 };
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
127 };
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
128
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
129
1882
4e80b8afd0da numpy in rendering plugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1877
diff changeset
130 static Orthanc::PixelFormat Convert(OrthancPluginPixelFormat format)
4e80b8afd0da numpy in rendering plugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1877
diff changeset
131 {
4e80b8afd0da numpy in rendering plugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1877
diff changeset
132 switch (format)
4e80b8afd0da numpy in rendering plugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1877
diff changeset
133 {
4e80b8afd0da numpy in rendering plugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1877
diff changeset
134 case OrthancPluginPixelFormat_RGB24:
4e80b8afd0da numpy in rendering plugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1877
diff changeset
135 return Orthanc::PixelFormat_RGB24;
4e80b8afd0da numpy in rendering plugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1877
diff changeset
136
4e80b8afd0da numpy in rendering plugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1877
diff changeset
137 case OrthancPluginPixelFormat_Grayscale8:
4e80b8afd0da numpy in rendering plugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1877
diff changeset
138 return Orthanc::PixelFormat_Grayscale8;
4e80b8afd0da numpy in rendering plugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1877
diff changeset
139
4e80b8afd0da numpy in rendering plugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1877
diff changeset
140 case OrthancPluginPixelFormat_Grayscale16:
4e80b8afd0da numpy in rendering plugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1877
diff changeset
141 return Orthanc::PixelFormat_Grayscale16;
4e80b8afd0da numpy in rendering plugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1877
diff changeset
142
4e80b8afd0da numpy in rendering plugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1877
diff changeset
143 case OrthancPluginPixelFormat_SignedGrayscale16:
4e80b8afd0da numpy in rendering plugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1877
diff changeset
144 return Orthanc::PixelFormat_SignedGrayscale16;
4e80b8afd0da numpy in rendering plugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1877
diff changeset
145
4e80b8afd0da numpy in rendering plugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1877
diff changeset
146 default:
4e80b8afd0da numpy in rendering plugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1877
diff changeset
147 throw Orthanc::OrthancException(Orthanc::ErrorCode_NotImplemented);
4e80b8afd0da numpy in rendering plugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1877
diff changeset
148 }
4e80b8afd0da numpy in rendering plugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1877
diff changeset
149 }
4e80b8afd0da numpy in rendering plugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1877
diff changeset
150
4e80b8afd0da numpy in rendering plugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1877
diff changeset
151
1884
0c6923982fdd rendering options from GET args
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1883
diff changeset
152 static bool ParseBoolean(const std::string& key,
0c6923982fdd rendering options from GET args
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1883
diff changeset
153 const std::string& value)
0c6923982fdd rendering options from GET args
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1883
diff changeset
154 {
0c6923982fdd rendering options from GET args
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1883
diff changeset
155 bool result;
0c6923982fdd rendering options from GET args
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1883
diff changeset
156
0c6923982fdd rendering options from GET args
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1883
diff changeset
157 if (Orthanc::SerializationToolbox::ParseBoolean(result, value))
0c6923982fdd rendering options from GET args
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1883
diff changeset
158 {
0c6923982fdd rendering options from GET args
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1883
diff changeset
159 return result;
0c6923982fdd rendering options from GET args
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1883
diff changeset
160 }
0c6923982fdd rendering options from GET args
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1883
diff changeset
161 else
0c6923982fdd rendering options from GET args
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1883
diff changeset
162 {
0c6923982fdd rendering options from GET args
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1883
diff changeset
163 throw Orthanc::OrthancException(Orthanc::ErrorCode_ParameterOutOfRange,
0c6923982fdd rendering options from GET args
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1883
diff changeset
164 "Bad value for " + key + ": " + value);
0c6923982fdd rendering options from GET args
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1883
diff changeset
165 }
0c6923982fdd rendering options from GET args
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1883
diff changeset
166 }
0c6923982fdd rendering options from GET args
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1883
diff changeset
167
0c6923982fdd rendering options from GET args
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1883
diff changeset
168
0c6923982fdd rendering options from GET args
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1883
diff changeset
169 static double ParseDouble(const std::string& key,
0c6923982fdd rendering options from GET args
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1883
diff changeset
170 const std::string& value)
0c6923982fdd rendering options from GET args
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1883
diff changeset
171 {
0c6923982fdd rendering options from GET args
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1883
diff changeset
172 double result;
0c6923982fdd rendering options from GET args
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1883
diff changeset
173
0c6923982fdd rendering options from GET args
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1883
diff changeset
174 if (Orthanc::SerializationToolbox::ParseDouble(result, value))
0c6923982fdd rendering options from GET args
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1883
diff changeset
175 {
0c6923982fdd rendering options from GET args
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1883
diff changeset
176 return result;
0c6923982fdd rendering options from GET args
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1883
diff changeset
177 }
0c6923982fdd rendering options from GET args
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1883
diff changeset
178 else
0c6923982fdd rendering options from GET args
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1883
diff changeset
179 {
0c6923982fdd rendering options from GET args
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1883
diff changeset
180 throw Orthanc::OrthancException(Orthanc::ErrorCode_ParameterOutOfRange,
0c6923982fdd rendering options from GET args
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1883
diff changeset
181 "Bad value for " + key + ": " + value);
0c6923982fdd rendering options from GET args
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1883
diff changeset
182 }
0c6923982fdd rendering options from GET args
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1883
diff changeset
183 }
0c6923982fdd rendering options from GET args
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1883
diff changeset
184
0c6923982fdd rendering options from GET args
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1883
diff changeset
185
1887
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
186 static unsigned int ParseUnsignedInteger(const std::string& key,
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
187 const std::string& value)
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
188 {
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
189 uint32_t result;
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
190
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
191 if (Orthanc::SerializationToolbox::ParseUnsignedInteger32(result, value))
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
192 {
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
193 return result;
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
194 }
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
195 else
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
196 {
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
197 throw Orthanc::OrthancException(Orthanc::ErrorCode_ParameterOutOfRange,
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
198 "Bad value for " + key + ": " + value);
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
199 }
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
200 }
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
201
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
202
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
203
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
204 class DataAugmentationParameters : public boost::noncopyable
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
205 {
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
206 private:
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
207 double angleRadians_;
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
208 double scaling_;
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
209 double offsetX_;
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
210 double offsetY_;
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
211 bool flipX_;
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
212 bool flipY_;
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
213 bool hasResize_;
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
214 unsigned int targetWidth_;
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
215 unsigned int targetHeight_;
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
216
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
217 void ApplyInternal(Orthanc::ImageAccessor& target,
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
218 const Orthanc::ImageAccessor& source)
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
219 {
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
220 if (source.GetWidth() == 0 ||
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
221 source.GetHeight() == 0)
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
222 {
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
223 Orthanc::ImageProcessing::Set(target, 0); // Clear the image
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
224 }
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
225 else if (target.GetWidth() == 0 ||
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
226 target.GetHeight() == 0)
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
227 {
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
228 // Nothing to do
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
229 }
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
230 else
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
231 {
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
232 OrthancStone::AffineTransform2D transform = ComputeTransform(source.GetWidth(), source.GetHeight());
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
233
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
234 OrthancStone::ImageInterpolation interpolation;
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
235
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
236 if (source.GetFormat() == Orthanc::PixelFormat_RGB24)
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
237 {
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
238 LOG(WARNING) << "Bilinear interpolation for color images is not implemented yet";
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
239 interpolation = OrthancStone::ImageInterpolation_Nearest;
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
240 }
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
241 else
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
242 {
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
243 interpolation = OrthancStone::ImageInterpolation_Bilinear;
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
244 }
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
245
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
246 transform.Apply(target, source, interpolation, true /* clear */);
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
247 }
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
248 }
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
249
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
250
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
251 Orthanc::ImageAccessor* ApplyUnchecked(const Orthanc::ImageAccessor& source)
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
252 {
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
253 std::unique_ptr<Orthanc::ImageAccessor> target;
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
254
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
255 if (hasResize_)
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
256 {
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
257 target.reset(new Orthanc::Image(source.GetFormat(), targetWidth_, targetHeight_, false));
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
258 }
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
259 else
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
260 {
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
261 target.reset(new Orthanc::Image(source.GetFormat(), source.GetWidth(), source.GetHeight(), false));
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
262 }
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
263
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
264 ApplyInternal(*target, source);
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
265 return target.release();
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
266 }
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
267
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
268
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
269 public:
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
270 DataAugmentationParameters()
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
271 {
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
272 Clear();
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
273 }
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
274
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
275
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
276 void Clear()
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
277 {
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
278 angleRadians_ = 0;
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
279 scaling_ = 1;
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
280 offsetX_ = 0;
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
281 offsetY_ = 0;
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
282 flipX_ = false;
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
283 flipY_ = false;
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
284 hasResize_ = false;
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
285 targetWidth_ = 0;
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
286 targetHeight_ = 0;
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
287 }
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
288
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
289
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
290 OrthancStone::AffineTransform2D ComputeTransform(unsigned int sourceWidth,
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
291 unsigned int sourceHeight) const
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
292 {
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
293 unsigned int w = (hasResize_ ? targetWidth_ : sourceWidth);
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
294 unsigned int h = (hasResize_ ? targetHeight_ : sourceHeight);
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
295
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
296 if (w == 0 ||
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
297 h == 0 ||
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
298 sourceWidth == 0 ||
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
299 sourceHeight == 0)
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
300 {
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
301 // Division by zero
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
302 throw Orthanc::OrthancException(Orthanc::ErrorCode_ParameterOutOfRange);
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
303 }
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
304
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
305 double r = std::min(static_cast<double>(w) / static_cast<double>(sourceWidth),
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
306 static_cast<double>(h) / static_cast<double>(sourceHeight));
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
307
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
308 OrthancStone::AffineTransform2D resize = OrthancStone::AffineTransform2D::Combine(
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
309 OrthancStone::AffineTransform2D::CreateOffset(static_cast<double>(w) / 2.0,
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
310 static_cast<double>(h) / 2.0),
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
311 OrthancStone::AffineTransform2D::CreateScaling(r, r));
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
312
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
313 OrthancStone::AffineTransform2D dataAugmentation = OrthancStone::AffineTransform2D::Combine(
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
314 OrthancStone::AffineTransform2D::CreateScaling(scaling_, scaling_),
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
315 OrthancStone::AffineTransform2D::CreateOffset(offsetX_, offsetY_),
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
316 OrthancStone::AffineTransform2D::CreateRotation(angleRadians_),
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
317 OrthancStone::AffineTransform2D::CreateOffset(-static_cast<double>(sourceWidth) / 2.0,
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
318 -static_cast<double>(sourceHeight) / 2.0),
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
319 OrthancStone::AffineTransform2D::CreateFlip(flipX_, flipY_, sourceWidth, sourceHeight));
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
320
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
321 return OrthancStone::AffineTransform2D::Combine(resize, dataAugmentation);
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
322 }
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
323
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
324
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
325 bool ParseParameter(const std::string& key,
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
326 const std::string& value)
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
327 {
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
328 if (key == "angle")
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
329 {
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
330 double angle = ParseDouble(key, value);
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
331 angleRadians_ = angle / 180.0 * boost::math::constants::pi<double>();
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
332 return true;
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
333 }
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
334 else if (key == "scaling")
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
335 {
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
336 scaling_ = ParseDouble(key, value);
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
337 return true;
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
338 }
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
339 else if (key == "offset-x")
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
340 {
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
341 offsetX_ = ParseDouble(key, value);
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
342 return true;
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
343 }
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
344 else if (key == "offset-y")
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
345 {
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
346 offsetY_ = ParseDouble(key, value);
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
347 return true;
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
348 }
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
349 else if (key == "flip-x")
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
350 {
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
351 flipX_ = ParseBoolean(key, value);
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
352 return true;
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
353 }
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
354 else if (key == "flip-y")
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
355 {
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
356 flipY_ = ParseBoolean(key, value);
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
357 return true;
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
358 }
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
359 else if (key == "resize")
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
360 {
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
361 std::vector<std::string> tokens;
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
362 Orthanc::Toolbox::TokenizeString(tokens, value, ',');
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
363 if (tokens.size() != 2)
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
364 {
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
365 throw Orthanc::OrthancException(Orthanc::ErrorCode_ParameterOutOfRange,
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
366 "Must provide two integers separated by commas in " + key + ": " + value);
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
367 }
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
368 else
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
369 {
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
370 targetWidth_ = ParseUnsignedInteger(key, tokens[0]);
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
371 targetHeight_ = ParseUnsignedInteger(key, tokens[1]);
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
372 hasResize_ = true;
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
373 return true;
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
374 }
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
375 }
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
376 else
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
377 {
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
378 return false;
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
379 }
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
380 }
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
381
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
382
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
383 Orthanc::ImageAccessor* Apply(const Orthanc::ImageAccessor& source)
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
384 {
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
385 if (source.GetFormat() != Orthanc::PixelFormat_RGB24 &&
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
386 source.GetFormat() != Orthanc::PixelFormat_Float32)
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
387 {
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
388 throw Orthanc::OrthancException(Orthanc::ErrorCode_IncompatibleImageFormat);
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
389 }
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
390 else
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
391 {
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
392 return ApplyUnchecked(source);
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
393 }
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
394 }
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
395
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
396
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
397 Orthanc::ImageAccessor* ApplyBinaryMask(const Orthanc::ImageAccessor& source)
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
398 {
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
399 if (source.GetFormat() != Orthanc::PixelFormat_Grayscale8)
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
400 {
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
401 throw Orthanc::OrthancException(Orthanc::ErrorCode_IncompatibleImageFormat,
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
402 "A segmentation mask should be a grayscale image");
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
403 }
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
404 else
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
405 {
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
406 std::unique_ptr<Orthanc::ImageAccessor> target(ApplyUnchecked(source));
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
407
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
408 const unsigned int h = target->GetHeight();
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
409 const unsigned int w = target->GetWidth();
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
410
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
411 for (unsigned int y = 0; y < h; y++)
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
412 {
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
413 uint8_t* p = reinterpret_cast<uint8_t*>(target->GetRow(y));
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
414 for (unsigned int x = 0; x < w; x++, p++)
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
415 {
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
416 if (*p < 128)
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
417 {
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
418 *p = 0;
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
419 }
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
420 else
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
421 {
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
422 *p = 255;
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
423 }
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
424 }
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
425 }
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
426
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
427 return target.release();
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
428 }
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
429 }
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
430 };
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
431
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
432
1882
4e80b8afd0da numpy in rendering plugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1877
diff changeset
433 static void RenderNumpyFrame(OrthancPluginRestOutput* output,
4e80b8afd0da numpy in rendering plugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1877
diff changeset
434 const char* url,
4e80b8afd0da numpy in rendering plugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1877
diff changeset
435 const OrthancPluginHttpRequest* request)
4e80b8afd0da numpy in rendering plugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1877
diff changeset
436 {
1887
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
437 DataAugmentationParameters dataAugmentation;
1884
0c6923982fdd rendering options from GET args
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1883
diff changeset
438 bool compress = false;
0c6923982fdd rendering options from GET args
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1883
diff changeset
439
0c6923982fdd rendering options from GET args
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1883
diff changeset
440 for (uint32_t i = 0; i < request->getCount; i++)
0c6923982fdd rendering options from GET args
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1883
diff changeset
441 {
0c6923982fdd rendering options from GET args
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1883
diff changeset
442 std::string key(request->getKeys[i]);
0c6923982fdd rendering options from GET args
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1883
diff changeset
443 std::string value(request->getValues[i]);
0c6923982fdd rendering options from GET args
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1883
diff changeset
444
1887
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
445 if (!dataAugmentation.ParseParameter(key, value))
1884
0c6923982fdd rendering options from GET args
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1883
diff changeset
446 {
1887
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
447 if (key == "compress")
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
448 {
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
449 compress = ParseBoolean(key, value);
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
450 }
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
451 else
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
452 {
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
453 LOG(WARNING) << "Unsupported option for data augmentation: " << key;
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
454 }
1884
0c6923982fdd rendering options from GET args
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1883
diff changeset
455 }
0c6923982fdd rendering options from GET args
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1883
diff changeset
456 }
1887
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
457
1883
fba6e550e0ee apply rescale slope/intercept in rendering plugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1882
diff changeset
458 OrthancPlugins::MemoryBuffer tags;
fba6e550e0ee apply rescale slope/intercept in rendering plugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1882
diff changeset
459 if (!tags.RestApiGet("/instances/" + std::string(request->groups[0]) + "/tags", false))
fba6e550e0ee apply rescale slope/intercept in rendering plugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1882
diff changeset
460 {
fba6e550e0ee apply rescale slope/intercept in rendering plugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1882
diff changeset
461 throw Orthanc::OrthancException(Orthanc::ErrorCode_InexistentItem);
fba6e550e0ee apply rescale slope/intercept in rendering plugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1882
diff changeset
462 }
fba6e550e0ee apply rescale slope/intercept in rendering plugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1882
diff changeset
463
fba6e550e0ee apply rescale slope/intercept in rendering plugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1882
diff changeset
464 Json::Value json;
fba6e550e0ee apply rescale slope/intercept in rendering plugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1882
diff changeset
465 tags.ToJson(json);
fba6e550e0ee apply rescale slope/intercept in rendering plugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1882
diff changeset
466
fba6e550e0ee apply rescale slope/intercept in rendering plugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1882
diff changeset
467 Orthanc::DicomMap m;
fba6e550e0ee apply rescale slope/intercept in rendering plugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1882
diff changeset
468 m.FromDicomAsJson(json);
fba6e550e0ee apply rescale slope/intercept in rendering plugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1882
diff changeset
469
fba6e550e0ee apply rescale slope/intercept in rendering plugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1882
diff changeset
470 OrthancStone::DicomInstanceParameters parameters(m);
1882
4e80b8afd0da numpy in rendering plugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1877
diff changeset
471
4e80b8afd0da numpy in rendering plugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1877
diff changeset
472 OrthancPlugins::MemoryBuffer dicom;
4e80b8afd0da numpy in rendering plugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1877
diff changeset
473 dicom.GetDicomInstance(request->groups[0]);
4e80b8afd0da numpy in rendering plugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1877
diff changeset
474
4e80b8afd0da numpy in rendering plugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1877
diff changeset
475 unsigned int frame = boost::lexical_cast<unsigned int>(request->groups[1]);
4e80b8afd0da numpy in rendering plugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1877
diff changeset
476
4e80b8afd0da numpy in rendering plugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1877
diff changeset
477 OrthancPlugins::OrthancImage image;
4e80b8afd0da numpy in rendering plugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1877
diff changeset
478 image.DecodeDicomImage(dicom.GetData(), dicom.GetSize(), frame);
4e80b8afd0da numpy in rendering plugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1877
diff changeset
479
4e80b8afd0da numpy in rendering plugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1877
diff changeset
480 Orthanc::ImageAccessor source;
4e80b8afd0da numpy in rendering plugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1877
diff changeset
481 source.AssignReadOnly(Convert(image.GetPixelFormat()), image.GetWidth(), image.GetHeight(),
4e80b8afd0da numpy in rendering plugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1877
diff changeset
482 image.GetPitch(), image.GetBuffer());
4e80b8afd0da numpy in rendering plugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1877
diff changeset
483
4e80b8afd0da numpy in rendering plugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1877
diff changeset
484 std::unique_ptr<Orthanc::ImageAccessor> modified;
1887
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
485
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
486 if (parameters.GetSopClassUid() == OrthancStone::SopClassUid_DicomSeg)
1882
4e80b8afd0da numpy in rendering plugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1877
diff changeset
487 {
1887
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
488 modified.reset(dataAugmentation.ApplyBinaryMask(source));
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
489 }
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
490 else if (source.GetFormat() == Orthanc::PixelFormat_RGB24)
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
491 {
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
492 modified.reset(dataAugmentation.Apply(source));
1882
4e80b8afd0da numpy in rendering plugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1877
diff changeset
493 }
4e80b8afd0da numpy in rendering plugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1877
diff changeset
494 else
4e80b8afd0da numpy in rendering plugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1877
diff changeset
495 {
1883
fba6e550e0ee apply rescale slope/intercept in rendering plugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1882
diff changeset
496 std::unique_ptr<Orthanc::ImageAccessor> converted(parameters.ConvertToFloat(source));
fba6e550e0ee apply rescale slope/intercept in rendering plugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1882
diff changeset
497 assert(converted.get() != NULL);
1887
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
498
aa4ed1cf4e8d refactoring using DataAugmentationParameters
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1886
diff changeset
499 modified.reset(dataAugmentation.Apply(*converted));
1882
4e80b8afd0da numpy in rendering plugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1877
diff changeset
500 }
4e80b8afd0da numpy in rendering plugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1877
diff changeset
501
4e80b8afd0da numpy in rendering plugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1877
diff changeset
502 assert(modified.get() != NULL);
4e80b8afd0da numpy in rendering plugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1877
diff changeset
503
4e80b8afd0da numpy in rendering plugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1877
diff changeset
504 std::string answer;
4e80b8afd0da numpy in rendering plugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1877
diff changeset
505 Orthanc::NumpyWriter writer;
1884
0c6923982fdd rendering options from GET args
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1883
diff changeset
506 writer.SetCompressed(compress);
1882
4e80b8afd0da numpy in rendering plugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1877
diff changeset
507 Orthanc::IImageWriter::WriteToMemory(writer, answer, *modified);
1884
0c6923982fdd rendering options from GET args
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1883
diff changeset
508
1882
4e80b8afd0da numpy in rendering plugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1877
diff changeset
509 OrthancPluginAnswerBuffer(OrthancPlugins::GetGlobalContext(), output,
1885
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
510 answer.c_str(), answer.size(), "application/octet-stream");
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
511 }
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
512
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
513
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
514 static bool IsRtStruct(const std::string& instanceId)
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
515 {
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
516 static const char* SOP_CLASS_UID = "0008,0016";
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
517 static const char* RT_STRUCT_IOD = "1.2.840.10008.5.1.4.1.1.481.3";
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
518
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
519 std::string s;
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
520 if (OrthancPlugins::RestApiGetString(s, "/instances/" + instanceId + "/content/" + SOP_CLASS_UID, false) &&
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
521 !s.empty())
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
522 {
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
523 if (s[s.size() - 1] == '\0') // Deal with DICOM padding
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
524 {
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
525 s.resize(s.size() - 1);
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
526 }
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
527
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
528 return s == RT_STRUCT_IOD;
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
529 }
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
530 else
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
531 {
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
532 return false;
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
533 }
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
534 }
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
535
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
536
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
537 static void ListRtStruct(OrthancPluginRestOutput* output,
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
538 const char* url,
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
539 const OrthancPluginHttpRequest* request)
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
540 {
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
541 // This is a quick version of "/tools/find" on "SOPClassUID" (the
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
542 // latter would load all the DICOM files from disk)
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
543
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
544 static const char* INSTANCES = "Instances";
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
545
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
546 Json::Value series;
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
547 OrthancPlugins::RestApiGet(series, "/series?expand", false);
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
548
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
549 if (series.type() != Json::arrayValue)
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
550 {
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
551 throw Orthanc::OrthancException(Orthanc::ErrorCode_InternalError);
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
552 }
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
553
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
554 Json::Value answer = Json::arrayValue;
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
555
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
556 for (Json::Value::ArrayIndex i = 0; i < series.size(); i++)
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
557 {
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
558 if (series[i].type() != Json::objectValue ||
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
559 !series[i].isMember(INSTANCES) ||
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
560 series[i][INSTANCES].type() != Json::arrayValue)
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
561 {
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
562 throw Orthanc::OrthancException(Orthanc::ErrorCode_InternalError);
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
563 }
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
564
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
565 const Json::Value& instances = series[i][INSTANCES];
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
566
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
567 for (Json::Value::ArrayIndex j = 0; j < instances.size(); j++)
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
568 {
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
569 if (instances[j].type() != Json::stringValue)
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
570 {
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
571 throw Orthanc::OrthancException(Orthanc::ErrorCode_InternalError);
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
572 }
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
573 }
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
574
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
575 if (instances.size() > 0 &&
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
576 IsRtStruct(instances[0].asString()))
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
577 {
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
578 for (Json::Value::ArrayIndex j = 0; j < instances.size(); j++)
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
579 {
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
580 answer.append(instances[j].asString());
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
581 }
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
582 }
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
583 }
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
584
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
585 std::string s = answer.toStyledString();
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
586 OrthancPluginAnswerBuffer(OrthancPlugins::GetGlobalContext(), output, s.c_str(), s.size(), "application/json");
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
587 }
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
588
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
589
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
590 static void GetRtStruct(OrthancPluginRestOutput* output,
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
591 const char* url,
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
592 const OrthancPluginHttpRequest* request)
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
593 {
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
594 static const char* STRUCTURES = "Structures";
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
595 static const char* INSTANCES = "Instances";
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
596
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
597 DicomStructureCache::Accessor accessor(DicomStructureCache::GetSingleton(), request->groups[0]);
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
598
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
599 if (!accessor.IsValid())
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
600 {
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
601 throw Orthanc::OrthancException(Orthanc::ErrorCode_InexistentItem);
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
602 }
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
603
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
604 Json::Value answer;
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
605 answer[STRUCTURES] = Json::arrayValue;
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
606
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
607 for (size_t i = 0; i < accessor.GetRtStruct().GetStructuresCount(); i++)
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
608 {
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
609 Json::Value color = Json::arrayValue;
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
610 color.append(accessor.GetRtStruct().GetStructureColor(i).GetRed());
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
611 color.append(accessor.GetRtStruct().GetStructureColor(i).GetGreen());
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
612 color.append(accessor.GetRtStruct().GetStructureColor(i).GetBlue());
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
613
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
614 Json::Value structure;
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
615 structure["Name"] = accessor.GetRtStruct().GetStructureName(i);
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
616 structure["Interpretation"] = accessor.GetRtStruct().GetStructureInterpretation(i);
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
617 structure["Color"] = color;
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
618
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
619 answer[STRUCTURES].append(structure);
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
620 }
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
621
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
622 std::set<std::string> sopInstanceUids;
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
623 accessor.GetRtStruct().GetReferencedInstances(sopInstanceUids);
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
624
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
625 answer[INSTANCES] = Json::arrayValue;
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
626 for (std::set<std::string>::const_iterator it = sopInstanceUids.begin(); it != sopInstanceUids.end(); ++it)
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
627 {
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
628 OrthancPlugins::OrthancString s;
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
629 s.Assign(OrthancPluginLookupInstance(OrthancPlugins::GetGlobalContext(), it->c_str()));
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
630
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
631 std::string t;
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
632 s.ToString(t);
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
633
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
634 answer[INSTANCES].append(t);
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
635 }
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
636
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
637 std::string s = answer.toStyledString();
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
638 OrthancPluginAnswerBuffer(OrthancPlugins::GetGlobalContext(), output, s.c_str(), s.size(), "application/json");
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
639 }
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
640
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
641
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
642 static void RenderRtStruct(OrthancPluginRestOutput* output,
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
643 const char* url,
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
644 const OrthancPluginHttpRequest* request)
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
645 {
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
646 static const char* MAIN_DICOM_TAGS = "MainDicomTags";
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
647
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
648 DicomStructureCache::Accessor accessor(DicomStructureCache::GetSingleton(), request->groups[0]);
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
649
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
650 if (!accessor.IsValid())
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
651 {
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
652 throw Orthanc::OrthancException(Orthanc::ErrorCode_InexistentItem);
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
653 }
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
654
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
655 std::string structureName;
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
656 std::string instanceId;
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
657
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
658 for (uint32_t i = 0; i < request->getCount; i++)
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
659 {
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
660 std::string key(request->getKeys[i]);
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
661 std::string value(request->getValues[i]);
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
662
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
663 if (key == "structure")
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
664 {
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
665 structureName = value;
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
666 }
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
667 else if (key == "instance")
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
668 {
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
669 instanceId = value;
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
670 }
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
671 else
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
672 {
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
673 LOG(WARNING) << "Unsupported option: " << key;
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
674 }
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
675 }
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
676
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
677 if (structureName.empty())
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
678 {
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
679 throw Orthanc::OrthancException(Orthanc::ErrorCode_NetworkProtocol,
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
680 "Missing option \"structure\" to provide the structure name");
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
681 }
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
682
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
683 if (instanceId.empty())
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
684 {
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
685 throw Orthanc::OrthancException(Orthanc::ErrorCode_NetworkProtocol,
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
686 "Missing option \"instance\" to provide the Orthanc identifier of the instance of interest");
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
687 }
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
688
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
689 size_t structureIndex;
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
690 bool found = false;
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
691 for (size_t i = 0; i < accessor.GetRtStruct().GetStructuresCount(); i++)
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
692 {
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
693 if (accessor.GetRtStruct().GetStructureName(i) == structureName)
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
694 {
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
695 structureIndex = i;
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
696 found = true;
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
697 break;
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
698 }
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
699 }
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
700
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
701 if (!found)
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
702 {
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
703 throw Orthanc::OrthancException(Orthanc::ErrorCode_InexistentItem,
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
704 "Unknown structure name: " + structureName);
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
705 }
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
706
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
707 Json::Value instance;
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
708 if (!OrthancPlugins::RestApiGet(instance, "/instances/" + instanceId, false))
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
709 {
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
710 throw Orthanc::OrthancException(Orthanc::ErrorCode_InexistentItem,
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
711 "Unknown instance with Orthanc ID: " + instanceId);
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
712 }
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
713
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
714 if (instance.type() != Json::objectValue ||
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
715 !instance.isMember(MAIN_DICOM_TAGS) ||
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
716 instance[MAIN_DICOM_TAGS].type() != Json::objectValue)
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
717 {
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
718 throw Orthanc::OrthancException(Orthanc::ErrorCode_InternalError);
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
719 }
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
720
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
721 std::string sopInstanceUid = Orthanc::SerializationToolbox::ReadString(instance[MAIN_DICOM_TAGS], "SOPInstanceUID");
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
722
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
723
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
724 std::list< std::vector<OrthancStone::Vector> > p;
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
725 accessor.GetRtStruct().GetStructurePoints(p, structureIndex, sopInstanceUid);
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
726
1886
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1885
diff changeset
727 // curl 'http://localhost:8042/stone/rt-struct/54460695-ba3885ee-ddf61ac0-f028e31d-a6e474d9/numpy?structure=BODY&instance=0dd13003-9b1b2712-0fb44325-48c28c99-430d111b'
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1885
diff changeset
728 unsigned int count = 0;
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1885
diff changeset
729 for (std::list< std::vector<OrthancStone::Vector> >::const_iterator
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1885
diff changeset
730 it = p.begin(); it != p.end(); ++it)
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1885
diff changeset
731 {
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1885
diff changeset
732 printf("%d: ", count++);
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1885
diff changeset
733 for (size_t i = 0; i < it->size(); i++)
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1885
diff changeset
734 {
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1885
diff changeset
735 printf("(%.1f,%.1f,%.1f) ", (*it)[i][0], (*it)[i][1], (*it)[i][2]);
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1885
diff changeset
736 }
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1885
diff changeset
737 printf("\n");
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1885
diff changeset
738 }
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1885
diff changeset
739
1885
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
740 std::string s = "hello";
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
741 OrthancPluginAnswerBuffer(OrthancPlugins::GetGlobalContext(), output, s.c_str(), s.size(), "application/json");
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
742 }
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
743
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
744
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
745 OrthancPluginErrorCode OnChangeCallback(OrthancPluginChangeType changeType,
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
746 OrthancPluginResourceType resourceType,
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
747 const char* resourceId)
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
748 {
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
749 switch (changeType)
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
750 {
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
751 case OrthancPluginChangeType_Deleted:
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
752 if (resourceType == OrthancPluginResourceType_Instance)
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
753 {
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
754 DicomStructureCache::GetSingleton().Invalidate(resourceId);
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
755 }
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
756
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
757 break;
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
758
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
759 default:
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
760 break;
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
761 }
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
762
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
763 return OrthancPluginErrorCode_Success;
1882
4e80b8afd0da numpy in rendering plugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1877
diff changeset
764 }
4e80b8afd0da numpy in rendering plugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1877
diff changeset
765
1877
a2955abe4c2e skeleton for the RenderingPlugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
766
a2955abe4c2e skeleton for the RenderingPlugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
767 extern "C"
a2955abe4c2e skeleton for the RenderingPlugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
768 {
a2955abe4c2e skeleton for the RenderingPlugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
769 ORTHANC_PLUGINS_API int32_t OrthancPluginInitialize(OrthancPluginContext* context)
a2955abe4c2e skeleton for the RenderingPlugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
770 {
a2955abe4c2e skeleton for the RenderingPlugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
771 OrthancPlugins::SetGlobalContext(context);
a2955abe4c2e skeleton for the RenderingPlugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
772
a2955abe4c2e skeleton for the RenderingPlugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
773 #if ORTHANC_FRAMEWORK_VERSION_IS_ABOVE(1, 7, 2)
a2955abe4c2e skeleton for the RenderingPlugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
774 Orthanc::Logging::InitializePluginContext(context);
a2955abe4c2e skeleton for the RenderingPlugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
775 #else
a2955abe4c2e skeleton for the RenderingPlugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
776 Orthanc::Logging::Initialize(context);
a2955abe4c2e skeleton for the RenderingPlugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
777 #endif
a2955abe4c2e skeleton for the RenderingPlugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
778
a2955abe4c2e skeleton for the RenderingPlugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
779 /* Check the version of the Orthanc core */
a2955abe4c2e skeleton for the RenderingPlugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
780 if (OrthancPluginCheckVersion(context) == 0)
a2955abe4c2e skeleton for the RenderingPlugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
781 {
a2955abe4c2e skeleton for the RenderingPlugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
782 char info[1024];
a2955abe4c2e skeleton for the RenderingPlugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
783 sprintf(info, "Your version of Orthanc (%s) must be above %d.%d.%d to run this plugin",
a2955abe4c2e skeleton for the RenderingPlugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
784 context->orthancVersion,
a2955abe4c2e skeleton for the RenderingPlugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
785 ORTHANC_PLUGINS_MINIMAL_MAJOR_NUMBER,
a2955abe4c2e skeleton for the RenderingPlugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
786 ORTHANC_PLUGINS_MINIMAL_MINOR_NUMBER,
a2955abe4c2e skeleton for the RenderingPlugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
787 ORTHANC_PLUGINS_MINIMAL_REVISION_NUMBER);
a2955abe4c2e skeleton for the RenderingPlugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
788 OrthancPluginLogError(context, info);
a2955abe4c2e skeleton for the RenderingPlugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
789 return -1;
a2955abe4c2e skeleton for the RenderingPlugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
790 }
a2955abe4c2e skeleton for the RenderingPlugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
791
a2955abe4c2e skeleton for the RenderingPlugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
792 try
a2955abe4c2e skeleton for the RenderingPlugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
793 {
1882
4e80b8afd0da numpy in rendering plugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1877
diff changeset
794 OrthancPlugins::RegisterRestCallback<RenderNumpyFrame>("/stone/instances/([^/]+)/frames/([0-9]+)/numpy", true);
1885
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
795 OrthancPlugins::RegisterRestCallback<ListRtStruct>("/stone/rt-struct", true);
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
796 OrthancPlugins::RegisterRestCallback<GetRtStruct>("/stone/rt-struct/([^/]+)/info", true);
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
797 OrthancPlugins::RegisterRestCallback<RenderRtStruct>("/stone/rt-struct/([^/]+)/numpy", true);
ddaee6b96501 retrieving rt-struct info
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 1884
diff changeset
798 OrthancPluginRegisterOnChangeCallback(context, OnChangeCallback);
1877
a2955abe4c2e skeleton for the RenderingPlugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
799 }
a2955abe4c2e skeleton for the RenderingPlugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
800 catch (...)
a2955abe4c2e skeleton for the RenderingPlugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
801 {
a2955abe4c2e skeleton for the RenderingPlugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
802 OrthancPlugins::LogError("Exception while initializing the Stone Web viewer plugin");
a2955abe4c2e skeleton for the RenderingPlugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
803 return -1;
a2955abe4c2e skeleton for the RenderingPlugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
804 }
a2955abe4c2e skeleton for the RenderingPlugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
805
a2955abe4c2e skeleton for the RenderingPlugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
806 return 0;
a2955abe4c2e skeleton for the RenderingPlugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
807 }
a2955abe4c2e skeleton for the RenderingPlugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
808
a2955abe4c2e skeleton for the RenderingPlugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
809
a2955abe4c2e skeleton for the RenderingPlugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
810 ORTHANC_PLUGINS_API void OrthancPluginFinalize()
a2955abe4c2e skeleton for the RenderingPlugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
811 {
a2955abe4c2e skeleton for the RenderingPlugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
812 }
a2955abe4c2e skeleton for the RenderingPlugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
813
a2955abe4c2e skeleton for the RenderingPlugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
814
a2955abe4c2e skeleton for the RenderingPlugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
815 ORTHANC_PLUGINS_API const char* OrthancPluginGetName()
a2955abe4c2e skeleton for the RenderingPlugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
816 {
a2955abe4c2e skeleton for the RenderingPlugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
817 return PLUGIN_NAME;
a2955abe4c2e skeleton for the RenderingPlugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
818 }
a2955abe4c2e skeleton for the RenderingPlugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
819
a2955abe4c2e skeleton for the RenderingPlugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
820
a2955abe4c2e skeleton for the RenderingPlugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
821 ORTHANC_PLUGINS_API const char* OrthancPluginGetVersion()
a2955abe4c2e skeleton for the RenderingPlugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
822 {
a2955abe4c2e skeleton for the RenderingPlugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
823 return PLUGIN_VERSION;
a2955abe4c2e skeleton for the RenderingPlugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
824 }
a2955abe4c2e skeleton for the RenderingPlugin
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
825 }