1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
|
// SPDX-FileCopyrightText: Copyright (c) Ken Martin, Will Schroeder, Bill Lorensen
// SPDX-License-Identifier: BSD-3-Clause
#include "vtkZLibDataCompressor.h"
#include "vtkObjectFactory.h"
#include "vtk_zlib.h"
VTK_ABI_NAMESPACE_BEGIN
vtkStandardNewMacro(vtkZLibDataCompressor);
//------------------------------------------------------------------------------
vtkZLibDataCompressor::vtkZLibDataCompressor()
{
this->CompressionLevel = Z_DEFAULT_COMPRESSION;
}
//------------------------------------------------------------------------------
vtkZLibDataCompressor::~vtkZLibDataCompressor() = default;
//------------------------------------------------------------------------------
void vtkZLibDataCompressor::PrintSelf(ostream& os, vtkIndent indent)
{
this->Superclass::PrintSelf(os, indent);
os << indent << "CompressionLevel: " << this->CompressionLevel << endl;
}
//------------------------------------------------------------------------------
size_t vtkZLibDataCompressor::CompressBuffer(unsigned char const* uncompressedData,
size_t uncompressedSize, unsigned char* compressedData, size_t compressionSpace)
{
uLongf cs = static_cast<uLongf>(compressionSpace);
Bytef* cd = reinterpret_cast<Bytef*>(compressedData);
const Bytef* ud = reinterpret_cast<const Bytef*>(uncompressedData);
uLong us = static_cast<uLong>(uncompressedSize);
// Call zlib's compress function.
if (compress2(cd, &cs, ud, us, this->CompressionLevel) != Z_OK)
{
vtkErrorMacro("Zlib error while compressing data.");
return 0;
}
return static_cast<size_t>(cs);
}
//------------------------------------------------------------------------------
size_t vtkZLibDataCompressor::UncompressBuffer(unsigned char const* compressedData,
size_t compressedSize, unsigned char* uncompressedData, size_t uncompressedSize)
{
uLongf us = static_cast<uLongf>(uncompressedSize);
Bytef* ud = reinterpret_cast<Bytef*>(uncompressedData);
const Bytef* cd = reinterpret_cast<const Bytef*>(compressedData);
uLong cs = static_cast<uLong>(compressedSize);
// Call zlib's uncompress function.
if (uncompress(ud, &us, cd, cs) != Z_OK)
{
vtkErrorMacro("Zlib error while uncompressing data.");
return 0;
}
// Make sure the output size matched that expected.
if (us != static_cast<uLongf>(uncompressedSize))
{
vtkErrorMacro("Decompression produced incorrect size.\n"
"Expected "
<< uncompressedSize << " and got " << us);
return 0;
}
return static_cast<size_t>(us);
}
//------------------------------------------------------------------------------
int vtkZLibDataCompressor::GetCompressionLevel()
{
vtkDebugMacro(<< this->GetClassName() << " (" << this << "): returning CompressionLevel "
<< this->CompressionLevel);
return this->CompressionLevel;
}
//------------------------------------------------------------------------------
void vtkZLibDataCompressor::SetCompressionLevel(int compressionLevel)
{
int min = 1;
int max = 9;
vtkDebugMacro(<< this->GetClassName() << " (" << this << "): setting CompressionLevel to "
<< compressionLevel);
if (this->CompressionLevel !=
(compressionLevel < min ? min : (compressionLevel > max ? max : compressionLevel)))
{
this->CompressionLevel =
(compressionLevel < min ? min : (compressionLevel > max ? max : compressionLevel));
this->Modified();
}
}
//------------------------------------------------------------------------------
size_t vtkZLibDataCompressor::GetMaximumCompressionSpace(size_t size)
{
// ZLib specifies that destination buffer must be 0.1% larger + 12 bytes.
return size + (size + 999) / 1000 + 12;
}
VTK_ABI_NAMESPACE_END
|
