blob: 2aa6e9b7468333976e97d03919a7a09df41ffded [file] [log] [blame]
Nick Kledzikf60a9272012-12-12 20:46:15 +00001//===- lib/Support/YAMLTraits.cpp -----------------------------------------===//
2//
3// The LLVM Linker
4//
5// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
7//
8//===----------------------------------------------------------------------===//
9
Benjamin Kramer16132e62015-03-23 18:07:13 +000010#include "llvm/Support/YAMLTraits.h"
11#include "llvm/ADT/SmallString.h"
Nick Kledzikf60a9272012-12-12 20:46:15 +000012#include "llvm/ADT/Twine.h"
13#include "llvm/Support/Casting.h"
Benjamin Kramer16132e62015-03-23 18:07:13 +000014#include "llvm/Support/Errc.h"
Nick Kledzikf60a9272012-12-12 20:46:15 +000015#include "llvm/Support/ErrorHandling.h"
Benjamin Kramercbe05842012-12-12 20:55:44 +000016#include "llvm/Support/Format.h"
Alex Lorenz68e787b2015-05-14 23:08:22 +000017#include "llvm/Support/LineIterator.h"
Nick Kledzikf60a9272012-12-12 20:46:15 +000018#include "llvm/Support/YAMLParser.h"
Benjamin Kramercbe05842012-12-12 20:55:44 +000019#include "llvm/Support/raw_ostream.h"
Rui Ueyama106eded2013-09-11 04:00:08 +000020#include <cctype>
Chandler Carruth8a8cd2b2014-01-07 11:48:04 +000021#include <cstring>
Benjamin Kramer36b0f122012-12-12 22:40:02 +000022using namespace llvm;
23using namespace yaml;
Nick Kledzikf60a9272012-12-12 20:46:15 +000024
25//===----------------------------------------------------------------------===//
26// IO
27//===----------------------------------------------------------------------===//
28
29IO::IO(void *Context) : Ctxt(Context) {
30}
31
32IO::~IO() {
33}
34
35void *IO::getContext() {
36 return Ctxt;
37}
38
39void IO::setContext(void *Context) {
40 Ctxt = Context;
41}
42
Nick Kledzikf60a9272012-12-12 20:46:15 +000043//===----------------------------------------------------------------------===//
44// Input
45//===----------------------------------------------------------------------===//
46
Alexander Kornienko681e37c2013-11-18 15:50:04 +000047Input::Input(StringRef InputContent,
48 void *Ctxt,
49 SourceMgr::DiagHandlerTy DiagHandler,
50 void *DiagHandlerCtxt)
Rui Ueyama38dfffa2013-09-11 00:53:07 +000051 : IO(Ctxt),
Nick Kledzik0dcef842013-01-08 21:04:44 +000052 Strm(new Stream(InputContent, SrcMgr)),
Craig Topperc10719f2014-04-07 04:17:22 +000053 CurrentNode(nullptr) {
Alexander Kornienko681e37c2013-11-18 15:50:04 +000054 if (DiagHandler)
55 SrcMgr.setDiagHandler(DiagHandler, DiagHandlerCtxt);
Nick Kledzikf60a9272012-12-12 20:46:15 +000056 DocIterator = Strm->begin();
57}
58
Nick Kledzik0dcef842013-01-08 21:04:44 +000059Input::~Input() {
Nick Kledzik0dcef842013-01-08 21:04:44 +000060}
61
Rafael Espindoladb4ed0b2014-06-13 02:24:39 +000062std::error_code Input::error() { return EC; }
Nick Kledzikf60a9272012-12-12 20:46:15 +000063
Juergen Ributzkad12ccbd2013-11-19 00:57:56 +000064// Pin the vtables to this file.
65void Input::HNode::anchor() {}
66void Input::EmptyHNode::anchor() {}
67void Input::ScalarHNode::anchor() {}
David Blaikied759fe52014-09-15 18:39:24 +000068void Input::MapHNode::anchor() {}
69void Input::SequenceHNode::anchor() {}
Juergen Ributzkad12ccbd2013-11-19 00:57:56 +000070
Nick Kledzik4761c602013-11-21 00:20:10 +000071bool Input::outputting() {
Nick Kledzikf60a9272012-12-12 20:46:15 +000072 return false;
73}
74
75bool Input::setCurrentDocument() {
Benjamin Kramer36b0f122012-12-12 22:40:02 +000076 if (DocIterator != Strm->end()) {
Nick Kledzikf60a9272012-12-12 20:46:15 +000077 Node *N = DocIterator->getRoot();
Alexander Kornienko681e37c2013-11-18 15:50:04 +000078 if (!N) {
Alex Lorenz7a38d752015-05-12 17:44:32 +000079 assert(Strm->failed() && "Root is NULL iff parsing failed");
Rafael Espindola2a826e42014-06-13 17:20:48 +000080 EC = make_error_code(errc::invalid_argument);
Alexander Kornienko681e37c2013-11-18 15:50:04 +000081 return false;
82 }
83
Benjamin Kramer36b0f122012-12-12 22:40:02 +000084 if (isa<NullNode>(N)) {
Nick Kledzikf60a9272012-12-12 20:46:15 +000085 // Empty files are allowed and ignored
86 ++DocIterator;
87 return setCurrentDocument();
88 }
David Blaikied759fe52014-09-15 18:39:24 +000089 TopNode = this->createHNodes(N);
Nick Kledzik0dcef842013-01-08 21:04:44 +000090 CurrentNode = TopNode.get();
Nick Kledzikf60a9272012-12-12 20:46:15 +000091 return true;
92 }
93 return false;
94}
95
Simon Atanasyanf97af8a2014-05-31 04:51:07 +000096bool Input::nextDocument() {
97 return ++DocIterator != Strm->end();
Nick Kledzikf60a9272012-12-12 20:46:15 +000098}
NAKAMURA Takumi9439c522013-11-14 07:08:49 +000099
Alex Lorenz2bdb4e12015-05-27 18:02:19 +0000100const Node *Input::getCurrentNode() const {
101 return CurrentNode ? CurrentNode->_node : nullptr;
102}
103
Nick Kledzik1e6033c2013-11-14 00:59:59 +0000104bool Input::mapTag(StringRef Tag, bool Default) {
NAKAMURA Takumi5b94d282013-11-14 07:08:56 +0000105 std::string foundTag = CurrentNode->_node->getVerbatimTag();
Nick Kledzik1e6033c2013-11-14 00:59:59 +0000106 if (foundTag.empty()) {
107 // If no tag found and 'Tag' is the default, say it was found.
108 return Default;
109 }
Alex Lorenz7a38d752015-05-12 17:44:32 +0000110 // Return true iff found tag matches supplied tag.
Nick Kledzik1e6033c2013-11-14 00:59:59 +0000111 return Tag.equals(foundTag);
112}
Nick Kledzikf60a9272012-12-12 20:46:15 +0000113
114void Input::beginMapping() {
Benjamin Kramer36b0f122012-12-12 22:40:02 +0000115 if (EC)
Nick Kledzikf60a9272012-12-12 20:46:15 +0000116 return;
Alexander Kornienko681e37c2013-11-18 15:50:04 +0000117 // CurrentNode can be null if the document is empty.
118 MapHNode *MN = dyn_cast_or_null<MapHNode>(CurrentNode);
Benjamin Kramer36b0f122012-12-12 22:40:02 +0000119 if (MN) {
Nick Kledzikf60a9272012-12-12 20:46:15 +0000120 MN->ValidKeys.clear();
121 }
122}
123
Benjamin Kramer36b0f122012-12-12 22:40:02 +0000124bool Input::preflightKey(const char *Key, bool Required, bool, bool &UseDefault,
125 void *&SaveInfo) {
Nick Kledzikf60a9272012-12-12 20:46:15 +0000126 UseDefault = false;
Benjamin Kramer36b0f122012-12-12 22:40:02 +0000127 if (EC)
Nick Kledzikf60a9272012-12-12 20:46:15 +0000128 return false;
Alexander Kornienko681e37c2013-11-18 15:50:04 +0000129
130 // CurrentNode is null for empty documents, which is an error in case required
131 // nodes are present.
132 if (!CurrentNode) {
133 if (Required)
Rafael Espindola2a826e42014-06-13 17:20:48 +0000134 EC = make_error_code(errc::invalid_argument);
Alexander Kornienko681e37c2013-11-18 15:50:04 +0000135 return false;
136 }
137
Benjamin Kramer36b0f122012-12-12 22:40:02 +0000138 MapHNode *MN = dyn_cast<MapHNode>(CurrentNode);
139 if (!MN) {
Nick Kledzikf60a9272012-12-12 20:46:15 +0000140 setError(CurrentNode, "not a mapping");
141 return false;
142 }
143 MN->ValidKeys.push_back(Key);
David Blaikied759fe52014-09-15 18:39:24 +0000144 HNode *Value = MN->Mapping[Key].get();
Benjamin Kramer36b0f122012-12-12 22:40:02 +0000145 if (!Value) {
146 if (Required)
Nick Kledzikf60a9272012-12-12 20:46:15 +0000147 setError(CurrentNode, Twine("missing required key '") + Key + "'");
148 else
149 UseDefault = true;
Benjamin Kramer36b0f122012-12-12 22:40:02 +0000150 return false;
Nick Kledzikf60a9272012-12-12 20:46:15 +0000151 }
152 SaveInfo = CurrentNode;
153 CurrentNode = Value;
154 return true;
155}
156
157void Input::postflightKey(void *saveInfo) {
Benjamin Kramer36b0f122012-12-12 22:40:02 +0000158 CurrentNode = reinterpret_cast<HNode *>(saveInfo);
Nick Kledzikf60a9272012-12-12 20:46:15 +0000159}
160
161void Input::endMapping() {
Benjamin Kramer36b0f122012-12-12 22:40:02 +0000162 if (EC)
Nick Kledzikf60a9272012-12-12 20:46:15 +0000163 return;
Alexander Kornienko681e37c2013-11-18 15:50:04 +0000164 // CurrentNode can be null if the document is empty.
165 MapHNode *MN = dyn_cast_or_null<MapHNode>(CurrentNode);
Benjamin Kramer36b0f122012-12-12 22:40:02 +0000166 if (!MN)
Nick Kledzikf60a9272012-12-12 20:46:15 +0000167 return;
Simon Atanasyan878bd8a2014-04-10 06:02:49 +0000168 for (const auto &NN : MN->Mapping) {
169 if (!MN->isValidKey(NN.first())) {
David Blaikied759fe52014-09-15 18:39:24 +0000170 setError(NN.second.get(), Twine("unknown key '") + NN.first() + "'");
Nick Kledzikf60a9272012-12-12 20:46:15 +0000171 break;
172 }
173 }
174}
175
Alex Lorenzb1225082015-05-04 20:11:40 +0000176void Input::beginFlowMapping() { beginMapping(); }
177
178void Input::endFlowMapping() { endMapping(); }
179
Nick Kledzikf60a9272012-12-12 20:46:15 +0000180unsigned Input::beginSequence() {
Justin Bogner64d2cdf2015-03-02 17:26:43 +0000181 if (SequenceHNode *SQ = dyn_cast<SequenceHNode>(CurrentNode))
Nick Kledzikf60a9272012-12-12 20:46:15 +0000182 return SQ->Entries.size();
Justin Bogner64d2cdf2015-03-02 17:26:43 +0000183 if (isa<EmptyHNode>(CurrentNode))
184 return 0;
185 // Treat case where there's a scalar "null" value as an empty sequence.
186 if (ScalarHNode *SN = dyn_cast<ScalarHNode>(CurrentNode)) {
187 if (isNull(SN->value()))
188 return 0;
Nick Kledzikf60a9272012-12-12 20:46:15 +0000189 }
Justin Bogner64d2cdf2015-03-02 17:26:43 +0000190 // Any other type of HNode is an error.
191 setError(CurrentNode, "not a sequence");
Nick Kledzikf60a9272012-12-12 20:46:15 +0000192 return 0;
193}
Benjamin Kramer36b0f122012-12-12 22:40:02 +0000194
Nick Kledzikf60a9272012-12-12 20:46:15 +0000195void Input::endSequence() {
196}
Benjamin Kramer36b0f122012-12-12 22:40:02 +0000197
Nick Kledzikf60a9272012-12-12 20:46:15 +0000198bool Input::preflightElement(unsigned Index, void *&SaveInfo) {
Benjamin Kramer36b0f122012-12-12 22:40:02 +0000199 if (EC)
Nick Kledzikf60a9272012-12-12 20:46:15 +0000200 return false;
Benjamin Kramer36b0f122012-12-12 22:40:02 +0000201 if (SequenceHNode *SQ = dyn_cast<SequenceHNode>(CurrentNode)) {
Nick Kledzikf60a9272012-12-12 20:46:15 +0000202 SaveInfo = CurrentNode;
David Blaikied759fe52014-09-15 18:39:24 +0000203 CurrentNode = SQ->Entries[Index].get();
Nick Kledzikf60a9272012-12-12 20:46:15 +0000204 return true;
205 }
206 return false;
207}
Benjamin Kramer36b0f122012-12-12 22:40:02 +0000208
Nick Kledzikf60a9272012-12-12 20:46:15 +0000209void Input::postflightElement(void *SaveInfo) {
Benjamin Kramer36b0f122012-12-12 22:40:02 +0000210 CurrentNode = reinterpret_cast<HNode *>(SaveInfo);
Nick Kledzikf60a9272012-12-12 20:46:15 +0000211}
212
Justin Bogner64d2cdf2015-03-02 17:26:43 +0000213unsigned Input::beginFlowSequence() { return beginSequence(); }
Benjamin Kramer36b0f122012-12-12 22:40:02 +0000214
Nick Kledzikf60a9272012-12-12 20:46:15 +0000215bool Input::preflightFlowElement(unsigned index, void *&SaveInfo) {
Benjamin Kramer36b0f122012-12-12 22:40:02 +0000216 if (EC)
Nick Kledzikf60a9272012-12-12 20:46:15 +0000217 return false;
Benjamin Kramer36b0f122012-12-12 22:40:02 +0000218 if (SequenceHNode *SQ = dyn_cast<SequenceHNode>(CurrentNode)) {
Nick Kledzikf60a9272012-12-12 20:46:15 +0000219 SaveInfo = CurrentNode;
David Blaikied759fe52014-09-15 18:39:24 +0000220 CurrentNode = SQ->Entries[index].get();
Nick Kledzikf60a9272012-12-12 20:46:15 +0000221 return true;
222 }
223 return false;
224}
Benjamin Kramer36b0f122012-12-12 22:40:02 +0000225
Nick Kledzikf60a9272012-12-12 20:46:15 +0000226void Input::postflightFlowElement(void *SaveInfo) {
Benjamin Kramer36b0f122012-12-12 22:40:02 +0000227 CurrentNode = reinterpret_cast<HNode *>(SaveInfo);
Nick Kledzikf60a9272012-12-12 20:46:15 +0000228}
229
Benjamin Kramer36b0f122012-12-12 22:40:02 +0000230void Input::endFlowSequence() {
231}
Nick Kledzikf60a9272012-12-12 20:46:15 +0000232
233void Input::beginEnumScalar() {
234 ScalarMatchFound = false;
235}
236
237bool Input::matchEnumScalar(const char *Str, bool) {
Benjamin Kramer36b0f122012-12-12 22:40:02 +0000238 if (ScalarMatchFound)
Nick Kledzikf60a9272012-12-12 20:46:15 +0000239 return false;
Benjamin Kramer36b0f122012-12-12 22:40:02 +0000240 if (ScalarHNode *SN = dyn_cast<ScalarHNode>(CurrentNode)) {
241 if (SN->value().equals(Str)) {
Nick Kledzikf60a9272012-12-12 20:46:15 +0000242 ScalarMatchFound = true;
243 return true;
244 }
245 }
246 return false;
247}
248
Michael J. Spencer731cae32015-01-23 21:57:50 +0000249bool Input::matchEnumFallback() {
250 if (ScalarMatchFound)
251 return false;
252 ScalarMatchFound = true;
253 return true;
254}
255
Nick Kledzikf60a9272012-12-12 20:46:15 +0000256void Input::endEnumScalar() {
Benjamin Kramer36b0f122012-12-12 22:40:02 +0000257 if (!ScalarMatchFound) {
Nick Kledzikf60a9272012-12-12 20:46:15 +0000258 setError(CurrentNode, "unknown enumerated scalar");
259 }
260}
261
Nick Kledzikf60a9272012-12-12 20:46:15 +0000262bool Input::beginBitSetScalar(bool &DoClear) {
263 BitValuesUsed.clear();
Benjamin Kramer36b0f122012-12-12 22:40:02 +0000264 if (SequenceHNode *SQ = dyn_cast<SequenceHNode>(CurrentNode)) {
Nick Kledzikf60a9272012-12-12 20:46:15 +0000265 BitValuesUsed.insert(BitValuesUsed.begin(), SQ->Entries.size(), false);
Benjamin Kramer36b0f122012-12-12 22:40:02 +0000266 } else {
Nick Kledzikf60a9272012-12-12 20:46:15 +0000267 setError(CurrentNode, "expected sequence of bit values");
268 }
269 DoClear = true;
270 return true;
271}
272
273bool Input::bitSetMatch(const char *Str, bool) {
Benjamin Kramer36b0f122012-12-12 22:40:02 +0000274 if (EC)
Nick Kledzikf60a9272012-12-12 20:46:15 +0000275 return false;
Benjamin Kramer36b0f122012-12-12 22:40:02 +0000276 if (SequenceHNode *SQ = dyn_cast<SequenceHNode>(CurrentNode)) {
Nick Kledzikf60a9272012-12-12 20:46:15 +0000277 unsigned Index = 0;
David Blaikied759fe52014-09-15 18:39:24 +0000278 for (auto &N : SQ->Entries) {
279 if (ScalarHNode *SN = dyn_cast<ScalarHNode>(N.get())) {
Benjamin Kramer36b0f122012-12-12 22:40:02 +0000280 if (SN->value().equals(Str)) {
Nick Kledzikf60a9272012-12-12 20:46:15 +0000281 BitValuesUsed[Index] = true;
282 return true;
283 }
Benjamin Kramer36b0f122012-12-12 22:40:02 +0000284 } else {
Nick Kledzikf60a9272012-12-12 20:46:15 +0000285 setError(CurrentNode, "unexpected scalar in sequence of bit values");
286 }
287 ++Index;
288 }
Benjamin Kramer36b0f122012-12-12 22:40:02 +0000289 } else {
Nick Kledzikf60a9272012-12-12 20:46:15 +0000290 setError(CurrentNode, "expected sequence of bit values");
291 }
292 return false;
293}
294
295void Input::endBitSetScalar() {
Benjamin Kramer36b0f122012-12-12 22:40:02 +0000296 if (EC)
Nick Kledzikf60a9272012-12-12 20:46:15 +0000297 return;
Benjamin Kramer36b0f122012-12-12 22:40:02 +0000298 if (SequenceHNode *SQ = dyn_cast<SequenceHNode>(CurrentNode)) {
Nick Kledzikf60a9272012-12-12 20:46:15 +0000299 assert(BitValuesUsed.size() == SQ->Entries.size());
Benjamin Kramer36b0f122012-12-12 22:40:02 +0000300 for (unsigned i = 0; i < SQ->Entries.size(); ++i) {
301 if (!BitValuesUsed[i]) {
David Blaikied759fe52014-09-15 18:39:24 +0000302 setError(SQ->Entries[i].get(), "unknown bit value");
Nick Kledzikf60a9272012-12-12 20:46:15 +0000303 return;
304 }
305 }
306 }
307}
308
David Majnemer77880332014-04-10 07:37:33 +0000309void Input::scalarString(StringRef &S, bool) {
Benjamin Kramer36b0f122012-12-12 22:40:02 +0000310 if (ScalarHNode *SN = dyn_cast<ScalarHNode>(CurrentNode)) {
Nick Kledzikf60a9272012-12-12 20:46:15 +0000311 S = SN->value();
Benjamin Kramer36b0f122012-12-12 22:40:02 +0000312 } else {
Nick Kledzikf60a9272012-12-12 20:46:15 +0000313 setError(CurrentNode, "unexpected scalar");
314 }
315}
316
Alex Lorenz68e787b2015-05-14 23:08:22 +0000317void Input::blockScalarString(StringRef &S) { scalarString(S, false); }
318
Nick Kledzikf60a9272012-12-12 20:46:15 +0000319void Input::setError(HNode *hnode, const Twine &message) {
Alexander Kornienko681e37c2013-11-18 15:50:04 +0000320 assert(hnode && "HNode must not be NULL");
Nick Kledzikf60a9272012-12-12 20:46:15 +0000321 this->setError(hnode->_node, message);
322}
323
324void Input::setError(Node *node, const Twine &message) {
325 Strm->printError(node, message);
Rafael Espindola2a826e42014-06-13 17:20:48 +0000326 EC = make_error_code(errc::invalid_argument);
Nick Kledzikf60a9272012-12-12 20:46:15 +0000327}
328
David Blaikied759fe52014-09-15 18:39:24 +0000329std::unique_ptr<Input::HNode> Input::createHNodes(Node *N) {
Benjamin Kramer36b0f122012-12-12 22:40:02 +0000330 SmallString<128> StringStorage;
331 if (ScalarNode *SN = dyn_cast<ScalarNode>(N)) {
Nick Kledzikf60a9272012-12-12 20:46:15 +0000332 StringRef KeyStr = SN->getValue(StringStorage);
Benjamin Kramer36b0f122012-12-12 22:40:02 +0000333 if (!StringStorage.empty()) {
Nick Kledzikf60a9272012-12-12 20:46:15 +0000334 // Copy string to permanent storage
Benjamin Kramer7a923772015-08-05 14:16:38 +0000335 KeyStr = StringStorage.str().copy(StringAllocator);
Nick Kledzikf60a9272012-12-12 20:46:15 +0000336 }
David Blaikied759fe52014-09-15 18:39:24 +0000337 return llvm::make_unique<ScalarHNode>(N, KeyStr);
Alex Lorenz68e787b2015-05-14 23:08:22 +0000338 } else if (BlockScalarNode *BSN = dyn_cast<BlockScalarNode>(N)) {
Benjamin Kramer7a923772015-08-05 14:16:38 +0000339 StringRef ValueCopy = BSN->getValue().copy(StringAllocator);
340 return llvm::make_unique<ScalarHNode>(N, ValueCopy);
Benjamin Kramer36b0f122012-12-12 22:40:02 +0000341 } else if (SequenceNode *SQ = dyn_cast<SequenceNode>(N)) {
David Blaikied759fe52014-09-15 18:39:24 +0000342 auto SQHNode = llvm::make_unique<SequenceHNode>(N);
Simon Atanasyan878bd8a2014-04-10 06:02:49 +0000343 for (Node &SN : *SQ) {
David Blaikied759fe52014-09-15 18:39:24 +0000344 auto Entry = this->createHNodes(&SN);
Benjamin Kramer36b0f122012-12-12 22:40:02 +0000345 if (EC)
Nick Kledzikf60a9272012-12-12 20:46:15 +0000346 break;
David Blaikied759fe52014-09-15 18:39:24 +0000347 SQHNode->Entries.push_back(std::move(Entry));
Nick Kledzikf60a9272012-12-12 20:46:15 +0000348 }
David Blaikied759fe52014-09-15 18:39:24 +0000349 return std::move(SQHNode);
Benjamin Kramer36b0f122012-12-12 22:40:02 +0000350 } else if (MappingNode *Map = dyn_cast<MappingNode>(N)) {
David Blaikied759fe52014-09-15 18:39:24 +0000351 auto mapHNode = llvm::make_unique<MapHNode>(N);
Simon Atanasyan878bd8a2014-04-10 06:02:49 +0000352 for (KeyValueNode &KVN : *Map) {
Rafael Espindolaa97373f2014-08-08 13:58:00 +0000353 Node *KeyNode = KVN.getKey();
354 ScalarNode *KeyScalar = dyn_cast<ScalarNode>(KeyNode);
355 if (!KeyScalar) {
356 setError(KeyNode, "Map key must be a scalar");
357 break;
358 }
Nick Kledzikf60a9272012-12-12 20:46:15 +0000359 StringStorage.clear();
Benjamin Kramer36b0f122012-12-12 22:40:02 +0000360 StringRef KeyStr = KeyScalar->getValue(StringStorage);
361 if (!StringStorage.empty()) {
Nick Kledzikf60a9272012-12-12 20:46:15 +0000362 // Copy string to permanent storage
Benjamin Kramer7a923772015-08-05 14:16:38 +0000363 KeyStr = StringStorage.str().copy(StringAllocator);
Nick Kledzikf60a9272012-12-12 20:46:15 +0000364 }
David Blaikied759fe52014-09-15 18:39:24 +0000365 auto ValueHNode = this->createHNodes(KVN.getValue());
Benjamin Kramer36b0f122012-12-12 22:40:02 +0000366 if (EC)
Nick Kledzikf60a9272012-12-12 20:46:15 +0000367 break;
David Blaikied759fe52014-09-15 18:39:24 +0000368 mapHNode->Mapping[KeyStr] = std::move(ValueHNode);
Nick Kledzikf60a9272012-12-12 20:46:15 +0000369 }
David Blaikied759fe52014-09-15 18:39:24 +0000370 return std::move(mapHNode);
Benjamin Kramer36b0f122012-12-12 22:40:02 +0000371 } else if (isa<NullNode>(N)) {
David Blaikied759fe52014-09-15 18:39:24 +0000372 return llvm::make_unique<EmptyHNode>(N);
Benjamin Kramer36b0f122012-12-12 22:40:02 +0000373 } else {
Nick Kledzikf60a9272012-12-12 20:46:15 +0000374 setError(N, "unknown node kind");
Craig Topperc10719f2014-04-07 04:17:22 +0000375 return nullptr;
Nick Kledzikf60a9272012-12-12 20:46:15 +0000376 }
377}
378
Nick Kledzikf60a9272012-12-12 20:46:15 +0000379bool Input::MapHNode::isValidKey(StringRef Key) {
Simon Atanasyan878bd8a2014-04-10 06:02:49 +0000380 for (const char *K : ValidKeys) {
381 if (Key.equals(K))
Nick Kledzikf60a9272012-12-12 20:46:15 +0000382 return true;
383 }
384 return false;
385}
386
387void Input::setError(const Twine &Message) {
388 this->setError(CurrentNode, Message);
389}
390
Aaron Ballman0e63e532013-08-15 23:17:53 +0000391bool Input::canElideEmptySequence() {
392 return false;
393}
394
Nick Kledzikf60a9272012-12-12 20:46:15 +0000395//===----------------------------------------------------------------------===//
396// Output
397//===----------------------------------------------------------------------===//
398
Frederic Riss4939e6a2015-05-29 17:56:28 +0000399Output::Output(raw_ostream &yout, void *context, int WrapColumn)
Benjamin Kramer36b0f122012-12-12 22:40:02 +0000400 : IO(context),
401 Out(yout),
Frederic Riss4939e6a2015-05-29 17:56:28 +0000402 WrapColumn(WrapColumn),
Benjamin Kramer36b0f122012-12-12 22:40:02 +0000403 Column(0),
404 ColumnAtFlowStart(0),
Alex Lorenzb1225082015-05-04 20:11:40 +0000405 ColumnAtMapFlowStart(0),
Benjamin Kramer36b0f122012-12-12 22:40:02 +0000406 NeedBitValueComma(false),
407 NeedFlowSequenceComma(false),
408 EnumerationMatchFound(false),
409 NeedsNewLine(false) {
Nick Kledzikf60a9272012-12-12 20:46:15 +0000410}
411
412Output::~Output() {
413}
414
Nick Kledzik4761c602013-11-21 00:20:10 +0000415bool Output::outputting() {
Nick Kledzikf60a9272012-12-12 20:46:15 +0000416 return true;
417}
418
419void Output::beginMapping() {
420 StateStack.push_back(inMapFirstKey);
421 NeedsNewLine = true;
422}
423
Nick Kledzik1e6033c2013-11-14 00:59:59 +0000424bool Output::mapTag(StringRef Tag, bool Use) {
425 if (Use) {
426 this->output(" ");
427 this->output(Tag);
428 }
429 return Use;
430}
431
Nick Kledzikf60a9272012-12-12 20:46:15 +0000432void Output::endMapping() {
433 StateStack.pop_back();
434}
435
Nick Kledzikf60a9272012-12-12 20:46:15 +0000436bool Output::preflightKey(const char *Key, bool Required, bool SameAsDefault,
Benjamin Kramer36b0f122012-12-12 22:40:02 +0000437 bool &UseDefault, void *&) {
Nick Kledzikf60a9272012-12-12 20:46:15 +0000438 UseDefault = false;
Benjamin Kramer36b0f122012-12-12 22:40:02 +0000439 if (Required || !SameAsDefault) {
Alex Lorenzb1225082015-05-04 20:11:40 +0000440 auto State = StateStack.back();
441 if (State == inFlowMapFirstKey || State == inFlowMapOtherKey) {
442 flowKey(Key);
443 } else {
444 this->newLineCheck();
445 this->paddedKey(Key);
446 }
Nick Kledzikf60a9272012-12-12 20:46:15 +0000447 return true;
448 }
449 return false;
450}
451
Benjamin Kramer36b0f122012-12-12 22:40:02 +0000452void Output::postflightKey(void *) {
453 if (StateStack.back() == inMapFirstKey) {
Nick Kledzikf60a9272012-12-12 20:46:15 +0000454 StateStack.pop_back();
455 StateStack.push_back(inMapOtherKey);
Alex Lorenzb1225082015-05-04 20:11:40 +0000456 } else if (StateStack.back() == inFlowMapFirstKey) {
457 StateStack.pop_back();
458 StateStack.push_back(inFlowMapOtherKey);
Nick Kledzikf60a9272012-12-12 20:46:15 +0000459 }
460}
461
Alex Lorenzb1225082015-05-04 20:11:40 +0000462void Output::beginFlowMapping() {
463 StateStack.push_back(inFlowMapFirstKey);
464 this->newLineCheck();
465 ColumnAtMapFlowStart = Column;
466 output("{ ");
467}
468
469void Output::endFlowMapping() {
470 StateStack.pop_back();
471 this->outputUpToEndOfLine(" }");
472}
473
Nick Kledzikf60a9272012-12-12 20:46:15 +0000474void Output::beginDocuments() {
475 this->outputUpToEndOfLine("---");
476}
477
478bool Output::preflightDocument(unsigned index) {
Benjamin Kramer36b0f122012-12-12 22:40:02 +0000479 if (index > 0)
Nick Kledzikf60a9272012-12-12 20:46:15 +0000480 this->outputUpToEndOfLine("\n---");
481 return true;
482}
483
484void Output::postflightDocument() {
485}
486
487void Output::endDocuments() {
488 output("\n...\n");
489}
490
491unsigned Output::beginSequence() {
492 StateStack.push_back(inSeq);
493 NeedsNewLine = true;
494 return 0;
495}
Benjamin Kramer36b0f122012-12-12 22:40:02 +0000496
Nick Kledzikf60a9272012-12-12 20:46:15 +0000497void Output::endSequence() {
498 StateStack.pop_back();
499}
Benjamin Kramer36b0f122012-12-12 22:40:02 +0000500
501bool Output::preflightElement(unsigned, void *&) {
Nick Kledzikf60a9272012-12-12 20:46:15 +0000502 return true;
503}
Benjamin Kramer36b0f122012-12-12 22:40:02 +0000504
505void Output::postflightElement(void *) {
Nick Kledzikf60a9272012-12-12 20:46:15 +0000506}
507
508unsigned Output::beginFlowSequence() {
Nick Kledzikf60a9272012-12-12 20:46:15 +0000509 StateStack.push_back(inFlowSeq);
Nick Kledzik11964f22013-01-04 19:32:00 +0000510 this->newLineCheck();
Nick Kledzikf60a9272012-12-12 20:46:15 +0000511 ColumnAtFlowStart = Column;
512 output("[ ");
513 NeedFlowSequenceComma = false;
514 return 0;
515}
Benjamin Kramer36b0f122012-12-12 22:40:02 +0000516
Nick Kledzikf60a9272012-12-12 20:46:15 +0000517void Output::endFlowSequence() {
518 StateStack.pop_back();
519 this->outputUpToEndOfLine(" ]");
520}
Benjamin Kramer36b0f122012-12-12 22:40:02 +0000521
522bool Output::preflightFlowElement(unsigned, void *&) {
523 if (NeedFlowSequenceComma)
Nick Kledzikf60a9272012-12-12 20:46:15 +0000524 output(", ");
Frederic Riss4939e6a2015-05-29 17:56:28 +0000525 if (WrapColumn && Column > WrapColumn) {
Nick Kledzikf60a9272012-12-12 20:46:15 +0000526 output("\n");
Benjamin Kramer36b0f122012-12-12 22:40:02 +0000527 for (int i = 0; i < ColumnAtFlowStart; ++i)
Nick Kledzikf60a9272012-12-12 20:46:15 +0000528 output(" ");
529 Column = ColumnAtFlowStart;
530 output(" ");
531 }
532 return true;
533}
Benjamin Kramer36b0f122012-12-12 22:40:02 +0000534
535void Output::postflightFlowElement(void *) {
Nick Kledzikf60a9272012-12-12 20:46:15 +0000536 NeedFlowSequenceComma = true;
537}
538
Nick Kledzikf60a9272012-12-12 20:46:15 +0000539void Output::beginEnumScalar() {
540 EnumerationMatchFound = false;
541}
542
543bool Output::matchEnumScalar(const char *Str, bool Match) {
Benjamin Kramer36b0f122012-12-12 22:40:02 +0000544 if (Match && !EnumerationMatchFound) {
Nick Kledzikf60a9272012-12-12 20:46:15 +0000545 this->newLineCheck();
546 this->outputUpToEndOfLine(Str);
547 EnumerationMatchFound = true;
548 }
549 return false;
550}
551
Michael J. Spencer731cae32015-01-23 21:57:50 +0000552bool Output::matchEnumFallback() {
553 if (EnumerationMatchFound)
554 return false;
555 EnumerationMatchFound = true;
556 return true;
557}
558
Nick Kledzikf60a9272012-12-12 20:46:15 +0000559void Output::endEnumScalar() {
Benjamin Kramer36b0f122012-12-12 22:40:02 +0000560 if (!EnumerationMatchFound)
Nick Kledzikf60a9272012-12-12 20:46:15 +0000561 llvm_unreachable("bad runtime enum value");
562}
563
Nick Kledzikf60a9272012-12-12 20:46:15 +0000564bool Output::beginBitSetScalar(bool &DoClear) {
565 this->newLineCheck();
566 output("[ ");
567 NeedBitValueComma = false;
568 DoClear = false;
569 return true;
570}
571
572bool Output::bitSetMatch(const char *Str, bool Matches) {
Benjamin Kramer36b0f122012-12-12 22:40:02 +0000573 if (Matches) {
574 if (NeedBitValueComma)
Nick Kledzikf60a9272012-12-12 20:46:15 +0000575 output(", ");
576 this->output(Str);
577 NeedBitValueComma = true;
578 }
579 return false;
580}
581
582void Output::endBitSetScalar() {
583 this->outputUpToEndOfLine(" ]");
584}
585
David Majnemer77880332014-04-10 07:37:33 +0000586void Output::scalarString(StringRef &S, bool MustQuote) {
Nick Kledzikf60a9272012-12-12 20:46:15 +0000587 this->newLineCheck();
Rui Ueyama106eded2013-09-11 04:00:08 +0000588 if (S.empty()) {
589 // Print '' for the empty string because leaving the field empty is not
590 // allowed.
591 this->outputUpToEndOfLine("''");
592 return;
593 }
David Majnemer77880332014-04-10 07:37:33 +0000594 if (!MustQuote) {
595 // Only quote if we must.
Nick Kledzikf60a9272012-12-12 20:46:15 +0000596 this->outputUpToEndOfLine(S);
597 return;
598 }
599 unsigned i = 0;
600 unsigned j = 0;
601 unsigned End = S.size();
602 output("'"); // Starting single quote.
603 const char *Base = S.data();
604 while (j < End) {
605 // Escape a single quote by doubling it.
606 if (S[j] == '\'') {
607 output(StringRef(&Base[i], j - i + 1));
608 output("'");
609 i = j + 1;
610 }
611 ++j;
612 }
613 output(StringRef(&Base[i], j - i));
614 this->outputUpToEndOfLine("'"); // Ending single quote.
615}
616
Alex Lorenz68e787b2015-05-14 23:08:22 +0000617void Output::blockScalarString(StringRef &S) {
618 if (!StateStack.empty())
619 newLineCheck();
620 output(" |");
621 outputNewLine();
622
623 unsigned Indent = StateStack.empty() ? 1 : StateStack.size();
624
625 auto Buffer = MemoryBuffer::getMemBuffer(S, "", false);
626 for (line_iterator Lines(*Buffer, false); !Lines.is_at_end(); ++Lines) {
627 for (unsigned I = 0; I < Indent; ++I) {
628 output(" ");
629 }
630 output(*Lines);
631 outputNewLine();
632 }
633}
634
Nick Kledzikf60a9272012-12-12 20:46:15 +0000635void Output::setError(const Twine &message) {
636}
637
Aaron Ballman0e63e532013-08-15 23:17:53 +0000638bool Output::canElideEmptySequence() {
639 // Normally, with an optional key/value where the value is an empty sequence,
640 // the whole key/value can be not written. But, that produces wrong yaml
641 // if the key/value is the only thing in the map and the map is used in
642 // a sequence. This detects if the this sequence is the first key/value
643 // in map that itself is embedded in a sequnce.
Rui Ueyama38dfffa2013-09-11 00:53:07 +0000644 if (StateStack.size() < 2)
Aaron Ballman0e63e532013-08-15 23:17:53 +0000645 return true;
Rui Ueyama38dfffa2013-09-11 00:53:07 +0000646 if (StateStack.back() != inMapFirstKey)
Aaron Ballman0e63e532013-08-15 23:17:53 +0000647 return true;
648 return (StateStack[StateStack.size()-2] != inSeq);
649}
650
Nick Kledzikf60a9272012-12-12 20:46:15 +0000651void Output::output(StringRef s) {
652 Column += s.size();
653 Out << s;
654}
655
656void Output::outputUpToEndOfLine(StringRef s) {
657 this->output(s);
Alex Lorenzb1225082015-05-04 20:11:40 +0000658 if (StateStack.empty() || (StateStack.back() != inFlowSeq &&
659 StateStack.back() != inFlowMapFirstKey &&
660 StateStack.back() != inFlowMapOtherKey))
Nick Kledzikf60a9272012-12-12 20:46:15 +0000661 NeedsNewLine = true;
662}
663
664void Output::outputNewLine() {
665 Out << "\n";
666 Column = 0;
667}
668
669// if seq at top, indent as if map, then add "- "
670// if seq in middle, use "- " if firstKey, else use " "
671//
672
673void Output::newLineCheck() {
Benjamin Kramer36b0f122012-12-12 22:40:02 +0000674 if (!NeedsNewLine)
Nick Kledzikf60a9272012-12-12 20:46:15 +0000675 return;
676 NeedsNewLine = false;
677
678 this->outputNewLine();
679
680 assert(StateStack.size() > 0);
681 unsigned Indent = StateStack.size() - 1;
682 bool OutputDash = false;
683
Benjamin Kramer36b0f122012-12-12 22:40:02 +0000684 if (StateStack.back() == inSeq) {
Nick Kledzikf60a9272012-12-12 20:46:15 +0000685 OutputDash = true;
Alex Lorenz42e91fa2015-05-01 18:34:25 +0000686 } else if ((StateStack.size() > 1) && ((StateStack.back() == inMapFirstKey) ||
Alex Lorenzb1225082015-05-04 20:11:40 +0000687 (StateStack.back() == inFlowSeq) ||
688 (StateStack.back() == inFlowMapFirstKey)) &&
Benjamin Kramer36b0f122012-12-12 22:40:02 +0000689 (StateStack[StateStack.size() - 2] == inSeq)) {
Nick Kledzikf60a9272012-12-12 20:46:15 +0000690 --Indent;
691 OutputDash = true;
692 }
693
Benjamin Kramer36b0f122012-12-12 22:40:02 +0000694 for (unsigned i = 0; i < Indent; ++i) {
Nick Kledzikf60a9272012-12-12 20:46:15 +0000695 output(" ");
696 }
Benjamin Kramer36b0f122012-12-12 22:40:02 +0000697 if (OutputDash) {
Nick Kledzikf60a9272012-12-12 20:46:15 +0000698 output("- ");
699 }
700
701}
702
703void Output::paddedKey(StringRef key) {
704 output(key);
705 output(":");
706 const char *spaces = " ";
Benjamin Kramer36b0f122012-12-12 22:40:02 +0000707 if (key.size() < strlen(spaces))
Nick Kledzikf60a9272012-12-12 20:46:15 +0000708 output(&spaces[key.size()]);
709 else
710 output(" ");
711}
712
Alex Lorenzb1225082015-05-04 20:11:40 +0000713void Output::flowKey(StringRef Key) {
714 if (StateStack.back() == inFlowMapOtherKey)
715 output(", ");
Frederic Riss4939e6a2015-05-29 17:56:28 +0000716 if (WrapColumn && Column > WrapColumn) {
Alex Lorenzb1225082015-05-04 20:11:40 +0000717 output("\n");
718 for (int I = 0; I < ColumnAtMapFlowStart; ++I)
719 output(" ");
720 Column = ColumnAtMapFlowStart;
721 output(" ");
722 }
723 output(Key);
724 output(": ");
725}
726
Nick Kledzikf60a9272012-12-12 20:46:15 +0000727//===----------------------------------------------------------------------===//
728// traits for built-in types
729//===----------------------------------------------------------------------===//
730
Benjamin Kramer36b0f122012-12-12 22:40:02 +0000731void ScalarTraits<bool>::output(const bool &Val, void *, raw_ostream &Out) {
732 Out << (Val ? "true" : "false");
733}
Nick Kledzikf60a9272012-12-12 20:46:15 +0000734
Benjamin Kramer36b0f122012-12-12 22:40:02 +0000735StringRef ScalarTraits<bool>::input(StringRef Scalar, void *, bool &Val) {
736 if (Scalar.equals("true")) {
737 Val = true;
738 return StringRef();
739 } else if (Scalar.equals("false")) {
740 Val = false;
Nick Kledzikf60a9272012-12-12 20:46:15 +0000741 return StringRef();
742 }
Benjamin Kramer36b0f122012-12-12 22:40:02 +0000743 return "invalid boolean";
744}
Nick Kledzikf60a9272012-12-12 20:46:15 +0000745
Benjamin Kramer36b0f122012-12-12 22:40:02 +0000746void ScalarTraits<StringRef>::output(const StringRef &Val, void *,
747 raw_ostream &Out) {
748 Out << Val;
749}
Nick Kledzikf60a9272012-12-12 20:46:15 +0000750
Benjamin Kramer36b0f122012-12-12 22:40:02 +0000751StringRef ScalarTraits<StringRef>::input(StringRef Scalar, void *,
752 StringRef &Val) {
753 Val = Scalar;
754 return StringRef();
755}
Alex Rosenbergf298f162015-01-26 18:02:18 +0000756
John Thompson48e018a2013-11-19 17:28:21 +0000757void ScalarTraits<std::string>::output(const std::string &Val, void *,
758 raw_ostream &Out) {
759 Out << Val;
760}
761
762StringRef ScalarTraits<std::string>::input(StringRef Scalar, void *,
763 std::string &Val) {
764 Val = Scalar.str();
765 return StringRef();
766}
Nick Kledzikf60a9272012-12-12 20:46:15 +0000767
Benjamin Kramer36b0f122012-12-12 22:40:02 +0000768void ScalarTraits<uint8_t>::output(const uint8_t &Val, void *,
769 raw_ostream &Out) {
770 // use temp uin32_t because ostream thinks uint8_t is a character
771 uint32_t Num = Val;
772 Out << Num;
773}
Nick Kledzikf60a9272012-12-12 20:46:15 +0000774
Benjamin Kramer36b0f122012-12-12 22:40:02 +0000775StringRef ScalarTraits<uint8_t>::input(StringRef Scalar, void *, uint8_t &Val) {
776 unsigned long long n;
777 if (getAsUnsignedInteger(Scalar, 0, n))
778 return "invalid number";
779 if (n > 0xFF)
780 return "out of range number";
781 Val = n;
782 return StringRef();
783}
Nick Kledzikf60a9272012-12-12 20:46:15 +0000784
Benjamin Kramer36b0f122012-12-12 22:40:02 +0000785void ScalarTraits<uint16_t>::output(const uint16_t &Val, void *,
786 raw_ostream &Out) {
787 Out << Val;
788}
Nick Kledzikf60a9272012-12-12 20:46:15 +0000789
Benjamin Kramer36b0f122012-12-12 22:40:02 +0000790StringRef ScalarTraits<uint16_t>::input(StringRef Scalar, void *,
791 uint16_t &Val) {
792 unsigned long long n;
793 if (getAsUnsignedInteger(Scalar, 0, n))
794 return "invalid number";
795 if (n > 0xFFFF)
796 return "out of range number";
797 Val = n;
798 return StringRef();
799}
Nick Kledzikf60a9272012-12-12 20:46:15 +0000800
Benjamin Kramer36b0f122012-12-12 22:40:02 +0000801void ScalarTraits<uint32_t>::output(const uint32_t &Val, void *,
802 raw_ostream &Out) {
803 Out << Val;
804}
Nick Kledzikf60a9272012-12-12 20:46:15 +0000805
Benjamin Kramer36b0f122012-12-12 22:40:02 +0000806StringRef ScalarTraits<uint32_t>::input(StringRef Scalar, void *,
807 uint32_t &Val) {
808 unsigned long long n;
809 if (getAsUnsignedInteger(Scalar, 0, n))
810 return "invalid number";
811 if (n > 0xFFFFFFFFUL)
812 return "out of range number";
813 Val = n;
814 return StringRef();
815}
Nick Kledzikf60a9272012-12-12 20:46:15 +0000816
Benjamin Kramer36b0f122012-12-12 22:40:02 +0000817void ScalarTraits<uint64_t>::output(const uint64_t &Val, void *,
818 raw_ostream &Out) {
819 Out << Val;
820}
Nick Kledzikf60a9272012-12-12 20:46:15 +0000821
Benjamin Kramer36b0f122012-12-12 22:40:02 +0000822StringRef ScalarTraits<uint64_t>::input(StringRef Scalar, void *,
823 uint64_t &Val) {
824 unsigned long long N;
825 if (getAsUnsignedInteger(Scalar, 0, N))
826 return "invalid number";
827 Val = N;
828 return StringRef();
829}
Nick Kledzikf60a9272012-12-12 20:46:15 +0000830
Benjamin Kramer36b0f122012-12-12 22:40:02 +0000831void ScalarTraits<int8_t>::output(const int8_t &Val, void *, raw_ostream &Out) {
832 // use temp in32_t because ostream thinks int8_t is a character
833 int32_t Num = Val;
834 Out << Num;
835}
Nick Kledzikf60a9272012-12-12 20:46:15 +0000836
Benjamin Kramer36b0f122012-12-12 22:40:02 +0000837StringRef ScalarTraits<int8_t>::input(StringRef Scalar, void *, int8_t &Val) {
838 long long N;
839 if (getAsSignedInteger(Scalar, 0, N))
840 return "invalid number";
841 if ((N > 127) || (N < -128))
842 return "out of range number";
843 Val = N;
844 return StringRef();
845}
Nick Kledzikf60a9272012-12-12 20:46:15 +0000846
Benjamin Kramer36b0f122012-12-12 22:40:02 +0000847void ScalarTraits<int16_t>::output(const int16_t &Val, void *,
848 raw_ostream &Out) {
849 Out << Val;
850}
Nick Kledzikf60a9272012-12-12 20:46:15 +0000851
Benjamin Kramer36b0f122012-12-12 22:40:02 +0000852StringRef ScalarTraits<int16_t>::input(StringRef Scalar, void *, int16_t &Val) {
853 long long N;
854 if (getAsSignedInteger(Scalar, 0, N))
855 return "invalid number";
856 if ((N > INT16_MAX) || (N < INT16_MIN))
857 return "out of range number";
858 Val = N;
859 return StringRef();
860}
Nick Kledzikf60a9272012-12-12 20:46:15 +0000861
Benjamin Kramer36b0f122012-12-12 22:40:02 +0000862void ScalarTraits<int32_t>::output(const int32_t &Val, void *,
863 raw_ostream &Out) {
864 Out << Val;
865}
866
867StringRef ScalarTraits<int32_t>::input(StringRef Scalar, void *, int32_t &Val) {
868 long long N;
869 if (getAsSignedInteger(Scalar, 0, N))
870 return "invalid number";
871 if ((N > INT32_MAX) || (N < INT32_MIN))
872 return "out of range number";
873 Val = N;
874 return StringRef();
875}
876
877void ScalarTraits<int64_t>::output(const int64_t &Val, void *,
878 raw_ostream &Out) {
879 Out << Val;
880}
881
882StringRef ScalarTraits<int64_t>::input(StringRef Scalar, void *, int64_t &Val) {
883 long long N;
884 if (getAsSignedInteger(Scalar, 0, N))
885 return "invalid number";
886 Val = N;
887 return StringRef();
888}
889
890void ScalarTraits<double>::output(const double &Val, void *, raw_ostream &Out) {
Nick Kledzikf60a9272012-12-12 20:46:15 +0000891 Out << format("%g", Val);
Benjamin Kramer36b0f122012-12-12 22:40:02 +0000892}
Nick Kledzikf60a9272012-12-12 20:46:15 +0000893
Benjamin Kramer36b0f122012-12-12 22:40:02 +0000894StringRef ScalarTraits<double>::input(StringRef Scalar, void *, double &Val) {
895 SmallString<32> buff(Scalar.begin(), Scalar.end());
896 char *end;
897 Val = strtod(buff.c_str(), &end);
898 if (*end != '\0')
899 return "invalid floating point number";
900 return StringRef();
901}
902
903void ScalarTraits<float>::output(const float &Val, void *, raw_ostream &Out) {
Nick Kledzikf60a9272012-12-12 20:46:15 +0000904 Out << format("%g", Val);
Benjamin Kramer36b0f122012-12-12 22:40:02 +0000905}
Nick Kledzikf60a9272012-12-12 20:46:15 +0000906
Benjamin Kramer36b0f122012-12-12 22:40:02 +0000907StringRef ScalarTraits<float>::input(StringRef Scalar, void *, float &Val) {
908 SmallString<32> buff(Scalar.begin(), Scalar.end());
909 char *end;
910 Val = strtod(buff.c_str(), &end);
911 if (*end != '\0')
912 return "invalid floating point number";
913 return StringRef();
914}
Nick Kledzikf60a9272012-12-12 20:46:15 +0000915
Benjamin Kramer36b0f122012-12-12 22:40:02 +0000916void ScalarTraits<Hex8>::output(const Hex8 &Val, void *, raw_ostream &Out) {
917 uint8_t Num = Val;
918 Out << format("0x%02X", Num);
919}
Nick Kledzikf60a9272012-12-12 20:46:15 +0000920
Benjamin Kramer36b0f122012-12-12 22:40:02 +0000921StringRef ScalarTraits<Hex8>::input(StringRef Scalar, void *, Hex8 &Val) {
922 unsigned long long n;
923 if (getAsUnsignedInteger(Scalar, 0, n))
924 return "invalid hex8 number";
925 if (n > 0xFF)
926 return "out of range hex8 number";
927 Val = n;
928 return StringRef();
929}
Nick Kledzikf60a9272012-12-12 20:46:15 +0000930
Benjamin Kramer36b0f122012-12-12 22:40:02 +0000931void ScalarTraits<Hex16>::output(const Hex16 &Val, void *, raw_ostream &Out) {
932 uint16_t Num = Val;
933 Out << format("0x%04X", Num);
934}
Nick Kledzikf60a9272012-12-12 20:46:15 +0000935
Benjamin Kramer36b0f122012-12-12 22:40:02 +0000936StringRef ScalarTraits<Hex16>::input(StringRef Scalar, void *, Hex16 &Val) {
937 unsigned long long n;
938 if (getAsUnsignedInteger(Scalar, 0, n))
939 return "invalid hex16 number";
940 if (n > 0xFFFF)
941 return "out of range hex16 number";
942 Val = n;
943 return StringRef();
944}
Nick Kledzikf60a9272012-12-12 20:46:15 +0000945
Benjamin Kramer36b0f122012-12-12 22:40:02 +0000946void ScalarTraits<Hex32>::output(const Hex32 &Val, void *, raw_ostream &Out) {
947 uint32_t Num = Val;
948 Out << format("0x%08X", Num);
949}
Nick Kledzikf60a9272012-12-12 20:46:15 +0000950
Benjamin Kramer36b0f122012-12-12 22:40:02 +0000951StringRef ScalarTraits<Hex32>::input(StringRef Scalar, void *, Hex32 &Val) {
952 unsigned long long n;
953 if (getAsUnsignedInteger(Scalar, 0, n))
954 return "invalid hex32 number";
955 if (n > 0xFFFFFFFFUL)
956 return "out of range hex32 number";
957 Val = n;
958 return StringRef();
959}
Nick Kledzikf60a9272012-12-12 20:46:15 +0000960
Benjamin Kramer36b0f122012-12-12 22:40:02 +0000961void ScalarTraits<Hex64>::output(const Hex64 &Val, void *, raw_ostream &Out) {
962 uint64_t Num = Val;
963 Out << format("0x%016llX", Num);
964}
Nick Kledzikf60a9272012-12-12 20:46:15 +0000965
Benjamin Kramer36b0f122012-12-12 22:40:02 +0000966StringRef ScalarTraits<Hex64>::input(StringRef Scalar, void *, Hex64 &Val) {
967 unsigned long long Num;
968 if (getAsUnsignedInteger(Scalar, 0, Num))
969 return "invalid hex64 number";
970 Val = Num;
971 return StringRef();
972}