comparison OrthancFramework/Sources/Compression/HierarchicalZipWriter.h @ 4044:d25f4c0fa160 framework

splitting code into OrthancFramework and OrthancServer
author Sebastien Jodogne <s.jodogne@gmail.com>
date Wed, 10 Jun 2020 20:30:34 +0200
parents Core/Compression/HierarchicalZipWriter.h@05a363186da6
children bf7b9edf6b81
comparison
equal deleted inserted replaced
4043:6c6239aec462 4044:d25f4c0fa160
1 /**
2 * Orthanc - A Lightweight, RESTful DICOM Store
3 * Copyright (C) 2012-2016 Sebastien Jodogne, Medical Physics
4 * Department, University Hospital of Liege, Belgium
5 * Copyright (C) 2017-2020 Osimis S.A., Belgium
6 *
7 * This program is free software: you can redistribute it and/or
8 * modify it under the terms of the GNU General Public License as
9 * published by the Free Software Foundation, either version 3 of the
10 * License, or (at your option) any later version.
11 *
12 * In addition, as a special exception, the copyright holders of this
13 * program give permission to link the code of its release with the
14 * OpenSSL project's "OpenSSL" library (or with modified versions of it
15 * that use the same license as the "OpenSSL" library), and distribute
16 * the linked executables. You must obey the GNU General Public License
17 * in all respects for all of the code used other than "OpenSSL". If you
18 * modify file(s) with this exception, you may extend this exception to
19 * your version of the file(s), but you are not obligated to do so. If
20 * you do not wish to do so, delete this exception statement from your
21 * version. If you delete this exception statement from all source files
22 * in the program, then also delete it here.
23 *
24 * This program is distributed in the hope that it will be useful, but
25 * WITHOUT ANY WARRANTY; without even the implied warranty of
26 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
27 * General Public License for more details.
28 *
29 * You should have received a copy of the GNU General Public License
30 * along with this program. If not, see <http://www.gnu.org/licenses/>.
31 **/
32
33
34 #pragma once
35
36 #include "ZipWriter.h"
37
38 #include <map>
39 #include <list>
40 #include <boost/lexical_cast.hpp>
41
42 #if ORTHANC_BUILD_UNIT_TESTS == 1
43 # include <gtest/gtest_prod.h>
44 #endif
45
46 namespace Orthanc
47 {
48 class ORTHANC_PUBLIC HierarchicalZipWriter : public boost::noncopyable
49 {
50 #if ORTHANC_BUILD_UNIT_TESTS == 1
51 FRIEND_TEST(HierarchicalZipWriter, Index);
52 FRIEND_TEST(HierarchicalZipWriter, Filenames);
53 #endif
54
55 private:
56 class ORTHANC_PUBLIC Index
57 {
58 private:
59 struct Directory
60 {
61 typedef std::map<std::string, unsigned int> Content;
62
63 std::string name_;
64 Content content_;
65 };
66
67 typedef std::list<Directory*> Stack;
68
69 Stack stack_;
70
71 std::string EnsureUniqueFilename(const char* filename);
72
73 public:
74 Index();
75
76 ~Index();
77
78 bool IsRoot() const
79 {
80 return stack_.size() == 1;
81 }
82
83 std::string OpenFile(const char* name);
84
85 void OpenDirectory(const char* name);
86
87 void CloseDirectory();
88
89 std::string GetCurrentDirectoryPath() const;
90
91 static std::string KeepAlphanumeric(const std::string& source);
92 };
93
94 Index indexer_;
95 ZipWriter writer_;
96
97 public:
98 HierarchicalZipWriter(const char* path);
99
100 ~HierarchicalZipWriter();
101
102 void SetZip64(bool isZip64)
103 {
104 writer_.SetZip64(isZip64);
105 }
106
107 bool IsZip64() const
108 {
109 return writer_.IsZip64();
110 }
111
112 void SetCompressionLevel(uint8_t level)
113 {
114 writer_.SetCompressionLevel(level);
115 }
116
117 uint8_t GetCompressionLevel() const
118 {
119 return writer_.GetCompressionLevel();
120 }
121
122 void SetAppendToExisting(bool append)
123 {
124 writer_.SetAppendToExisting(append);
125 }
126
127 bool IsAppendToExisting() const
128 {
129 return writer_.IsAppendToExisting();
130 }
131
132 void OpenFile(const char* name);
133
134 void OpenDirectory(const char* name);
135
136 void CloseDirectory();
137
138 std::string GetCurrentDirectoryPath() const
139 {
140 return indexer_.GetCurrentDirectoryPath();
141 }
142
143 void Write(const void* data, size_t length)
144 {
145 writer_.Write(data, length);
146 }
147
148 void Write(const std::string& data)
149 {
150 writer_.Write(data);
151 }
152 };
153 }