isotree 0.2.2 → 0.3.0
Sign up to get free protection for your applications and to get access to all the features.
- checksums.yaml +4 -4
- data/CHANGELOG.md +8 -1
- data/LICENSE.txt +2 -2
- data/README.md +32 -14
- data/ext/isotree/ext.cpp +144 -31
- data/ext/isotree/extconf.rb +7 -7
- data/lib/isotree/isolation_forest.rb +110 -30
- data/lib/isotree/version.rb +1 -1
- data/vendor/isotree/LICENSE +1 -1
- data/vendor/isotree/README.md +165 -27
- data/vendor/isotree/include/isotree.hpp +2111 -0
- data/vendor/isotree/include/isotree_oop.hpp +394 -0
- data/vendor/isotree/inst/COPYRIGHTS +62 -0
- data/vendor/isotree/src/RcppExports.cpp +525 -52
- data/vendor/isotree/src/Rwrapper.cpp +1931 -268
- data/vendor/isotree/src/c_interface.cpp +953 -0
- data/vendor/isotree/src/crit.hpp +4232 -0
- data/vendor/isotree/src/dist.hpp +1886 -0
- data/vendor/isotree/src/exp_depth_table.hpp +134 -0
- data/vendor/isotree/src/extended.hpp +1444 -0
- data/vendor/isotree/src/external_facing_generic.hpp +399 -0
- data/vendor/isotree/src/fit_model.hpp +2401 -0
- data/vendor/isotree/src/{dealloc.cpp → headers_joined.hpp} +38 -22
- data/vendor/isotree/src/helpers_iforest.hpp +813 -0
- data/vendor/isotree/src/{impute.cpp → impute.hpp} +353 -122
- data/vendor/isotree/src/indexer.cpp +515 -0
- data/vendor/isotree/src/instantiate_template_headers.cpp +118 -0
- data/vendor/isotree/src/instantiate_template_headers.hpp +240 -0
- data/vendor/isotree/src/isoforest.hpp +1659 -0
- data/vendor/isotree/src/isotree.hpp +1804 -392
- data/vendor/isotree/src/isotree_exportable.hpp +99 -0
- data/vendor/isotree/src/merge_models.cpp +159 -16
- data/vendor/isotree/src/mult.hpp +1321 -0
- data/vendor/isotree/src/oop_interface.cpp +842 -0
- data/vendor/isotree/src/oop_interface.hpp +278 -0
- data/vendor/isotree/src/other_helpers.hpp +219 -0
- data/vendor/isotree/src/predict.hpp +1932 -0
- data/vendor/isotree/src/python_helpers.hpp +134 -0
- data/vendor/isotree/src/ref_indexer.hpp +154 -0
- data/vendor/isotree/src/robinmap/LICENSE +21 -0
- data/vendor/isotree/src/robinmap/README.md +483 -0
- data/vendor/isotree/src/robinmap/include/tsl/robin_growth_policy.h +406 -0
- data/vendor/isotree/src/robinmap/include/tsl/robin_hash.h +1620 -0
- data/vendor/isotree/src/robinmap/include/tsl/robin_map.h +807 -0
- data/vendor/isotree/src/robinmap/include/tsl/robin_set.h +660 -0
- data/vendor/isotree/src/serialize.cpp +4300 -139
- data/vendor/isotree/src/sql.cpp +141 -59
- data/vendor/isotree/src/subset_models.cpp +174 -0
- data/vendor/isotree/src/utils.hpp +3808 -0
- data/vendor/isotree/src/xoshiro.hpp +467 -0
- data/vendor/isotree/src/ziggurat.hpp +405 -0
- metadata +38 -104
- data/vendor/cereal/LICENSE +0 -24
- data/vendor/cereal/README.md +0 -85
- data/vendor/cereal/include/cereal/access.hpp +0 -351
- data/vendor/cereal/include/cereal/archives/adapters.hpp +0 -163
- data/vendor/cereal/include/cereal/archives/binary.hpp +0 -169
- data/vendor/cereal/include/cereal/archives/json.hpp +0 -1019
- data/vendor/cereal/include/cereal/archives/portable_binary.hpp +0 -334
- data/vendor/cereal/include/cereal/archives/xml.hpp +0 -956
- data/vendor/cereal/include/cereal/cereal.hpp +0 -1089
- data/vendor/cereal/include/cereal/details/helpers.hpp +0 -422
- data/vendor/cereal/include/cereal/details/polymorphic_impl.hpp +0 -796
- data/vendor/cereal/include/cereal/details/polymorphic_impl_fwd.hpp +0 -65
- data/vendor/cereal/include/cereal/details/static_object.hpp +0 -127
- data/vendor/cereal/include/cereal/details/traits.hpp +0 -1411
- data/vendor/cereal/include/cereal/details/util.hpp +0 -84
- data/vendor/cereal/include/cereal/external/base64.hpp +0 -134
- data/vendor/cereal/include/cereal/external/rapidjson/allocators.h +0 -284
- data/vendor/cereal/include/cereal/external/rapidjson/cursorstreamwrapper.h +0 -78
- data/vendor/cereal/include/cereal/external/rapidjson/document.h +0 -2652
- data/vendor/cereal/include/cereal/external/rapidjson/encodedstream.h +0 -299
- data/vendor/cereal/include/cereal/external/rapidjson/encodings.h +0 -716
- data/vendor/cereal/include/cereal/external/rapidjson/error/en.h +0 -74
- data/vendor/cereal/include/cereal/external/rapidjson/error/error.h +0 -161
- data/vendor/cereal/include/cereal/external/rapidjson/filereadstream.h +0 -99
- data/vendor/cereal/include/cereal/external/rapidjson/filewritestream.h +0 -104
- data/vendor/cereal/include/cereal/external/rapidjson/fwd.h +0 -151
- data/vendor/cereal/include/cereal/external/rapidjson/internal/biginteger.h +0 -290
- data/vendor/cereal/include/cereal/external/rapidjson/internal/diyfp.h +0 -271
- data/vendor/cereal/include/cereal/external/rapidjson/internal/dtoa.h +0 -245
- data/vendor/cereal/include/cereal/external/rapidjson/internal/ieee754.h +0 -78
- data/vendor/cereal/include/cereal/external/rapidjson/internal/itoa.h +0 -308
- data/vendor/cereal/include/cereal/external/rapidjson/internal/meta.h +0 -186
- data/vendor/cereal/include/cereal/external/rapidjson/internal/pow10.h +0 -55
- data/vendor/cereal/include/cereal/external/rapidjson/internal/regex.h +0 -740
- data/vendor/cereal/include/cereal/external/rapidjson/internal/stack.h +0 -232
- data/vendor/cereal/include/cereal/external/rapidjson/internal/strfunc.h +0 -69
- data/vendor/cereal/include/cereal/external/rapidjson/internal/strtod.h +0 -290
- data/vendor/cereal/include/cereal/external/rapidjson/internal/swap.h +0 -46
- data/vendor/cereal/include/cereal/external/rapidjson/istreamwrapper.h +0 -128
- data/vendor/cereal/include/cereal/external/rapidjson/memorybuffer.h +0 -70
- data/vendor/cereal/include/cereal/external/rapidjson/memorystream.h +0 -71
- data/vendor/cereal/include/cereal/external/rapidjson/msinttypes/inttypes.h +0 -316
- data/vendor/cereal/include/cereal/external/rapidjson/msinttypes/stdint.h +0 -300
- data/vendor/cereal/include/cereal/external/rapidjson/ostreamwrapper.h +0 -81
- data/vendor/cereal/include/cereal/external/rapidjson/pointer.h +0 -1414
- data/vendor/cereal/include/cereal/external/rapidjson/prettywriter.h +0 -277
- data/vendor/cereal/include/cereal/external/rapidjson/rapidjson.h +0 -656
- data/vendor/cereal/include/cereal/external/rapidjson/reader.h +0 -2230
- data/vendor/cereal/include/cereal/external/rapidjson/schema.h +0 -2497
- data/vendor/cereal/include/cereal/external/rapidjson/stream.h +0 -223
- data/vendor/cereal/include/cereal/external/rapidjson/stringbuffer.h +0 -121
- data/vendor/cereal/include/cereal/external/rapidjson/writer.h +0 -709
- data/vendor/cereal/include/cereal/external/rapidxml/license.txt +0 -52
- data/vendor/cereal/include/cereal/external/rapidxml/manual.html +0 -406
- data/vendor/cereal/include/cereal/external/rapidxml/rapidxml.hpp +0 -2624
- data/vendor/cereal/include/cereal/external/rapidxml/rapidxml_iterators.hpp +0 -175
- data/vendor/cereal/include/cereal/external/rapidxml/rapidxml_print.hpp +0 -428
- data/vendor/cereal/include/cereal/external/rapidxml/rapidxml_utils.hpp +0 -123
- data/vendor/cereal/include/cereal/macros.hpp +0 -154
- data/vendor/cereal/include/cereal/specialize.hpp +0 -139
- data/vendor/cereal/include/cereal/types/array.hpp +0 -79
- data/vendor/cereal/include/cereal/types/atomic.hpp +0 -55
- data/vendor/cereal/include/cereal/types/base_class.hpp +0 -203
- data/vendor/cereal/include/cereal/types/bitset.hpp +0 -176
- data/vendor/cereal/include/cereal/types/boost_variant.hpp +0 -164
- data/vendor/cereal/include/cereal/types/chrono.hpp +0 -72
- data/vendor/cereal/include/cereal/types/common.hpp +0 -129
- data/vendor/cereal/include/cereal/types/complex.hpp +0 -56
- data/vendor/cereal/include/cereal/types/concepts/pair_associative_container.hpp +0 -73
- data/vendor/cereal/include/cereal/types/deque.hpp +0 -62
- data/vendor/cereal/include/cereal/types/forward_list.hpp +0 -68
- data/vendor/cereal/include/cereal/types/functional.hpp +0 -43
- data/vendor/cereal/include/cereal/types/list.hpp +0 -62
- data/vendor/cereal/include/cereal/types/map.hpp +0 -36
- data/vendor/cereal/include/cereal/types/memory.hpp +0 -425
- data/vendor/cereal/include/cereal/types/optional.hpp +0 -66
- data/vendor/cereal/include/cereal/types/polymorphic.hpp +0 -483
- data/vendor/cereal/include/cereal/types/queue.hpp +0 -132
- data/vendor/cereal/include/cereal/types/set.hpp +0 -103
- data/vendor/cereal/include/cereal/types/stack.hpp +0 -76
- data/vendor/cereal/include/cereal/types/string.hpp +0 -61
- data/vendor/cereal/include/cereal/types/tuple.hpp +0 -123
- data/vendor/cereal/include/cereal/types/unordered_map.hpp +0 -36
- data/vendor/cereal/include/cereal/types/unordered_set.hpp +0 -99
- data/vendor/cereal/include/cereal/types/utility.hpp +0 -47
- data/vendor/cereal/include/cereal/types/valarray.hpp +0 -89
- data/vendor/cereal/include/cereal/types/variant.hpp +0 -109
- data/vendor/cereal/include/cereal/types/vector.hpp +0 -112
- data/vendor/cereal/include/cereal/version.hpp +0 -52
- data/vendor/isotree/src/Makevars +0 -4
- data/vendor/isotree/src/crit.cpp +0 -912
- data/vendor/isotree/src/dist.cpp +0 -749
- data/vendor/isotree/src/extended.cpp +0 -790
- data/vendor/isotree/src/fit_model.cpp +0 -1090
- data/vendor/isotree/src/helpers_iforest.cpp +0 -324
- data/vendor/isotree/src/isoforest.cpp +0 -771
- data/vendor/isotree/src/mult.cpp +0 -607
- data/vendor/isotree/src/predict.cpp +0 -853
- data/vendor/isotree/src/utils.cpp +0 -1566
@@ -1,709 +0,0 @@
|
|
1
|
-
// Tencent is pleased to support the open source community by making RapidJSON available.
|
2
|
-
//
|
3
|
-
// Copyright (C) 2015 THL A29 Limited, a Tencent company, and Milo Yip. All rights reserved.
|
4
|
-
//
|
5
|
-
// Licensed under the MIT License (the "License"); you may not use this file except
|
6
|
-
// in compliance with the License. You may obtain a copy of the License at
|
7
|
-
//
|
8
|
-
// http://opensource.org/licenses/MIT
|
9
|
-
//
|
10
|
-
// Unless required by applicable law or agreed to in writing, software distributed
|
11
|
-
// under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR
|
12
|
-
// CONDITIONS OF ANY KIND, either express or implied. See the License for the
|
13
|
-
// specific language governing permissions and limitations under the License.
|
14
|
-
|
15
|
-
#ifndef CEREAL_RAPIDJSON_WRITER_H_
|
16
|
-
#define CEREAL_RAPIDJSON_WRITER_H_
|
17
|
-
|
18
|
-
#include "stream.h"
|
19
|
-
#include "internal/meta.h"
|
20
|
-
#include "internal/stack.h"
|
21
|
-
#include "internal/strfunc.h"
|
22
|
-
#include "internal/dtoa.h"
|
23
|
-
#include "internal/itoa.h"
|
24
|
-
#include "stringbuffer.h"
|
25
|
-
#include <new> // placement new
|
26
|
-
|
27
|
-
#if defined(CEREAL_RAPIDJSON_SIMD) && defined(_MSC_VER)
|
28
|
-
#include <intrin.h>
|
29
|
-
#pragma intrinsic(_BitScanForward)
|
30
|
-
#endif
|
31
|
-
#ifdef CEREAL_RAPIDJSON_SSE42
|
32
|
-
#include <nmmintrin.h>
|
33
|
-
#elif defined(CEREAL_RAPIDJSON_SSE2)
|
34
|
-
#include <emmintrin.h>
|
35
|
-
#elif defined(CEREAL_RAPIDJSON_NEON)
|
36
|
-
#include <arm_neon.h>
|
37
|
-
#endif
|
38
|
-
|
39
|
-
#ifdef __clang__
|
40
|
-
CEREAL_RAPIDJSON_DIAG_PUSH
|
41
|
-
CEREAL_RAPIDJSON_DIAG_OFF(padded)
|
42
|
-
CEREAL_RAPIDJSON_DIAG_OFF(unreachable-code)
|
43
|
-
CEREAL_RAPIDJSON_DIAG_OFF(c++98-compat)
|
44
|
-
#elif defined(_MSC_VER)
|
45
|
-
CEREAL_RAPIDJSON_DIAG_PUSH
|
46
|
-
CEREAL_RAPIDJSON_DIAG_OFF(4127) // conditional expression is constant
|
47
|
-
#endif
|
48
|
-
|
49
|
-
CEREAL_RAPIDJSON_NAMESPACE_BEGIN
|
50
|
-
|
51
|
-
///////////////////////////////////////////////////////////////////////////////
|
52
|
-
// WriteFlag
|
53
|
-
|
54
|
-
/*! \def CEREAL_RAPIDJSON_WRITE_DEFAULT_FLAGS
|
55
|
-
\ingroup CEREAL_RAPIDJSON_CONFIG
|
56
|
-
\brief User-defined kWriteDefaultFlags definition.
|
57
|
-
|
58
|
-
User can define this as any \c WriteFlag combinations.
|
59
|
-
*/
|
60
|
-
#ifndef CEREAL_RAPIDJSON_WRITE_DEFAULT_FLAGS
|
61
|
-
#define CEREAL_RAPIDJSON_WRITE_DEFAULT_FLAGS kWriteNoFlags
|
62
|
-
#endif
|
63
|
-
|
64
|
-
//! Combination of writeFlags
|
65
|
-
enum WriteFlag {
|
66
|
-
kWriteNoFlags = 0, //!< No flags are set.
|
67
|
-
kWriteValidateEncodingFlag = 1, //!< Validate encoding of JSON strings.
|
68
|
-
kWriteNanAndInfFlag = 2, //!< Allow writing of Infinity, -Infinity and NaN.
|
69
|
-
kWriteDefaultFlags = CEREAL_RAPIDJSON_WRITE_DEFAULT_FLAGS //!< Default write flags. Can be customized by defining CEREAL_RAPIDJSON_WRITE_DEFAULT_FLAGS
|
70
|
-
};
|
71
|
-
|
72
|
-
//! JSON writer
|
73
|
-
/*! Writer implements the concept Handler.
|
74
|
-
It generates JSON text by events to an output os.
|
75
|
-
|
76
|
-
User may programmatically calls the functions of a writer to generate JSON text.
|
77
|
-
|
78
|
-
On the other side, a writer can also be passed to objects that generates events,
|
79
|
-
|
80
|
-
for example Reader::Parse() and Document::Accept().
|
81
|
-
|
82
|
-
\tparam OutputStream Type of output stream.
|
83
|
-
\tparam SourceEncoding Encoding of source string.
|
84
|
-
\tparam TargetEncoding Encoding of output stream.
|
85
|
-
\tparam StackAllocator Type of allocator for allocating memory of stack.
|
86
|
-
\note implements Handler concept
|
87
|
-
*/
|
88
|
-
template<typename OutputStream, typename SourceEncoding = UTF8<>, typename TargetEncoding = UTF8<>, typename StackAllocator = CrtAllocator, unsigned writeFlags = kWriteDefaultFlags>
|
89
|
-
class Writer {
|
90
|
-
public:
|
91
|
-
typedef typename SourceEncoding::Ch Ch;
|
92
|
-
|
93
|
-
static const int kDefaultMaxDecimalPlaces = 324;
|
94
|
-
|
95
|
-
//! Constructor
|
96
|
-
/*! \param os Output stream.
|
97
|
-
\param stackAllocator User supplied allocator. If it is null, it will create a private one.
|
98
|
-
\param levelDepth Initial capacity of stack.
|
99
|
-
*/
|
100
|
-
explicit
|
101
|
-
Writer(OutputStream& os, StackAllocator* stackAllocator = 0, size_t levelDepth = kDefaultLevelDepth) :
|
102
|
-
os_(&os), level_stack_(stackAllocator, levelDepth * sizeof(Level)), maxDecimalPlaces_(kDefaultMaxDecimalPlaces), hasRoot_(false) {}
|
103
|
-
|
104
|
-
explicit
|
105
|
-
Writer(StackAllocator* allocator = 0, size_t levelDepth = kDefaultLevelDepth) :
|
106
|
-
os_(0), level_stack_(allocator, levelDepth * sizeof(Level)), maxDecimalPlaces_(kDefaultMaxDecimalPlaces), hasRoot_(false) {}
|
107
|
-
|
108
|
-
#if CEREAL_RAPIDJSON_HAS_CXX11_RVALUE_REFS
|
109
|
-
Writer(Writer&& rhs) :
|
110
|
-
os_(rhs.os_), level_stack_(std::move(rhs.level_stack_)), maxDecimalPlaces_(rhs.maxDecimalPlaces_), hasRoot_(rhs.hasRoot_) {
|
111
|
-
rhs.os_ = 0;
|
112
|
-
}
|
113
|
-
#endif
|
114
|
-
|
115
|
-
//! Reset the writer with a new stream.
|
116
|
-
/*!
|
117
|
-
This function reset the writer with a new stream and default settings,
|
118
|
-
in order to make a Writer object reusable for output multiple JSONs.
|
119
|
-
|
120
|
-
\param os New output stream.
|
121
|
-
\code
|
122
|
-
Writer<OutputStream> writer(os1);
|
123
|
-
writer.StartObject();
|
124
|
-
// ...
|
125
|
-
writer.EndObject();
|
126
|
-
|
127
|
-
writer.Reset(os2);
|
128
|
-
writer.StartObject();
|
129
|
-
// ...
|
130
|
-
writer.EndObject();
|
131
|
-
\endcode
|
132
|
-
*/
|
133
|
-
void Reset(OutputStream& os) {
|
134
|
-
os_ = &os;
|
135
|
-
hasRoot_ = false;
|
136
|
-
level_stack_.Clear();
|
137
|
-
}
|
138
|
-
|
139
|
-
//! Checks whether the output is a complete JSON.
|
140
|
-
/*!
|
141
|
-
A complete JSON has a complete root object or array.
|
142
|
-
*/
|
143
|
-
bool IsComplete() const {
|
144
|
-
return hasRoot_ && level_stack_.Empty();
|
145
|
-
}
|
146
|
-
|
147
|
-
int GetMaxDecimalPlaces() const {
|
148
|
-
return maxDecimalPlaces_;
|
149
|
-
}
|
150
|
-
|
151
|
-
//! Sets the maximum number of decimal places for double output.
|
152
|
-
/*!
|
153
|
-
This setting truncates the output with specified number of decimal places.
|
154
|
-
|
155
|
-
For example,
|
156
|
-
|
157
|
-
\code
|
158
|
-
writer.SetMaxDecimalPlaces(3);
|
159
|
-
writer.StartArray();
|
160
|
-
writer.Double(0.12345); // "0.123"
|
161
|
-
writer.Double(0.0001); // "0.0"
|
162
|
-
writer.Double(1.234567890123456e30); // "1.234567890123456e30" (do not truncate significand for positive exponent)
|
163
|
-
writer.Double(1.23e-4); // "0.0" (do truncate significand for negative exponent)
|
164
|
-
writer.EndArray();
|
165
|
-
\endcode
|
166
|
-
|
167
|
-
The default setting does not truncate any decimal places. You can restore to this setting by calling
|
168
|
-
\code
|
169
|
-
writer.SetMaxDecimalPlaces(Writer::kDefaultMaxDecimalPlaces);
|
170
|
-
\endcode
|
171
|
-
*/
|
172
|
-
void SetMaxDecimalPlaces(int maxDecimalPlaces) {
|
173
|
-
maxDecimalPlaces_ = maxDecimalPlaces;
|
174
|
-
}
|
175
|
-
|
176
|
-
/*!@name Implementation of Handler
|
177
|
-
\see Handler
|
178
|
-
*/
|
179
|
-
//@{
|
180
|
-
|
181
|
-
bool Null() { Prefix(kNullType); return EndValue(WriteNull()); }
|
182
|
-
bool Bool(bool b) { Prefix(b ? kTrueType : kFalseType); return EndValue(WriteBool(b)); }
|
183
|
-
bool Int(int i) { Prefix(kNumberType); return EndValue(WriteInt(i)); }
|
184
|
-
bool Uint(unsigned u) { Prefix(kNumberType); return EndValue(WriteUint(u)); }
|
185
|
-
bool Int64(int64_t i64) { Prefix(kNumberType); return EndValue(WriteInt64(i64)); }
|
186
|
-
bool Uint64(uint64_t u64) { Prefix(kNumberType); return EndValue(WriteUint64(u64)); }
|
187
|
-
|
188
|
-
//! Writes the given \c double value to the stream
|
189
|
-
/*!
|
190
|
-
\param d The value to be written.
|
191
|
-
\return Whether it is succeed.
|
192
|
-
*/
|
193
|
-
bool Double(double d) { Prefix(kNumberType); return EndValue(WriteDouble(d)); }
|
194
|
-
|
195
|
-
bool RawNumber(const Ch* str, SizeType length, bool copy = false) {
|
196
|
-
CEREAL_RAPIDJSON_ASSERT(str != 0);
|
197
|
-
(void)copy;
|
198
|
-
Prefix(kNumberType);
|
199
|
-
return EndValue(WriteString(str, length));
|
200
|
-
}
|
201
|
-
|
202
|
-
bool String(const Ch* str, SizeType length, bool copy = false) {
|
203
|
-
CEREAL_RAPIDJSON_ASSERT(str != 0);
|
204
|
-
(void)copy;
|
205
|
-
Prefix(kStringType);
|
206
|
-
return EndValue(WriteString(str, length));
|
207
|
-
}
|
208
|
-
|
209
|
-
#if CEREAL_RAPIDJSON_HAS_STDSTRING
|
210
|
-
bool String(const std::basic_string<Ch>& str) {
|
211
|
-
return String(str.data(), SizeType(str.size()));
|
212
|
-
}
|
213
|
-
#endif
|
214
|
-
|
215
|
-
bool StartObject() {
|
216
|
-
Prefix(kObjectType);
|
217
|
-
new (level_stack_.template Push<Level>()) Level(false);
|
218
|
-
return WriteStartObject();
|
219
|
-
}
|
220
|
-
|
221
|
-
bool Key(const Ch* str, SizeType length, bool copy = false) { return String(str, length, copy); }
|
222
|
-
|
223
|
-
#if CEREAL_RAPIDJSON_HAS_STDSTRING
|
224
|
-
bool Key(const std::basic_string<Ch>& str)
|
225
|
-
{
|
226
|
-
return Key(str.data(), SizeType(str.size()));
|
227
|
-
}
|
228
|
-
#endif
|
229
|
-
|
230
|
-
bool EndObject(SizeType memberCount = 0) {
|
231
|
-
(void)memberCount;
|
232
|
-
CEREAL_RAPIDJSON_ASSERT(level_stack_.GetSize() >= sizeof(Level)); // not inside an Object
|
233
|
-
CEREAL_RAPIDJSON_ASSERT(!level_stack_.template Top<Level>()->inArray); // currently inside an Array, not Object
|
234
|
-
CEREAL_RAPIDJSON_ASSERT(0 == level_stack_.template Top<Level>()->valueCount % 2); // Object has a Key without a Value
|
235
|
-
level_stack_.template Pop<Level>(1);
|
236
|
-
return EndValue(WriteEndObject());
|
237
|
-
}
|
238
|
-
|
239
|
-
bool StartArray() {
|
240
|
-
Prefix(kArrayType);
|
241
|
-
new (level_stack_.template Push<Level>()) Level(true);
|
242
|
-
return WriteStartArray();
|
243
|
-
}
|
244
|
-
|
245
|
-
bool EndArray(SizeType elementCount = 0) {
|
246
|
-
(void)elementCount;
|
247
|
-
CEREAL_RAPIDJSON_ASSERT(level_stack_.GetSize() >= sizeof(Level));
|
248
|
-
CEREAL_RAPIDJSON_ASSERT(level_stack_.template Top<Level>()->inArray);
|
249
|
-
level_stack_.template Pop<Level>(1);
|
250
|
-
return EndValue(WriteEndArray());
|
251
|
-
}
|
252
|
-
//@}
|
253
|
-
|
254
|
-
/*! @name Convenience extensions */
|
255
|
-
//@{
|
256
|
-
|
257
|
-
//! Simpler but slower overload.
|
258
|
-
bool String(const Ch* const& str) { return String(str, internal::StrLen(str)); }
|
259
|
-
bool Key(const Ch* const& str) { return Key(str, internal::StrLen(str)); }
|
260
|
-
|
261
|
-
//@}
|
262
|
-
|
263
|
-
//! Write a raw JSON value.
|
264
|
-
/*!
|
265
|
-
For user to write a stringified JSON as a value.
|
266
|
-
|
267
|
-
\param json A well-formed JSON value. It should not contain null character within [0, length - 1] range.
|
268
|
-
\param length Length of the json.
|
269
|
-
\param type Type of the root of json.
|
270
|
-
*/
|
271
|
-
bool RawValue(const Ch* json, size_t length, Type type) {
|
272
|
-
CEREAL_RAPIDJSON_ASSERT(json != 0);
|
273
|
-
Prefix(type);
|
274
|
-
return EndValue(WriteRawValue(json, length));
|
275
|
-
}
|
276
|
-
|
277
|
-
//! Flush the output stream.
|
278
|
-
/*!
|
279
|
-
Allows the user to flush the output stream immediately.
|
280
|
-
*/
|
281
|
-
void Flush() {
|
282
|
-
os_->Flush();
|
283
|
-
}
|
284
|
-
|
285
|
-
protected:
|
286
|
-
//! Information for each nested level
|
287
|
-
struct Level {
|
288
|
-
Level(bool inArray_) : valueCount(0), inArray(inArray_) {}
|
289
|
-
size_t valueCount; //!< number of values in this level
|
290
|
-
bool inArray; //!< true if in array, otherwise in object
|
291
|
-
};
|
292
|
-
|
293
|
-
static const size_t kDefaultLevelDepth = 32;
|
294
|
-
|
295
|
-
bool WriteNull() {
|
296
|
-
PutReserve(*os_, 4);
|
297
|
-
PutUnsafe(*os_, 'n'); PutUnsafe(*os_, 'u'); PutUnsafe(*os_, 'l'); PutUnsafe(*os_, 'l'); return true;
|
298
|
-
}
|
299
|
-
|
300
|
-
bool WriteBool(bool b) {
|
301
|
-
if (b) {
|
302
|
-
PutReserve(*os_, 4);
|
303
|
-
PutUnsafe(*os_, 't'); PutUnsafe(*os_, 'r'); PutUnsafe(*os_, 'u'); PutUnsafe(*os_, 'e');
|
304
|
-
}
|
305
|
-
else {
|
306
|
-
PutReserve(*os_, 5);
|
307
|
-
PutUnsafe(*os_, 'f'); PutUnsafe(*os_, 'a'); PutUnsafe(*os_, 'l'); PutUnsafe(*os_, 's'); PutUnsafe(*os_, 'e');
|
308
|
-
}
|
309
|
-
return true;
|
310
|
-
}
|
311
|
-
|
312
|
-
bool WriteInt(int i) {
|
313
|
-
char buffer[11];
|
314
|
-
const char* end = internal::i32toa(i, buffer);
|
315
|
-
PutReserve(*os_, static_cast<size_t>(end - buffer));
|
316
|
-
for (const char* p = buffer; p != end; ++p)
|
317
|
-
PutUnsafe(*os_, static_cast<typename OutputStream::Ch>(*p));
|
318
|
-
return true;
|
319
|
-
}
|
320
|
-
|
321
|
-
bool WriteUint(unsigned u) {
|
322
|
-
char buffer[10];
|
323
|
-
const char* end = internal::u32toa(u, buffer);
|
324
|
-
PutReserve(*os_, static_cast<size_t>(end - buffer));
|
325
|
-
for (const char* p = buffer; p != end; ++p)
|
326
|
-
PutUnsafe(*os_, static_cast<typename OutputStream::Ch>(*p));
|
327
|
-
return true;
|
328
|
-
}
|
329
|
-
|
330
|
-
bool WriteInt64(int64_t i64) {
|
331
|
-
char buffer[21];
|
332
|
-
const char* end = internal::i64toa(i64, buffer);
|
333
|
-
PutReserve(*os_, static_cast<size_t>(end - buffer));
|
334
|
-
for (const char* p = buffer; p != end; ++p)
|
335
|
-
PutUnsafe(*os_, static_cast<typename OutputStream::Ch>(*p));
|
336
|
-
return true;
|
337
|
-
}
|
338
|
-
|
339
|
-
bool WriteUint64(uint64_t u64) {
|
340
|
-
char buffer[20];
|
341
|
-
char* end = internal::u64toa(u64, buffer);
|
342
|
-
PutReserve(*os_, static_cast<size_t>(end - buffer));
|
343
|
-
for (char* p = buffer; p != end; ++p)
|
344
|
-
PutUnsafe(*os_, static_cast<typename OutputStream::Ch>(*p));
|
345
|
-
return true;
|
346
|
-
}
|
347
|
-
|
348
|
-
bool WriteDouble(double d) {
|
349
|
-
if (internal::Double(d).IsNanOrInf()) {
|
350
|
-
if (!(writeFlags & kWriteNanAndInfFlag))
|
351
|
-
return false;
|
352
|
-
if (internal::Double(d).IsNan()) {
|
353
|
-
PutReserve(*os_, 3);
|
354
|
-
PutUnsafe(*os_, 'N'); PutUnsafe(*os_, 'a'); PutUnsafe(*os_, 'N');
|
355
|
-
return true;
|
356
|
-
}
|
357
|
-
if (internal::Double(d).Sign()) {
|
358
|
-
PutReserve(*os_, 9);
|
359
|
-
PutUnsafe(*os_, '-');
|
360
|
-
}
|
361
|
-
else
|
362
|
-
PutReserve(*os_, 8);
|
363
|
-
PutUnsafe(*os_, 'I'); PutUnsafe(*os_, 'n'); PutUnsafe(*os_, 'f');
|
364
|
-
PutUnsafe(*os_, 'i'); PutUnsafe(*os_, 'n'); PutUnsafe(*os_, 'i'); PutUnsafe(*os_, 't'); PutUnsafe(*os_, 'y');
|
365
|
-
return true;
|
366
|
-
}
|
367
|
-
|
368
|
-
char buffer[25];
|
369
|
-
char* end = internal::dtoa(d, buffer, maxDecimalPlaces_);
|
370
|
-
PutReserve(*os_, static_cast<size_t>(end - buffer));
|
371
|
-
for (char* p = buffer; p != end; ++p)
|
372
|
-
PutUnsafe(*os_, static_cast<typename OutputStream::Ch>(*p));
|
373
|
-
return true;
|
374
|
-
}
|
375
|
-
|
376
|
-
bool WriteString(const Ch* str, SizeType length) {
|
377
|
-
static const typename OutputStream::Ch hexDigits[16] = { '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', 'A', 'B', 'C', 'D', 'E', 'F' };
|
378
|
-
static const char escape[256] = {
|
379
|
-
#define Z16 0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0
|
380
|
-
//0 1 2 3 4 5 6 7 8 9 A B C D E F
|
381
|
-
'u', 'u', 'u', 'u', 'u', 'u', 'u', 'u', 'b', 't', 'n', 'u', 'f', 'r', 'u', 'u', // 00
|
382
|
-
'u', 'u', 'u', 'u', 'u', 'u', 'u', 'u', 'u', 'u', 'u', 'u', 'u', 'u', 'u', 'u', // 10
|
383
|
-
0, 0, '"', 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, // 20
|
384
|
-
Z16, Z16, // 30~4F
|
385
|
-
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,'\\', 0, 0, 0, // 50
|
386
|
-
Z16, Z16, Z16, Z16, Z16, Z16, Z16, Z16, Z16, Z16 // 60~FF
|
387
|
-
#undef Z16
|
388
|
-
};
|
389
|
-
|
390
|
-
if (TargetEncoding::supportUnicode)
|
391
|
-
PutReserve(*os_, 2 + length * 6); // "\uxxxx..."
|
392
|
-
else
|
393
|
-
PutReserve(*os_, 2 + length * 12); // "\uxxxx\uyyyy..."
|
394
|
-
|
395
|
-
PutUnsafe(*os_, '\"');
|
396
|
-
GenericStringStream<SourceEncoding> is(str);
|
397
|
-
while (ScanWriteUnescapedString(is, length)) {
|
398
|
-
const Ch c = is.Peek();
|
399
|
-
if (!TargetEncoding::supportUnicode && static_cast<unsigned>(c) >= 0x80) {
|
400
|
-
// Unicode escaping
|
401
|
-
unsigned codepoint;
|
402
|
-
if (CEREAL_RAPIDJSON_UNLIKELY(!SourceEncoding::Decode(is, &codepoint)))
|
403
|
-
return false;
|
404
|
-
PutUnsafe(*os_, '\\');
|
405
|
-
PutUnsafe(*os_, 'u');
|
406
|
-
if (codepoint <= 0xD7FF || (codepoint >= 0xE000 && codepoint <= 0xFFFF)) {
|
407
|
-
PutUnsafe(*os_, hexDigits[(codepoint >> 12) & 15]);
|
408
|
-
PutUnsafe(*os_, hexDigits[(codepoint >> 8) & 15]);
|
409
|
-
PutUnsafe(*os_, hexDigits[(codepoint >> 4) & 15]);
|
410
|
-
PutUnsafe(*os_, hexDigits[(codepoint ) & 15]);
|
411
|
-
}
|
412
|
-
else {
|
413
|
-
CEREAL_RAPIDJSON_ASSERT(codepoint >= 0x010000 && codepoint <= 0x10FFFF);
|
414
|
-
// Surrogate pair
|
415
|
-
unsigned s = codepoint - 0x010000;
|
416
|
-
unsigned lead = (s >> 10) + 0xD800;
|
417
|
-
unsigned trail = (s & 0x3FF) + 0xDC00;
|
418
|
-
PutUnsafe(*os_, hexDigits[(lead >> 12) & 15]);
|
419
|
-
PutUnsafe(*os_, hexDigits[(lead >> 8) & 15]);
|
420
|
-
PutUnsafe(*os_, hexDigits[(lead >> 4) & 15]);
|
421
|
-
PutUnsafe(*os_, hexDigits[(lead ) & 15]);
|
422
|
-
PutUnsafe(*os_, '\\');
|
423
|
-
PutUnsafe(*os_, 'u');
|
424
|
-
PutUnsafe(*os_, hexDigits[(trail >> 12) & 15]);
|
425
|
-
PutUnsafe(*os_, hexDigits[(trail >> 8) & 15]);
|
426
|
-
PutUnsafe(*os_, hexDigits[(trail >> 4) & 15]);
|
427
|
-
PutUnsafe(*os_, hexDigits[(trail ) & 15]);
|
428
|
-
}
|
429
|
-
}
|
430
|
-
else if ((sizeof(Ch) == 1 || static_cast<unsigned>(c) < 256) && CEREAL_RAPIDJSON_UNLIKELY(escape[static_cast<unsigned char>(c)])) {
|
431
|
-
is.Take();
|
432
|
-
PutUnsafe(*os_, '\\');
|
433
|
-
PutUnsafe(*os_, static_cast<typename OutputStream::Ch>(escape[static_cast<unsigned char>(c)]));
|
434
|
-
if (escape[static_cast<unsigned char>(c)] == 'u') {
|
435
|
-
PutUnsafe(*os_, '0');
|
436
|
-
PutUnsafe(*os_, '0');
|
437
|
-
PutUnsafe(*os_, hexDigits[static_cast<unsigned char>(c) >> 4]);
|
438
|
-
PutUnsafe(*os_, hexDigits[static_cast<unsigned char>(c) & 0xF]);
|
439
|
-
}
|
440
|
-
}
|
441
|
-
else if (CEREAL_RAPIDJSON_UNLIKELY(!(writeFlags & kWriteValidateEncodingFlag ?
|
442
|
-
Transcoder<SourceEncoding, TargetEncoding>::Validate(is, *os_) :
|
443
|
-
Transcoder<SourceEncoding, TargetEncoding>::TranscodeUnsafe(is, *os_))))
|
444
|
-
return false;
|
445
|
-
}
|
446
|
-
PutUnsafe(*os_, '\"');
|
447
|
-
return true;
|
448
|
-
}
|
449
|
-
|
450
|
-
bool ScanWriteUnescapedString(GenericStringStream<SourceEncoding>& is, size_t length) {
|
451
|
-
return CEREAL_RAPIDJSON_LIKELY(is.Tell() < length);
|
452
|
-
}
|
453
|
-
|
454
|
-
bool WriteStartObject() { os_->Put('{'); return true; }
|
455
|
-
bool WriteEndObject() { os_->Put('}'); return true; }
|
456
|
-
bool WriteStartArray() { os_->Put('['); return true; }
|
457
|
-
bool WriteEndArray() { os_->Put(']'); return true; }
|
458
|
-
|
459
|
-
bool WriteRawValue(const Ch* json, size_t length) {
|
460
|
-
PutReserve(*os_, length);
|
461
|
-
GenericStringStream<SourceEncoding> is(json);
|
462
|
-
while (CEREAL_RAPIDJSON_LIKELY(is.Tell() < length)) {
|
463
|
-
CEREAL_RAPIDJSON_ASSERT(is.Peek() != '\0');
|
464
|
-
if (CEREAL_RAPIDJSON_UNLIKELY(!(writeFlags & kWriteValidateEncodingFlag ?
|
465
|
-
Transcoder<SourceEncoding, TargetEncoding>::Validate(is, *os_) :
|
466
|
-
Transcoder<SourceEncoding, TargetEncoding>::TranscodeUnsafe(is, *os_))))
|
467
|
-
return false;
|
468
|
-
}
|
469
|
-
return true;
|
470
|
-
}
|
471
|
-
|
472
|
-
void Prefix(Type type) {
|
473
|
-
(void)type;
|
474
|
-
if (CEREAL_RAPIDJSON_LIKELY(level_stack_.GetSize() != 0)) { // this value is not at root
|
475
|
-
Level* level = level_stack_.template Top<Level>();
|
476
|
-
if (level->valueCount > 0) {
|
477
|
-
if (level->inArray)
|
478
|
-
os_->Put(','); // add comma if it is not the first element in array
|
479
|
-
else // in object
|
480
|
-
os_->Put((level->valueCount % 2 == 0) ? ',' : ':');
|
481
|
-
}
|
482
|
-
if (!level->inArray && level->valueCount % 2 == 0)
|
483
|
-
CEREAL_RAPIDJSON_ASSERT(type == kStringType); // if it's in object, then even number should be a name
|
484
|
-
level->valueCount++;
|
485
|
-
}
|
486
|
-
else {
|
487
|
-
CEREAL_RAPIDJSON_ASSERT(!hasRoot_); // Should only has one and only one root.
|
488
|
-
hasRoot_ = true;
|
489
|
-
}
|
490
|
-
}
|
491
|
-
|
492
|
-
// Flush the value if it is the top level one.
|
493
|
-
bool EndValue(bool ret) {
|
494
|
-
if (CEREAL_RAPIDJSON_UNLIKELY(level_stack_.Empty())) // end of json text
|
495
|
-
Flush();
|
496
|
-
return ret;
|
497
|
-
}
|
498
|
-
|
499
|
-
OutputStream* os_;
|
500
|
-
internal::Stack<StackAllocator> level_stack_;
|
501
|
-
int maxDecimalPlaces_;
|
502
|
-
bool hasRoot_;
|
503
|
-
|
504
|
-
private:
|
505
|
-
// Prohibit copy constructor & assignment operator.
|
506
|
-
Writer(const Writer&);
|
507
|
-
Writer& operator=(const Writer&);
|
508
|
-
};
|
509
|
-
|
510
|
-
// Full specialization for StringStream to prevent memory copying
|
511
|
-
|
512
|
-
template<>
|
513
|
-
inline bool Writer<StringBuffer>::WriteInt(int i) {
|
514
|
-
char *buffer = os_->Push(11);
|
515
|
-
const char* end = internal::i32toa(i, buffer);
|
516
|
-
os_->Pop(static_cast<size_t>(11 - (end - buffer)));
|
517
|
-
return true;
|
518
|
-
}
|
519
|
-
|
520
|
-
template<>
|
521
|
-
inline bool Writer<StringBuffer>::WriteUint(unsigned u) {
|
522
|
-
char *buffer = os_->Push(10);
|
523
|
-
const char* end = internal::u32toa(u, buffer);
|
524
|
-
os_->Pop(static_cast<size_t>(10 - (end - buffer)));
|
525
|
-
return true;
|
526
|
-
}
|
527
|
-
|
528
|
-
template<>
|
529
|
-
inline bool Writer<StringBuffer>::WriteInt64(int64_t i64) {
|
530
|
-
char *buffer = os_->Push(21);
|
531
|
-
const char* end = internal::i64toa(i64, buffer);
|
532
|
-
os_->Pop(static_cast<size_t>(21 - (end - buffer)));
|
533
|
-
return true;
|
534
|
-
}
|
535
|
-
|
536
|
-
template<>
|
537
|
-
inline bool Writer<StringBuffer>::WriteUint64(uint64_t u) {
|
538
|
-
char *buffer = os_->Push(20);
|
539
|
-
const char* end = internal::u64toa(u, buffer);
|
540
|
-
os_->Pop(static_cast<size_t>(20 - (end - buffer)));
|
541
|
-
return true;
|
542
|
-
}
|
543
|
-
|
544
|
-
template<>
|
545
|
-
inline bool Writer<StringBuffer>::WriteDouble(double d) {
|
546
|
-
if (internal::Double(d).IsNanOrInf()) {
|
547
|
-
// Note: This code path can only be reached if (CEREAL_RAPIDJSON_WRITE_DEFAULT_FLAGS & kWriteNanAndInfFlag).
|
548
|
-
if (!(kWriteDefaultFlags & kWriteNanAndInfFlag))
|
549
|
-
return false;
|
550
|
-
if (internal::Double(d).IsNan()) {
|
551
|
-
PutReserve(*os_, 3);
|
552
|
-
PutUnsafe(*os_, 'N'); PutUnsafe(*os_, 'a'); PutUnsafe(*os_, 'N');
|
553
|
-
return true;
|
554
|
-
}
|
555
|
-
if (internal::Double(d).Sign()) {
|
556
|
-
PutReserve(*os_, 9);
|
557
|
-
PutUnsafe(*os_, '-');
|
558
|
-
}
|
559
|
-
else
|
560
|
-
PutReserve(*os_, 8);
|
561
|
-
PutUnsafe(*os_, 'I'); PutUnsafe(*os_, 'n'); PutUnsafe(*os_, 'f');
|
562
|
-
PutUnsafe(*os_, 'i'); PutUnsafe(*os_, 'n'); PutUnsafe(*os_, 'i'); PutUnsafe(*os_, 't'); PutUnsafe(*os_, 'y');
|
563
|
-
return true;
|
564
|
-
}
|
565
|
-
|
566
|
-
char *buffer = os_->Push(25);
|
567
|
-
char* end = internal::dtoa(d, buffer, maxDecimalPlaces_);
|
568
|
-
os_->Pop(static_cast<size_t>(25 - (end - buffer)));
|
569
|
-
return true;
|
570
|
-
}
|
571
|
-
|
572
|
-
#if defined(CEREAL_RAPIDJSON_SSE2) || defined(CEREAL_RAPIDJSON_SSE42)
|
573
|
-
template<>
|
574
|
-
inline bool Writer<StringBuffer>::ScanWriteUnescapedString(StringStream& is, size_t length) {
|
575
|
-
if (length < 16)
|
576
|
-
return CEREAL_RAPIDJSON_LIKELY(is.Tell() < length);
|
577
|
-
|
578
|
-
if (!CEREAL_RAPIDJSON_LIKELY(is.Tell() < length))
|
579
|
-
return false;
|
580
|
-
|
581
|
-
const char* p = is.src_;
|
582
|
-
const char* end = is.head_ + length;
|
583
|
-
const char* nextAligned = reinterpret_cast<const char*>((reinterpret_cast<size_t>(p) + 15) & static_cast<size_t>(~15));
|
584
|
-
const char* endAligned = reinterpret_cast<const char*>(reinterpret_cast<size_t>(end) & static_cast<size_t>(~15));
|
585
|
-
if (nextAligned > end)
|
586
|
-
return true;
|
587
|
-
|
588
|
-
while (p != nextAligned)
|
589
|
-
if (*p < 0x20 || *p == '\"' || *p == '\\') {
|
590
|
-
is.src_ = p;
|
591
|
-
return CEREAL_RAPIDJSON_LIKELY(is.Tell() < length);
|
592
|
-
}
|
593
|
-
else
|
594
|
-
os_->PutUnsafe(*p++);
|
595
|
-
|
596
|
-
// The rest of string using SIMD
|
597
|
-
static const char dquote[16] = { '\"', '\"', '\"', '\"', '\"', '\"', '\"', '\"', '\"', '\"', '\"', '\"', '\"', '\"', '\"', '\"' };
|
598
|
-
static const char bslash[16] = { '\\', '\\', '\\', '\\', '\\', '\\', '\\', '\\', '\\', '\\', '\\', '\\', '\\', '\\', '\\', '\\' };
|
599
|
-
static const char space[16] = { 0x1F, 0x1F, 0x1F, 0x1F, 0x1F, 0x1F, 0x1F, 0x1F, 0x1F, 0x1F, 0x1F, 0x1F, 0x1F, 0x1F, 0x1F, 0x1F };
|
600
|
-
const __m128i dq = _mm_loadu_si128(reinterpret_cast<const __m128i *>(&dquote[0]));
|
601
|
-
const __m128i bs = _mm_loadu_si128(reinterpret_cast<const __m128i *>(&bslash[0]));
|
602
|
-
const __m128i sp = _mm_loadu_si128(reinterpret_cast<const __m128i *>(&space[0]));
|
603
|
-
|
604
|
-
for (; p != endAligned; p += 16) {
|
605
|
-
const __m128i s = _mm_load_si128(reinterpret_cast<const __m128i *>(p));
|
606
|
-
const __m128i t1 = _mm_cmpeq_epi8(s, dq);
|
607
|
-
const __m128i t2 = _mm_cmpeq_epi8(s, bs);
|
608
|
-
const __m128i t3 = _mm_cmpeq_epi8(_mm_max_epu8(s, sp), sp); // s < 0x20 <=> max(s, 0x1F) == 0x1F
|
609
|
-
const __m128i x = _mm_or_si128(_mm_or_si128(t1, t2), t3);
|
610
|
-
unsigned short r = static_cast<unsigned short>(_mm_movemask_epi8(x));
|
611
|
-
if (CEREAL_RAPIDJSON_UNLIKELY(r != 0)) { // some of characters is escaped
|
612
|
-
SizeType len;
|
613
|
-
#ifdef _MSC_VER // Find the index of first escaped
|
614
|
-
unsigned long offset;
|
615
|
-
_BitScanForward(&offset, r);
|
616
|
-
len = offset;
|
617
|
-
#else
|
618
|
-
len = static_cast<SizeType>(__builtin_ffs(r) - 1);
|
619
|
-
#endif
|
620
|
-
char* q = reinterpret_cast<char*>(os_->PushUnsafe(len));
|
621
|
-
for (size_t i = 0; i < len; i++)
|
622
|
-
q[i] = p[i];
|
623
|
-
|
624
|
-
p += len;
|
625
|
-
break;
|
626
|
-
}
|
627
|
-
_mm_storeu_si128(reinterpret_cast<__m128i *>(os_->PushUnsafe(16)), s);
|
628
|
-
}
|
629
|
-
|
630
|
-
is.src_ = p;
|
631
|
-
return CEREAL_RAPIDJSON_LIKELY(is.Tell() < length);
|
632
|
-
}
|
633
|
-
#elif defined(CEREAL_RAPIDJSON_NEON)
|
634
|
-
template<>
|
635
|
-
inline bool Writer<StringBuffer>::ScanWriteUnescapedString(StringStream& is, size_t length) {
|
636
|
-
if (length < 16)
|
637
|
-
return CEREAL_RAPIDJSON_LIKELY(is.Tell() < length);
|
638
|
-
|
639
|
-
if (!CEREAL_RAPIDJSON_LIKELY(is.Tell() < length))
|
640
|
-
return false;
|
641
|
-
|
642
|
-
const char* p = is.src_;
|
643
|
-
const char* end = is.head_ + length;
|
644
|
-
const char* nextAligned = reinterpret_cast<const char*>((reinterpret_cast<size_t>(p) + 15) & static_cast<size_t>(~15));
|
645
|
-
const char* endAligned = reinterpret_cast<const char*>(reinterpret_cast<size_t>(end) & static_cast<size_t>(~15));
|
646
|
-
if (nextAligned > end)
|
647
|
-
return true;
|
648
|
-
|
649
|
-
while (p != nextAligned)
|
650
|
-
if (*p < 0x20 || *p == '\"' || *p == '\\') {
|
651
|
-
is.src_ = p;
|
652
|
-
return CEREAL_RAPIDJSON_LIKELY(is.Tell() < length);
|
653
|
-
}
|
654
|
-
else
|
655
|
-
os_->PutUnsafe(*p++);
|
656
|
-
|
657
|
-
// The rest of string using SIMD
|
658
|
-
const uint8x16_t s0 = vmovq_n_u8('"');
|
659
|
-
const uint8x16_t s1 = vmovq_n_u8('\\');
|
660
|
-
const uint8x16_t s2 = vmovq_n_u8('\b');
|
661
|
-
const uint8x16_t s3 = vmovq_n_u8(32);
|
662
|
-
|
663
|
-
for (; p != endAligned; p += 16) {
|
664
|
-
const uint8x16_t s = vld1q_u8(reinterpret_cast<const uint8_t *>(p));
|
665
|
-
uint8x16_t x = vceqq_u8(s, s0);
|
666
|
-
x = vorrq_u8(x, vceqq_u8(s, s1));
|
667
|
-
x = vorrq_u8(x, vceqq_u8(s, s2));
|
668
|
-
x = vorrq_u8(x, vcltq_u8(s, s3));
|
669
|
-
|
670
|
-
x = vrev64q_u8(x); // Rev in 64
|
671
|
-
uint64_t low = vgetq_lane_u64(reinterpret_cast<uint64x2_t>(x), 0); // extract
|
672
|
-
uint64_t high = vgetq_lane_u64(reinterpret_cast<uint64x2_t>(x), 1); // extract
|
673
|
-
|
674
|
-
SizeType len = 0;
|
675
|
-
bool escaped = false;
|
676
|
-
if (low == 0) {
|
677
|
-
if (high != 0) {
|
678
|
-
unsigned lz = (unsigned)__builtin_clzll(high);
|
679
|
-
len = 8 + (lz >> 3);
|
680
|
-
escaped = true;
|
681
|
-
}
|
682
|
-
} else {
|
683
|
-
unsigned lz = (unsigned)__builtin_clzll(low);
|
684
|
-
len = lz >> 3;
|
685
|
-
escaped = true;
|
686
|
-
}
|
687
|
-
if (CEREAL_RAPIDJSON_UNLIKELY(escaped)) { // some of characters is escaped
|
688
|
-
char* q = reinterpret_cast<char*>(os_->PushUnsafe(len));
|
689
|
-
for (size_t i = 0; i < len; i++)
|
690
|
-
q[i] = p[i];
|
691
|
-
|
692
|
-
p += len;
|
693
|
-
break;
|
694
|
-
}
|
695
|
-
vst1q_u8(reinterpret_cast<uint8_t *>(os_->PushUnsafe(16)), s);
|
696
|
-
}
|
697
|
-
|
698
|
-
is.src_ = p;
|
699
|
-
return CEREAL_RAPIDJSON_LIKELY(is.Tell() < length);
|
700
|
-
}
|
701
|
-
#endif // CEREAL_RAPIDJSON_NEON
|
702
|
-
|
703
|
-
CEREAL_RAPIDJSON_NAMESPACE_END
|
704
|
-
|
705
|
-
#if defined(_MSC_VER) || defined(__clang__)
|
706
|
-
CEREAL_RAPIDJSON_DIAG_POP
|
707
|
-
#endif
|
708
|
-
|
709
|
-
#endif // CEREAL_RAPIDJSON_CEREAL_RAPIDJSON_H_
|