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
103
104
105
106
107
108
109
110
111
112
113
114
115
116
|
// SPDX-FileCopyrightText: Copyright (c) Ken Martin, Will Schroeder, Bill Lorensen
// SPDX-License-Identifier: BSD-3-Clause
#include "vtkMemoryResourceStream.h"
#include "vtkObjectFactory.h"
#include <algorithm> // std::min
#include <cstring> // std::memcpy
VTK_ABI_NAMESPACE_BEGIN
vtkStandardNewMacro(vtkMemoryResourceStream);
//------------------------------------------------------------------------------
vtkMemoryResourceStream::vtkMemoryResourceStream()
: vtkResourceStream{ true }
{
}
//------------------------------------------------------------------------------
void vtkMemoryResourceStream::SetBuffer(const void* buffer, std::size_t size, bool copy)
{
if (buffer == nullptr && size != 0)
{
vtkErrorMacro("buffer must not be nullptr if size > 0");
return;
}
this->Size = size;
this->Pos = 0;
this->Eos = (this->Size == 0);
this->Holder.reset();
if (copy && size > 0)
{
std::unique_ptr<unsigned char[]> ptr{ new unsigned char[size] };
std::memcpy(ptr.get(), buffer, size);
this->Buffer = ptr.get();
this->Holder = MakeHolder(std::move(ptr));
}
else
{
this->Buffer = static_cast<const unsigned char*>(buffer);
}
this->Modified();
}
//------------------------------------------------------------------------------
std::size_t vtkMemoryResourceStream::Read(void* buffer, std::size_t bytes)
{
if (bytes == 0)
{
return 0;
}
const auto sbytes = static_cast<vtkTypeInt64>(bytes);
const auto ssize = static_cast<vtkTypeInt64>(this->Size);
const auto read = std::min(sbytes, ssize - this->Pos);
if (read <= 0)
{
this->Eos = true;
return 0;
}
std::memcpy(buffer, this->Buffer + this->Pos, static_cast<std::size_t>(read));
this->Pos += read;
this->Eos = read != sbytes;
return read;
}
//------------------------------------------------------------------------------
bool vtkMemoryResourceStream::EndOfStream()
{
return this->Eos;
}
//------------------------------------------------------------------------------
vtkTypeInt64 vtkMemoryResourceStream::Seek(vtkTypeInt64 pos, SeekDirection dir)
{
if (dir == SeekDirection::Begin)
{
this->Pos = pos;
}
else if (dir == SeekDirection::Current)
{
this->Pos += pos;
}
else
{
this->Pos = static_cast<vtkTypeInt64>(this->Size) + pos;
}
this->Eos = false;
return this->Pos;
}
//------------------------------------------------------------------------------
vtkTypeInt64 vtkMemoryResourceStream::Tell()
{
return this->Pos;
}
//------------------------------------------------------------------------------
void vtkMemoryResourceStream::PrintSelf(ostream& os, vtkIndent indent)
{
Superclass::PrintSelf(os, indent);
os << indent << "Buffer: " << this->Buffer << "\n";
os << indent << "Size: " << this->Size << "o\n";
os << indent << "Position: " << this->Pos << "\n";
}
VTK_ABI_NAMESPACE_END
|
