diff Framework/DownloadArea.h @ 0:95226b754d9e

initial release
author Sebastien Jodogne <s.jodogne@gmail.com>
date Mon, 17 Sep 2018 11:34:55 +0200
parents
children 4c3437217518
line wrap: on
line diff
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/Framework/DownloadArea.h	Mon Sep 17 11:34:55 2018 +0200
@@ -0,0 +1,107 @@
+/**
+ * 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 "TransferScheduler.h"
+
+#include <Core/TemporaryFile.h>
+
+namespace OrthancPlugins
+{
+  class DownloadArea : public boost::noncopyable
+  {
+  private:
+    class Instance : public boost::noncopyable
+    {
+    private:
+      DicomInstanceInfo       info_;
+      Orthanc::TemporaryFile  file_;
+
+      class Writer;
+
+    public:
+      Instance(const DicomInstanceInfo& info);
+
+      const DicomInstanceInfo& GetInfo() const
+      {
+        return info_;
+      }
+
+      void WriteChunk(size_t offset,
+                      const void* data,
+                      size_t size);
+
+      void Commit(OrthancPluginContext* context,
+                  bool simulate) const;
+    };
+
+
+    typedef std::map<std::string, Instance*>   Instances;
+
+    boost::mutex  mutex_;
+    Instances     instances_;
+    size_t        totalSize_;
+
+
+    void Clear();
+
+    Instance& LookupInstance(const std::string& id);
+
+    void WriteUncompressedBucket(const TransferBucket& bucket,
+                                 const void* data,
+                                 size_t size);
+
+    void Setup(const std::vector<DicomInstanceInfo>& instances);
+    
+    void CommitInternal(OrthancPluginContext* context,
+                        bool simulate);
+
+  public:
+    DownloadArea(const TransferScheduler& scheduler);
+
+    DownloadArea(const std::vector<DicomInstanceInfo>& instances)
+    {
+      Setup(instances);
+    }
+
+    ~DownloadArea()
+    {
+      Clear();
+    }
+
+    size_t GetTotalSize() const
+    {
+      return totalSize_;
+    }
+
+    void WriteBucket(const TransferBucket& bucket,
+                     const void* data,
+                     size_t size,
+                     BucketCompression compression);
+
+    void WriteInstance(const std::string& instanceId,
+                       const void* data,
+                       size_t size);
+
+    void CheckMD5();
+
+    void Commit(OrthancPluginContext* context);
+  };
+}