emitter.h 7.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281
  1. #ifndef EMITTER_H_62B23520_7C8E_11DE_8A39_0800200C9A66
  2. #define EMITTER_H_62B23520_7C8E_11DE_8A39_0800200C9A66
  3. #if defined(_MSC_VER) || \
  4. (defined(__GNUC__) && (__GNUC__ == 3 && __GNUC_MINOR__ >= 4) || \
  5. (__GNUC__ >= 4)) // GCC supports "pragma once" correctly since 3.4
  6. #pragma once
  7. #endif
  8. #include <cmath>
  9. #include <cstddef>
  10. #include <limits>
  11. #include <memory>
  12. #include <sstream>
  13. #include <string>
  14. #include <type_traits>
  15. #include "yaml-cpp/binary.h"
  16. #include "yaml-cpp/dll.h"
  17. #include "yaml-cpp/emitterdef.h"
  18. #include "yaml-cpp/emittermanip.h"
  19. #include "yaml-cpp/null.h"
  20. #include "yaml-cpp/ostream_wrapper.h"
  21. namespace YAML {
  22. class Binary;
  23. struct _Null;
  24. } // namespace YAML
  25. namespace YAML {
  26. class EmitterState;
  27. class YAML_CPP_API Emitter {
  28. public:
  29. Emitter();
  30. explicit Emitter(std::ostream& stream);
  31. Emitter(const Emitter&) = delete;
  32. Emitter& operator=(const Emitter&) = delete;
  33. ~Emitter();
  34. // output
  35. const char* c_str() const;
  36. std::size_t size() const;
  37. // state checking
  38. bool good() const;
  39. const std::string GetLastError() const;
  40. // global setters
  41. bool SetOutputCharset(EMITTER_MANIP value);
  42. bool SetStringFormat(EMITTER_MANIP value);
  43. bool SetBoolFormat(EMITTER_MANIP value);
  44. bool SetNullFormat(EMITTER_MANIP value);
  45. bool SetIntBase(EMITTER_MANIP value);
  46. bool SetSeqFormat(EMITTER_MANIP value);
  47. bool SetMapFormat(EMITTER_MANIP value);
  48. bool SetIndent(std::size_t n);
  49. bool SetPreCommentIndent(std::size_t n);
  50. bool SetPostCommentIndent(std::size_t n);
  51. bool SetFloatPrecision(std::size_t n);
  52. bool SetDoublePrecision(std::size_t n);
  53. void RestoreGlobalModifiedSettings();
  54. // local setters
  55. Emitter& SetLocalValue(EMITTER_MANIP value);
  56. Emitter& SetLocalIndent(const _Indent& indent);
  57. Emitter& SetLocalPrecision(const _Precision& precision);
  58. // overloads of write
  59. Emitter& Write(const std::string& str);
  60. Emitter& Write(bool b);
  61. Emitter& Write(char ch);
  62. Emitter& Write(const _Alias& alias);
  63. Emitter& Write(const _Anchor& anchor);
  64. Emitter& Write(const _Tag& tag);
  65. Emitter& Write(const _Comment& comment);
  66. Emitter& Write(const _Null& n);
  67. Emitter& Write(const Binary& binary);
  68. template <typename T>
  69. Emitter& WriteIntegralType(T value);
  70. template <typename T>
  71. Emitter& WriteStreamable(T value);
  72. private:
  73. template <typename T>
  74. void SetStreamablePrecision(std::stringstream&) {}
  75. std::size_t GetFloatPrecision() const;
  76. std::size_t GetDoublePrecision() const;
  77. void PrepareIntegralStream(std::stringstream& stream) const;
  78. void StartedScalar();
  79. private:
  80. void EmitBeginDoc();
  81. void EmitEndDoc();
  82. void EmitBeginSeq();
  83. void EmitEndSeq();
  84. void EmitBeginMap();
  85. void EmitEndMap();
  86. void EmitNewline();
  87. void EmitKindTag();
  88. void EmitTag(bool verbatim, const _Tag& tag);
  89. void PrepareNode(EmitterNodeType::value child);
  90. void PrepareTopNode(EmitterNodeType::value child);
  91. void FlowSeqPrepareNode(EmitterNodeType::value child);
  92. void BlockSeqPrepareNode(EmitterNodeType::value child);
  93. void FlowMapPrepareNode(EmitterNodeType::value child);
  94. void FlowMapPrepareLongKey(EmitterNodeType::value child);
  95. void FlowMapPrepareLongKeyValue(EmitterNodeType::value child);
  96. void FlowMapPrepareSimpleKey(EmitterNodeType::value child);
  97. void FlowMapPrepareSimpleKeyValue(EmitterNodeType::value child);
  98. void BlockMapPrepareNode(EmitterNodeType::value child);
  99. void BlockMapPrepareLongKey(EmitterNodeType::value child);
  100. void BlockMapPrepareLongKeyValue(EmitterNodeType::value child);
  101. void BlockMapPrepareSimpleKey(EmitterNodeType::value child);
  102. void BlockMapPrepareSimpleKeyValue(EmitterNodeType::value child);
  103. void SpaceOrIndentTo(bool requireSpace, std::size_t indent);
  104. const char* ComputeFullBoolName(bool b) const;
  105. const char* ComputeNullName() const;
  106. bool CanEmitNewline() const;
  107. private:
  108. std::unique_ptr<EmitterState> m_pState;
  109. ostream_wrapper m_stream;
  110. };
  111. template <typename T>
  112. inline Emitter& Emitter::WriteIntegralType(T value) {
  113. if (!good())
  114. return *this;
  115. PrepareNode(EmitterNodeType::Scalar);
  116. std::stringstream stream;
  117. PrepareIntegralStream(stream);
  118. stream << value;
  119. m_stream << stream.str();
  120. StartedScalar();
  121. return *this;
  122. }
  123. template <typename T>
  124. inline Emitter& Emitter::WriteStreamable(T value) {
  125. if (!good())
  126. return *this;
  127. PrepareNode(EmitterNodeType::Scalar);
  128. std::stringstream stream;
  129. SetStreamablePrecision<T>(stream);
  130. bool special = false;
  131. if (std::is_floating_point<T>::value) {
  132. if ((std::numeric_limits<T>::has_quiet_NaN ||
  133. std::numeric_limits<T>::has_signaling_NaN) &&
  134. std::isnan(value)) {
  135. special = true;
  136. stream << ".nan";
  137. } else if (std::numeric_limits<T>::has_infinity && std::isinf(value)) {
  138. special = true;
  139. if (std::signbit(value)) {
  140. stream << "-.inf";
  141. } else {
  142. stream << ".inf";
  143. }
  144. }
  145. }
  146. if (!special) {
  147. stream << value;
  148. }
  149. m_stream << stream.str();
  150. StartedScalar();
  151. return *this;
  152. }
  153. template <>
  154. inline void Emitter::SetStreamablePrecision<float>(std::stringstream& stream) {
  155. stream.precision(static_cast<std::streamsize>(GetFloatPrecision()));
  156. }
  157. template <>
  158. inline void Emitter::SetStreamablePrecision<double>(std::stringstream& stream) {
  159. stream.precision(static_cast<std::streamsize>(GetDoublePrecision()));
  160. }
  161. // overloads of insertion
  162. inline Emitter& operator<<(Emitter& emitter, const std::string& v) {
  163. return emitter.Write(v);
  164. }
  165. inline Emitter& operator<<(Emitter& emitter, bool v) {
  166. return emitter.Write(v);
  167. }
  168. inline Emitter& operator<<(Emitter& emitter, char v) {
  169. return emitter.Write(v);
  170. }
  171. inline Emitter& operator<<(Emitter& emitter, unsigned char v) {
  172. return emitter.Write(static_cast<char>(v));
  173. }
  174. inline Emitter& operator<<(Emitter& emitter, const _Alias& v) {
  175. return emitter.Write(v);
  176. }
  177. inline Emitter& operator<<(Emitter& emitter, const _Anchor& v) {
  178. return emitter.Write(v);
  179. }
  180. inline Emitter& operator<<(Emitter& emitter, const _Tag& v) {
  181. return emitter.Write(v);
  182. }
  183. inline Emitter& operator<<(Emitter& emitter, const _Comment& v) {
  184. return emitter.Write(v);
  185. }
  186. inline Emitter& operator<<(Emitter& emitter, const _Null& v) {
  187. return emitter.Write(v);
  188. }
  189. inline Emitter& operator<<(Emitter& emitter, const Binary& b) {
  190. return emitter.Write(b);
  191. }
  192. inline Emitter& operator<<(Emitter& emitter, const char* v) {
  193. return emitter.Write(std::string(v));
  194. }
  195. inline Emitter& operator<<(Emitter& emitter, int v) {
  196. return emitter.WriteIntegralType(v);
  197. }
  198. inline Emitter& operator<<(Emitter& emitter, unsigned int v) {
  199. return emitter.WriteIntegralType(v);
  200. }
  201. inline Emitter& operator<<(Emitter& emitter, short v) {
  202. return emitter.WriteIntegralType(v);
  203. }
  204. inline Emitter& operator<<(Emitter& emitter, unsigned short v) {
  205. return emitter.WriteIntegralType(v);
  206. }
  207. inline Emitter& operator<<(Emitter& emitter, long v) {
  208. return emitter.WriteIntegralType(v);
  209. }
  210. inline Emitter& operator<<(Emitter& emitter, unsigned long v) {
  211. return emitter.WriteIntegralType(v);
  212. }
  213. inline Emitter& operator<<(Emitter& emitter, long long v) {
  214. return emitter.WriteIntegralType(v);
  215. }
  216. inline Emitter& operator<<(Emitter& emitter, unsigned long long v) {
  217. return emitter.WriteIntegralType(v);
  218. }
  219. inline Emitter& operator<<(Emitter& emitter, float v) {
  220. return emitter.WriteStreamable(v);
  221. }
  222. inline Emitter& operator<<(Emitter& emitter, double v) {
  223. return emitter.WriteStreamable(v);
  224. }
  225. inline Emitter& operator<<(Emitter& emitter, EMITTER_MANIP value) {
  226. return emitter.SetLocalValue(value);
  227. }
  228. inline Emitter& operator<<(Emitter& emitter, _Indent indent) {
  229. return emitter.SetLocalIndent(indent);
  230. }
  231. inline Emitter& operator<<(Emitter& emitter, _Precision precision) {
  232. return emitter.SetLocalPrecision(precision);
  233. }
  234. } // namespace YAML
  235. #endif // EMITTER_H_62B23520_7C8E_11DE_8A39_0800200C9A66