Mercurial > hg > orthanc

comparison Core/Compression/HierarchicalZipWriter.h @ 247:c9b3ba0fd140

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
path management in zip files
author Sebastien Jodogne <s.jodogne@gmail.com>
date Tue, 04 Dec 2012 17:27:23 +0100
parents
children 5694365ecb96
comparison
equal deleted inserted replaced
246:fe6ba20d00a8 247:c9b3ba0fd140
1 /**
2 * Orthanc - A Lightweight, RESTful DICOM Store
3 * Copyright (C) 2012 Medical Physics Department, CHU of Liege,
4 * Belgium
5 *
6 * This program is free software: you can redistribute it and/or
7 * modify it under the terms of the GNU General Public License as
8 * published by the Free Software Foundation, either version 3 of the
9 * License, or (at your option) any later version.
10 *
11 * In addition, as a special exception, the copyright holders of this
12 * program give permission to link the code of its release with the
13 * OpenSSL project's "OpenSSL" library (or with modified versions of it
14 * that use the same license as the "OpenSSL" library), and distribute
15 * the linked executables. You must obey the GNU General Public License
16 * in all respects for all of the code used other than "OpenSSL". If you
17 * modify file(s) with this exception, you may extend this exception to
18 * your version of the file(s), but you are not obligated to do so. If
19 * you do not wish to do so, delete this exception statement from your
20 * version. If you delete this exception statement from all source files
21 * in the program, then also delete it here.
22 *
23 * This program is distributed in the hope that it will be useful, but
24 * WITHOUT ANY WARRANTY; without even the implied warranty of
25 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
26 * General Public License for more details.
27 *
28 * You should have received a copy of the GNU General Public License
29 * along with this program. If not, see <http://www.gnu.org/licenses/>.
30 **/
31
32
33 #pragma once
34
35 #include "ZipWriter.h"
36
37 #include <map>
38 #include <list>
39 #include <boost/lexical_cast.hpp>
40
41 namespace Orthanc
42 {
43 class HierarchicalZipWriter
44 {
45 FRIEND_TEST(HierarchicalZipWriter, Index);
46 FRIEND_TEST(HierarchicalZipWriter, Filenames);
47
48 private:
49 class Index
50 {
51 private:
52 struct Directory
53 {
54 typedef std::map<std::string, unsigned int> Content;
55
56 std::string name_;
57 Content content_;
58 };
59
60 typedef std::list<Directory*> Stack;
61
62 Stack stack_;
63
64 std::string GetCurrentDirectoryPath() const;
65
66 std::string EnsureUniqueFilename(const char* filename);
67
68 public:
69 Index();
70
71 ~Index();
72
73 bool IsRoot() const
74 {
75 return stack_.size() == 1;
76 }
77
78 std::string CreateFile(const char* name);
79
80 void CreateDirectory(const char* name);
81
82 void CloseDirectory();
83
84 static std::string KeepAlphanumeric(const std::string& source);
85 };
86
87 Index indexer_;
88 ZipWriter writer_;
89
90 public:
91 HierarchicalZipWriter(const char* path);
92
93 ~HierarchicalZipWriter();
94
95 void SetCompressionLevel(uint8_t level)
96 {
97 writer_.SetCompressionLevel(level);
98 }
99
100 uint8_t GetCompressionLevel() const
101 {
102 return writer_.GetCompressionLevel();
103 }
104
105 void CreateFile(const char* name);
106
107 void CreateDirectory(const char* name);
108
109 void CloseDirectory();
110
111 void Write(const char* data, size_t length)
112 {
113 writer_.Write(data, length);
114 }
115
116 void Write(const std::string& data)
117 {
118 writer_.Write(data);
119 }
120 };
121 }