diff Framework/TransferScheduler.h @ 0:95226b754d9e

initial release
author Sebastien Jodogne <s.jodogne@gmail.com>
date Mon, 17 Sep 2018 11:34:55 +0200
parents
children 7e207ade2f1a
line wrap: on
line diff
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/Framework/TransferScheduler.h	Mon Sep 17 11:34:55 2018 +0200
@@ -0,0 +1,93 @@
+/**
+ * Transfers accelerator plugin for Orthanc
+ * Copyright (C) 2018 Osimis, Belgium
+ *
+ * This program is free software: you can redistribute it and/or
+ * modify it under the terms of the GNU Affero General Public License
+ * as published by the Free Software Foundation, either version 3 of
+ * the License, or (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful, but
+ * WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+ * Affero General Public License for more details.
+ * 
+ * You should have received a copy of the GNU Affero General Public License
+ * along with this program. If not, see <http://www.gnu.org/licenses/>.
+ **/
+
+
+#pragma once
+
+#include "OrthancInstancesCache.h"
+
+
+namespace OrthancPlugins
+{
+  class TransferScheduler : public boost::noncopyable
+  {
+  private:
+    void AddResource(OrthancInstancesCache& cache, 
+                     Orthanc::ResourceType level,
+                     const std::string& id);
+
+    void ComputeBucketsInternal(std::vector<TransferBucket>& target,
+                                size_t groupThreshold,
+                                size_t separateThreshold,
+                                const std::string& baseUrl,  /* only needed in pull mode */
+                                BucketCompression compression /* only needed in pull mode */) const;
+
+    typedef std::map<std::string, DicomInstanceInfo>   Instances;
+
+    Instances    instances_;
+
+
+  public:
+    void AddPatient(OrthancInstancesCache& cache, 
+                    const std::string& patient)
+    {
+      AddResource(cache, Orthanc::ResourceType_Patient, patient);
+    }
+
+    void AddStudy(OrthancInstancesCache& cache, 
+                  const std::string& study)
+    {
+      AddResource(cache, Orthanc::ResourceType_Study, study);
+    }
+
+    void AddSeries(OrthancInstancesCache& cache, 
+                   const std::string& series)
+    {
+      AddResource(cache, Orthanc::ResourceType_Series, series);
+    }
+
+    void AddInstance(OrthancInstancesCache& cache, 
+                     const std::string& instanceId);
+
+    void AddInstance(const DicomInstanceInfo& info);
+
+    void ParseListOfResources(OrthancInstancesCache& cache, 
+                              const Json::Value& resources);
+
+    void ListInstances(std::vector<DicomInstanceInfo>& target) const;
+
+    size_t GetInstancesCount() const
+    {
+      return instances_.size();
+    }
+
+    size_t GetTotalSize() const;
+
+    void ComputePullBuckets(std::vector<TransferBucket>& target,
+                            size_t groupThreshold,
+                            size_t separateThreshold,
+                            const std::string& baseUrl,
+                            BucketCompression compression) const;
+
+    void FormatPushTransaction(Json::Value& target,
+                               std::vector<TransferBucket>& buckets,
+                               size_t groupThreshold,
+                               size_t separateThreshold,
+                               BucketCompression compression) const;
+  };
+}