isotree 0.1.4 → 0.1.5

Sign up to get free protection for your applications and to get access to all the features.
Files changed (118) hide show
  1. checksums.yaml +4 -4
  2. data/CHANGELOG.md +5 -0
  3. data/LICENSE.txt +2 -2
  4. data/README.md +22 -1
  5. data/ext/isotree/ext.cpp +26 -0
  6. data/ext/isotree/extconf.rb +3 -3
  7. data/lib/isotree.rb +1 -0
  8. data/lib/isotree/isolation_forest.rb +86 -1
  9. data/lib/isotree/version.rb +1 -1
  10. data/vendor/cereal/LICENSE +24 -0
  11. data/vendor/cereal/README.md +85 -0
  12. data/vendor/cereal/include/cereal/access.hpp +351 -0
  13. data/vendor/cereal/include/cereal/archives/adapters.hpp +163 -0
  14. data/vendor/cereal/include/cereal/archives/binary.hpp +169 -0
  15. data/vendor/cereal/include/cereal/archives/json.hpp +1019 -0
  16. data/vendor/cereal/include/cereal/archives/portable_binary.hpp +334 -0
  17. data/vendor/cereal/include/cereal/archives/xml.hpp +956 -0
  18. data/vendor/cereal/include/cereal/cereal.hpp +1089 -0
  19. data/vendor/cereal/include/cereal/details/helpers.hpp +422 -0
  20. data/vendor/cereal/include/cereal/details/polymorphic_impl.hpp +796 -0
  21. data/vendor/cereal/include/cereal/details/polymorphic_impl_fwd.hpp +65 -0
  22. data/vendor/cereal/include/cereal/details/static_object.hpp +127 -0
  23. data/vendor/cereal/include/cereal/details/traits.hpp +1411 -0
  24. data/vendor/cereal/include/cereal/details/util.hpp +84 -0
  25. data/vendor/cereal/include/cereal/external/base64.hpp +134 -0
  26. data/vendor/cereal/include/cereal/external/rapidjson/allocators.h +284 -0
  27. data/vendor/cereal/include/cereal/external/rapidjson/cursorstreamwrapper.h +78 -0
  28. data/vendor/cereal/include/cereal/external/rapidjson/document.h +2652 -0
  29. data/vendor/cereal/include/cereal/external/rapidjson/encodedstream.h +299 -0
  30. data/vendor/cereal/include/cereal/external/rapidjson/encodings.h +716 -0
  31. data/vendor/cereal/include/cereal/external/rapidjson/error/en.h +74 -0
  32. data/vendor/cereal/include/cereal/external/rapidjson/error/error.h +161 -0
  33. data/vendor/cereal/include/cereal/external/rapidjson/filereadstream.h +99 -0
  34. data/vendor/cereal/include/cereal/external/rapidjson/filewritestream.h +104 -0
  35. data/vendor/cereal/include/cereal/external/rapidjson/fwd.h +151 -0
  36. data/vendor/cereal/include/cereal/external/rapidjson/internal/biginteger.h +290 -0
  37. data/vendor/cereal/include/cereal/external/rapidjson/internal/diyfp.h +271 -0
  38. data/vendor/cereal/include/cereal/external/rapidjson/internal/dtoa.h +245 -0
  39. data/vendor/cereal/include/cereal/external/rapidjson/internal/ieee754.h +78 -0
  40. data/vendor/cereal/include/cereal/external/rapidjson/internal/itoa.h +308 -0
  41. data/vendor/cereal/include/cereal/external/rapidjson/internal/meta.h +186 -0
  42. data/vendor/cereal/include/cereal/external/rapidjson/internal/pow10.h +55 -0
  43. data/vendor/cereal/include/cereal/external/rapidjson/internal/regex.h +740 -0
  44. data/vendor/cereal/include/cereal/external/rapidjson/internal/stack.h +232 -0
  45. data/vendor/cereal/include/cereal/external/rapidjson/internal/strfunc.h +69 -0
  46. data/vendor/cereal/include/cereal/external/rapidjson/internal/strtod.h +290 -0
  47. data/vendor/cereal/include/cereal/external/rapidjson/internal/swap.h +46 -0
  48. data/vendor/cereal/include/cereal/external/rapidjson/istreamwrapper.h +128 -0
  49. data/vendor/cereal/include/cereal/external/rapidjson/memorybuffer.h +70 -0
  50. data/vendor/cereal/include/cereal/external/rapidjson/memorystream.h +71 -0
  51. data/vendor/cereal/include/cereal/external/rapidjson/msinttypes/inttypes.h +316 -0
  52. data/vendor/cereal/include/cereal/external/rapidjson/msinttypes/stdint.h +300 -0
  53. data/vendor/cereal/include/cereal/external/rapidjson/ostreamwrapper.h +81 -0
  54. data/vendor/cereal/include/cereal/external/rapidjson/pointer.h +1414 -0
  55. data/vendor/cereal/include/cereal/external/rapidjson/prettywriter.h +277 -0
  56. data/vendor/cereal/include/cereal/external/rapidjson/rapidjson.h +656 -0
  57. data/vendor/cereal/include/cereal/external/rapidjson/reader.h +2230 -0
  58. data/vendor/cereal/include/cereal/external/rapidjson/schema.h +2497 -0
  59. data/vendor/cereal/include/cereal/external/rapidjson/stream.h +223 -0
  60. data/vendor/cereal/include/cereal/external/rapidjson/stringbuffer.h +121 -0
  61. data/vendor/cereal/include/cereal/external/rapidjson/writer.h +709 -0
  62. data/vendor/cereal/include/cereal/external/rapidxml/license.txt +52 -0
  63. data/vendor/cereal/include/cereal/external/rapidxml/manual.html +406 -0
  64. data/vendor/cereal/include/cereal/external/rapidxml/rapidxml.hpp +2624 -0
  65. data/vendor/cereal/include/cereal/external/rapidxml/rapidxml_iterators.hpp +175 -0
  66. data/vendor/cereal/include/cereal/external/rapidxml/rapidxml_print.hpp +428 -0
  67. data/vendor/cereal/include/cereal/external/rapidxml/rapidxml_utils.hpp +123 -0
  68. data/vendor/cereal/include/cereal/macros.hpp +154 -0
  69. data/vendor/cereal/include/cereal/specialize.hpp +139 -0
  70. data/vendor/cereal/include/cereal/types/array.hpp +79 -0
  71. data/vendor/cereal/include/cereal/types/atomic.hpp +55 -0
  72. data/vendor/cereal/include/cereal/types/base_class.hpp +203 -0
  73. data/vendor/cereal/include/cereal/types/bitset.hpp +176 -0
  74. data/vendor/cereal/include/cereal/types/boost_variant.hpp +164 -0
  75. data/vendor/cereal/include/cereal/types/chrono.hpp +72 -0
  76. data/vendor/cereal/include/cereal/types/common.hpp +129 -0
  77. data/vendor/cereal/include/cereal/types/complex.hpp +56 -0
  78. data/vendor/cereal/include/cereal/types/concepts/pair_associative_container.hpp +73 -0
  79. data/vendor/cereal/include/cereal/types/deque.hpp +62 -0
  80. data/vendor/cereal/include/cereal/types/forward_list.hpp +68 -0
  81. data/vendor/cereal/include/cereal/types/functional.hpp +43 -0
  82. data/vendor/cereal/include/cereal/types/list.hpp +62 -0
  83. data/vendor/cereal/include/cereal/types/map.hpp +36 -0
  84. data/vendor/cereal/include/cereal/types/memory.hpp +425 -0
  85. data/vendor/cereal/include/cereal/types/optional.hpp +66 -0
  86. data/vendor/cereal/include/cereal/types/polymorphic.hpp +483 -0
  87. data/vendor/cereal/include/cereal/types/queue.hpp +132 -0
  88. data/vendor/cereal/include/cereal/types/set.hpp +103 -0
  89. data/vendor/cereal/include/cereal/types/stack.hpp +76 -0
  90. data/vendor/cereal/include/cereal/types/string.hpp +61 -0
  91. data/vendor/cereal/include/cereal/types/tuple.hpp +123 -0
  92. data/vendor/cereal/include/cereal/types/unordered_map.hpp +36 -0
  93. data/vendor/cereal/include/cereal/types/unordered_set.hpp +99 -0
  94. data/vendor/cereal/include/cereal/types/utility.hpp +47 -0
  95. data/vendor/cereal/include/cereal/types/valarray.hpp +89 -0
  96. data/vendor/cereal/include/cereal/types/variant.hpp +109 -0
  97. data/vendor/cereal/include/cereal/types/vector.hpp +112 -0
  98. data/vendor/cereal/include/cereal/version.hpp +52 -0
  99. data/vendor/isotree/LICENSE +1 -1
  100. data/vendor/isotree/README.md +2 -1
  101. data/vendor/isotree/src/RcppExports.cpp +44 -4
  102. data/vendor/isotree/src/Rwrapper.cpp +141 -51
  103. data/vendor/isotree/src/crit.cpp +1 -1
  104. data/vendor/isotree/src/dealloc.cpp +1 -1
  105. data/vendor/isotree/src/dist.cpp +6 -6
  106. data/vendor/isotree/src/extended.cpp +5 -5
  107. data/vendor/isotree/src/fit_model.cpp +30 -19
  108. data/vendor/isotree/src/helpers_iforest.cpp +26 -11
  109. data/vendor/isotree/src/impute.cpp +7 -7
  110. data/vendor/isotree/src/isoforest.cpp +7 -7
  111. data/vendor/isotree/src/isotree.hpp +27 -5
  112. data/vendor/isotree/src/merge_models.cpp +1 -1
  113. data/vendor/isotree/src/mult.cpp +1 -1
  114. data/vendor/isotree/src/predict.cpp +20 -16
  115. data/vendor/isotree/src/serialize.cpp +1 -1
  116. data/vendor/isotree/src/sql.cpp +545 -0
  117. data/vendor/isotree/src/utils.cpp +36 -44
  118. metadata +98 -92
@@ -0,0 +1,223 @@
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
+ #include "rapidjson.h"
16
+
17
+ #ifndef CEREAL_RAPIDJSON_STREAM_H_
18
+ #define CEREAL_RAPIDJSON_STREAM_H_
19
+
20
+ #include "encodings.h"
21
+
22
+ CEREAL_RAPIDJSON_NAMESPACE_BEGIN
23
+
24
+ ///////////////////////////////////////////////////////////////////////////////
25
+ // Stream
26
+
27
+ /*! \class rapidjson::Stream
28
+ \brief Concept for reading and writing characters.
29
+
30
+ For read-only stream, no need to implement PutBegin(), Put(), Flush() and PutEnd().
31
+
32
+ For write-only stream, only need to implement Put() and Flush().
33
+
34
+ \code
35
+ concept Stream {
36
+ typename Ch; //!< Character type of the stream.
37
+
38
+ //! Read the current character from stream without moving the read cursor.
39
+ Ch Peek() const;
40
+
41
+ //! Read the current character from stream and moving the read cursor to next character.
42
+ Ch Take();
43
+
44
+ //! Get the current read cursor.
45
+ //! \return Number of characters read from start.
46
+ size_t Tell();
47
+
48
+ //! Begin writing operation at the current read pointer.
49
+ //! \return The begin writer pointer.
50
+ Ch* PutBegin();
51
+
52
+ //! Write a character.
53
+ void Put(Ch c);
54
+
55
+ //! Flush the buffer.
56
+ void Flush();
57
+
58
+ //! End the writing operation.
59
+ //! \param begin The begin write pointer returned by PutBegin().
60
+ //! \return Number of characters written.
61
+ size_t PutEnd(Ch* begin);
62
+ }
63
+ \endcode
64
+ */
65
+
66
+ //! Provides additional information for stream.
67
+ /*!
68
+ By using traits pattern, this type provides a default configuration for stream.
69
+ For custom stream, this type can be specialized for other configuration.
70
+ See TEST(Reader, CustomStringStream) in readertest.cpp for example.
71
+ */
72
+ template<typename Stream>
73
+ struct StreamTraits {
74
+ //! Whether to make local copy of stream for optimization during parsing.
75
+ /*!
76
+ By default, for safety, streams do not use local copy optimization.
77
+ Stream that can be copied fast should specialize this, like StreamTraits<StringStream>.
78
+ */
79
+ enum { copyOptimization = 0 };
80
+ };
81
+
82
+ //! Reserve n characters for writing to a stream.
83
+ template<typename Stream>
84
+ inline void PutReserve(Stream& stream, size_t count) {
85
+ (void)stream;
86
+ (void)count;
87
+ }
88
+
89
+ //! Write character to a stream, presuming buffer is reserved.
90
+ template<typename Stream>
91
+ inline void PutUnsafe(Stream& stream, typename Stream::Ch c) {
92
+ stream.Put(c);
93
+ }
94
+
95
+ //! Put N copies of a character to a stream.
96
+ template<typename Stream, typename Ch>
97
+ inline void PutN(Stream& stream, Ch c, size_t n) {
98
+ PutReserve(stream, n);
99
+ for (size_t i = 0; i < n; i++)
100
+ PutUnsafe(stream, c);
101
+ }
102
+
103
+ ///////////////////////////////////////////////////////////////////////////////
104
+ // GenericStreamWrapper
105
+
106
+ //! A Stream Wrapper
107
+ /*! \tThis string stream is a wrapper for any stream by just forwarding any
108
+ \treceived message to the origin stream.
109
+ \note implements Stream concept
110
+ */
111
+
112
+ #if defined(_MSC_VER) && _MSC_VER <= 1800
113
+ CEREAL_RAPIDJSON_DIAG_PUSH
114
+ CEREAL_RAPIDJSON_DIAG_OFF(4702) // unreachable code
115
+ CEREAL_RAPIDJSON_DIAG_OFF(4512) // assignment operator could not be generated
116
+ #endif
117
+
118
+ template <typename InputStream, typename Encoding = UTF8<> >
119
+ class GenericStreamWrapper {
120
+ public:
121
+ typedef typename Encoding::Ch Ch;
122
+ GenericStreamWrapper(InputStream& is): is_(is) {}
123
+
124
+ Ch Peek() const { return is_.Peek(); }
125
+ Ch Take() { return is_.Take(); }
126
+ size_t Tell() { return is_.Tell(); }
127
+ Ch* PutBegin() { return is_.PutBegin(); }
128
+ void Put(Ch ch) { is_.Put(ch); }
129
+ void Flush() { is_.Flush(); }
130
+ size_t PutEnd(Ch* ch) { return is_.PutEnd(ch); }
131
+
132
+ // wrapper for MemoryStream
133
+ const Ch* Peek4() const { return is_.Peek4(); }
134
+
135
+ // wrapper for AutoUTFInputStream
136
+ UTFType GetType() const { return is_.GetType(); }
137
+ bool HasBOM() const { return is_.HasBOM(); }
138
+
139
+ protected:
140
+ InputStream& is_;
141
+ };
142
+
143
+ #if defined(_MSC_VER) && _MSC_VER <= 1800
144
+ CEREAL_RAPIDJSON_DIAG_POP
145
+ #endif
146
+
147
+ ///////////////////////////////////////////////////////////////////////////////
148
+ // StringStream
149
+
150
+ //! Read-only string stream.
151
+ /*! \note implements Stream concept
152
+ */
153
+ template <typename Encoding>
154
+ struct GenericStringStream {
155
+ typedef typename Encoding::Ch Ch;
156
+
157
+ GenericStringStream(const Ch *src) : src_(src), head_(src) {}
158
+
159
+ Ch Peek() const { return *src_; }
160
+ Ch Take() { return *src_++; }
161
+ size_t Tell() const { return static_cast<size_t>(src_ - head_); }
162
+
163
+ Ch* PutBegin() { CEREAL_RAPIDJSON_ASSERT(false); return 0; }
164
+ void Put(Ch) { CEREAL_RAPIDJSON_ASSERT(false); }
165
+ void Flush() { CEREAL_RAPIDJSON_ASSERT(false); }
166
+ size_t PutEnd(Ch*) { CEREAL_RAPIDJSON_ASSERT(false); return 0; }
167
+
168
+ const Ch* src_; //!< Current read position.
169
+ const Ch* head_; //!< Original head of the string.
170
+ };
171
+
172
+ template <typename Encoding>
173
+ struct StreamTraits<GenericStringStream<Encoding> > {
174
+ enum { copyOptimization = 1 };
175
+ };
176
+
177
+ //! String stream with UTF8 encoding.
178
+ typedef GenericStringStream<UTF8<> > StringStream;
179
+
180
+ ///////////////////////////////////////////////////////////////////////////////
181
+ // InsituStringStream
182
+
183
+ //! A read-write string stream.
184
+ /*! This string stream is particularly designed for in-situ parsing.
185
+ \note implements Stream concept
186
+ */
187
+ template <typename Encoding>
188
+ struct GenericInsituStringStream {
189
+ typedef typename Encoding::Ch Ch;
190
+
191
+ GenericInsituStringStream(Ch *src) : src_(src), dst_(0), head_(src) {}
192
+
193
+ // Read
194
+ Ch Peek() { return *src_; }
195
+ Ch Take() { return *src_++; }
196
+ size_t Tell() { return static_cast<size_t>(src_ - head_); }
197
+
198
+ // Write
199
+ void Put(Ch c) { CEREAL_RAPIDJSON_ASSERT(dst_ != 0); *dst_++ = c; }
200
+
201
+ Ch* PutBegin() { return dst_ = src_; }
202
+ size_t PutEnd(Ch* begin) { return static_cast<size_t>(dst_ - begin); }
203
+ void Flush() {}
204
+
205
+ Ch* Push(size_t count) { Ch* begin = dst_; dst_ += count; return begin; }
206
+ void Pop(size_t count) { dst_ -= count; }
207
+
208
+ Ch* src_;
209
+ Ch* dst_;
210
+ Ch* head_;
211
+ };
212
+
213
+ template <typename Encoding>
214
+ struct StreamTraits<GenericInsituStringStream<Encoding> > {
215
+ enum { copyOptimization = 1 };
216
+ };
217
+
218
+ //! Insitu string stream with UTF8 encoding.
219
+ typedef GenericInsituStringStream<UTF8<> > InsituStringStream;
220
+
221
+ CEREAL_RAPIDJSON_NAMESPACE_END
222
+
223
+ #endif // CEREAL_RAPIDJSON_STREAM_H_
@@ -0,0 +1,121 @@
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_STRINGBUFFER_H_
16
+ #define CEREAL_RAPIDJSON_STRINGBUFFER_H_
17
+
18
+ #include "stream.h"
19
+ #include "internal/stack.h"
20
+
21
+ #if CEREAL_RAPIDJSON_HAS_CXX11_RVALUE_REFS
22
+ #include <utility> // std::move
23
+ #endif
24
+
25
+ #include "internal/stack.h"
26
+
27
+ #if defined(__clang__)
28
+ CEREAL_RAPIDJSON_DIAG_PUSH
29
+ CEREAL_RAPIDJSON_DIAG_OFF(c++98-compat)
30
+ #endif
31
+
32
+ CEREAL_RAPIDJSON_NAMESPACE_BEGIN
33
+
34
+ //! Represents an in-memory output stream.
35
+ /*!
36
+ \tparam Encoding Encoding of the stream.
37
+ \tparam Allocator type for allocating memory buffer.
38
+ \note implements Stream concept
39
+ */
40
+ template <typename Encoding, typename Allocator = CrtAllocator>
41
+ class GenericStringBuffer {
42
+ public:
43
+ typedef typename Encoding::Ch Ch;
44
+
45
+ GenericStringBuffer(Allocator* allocator = 0, size_t capacity = kDefaultCapacity) : stack_(allocator, capacity) {}
46
+
47
+ #if CEREAL_RAPIDJSON_HAS_CXX11_RVALUE_REFS
48
+ GenericStringBuffer(GenericStringBuffer&& rhs) : stack_(std::move(rhs.stack_)) {}
49
+ GenericStringBuffer& operator=(GenericStringBuffer&& rhs) {
50
+ if (&rhs != this)
51
+ stack_ = std::move(rhs.stack_);
52
+ return *this;
53
+ }
54
+ #endif
55
+
56
+ void Put(Ch c) { *stack_.template Push<Ch>() = c; }
57
+ void PutUnsafe(Ch c) { *stack_.template PushUnsafe<Ch>() = c; }
58
+ void Flush() {}
59
+
60
+ void Clear() { stack_.Clear(); }
61
+ void ShrinkToFit() {
62
+ // Push and pop a null terminator. This is safe.
63
+ *stack_.template Push<Ch>() = '\0';
64
+ stack_.ShrinkToFit();
65
+ stack_.template Pop<Ch>(1);
66
+ }
67
+
68
+ void Reserve(size_t count) { stack_.template Reserve<Ch>(count); }
69
+ Ch* Push(size_t count) { return stack_.template Push<Ch>(count); }
70
+ Ch* PushUnsafe(size_t count) { return stack_.template PushUnsafe<Ch>(count); }
71
+ void Pop(size_t count) { stack_.template Pop<Ch>(count); }
72
+
73
+ const Ch* GetString() const {
74
+ // Push and pop a null terminator. This is safe.
75
+ *stack_.template Push<Ch>() = '\0';
76
+ stack_.template Pop<Ch>(1);
77
+
78
+ return stack_.template Bottom<Ch>();
79
+ }
80
+
81
+ //! Get the size of string in bytes in the string buffer.
82
+ size_t GetSize() const { return stack_.GetSize(); }
83
+
84
+ //! Get the length of string in Ch in the string buffer.
85
+ size_t GetLength() const { return stack_.GetSize() / sizeof(Ch); }
86
+
87
+ static const size_t kDefaultCapacity = 256;
88
+ mutable internal::Stack<Allocator> stack_;
89
+
90
+ private:
91
+ // Prohibit copy constructor & assignment operator.
92
+ GenericStringBuffer(const GenericStringBuffer&);
93
+ GenericStringBuffer& operator=(const GenericStringBuffer&);
94
+ };
95
+
96
+ //! String buffer with UTF8 encoding
97
+ typedef GenericStringBuffer<UTF8<> > StringBuffer;
98
+
99
+ template<typename Encoding, typename Allocator>
100
+ inline void PutReserve(GenericStringBuffer<Encoding, Allocator>& stream, size_t count) {
101
+ stream.Reserve(count);
102
+ }
103
+
104
+ template<typename Encoding, typename Allocator>
105
+ inline void PutUnsafe(GenericStringBuffer<Encoding, Allocator>& stream, typename Encoding::Ch c) {
106
+ stream.PutUnsafe(c);
107
+ }
108
+
109
+ //! Implement specialized version of PutN() with memset() for better performance.
110
+ template<>
111
+ inline void PutN(GenericStringBuffer<UTF8<> >& stream, char c, size_t n) {
112
+ std::memset(stream.stack_.Push<char>(n), c, n * sizeof(c));
113
+ }
114
+
115
+ CEREAL_RAPIDJSON_NAMESPACE_END
116
+
117
+ #if defined(__clang__)
118
+ CEREAL_RAPIDJSON_DIAG_POP
119
+ #endif
120
+
121
+ #endif // CEREAL_RAPIDJSON_STRINGBUFFER_H_
@@ -0,0 +1,709 @@
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_