maxjhandsome
/
llvm-project
forked from OSchip/llvm-project
1
0
Fork 0
Code Issues Pull Requests Packages Releases Wiki Activity

Initial implementation of a utility for converting native data 

structures to and from YAML using traits.  The first client will
be the test suite of lld.  The documentation will show up at:

   http://llvm.org/docs/YamlIO.html

llvm-svn: 170019
This commit is contained in:
Nick Kledzik 2012-12-12 20:46:15 +00:00
parent b7d3d03bf9
commit f60a9279ea
7 changed files with 4152 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

862
llvm/docs/YamlIO.rst Normal file
View File

@ -0,0 +1,862 @@
.. _yamlio:
=====================
YAML I/O
=====================
.. contents::
:local:
Introduction to YAML
====================
YAML is a human readable data serialization language. The full YAML language
spec can be read at `yaml.org
<http://www.yaml.org/spec/1.2/spec.html#Introduction>`_. The simplest form of
yaml is just "scalars", "mappings", and "sequences". A scalar is any number
or string. The pound/hash symbol (#) begins a comment line. A mapping is
a set of key-value pairs where the key ends with a colon. For example:
.. code-block:: yaml
# a mapping
name: Tom
hat-size: 7
A sequence is a list of items where each item starts with a leading dash ('-').
For example:
.. code-block:: yaml
# a sequence
- x86
- x86_64
- PowerPC
You can combine mappings and sequences by indenting. For example a sequence
of mappings in which one of the mapping values is itself a sequence:
.. code-block:: yaml
# a sequence of mappings with one key's value being a sequence
- name: Tom
cpus:
- x86
- x86_64
- name: Bob
cpus:
- x86
- name: Dan
cpus:
- PowerPC
- x86
Sometime sequences are known to be short and the one entry per line is too
verbose, so YAML offers an alternate syntax for sequences called a "Flow
Sequence" in which you put comma separated sequence elements into square
brackets. The above example could then be simplified to :
.. code-block:: yaml
# a sequence of mappings with one key's value being a flow sequence
- name: Tom
cpus: [ x86, x86_64 ]
- name: Bob
cpus: [ x86 ]
- name: Dan
cpus: [ PowerPC, x86 ]
Introduction to YAML I/O
========================
The use of indenting makes the YAML easy for a human to read and understand,
but having a program read and write YAML involves a lot of tedious details.
The YAML I/O library structures and simplifies reading and writing YAML
documents.
YAML I/O assumes you have some "native" data structures which you want to be
able to dump as YAML and recreate from YAML. The first step is to try
writing example YAML for your data structures. You may find after looking at
possible YAML representations that a direct mapping of your data structures
to YAML is not very readable. Often the fields are not in the order that
a human would find readable. Or the same information is replicated in multiple
locations, making it hard for a human to write such YAML correctly.
In relational database theory there is a design step called normalization in
which you reorganize fields and tables. The same considerations need to
go into the design of your YAML encoding. But, you may not want to change
your exisiting native data structures. Therefore, when writing out YAML
there may be a normalization step, and when reading YAML there would be a
corresponding denormalization step.
YAML I/O uses a non-invasive, traits based design. YAML I/O defines some
abstract base templates. You specialize those templates on your data types.
For instance, if you have an eumerated type FooBar you could specialize
ScalarEnumerationTraits on that type and define the enumeration() method:
.. code-block:: c++
using llvm::yaml::ScalarEnumerationTraits;
using llvm::yaml::IO;
template <>
struct ScalarEnumerationTraits<FooBar> {
static void enumeration(IO &io, FooBar &value) {
...
}
};
As with all YAML I/O template specializations, the ScalarEnumerationTraits is used for
both reading and writing YAML. That is, the mapping between in-memory enum
values and the YAML string representation is only in place.
This assures that the code for writing and parsing of YAML stays in sync.
To specify a YAML mappings, you define a specialization on
llvm::yaml::MapppingTraits.
If your native data structure happens to be a struct that is already normalized,
then the specialization is simple. For example:
.. code-block:: c++
using llvm::yaml::MapppingTraits;
using llvm::yaml::IO;
template <>
struct MapppingTraits<Person> {
static void mapping(IO &io, Person &info) {
io.mapRequired("name", info.name);
io.mapOptional("hat-size", info.hatSize);
}
};
A YAML sequence is automatically infered if you data type has begin()/end()
iterators and a push_back() method. Therefore any of the STL containers
(such as std::vector<>) will automatically translate to YAML sequences.
Once you have defined specializations for your data types, you can
programmatically use YAML I/O to write a YAML document:
.. code-block:: c++
using llvm::yaml::Output;
Person tom;
tom.name = "Tom";
tom.hatSize = 8;
Person dan;
dan.name = "Dan";
dan.hatSize = 7;
std::vector<Person> persons;
persons.push_back(tom);
persons.push_back(dan);
Output yout(llvm::outs());
yout << persons;
This would write the following:
.. code-block:: yaml
- name: Tom
hat-size: 8
- name: Dan
hat-size: 7
And you can also read such YAML documents with the following code:
.. code-block:: c++
using llvm::yaml::Input;
typedef std::vector<Person> PersonList;
std::vector<PersonList> docs;
Input yin(document.getBuffer());
yin >> docs;
if ( yin.error() )
return;
// Process read document
for ( PersonList &pl : docs ) {
for ( Person &person : pl ) {
cout << "name=" << person.name;
}
}
One other feature of YAML is the ability to define multiple documents in a
single file. That is why reading YAML produces a vector of your document type.
Error Handling
==============
When parsing a YAML document, if the input does not match your schema (as
expressed in your XxxTraits<> specializations). YAML I/O
will print out an error message and your Input object's error() method will
return true. For instance the following document:
.. code-block:: yaml
- name: Tom
shoe-size: 12
- name: Dan
hat-size: 7
Has a key (shoe-size) that is not defined in the schema. YAML I/O will
automatically generate this error:
.. code-block:: yaml
YAML:2:2: error: unknown key 'shoe-size'
shoe-size: 12
^~~~~~~~~
Similar errors are produced for other input not conforming to the schema.
Scalars
=======
YAML scalars are just strings (i.e. not a sequence or mapping). The YAML I/O
library provides support for translating between YAML scalars and specific
C++ types.
Built-in types
--------------
The following types have built-in support in YAML I/O:
* bool
* float
* double
* StringRef
* int64_t
* int32_t
* int16_t
* int8_t
* uint64_t
* uint32_t
* uint16_t
* uint8_t
That is, you can use those types in fields of MapppingTraits or as element type
in sequence. When reading, YAML I/O will validate that the string found
is convertible to that type and error out if not.
Unique types
------------
Given that YAML I/O is trait based, the selection of how to convert your data
to YAML is based on the type of your data. But in C++ type matching, typedefs
do not generate unique type names. That means if you have two typedefs of
unsigned int, to YAML I/O both types look exactly like unsigned int. To
facilitate make unique type names, YAML I/O provides a macro which is used
like a typedef on built-in types, but expands to create a class with conversion
operators to and from the base type. For example:
.. code-block:: c++
LLVM_YAML_STRONG_TYPEDEF(uint32_t, MyFooFlags)
LLVM_YAML_STRONG_TYPEDEF(uint32_t, MyBarFlags)
This generates two classes MyFooFlags and MyBarFlags which you can use in your
native data structures instead of uint32_t. They are implicitly
converted to and from uint32_t. The point of creating these unique types
is that you can now specify traits on them to get different YAML conversions.
Hex types
---------
An example use of a unique type is that YAML I/O provides fixed sized unsigned
integers that are written with YAML I/O as hexadecimal instead of the decimal
format used by the built-in integer types:
* Hex64
* Hex32
* Hex16
* Hex8
You can use llvm::yaml::Hex32 instead of uint32_t and the only different will
be that when YAML I/O writes out that type it will be formatted in hexadecimal.
ScalarEnumerationTraits
-----------------------
YAML I/O supports translating between in-memory enumerations and a set of string
values in YAML documents. This is done by specializing ScalarEnumerationTraits<>
on your enumeration type and define a enumeration() method.
For instance, suppose you had an enumeration of CPUs and a struct with it as
a field:
.. code-block:: c++
enum CPUs {
cpu_x86_64 = 5,
cpu_x86 = 7,
cpu_PowerPC = 8
};
struct Info {
CPUs cpu;
uint32_t flags;
};
To support reading and writing of this enumeration, you can define a
ScalarEnumerationTraits specialization on CPUs, which can then be used
as a field type:
.. code-block:: c++
using llvm::yaml::ScalarEnumerationTraits;
using llvm::yaml::MapppingTraits;
using llvm::yaml::IO;
template <>
struct ScalarEnumerationTraits<CPUs> {
static void enumeration(IO &io, CPUs &value) {
io.enumCase(value, "x86_64", cpu_x86_64);
io.enumCase(value, "x86", cpu_x86);
io.enumCase(value, "PowerPC", cpu_PowerPC);
}
};
template <>
struct MapppingTraits<Info> {
static void mapping(IO &io, Info &info) {
io.mapRequired("cpu", info.cpu);
io.mapOptional("flags", info.flags, 0);
}
};
When reading YAML, if the string found does not match any of the the strings
specified by enumCase() methods, an error is automatically generated.
When writing YAML, if the value being written does not match any of the values
specified by the enumCase() methods, a runtime assertion is triggered.
BitValue
--------
Another common data structure in C++ is a field where each bit has a unique
meaning. This is often used in a "flags" field. YAML I/O has support for
converting such fields to a flow sequence. For instance suppose you
had the following bit flags defined:
.. code-block:: c++
enum {
flagsPointy = 1
flagsHollow = 2
flagsFlat = 4
flagsRound = 8
};
LLVM_YAML_UNIQUE_TYPE(MyFlags, uint32_t)
To support reading and writing of MyFlags, you specialize ScalarBitSetTraits<>
on MyFlags and provide the bit values and their names.
.. code-block:: c++
using llvm::yaml::ScalarBitSetTraits;
using llvm::yaml::MapppingTraits;
using llvm::yaml::IO;
template <>
struct ScalarBitSetTraits<MyFlags> {
static void bitset(IO &io, MyFlags &value) {
io.bitSetCase(value, "hollow", flagHollow);
io.bitSetCase(value, "flat", flagFlat);
io.bitSetCase(value, "round", flagRound);
io.bitSetCase(value, "pointy", flagPointy);
}
};
struct Info {
StringRef name;
MyFlags flags;
};
template <>
struct MapppingTraits<Info> {
static void mapping(IO &io, Info& info) {
io.mapRequired("name", info.name);
io.mapRequired("flags", info.flags);
}
};
With the above, YAML I/O (when writing) will test mask each value in the
bitset trait against the flags field, and each that matches will
cause the corresponding string to be added to the flow sequence. The opposite
is done when reading and any unknown string values will result in a error. With
the above schema, a same valid YAML document is:
.. code-block:: yaml
name: Tom
flags: [ pointy, flat ]
Custom Scalar
-------------
Sometimes for readability a scalar needs to be formatted in a custom way. For
instance your internal data structure may use a integer for time (seconds since
some epoch), but in YAML it would be much nicer to express that integer in
some time format (e.g. 4-May-2012 10:30pm). YAML I/O has a way to support
custom formatting and parsing of scalar types by specializing ScalarTraits<> on
your data type. When writing, YAML I/O will provide the native type and
your specialization must create a temporary llvm::StringRef. When reading,
YAML I/O will provide a llvm::StringRef of scalar and your specialization
must convert that to your native data type. An outline of a custom scalar type
looks like:
.. code-block:: c++
using llvm::yaml::ScalarTraits;
using llvm::yaml::IO;
template <>
struct ScalarTraits<MyCustomType> {
static void output(const T &value, llvm::raw_ostream &out) {
out << value; // do custom formatting here
}
static StringRef input(StringRef scalar, T &value) {
// do custom parsing here. Return the empty string on success,
// or an error message on failure.
return StringRef();
}
};
Mappings
========
To be translated to or from a YAML mapping for your type T you must specialize
llvm::yaml::MapppingTraits on T and implement the "void mapping(IO &io, T&)"
method. If your native data structures use pointers to a class everywhere,
you can specialize on the class pointer. Examples:
.. code-block:: c++
using llvm::yaml::MapppingTraits;
using llvm::yaml::IO;
// Example of struct Foo which is used by value
template <>
struct MapppingTraits<Foo> {
static void mapping(IO &io, Foo &foo) {
io.mapOptional("size", foo.size);
...
}
};
// Example of struct Bar which is natively always a pointer
template <>
struct MapppingTraits<Bar*> {
static void mapping(IO &io, Bar *&bar) {
io.mapOptional("size", bar->size);
...
}
};
No Normalization
----------------
The mapping() method is responsible, if needed, for normalizing and
denormalizing. In a simple case where the native data structure requires no
normalization, the mapping method just uses mapOptional() or mapRequired() to
bind the struct's fields to YAML key names. For example:
.. code-block:: c++
using llvm::yaml::MapppingTraits;
using llvm::yaml::IO;
template <>
struct MapppingTraits<Person> {
static void mapping(IO &io, Person &info) {
io.mapRequired("name", info.name);
io.mapOptional("hat-size", info.hatSize);
}
};
Normalization
----------------
When [de]normalization is required, the mapping() method needs a way to access
normalized values as fields. To help with this, there is
a template MappingNormalization<> which you can then use to automatically
do the normalization and denormalization. The template is used to create
a local variable in your mapping() method which contains the normalized keys.
Suppose you have native data type
Polar which specifies a position in polar coordinates (distance, angle):
.. code-block:: c++
struct Polar {
float distance;
float angle;
};
but you've decided the normalized YAML for should be in x,y coordinates. That
is, you want the yaml to look like:
.. code-block:: yaml
x: 10.3
y: -4.7
You can support this by defining a MapppingTraits that normalizes the polar
coordinates to x,y coordinates when writing YAML and denormalizes x,y
coordindates into polar when reading YAML.
.. code-block:: c++
using llvm::yaml::MapppingTraits;
using llvm::yaml::IO;
template <>
struct MapppingTraits<Polar> {
class NormalizedPolar {
public:
NormalizedPolar(IO &io)
: x(0.0), y(0.0) {
}
NormalizedPolar(IO &, Polar &polar)
: x(polar.distance * cos(polar.angle)),
y(polar.distance * sin(polar.angle)) {
}
Polar denormalize(IO &) {
return Polar(sqrt(x*x+y*y, arctan(x,y));
}
float x;
float y;
};
static void mapping(IO &io, Polar &polar) {
MappingNormalization<NormalizedPolar, Polar> keys(io, polar);
io.mapRequired("x", keys->x);
io.mapRequired("y", keys->y);
}
};
When writing YAML, the local variable "keys" will be a stack allocated
instance of NormalizedPolar, constructed from the suppled polar object which
initializes it x and y fields. The mapRequired() methods then write out the x
and y values as key/value pairs.
When reading YAML, the local variable "keys" will be a stack allocated instance
of NormalizedPolar, constructed by the empty constructor. The mapRequired
methods will find the matching key in the YAML document and fill in the x and y
fields of the NormalizedPolar object keys. At the end of the mapping() method
when the local keys variable goes out of scope, the denormalize() method will
automatically be called to convert the read values back to polar coordinates,
and then assigned back to the second parameter to mapping().
In some cases, the normalized class may be a subclass of the native type and
could be returned by the denormalize() method, except that the temporary
normalized instance is stack allocated. In these cases, the utility template
MappingNormalizationHeap<> can be used instead. It just like
MappingNormalization<> except that it heap allocates the normalized object
when reading YAML. It never destroyes the normalized object. The denormalize()
method can this return "this".
Default values
--------------
Within a mapping() method, calls to io.mapRequired() mean that that key is
required to exist when parsing YAML documents, otherwise YAML I/O will issue an
error.
On the other hand, keys registered with io.mapOptional() are allowed to not
exist in the YAML document being read. So what value is put in the field
for those optional keys?
There are two steps to how those optional fields are filled in. First, the
second parameter to the mapping() method is a reference to a native class. That
native class must have a default constructor. Whatever value the default
constructor initially sets for an optional field will be that field's value.
Second, the mapOptional() method has an optional third parameter. If provided
it is the value that mapOptional() should set that field to if the YAML document
does not have that key.
There is one important difference between those two ways (default constructor
and third parameter to mapOptional). When YAML I/O generates a YAML document,
if the mapOptional() third parameter is used, if the actual value being written
is the same as (using ==) the default value, then that key/value is not written.
Order of Keys
--------------
When writing out a YAML document, the keys are written in the order that the
calls to mapRequired()/mapOptional() are made in the mapping() method. This
gives you a chance to write the fields in an order that a human reader of
the YAML document would find natural. This may be different that the order
of the fields in the native class.
When reading in a YAML document, the keys in the document can be in any order,
but they are processed in the order that the calls to mapRequired()/mapOptional()
are made in the mapping() method. That enables some interesting
functionality. For instance, if the first field bound is the cpu and the second
field bound is flags, and the flags are cpu specific, you can programmatically
switch how the flags are converted to and from YAML based on the cpu.
This works for both reading and writing. For example:
.. code-block:: c++
using llvm::yaml::MapppingTraits;
using llvm::yaml::IO;
struct Info {
CPUs cpu;
uint32_t flags;
};
template <>
struct MapppingTraits<Info> {
static void mapping(IO &io, Info &info) {
io.mapRequired("cpu", info.cpu);
// flags must come after cpu for this to work when reading yaml
if ( info.cpu == cpu_x86_64 )
io.mapRequired("flags", *(My86_64Flags*)info.flags);
else
io.mapRequired("flags", *(My86Flags*)info.flags);
}
};
Sequence
========
To be translated to or from a YAML sequence for your type T you must specialize
llvm::yaml::SequenceTraits on T and implement two methods:
“size_t size(IO &io, T&)” and “T::value_type& element(IO &io, T&, size_t indx)”.
For example:
.. code-block:: c++
template <>
struct SequenceTraits<MySeq> {
static size_t size(IO &io, MySeq &list) { ... }
static MySeqEl element(IO &io, MySeq &list, size_t index) { ... }
};
The size() method returns how many elements are currently in your sequence.
The element() method returns a reference to the i'th element in the sequence.
When parsing YAML, the element() method may be called with an index one bigger
than the current size. Your element() method should allocate space for one
more element (using default constructor if element is a C++ object) and returns
a reference to that new allocated space.
Flow Sequence
-------------
A YAML "flow sequence" is a sequence that when written to YAML it uses the
inline notation (e.g [ foo, bar ] ). To specify that a sequence type should
be written in YAML as a flow sequence, your SequenceTraits specialization should
add "static const bool flow = true;". For instance:
.. code-block:: c++
template <>
struct SequenceTraits<MyList> {
static size_t size(IO &io, MyList &list) { ... }
static MyListEl element(IO &io, MyList &list, size_t index) { ... }
// The existence of this member causes YAML I/O to use a flow sequence
static const bool flow = true;
};
With the above, if you used MyList as the data type in your native data
strucutures, then then when converted to YAML, a flow sequence of integers
will be used (e.g. [ 10, -3, 4 ]).
Utility Macros
--------------
Since a common source of sequences is std::vector<>, YAML I/O provids macros:
LLVM_YAML_IS_SEQUENCE_VECTOR() and LLVM_YAML_IS_FLOW_SEQUENCE_VECTOR() which
can be used to easily specify SequenceTraits<> on a std::vector type. YAML
I/O does not partial specialize SequenceTraits on std::vector<> because that
would force all vectors to be sequences. An example use of the macros:
.. code-block:: c++
std::vector<MyType1>;
std::vector<MyType2>;
LLVM_YAML_IS_SEQUENCE_VECTOR(MyType1)
LLVM_YAML_IS_FLOW_SEQUENCE_VECTOR(MyType2)
Document List
=============
YAML allows you to define multiple "documents" in a single YAML file. Each
new document starts with a left aligned "---" token. The end of all documents
is denoted with a left aligned "..." token. Many users of YAML will never
have need for multiple documents. The top level node in their YAML schema
will be a mapping or sequence. For those cases, the following is not needed.
But for cases where you do want multiple documents, you can specify a
trait for you document list type. The trait has the same methods as
SequenceTraits but is named DocumentListTraits. For example:
.. code-block:: c++
template <>
struct DocumentListTraits<MyDocList> {
static size_t size(IO &io, MyDocList &list) { ... }
static MyDocType element(IO &io, MyDocList &list, size_t index) { ... }
};
User Context Data
=================
When an llvm::yaml::Input or llvm::yaml::Output object is created their
constructors take an optional "context" parameter. This is a pointer to
whatever state information you might need.
For instance, in a previous example we showed how the conversion type for a
flags field could be determined at runtime based on the value of another field
in the mapping. But what if an inner mapping needs to know some field value
of an outer mapping? That is where the "context" parameter comes in. You
can set values in the context in the outer map's mapping() method and
retrieve those values in the inner map's mapping() method.
The context value is just a void*. All your traits which use the context
and operate on your native data types, need to agree what the context value
actually is. It could be a pointer to an object or struct which your various
traits use to shared context sensitive information.
Output
======
The llvm::yaml::Output class is used to generate a YAML document from your
in-memory data structures, using traits defined on your data types.
To instantiate an Output object you need an llvm::raw_ostream, and optionally
a context pointer:
.. code-block:: c++
class Output : public IO {
public:
Output(llvm::raw_ostream &, void *context=NULL);
Once you have an Output object, you can use the C++ stream operator on it
to write your native data as YAML. One thing to recall is that a YAML file
can contain multiple "documents". If the top level data structure you are
streaming as YAML is a mapping, scalar, or sequence, then Output assumes you
are generating one document and wraps the mapping output
with "``---``" and trailing "``...``".
.. code-block:: c++
using llvm::yaml::Output;
void dumpMyMapDoc(const MyMapType &info) {
Output yout(llvm::outs());
yout << info;
}
The above could produce output like:
.. code-block:: yaml
---
name: Tom
hat-size: 7
...
On the other hand, if the top level data structure you are streaming as YAML
has a DocumentListTraits specialization, then Output walks through each element
of your DocumentList and generates a "---" before the start of each element
and ends with a "...".
.. code-block:: c++
using llvm::yaml::Output;
void dumpMyMapDoc(const MyDocListType &docList) {
Output yout(llvm::outs());
yout << docList;
}
The above could produce output like:
.. code-block:: yaml
---
name: Tom
hat-size: 7
---
name: Tom
shoe-size: 11
...
Input
=====
The llvm::yaml::Input class is used to parse YAML document(s) into your native
data structures. To instantiate an Input
object you need a StringRef to the entire YAML file, and optionally a context
pointer:
.. code-block:: c++
class Input : public IO {
public:
Input(StringRef inputContent, void *context=NULL);
Once you have an Input object, you can use the C++ stream operator to read
the document(s). If you expect there might be multiple YAML documents in
one file, you'll need to specialize DocumentListTraits on a list of your
document type and stream in that document list type. Otherwise you can
just stream in the document type. Also, you can check if there was
any syntax errors in the YAML be calling the error() method on the Input
object. For example:
.. code-block:: c++
// Reading a single document
using llvm::yaml::Input;
Input yin(mb.getBuffer());
// Parse the YAML file
MyDocType theDoc;
yin >> theDoc;
// Check for error
if ( yin.error() )
return;
.. code-block:: c++
// Reading multiple documents in one file
using llvm::yaml::Input;
LLVM_YAML_IS_DOCUMENT_LIST_VECTOR(std::vector<MyDocType>)
Input yin(mb.getBuffer());
// Parse the YAML file
std::vector<MyDocType> theDocList;
yin >> theDocList;
// Check for error
if ( yin.error() )
return;

5
llvm/docs/userguides.rst
View File

@ -24,6 +24,7 @@ User Guides
tutorial/index
ReleaseNotes
Passes
YamlIO
* :ref:`getting_started`
@ -100,6 +101,10 @@ User Guides
Instructions for adding new builder to LLVM buildbot master.
* :ref:`yamlio`
A reference guide for using LLVM's YAML I/O library.
* **IRC** -- You can probably find help on the unofficial LLVM IRC.
We often are on irc.oftc.net in the #llvm channel. If you are using the

1114
llvm/include/llvm/Support/YAMLTraits.h Normal file
View File

File diff suppressed because it is too large Load Diff

1
llvm/lib/Support/CMakeLists.txt
View File

@ -50,6 +50,7 @@ add_llvm_library(LLVMSupport
Triple.cpp
Twine.cpp
YAMLParser.cpp
YAMLTraits.cpp
raw_os_ostream.cpp
raw_ostream.cpp
regcomp.c

881
llvm/lib/Support/YAMLTraits.cpp Normal file
View File

@ -0,0 +1,881 @@
//===- lib/Support/YAMLTraits.cpp -----------------------------------------===//
//
// The LLVM Linker
//
// This file is distributed under the University of Illinois Open Source
// License. See LICENSE.TXT for details.
//
//===----------------------------------------------------------------------===//
#define BUILDING_YAMLIO
#include "llvm/Support/YAMLTraits.h"
#include "llvm/ADT/Twine.h"
#include "llvm/Support/Casting.h"
#include "llvm/Support/ErrorHandling.h"
#include "llvm/Support/format.h"
#include "llvm/Support/raw_ostream.h"
#include "llvm/Support/YAMLParser.h"
#include <cstring>
namespace llvm {
namespace yaml {
//===----------------------------------------------------------------------===//
// IO
//===----------------------------------------------------------------------===//
IO::IO(void *Context) : Ctxt(Context) {
}
IO::~IO() {
}
void *IO::getContext() {
return Ctxt;
}
void IO::setContext(void *Context) {
Ctxt = Context;
}
//===----------------------------------------------------------------------===//
// Input
//===----------------------------------------------------------------------===//
Input::Input(StringRef InputContent, void *Ctxt)
: IO(Ctxt), CurrentNode(NULL) {
Strm = new Stream(InputContent, SrcMgr);
DocIterator = Strm->begin();
}
llvm::error_code Input::error() {
return EC;
}
void Input::setDiagHandler(llvm::SourceMgr::DiagHandlerTy Handler, void *Ctxt) {
SrcMgr.setDiagHandler(Handler, Ctxt);
}
bool Input::outputting() {
return false;
}
bool Input::setCurrentDocument() {
if ( DocIterator != Strm->end() ) {
Node *N = DocIterator->getRoot();
if (llvm::isa<NullNode>(N)) {
// Empty files are allowed and ignored
++DocIterator;
return setCurrentDocument();
}
CurrentNode = this->createHNodes(N);
return true;
}
return false;
}
void Input::nextDocument() {
++DocIterator;
}
void Input::beginMapping() {
if ( EC )
return;
MapHNode *MN = llvm::dyn_cast<MapHNode>(CurrentNode);
if ( MN ) {
MN->ValidKeys.clear();
}
}
bool Input::preflightKey(const char *Key, bool Required, bool,
bool &UseDefault, void *&SaveInfo) {
UseDefault = false;
if ( EC )
return false;
MapHNode *MN = llvm::dyn_cast<MapHNode>(CurrentNode);
if ( !MN ) {
setError(CurrentNode, "not a mapping");
return false;
}
MN->ValidKeys.push_back(Key);
HNode *Value = MN->Mapping[Key];
if ( !Value ) {
if ( Required )
setError(CurrentNode, Twine("missing required key '") + Key + "'");
else
UseDefault = true;
return false;
}
SaveInfo = CurrentNode;
CurrentNode = Value;
return true;
}
void Input::postflightKey(void *saveInfo) {
CurrentNode = reinterpret_cast<HNode*>(saveInfo);
}
void Input::endMapping() {
if ( EC )
return;
MapHNode *MN = llvm::dyn_cast<MapHNode>(CurrentNode);
if ( !MN )
return;
for (MapHNode::NameToNode::iterator i=MN->Mapping.begin(),
End=MN->Mapping.end(); i != End; ++i) {
if ( ! MN->isValidKey(i->first) ) {
setError(i->second, Twine("unknown key '") + i->first + "'" );
break;
}
}
}
unsigned Input::beginSequence() {
if ( SequenceHNode *SQ = llvm::dyn_cast<SequenceHNode>(CurrentNode) ) {
return SQ->Entries.size();
}
return 0;
}
void Input::endSequence() {
}
bool Input::preflightElement(unsigned Index, void *&SaveInfo) {
if ( EC )
return false;
if ( SequenceHNode *SQ = llvm::dyn_cast<SequenceHNode>(CurrentNode) ) {
SaveInfo = CurrentNode;
CurrentNode = SQ->Entries[Index];
return true;
}
return false;
}
void Input::postflightElement(void *SaveInfo) {
CurrentNode = reinterpret_cast<HNode*>(SaveInfo);
}
unsigned Input::beginFlowSequence() {
if ( SequenceHNode *SQ = llvm::dyn_cast<SequenceHNode>(CurrentNode) ) {
return SQ->Entries.size();
}
return 0;
}
bool Input::preflightFlowElement(unsigned index, void *&SaveInfo) {
if ( EC )
return false;
if ( SequenceHNode *SQ = llvm::dyn_cast<SequenceHNode>(CurrentNode) ) {
SaveInfo = CurrentNode;
CurrentNode = SQ->Entries[index];
return true;
}
return false;
}
void Input::postflightFlowElement(void *SaveInfo) {
CurrentNode = reinterpret_cast<HNode*>(SaveInfo);
}
void Input::endFlowSequence() {
}
void Input::beginEnumScalar() {
ScalarMatchFound = false;
}
bool Input::matchEnumScalar(const char *Str, bool) {
if ( ScalarMatchFound )
return false;
if ( ScalarHNode *SN = llvm::dyn_cast<ScalarHNode>(CurrentNode) ) {
if ( SN->value().equals(Str) ) {
ScalarMatchFound = true;
return true;
}
}
return false;
}
void Input::endEnumScalar() {
if ( !ScalarMatchFound ) {
setError(CurrentNode, "unknown enumerated scalar");
}
}
bool Input::beginBitSetScalar(bool &DoClear) {
BitValuesUsed.clear();
if ( SequenceHNode *SQ = llvm::dyn_cast<SequenceHNode>(CurrentNode) ) {
BitValuesUsed.insert(BitValuesUsed.begin(), SQ->Entries.size(), false);
}
else {
setError(CurrentNode, "expected sequence of bit values");
}
DoClear = true;
return true;
}
bool Input::bitSetMatch(const char *Str, bool) {
if ( EC )
return false;
if ( SequenceHNode *SQ = llvm::dyn_cast<SequenceHNode>(CurrentNode) ) {
unsigned Index = 0;
for (std::vector<HNode*>::iterator i=SQ->Entries.begin(),
End=SQ->Entries.end(); i != End; ++i) {
if ( ScalarHNode *SN = llvm::dyn_cast<ScalarHNode>(*i) ) {
if ( SN->value().equals(Str) ) {
BitValuesUsed[Index] = true;
return true;
}
}
else {
setError(CurrentNode, "unexpected scalar in sequence of bit values");
}
++Index;
}
}
else {
setError(CurrentNode, "expected sequence of bit values");
}
return false;
}
void Input::endBitSetScalar() {
if ( EC )
return;
if ( SequenceHNode *SQ = llvm::dyn_cast<SequenceHNode>(CurrentNode) ) {
assert(BitValuesUsed.size() == SQ->Entries.size());
for ( unsigned i=0; i < SQ->Entries.size(); ++i ) {
if ( !BitValuesUsed[i] ) {
setError(SQ->Entries[i], "unknown bit value");
return;
}
}
}
}
void Input::scalarString(StringRef &S) {
if ( ScalarHNode *SN = llvm::dyn_cast<ScalarHNode>(CurrentNode) ) {
S = SN->value();
}
else {
setError(CurrentNode, "unexpected scalar");
}
}
void Input::setError(HNode *hnode, const Twine &message) {
this->setError(hnode->_node, message);
}
void Input::setError(Node *node, const Twine &message) {
Strm->printError(node, message);
EC = make_error_code(errc::invalid_argument);
}
Input::HNode *Input::createHNodes(Node *N) {
llvm::SmallString<128> StringStorage;
if ( ScalarNode *SN = llvm::dyn_cast<ScalarNode>(N) ) {
StringRef KeyStr = SN->getValue(StringStorage);
if ( !StringStorage.empty() ) {
// Copy string to permanent storage
unsigned Len = StringStorage.size();
char* Buf = Allocator.Allocate<char>(Len);
memcpy(Buf, &StringStorage[0], Len);
KeyStr = StringRef(Buf, Len);
}
return new (Allocator) ScalarHNode(N, KeyStr);
}
else if ( SequenceNode *SQ = llvm::dyn_cast<SequenceNode>(N) ) {
SequenceHNode *SQHNode = new (Allocator) SequenceHNode(N);
for (SequenceNode::iterator i=SQ->begin(),End=SQ->end(); i != End; ++i ) {
HNode *Entry = this->createHNodes(i);
if ( EC )
break;
SQHNode->Entries.push_back(Entry);
}
return SQHNode;
}
else if ( MappingNode *Map = llvm::dyn_cast<MappingNode>(N) ) {
MapHNode *mapHNode = new (Allocator) MapHNode(N);
for (MappingNode::iterator i=Map->begin(), End=Map->end(); i != End; ++i ) {
ScalarNode *KeyScalar = llvm::dyn_cast<ScalarNode>(i->getKey());
StringStorage.clear();
llvm::StringRef KeyStr = KeyScalar->getValue(StringStorage);
if ( !StringStorage.empty() ) {
// Copy string to permanent storage
unsigned Len = StringStorage.size();
char* Buf = Allocator.Allocate<char>(Len);
memcpy(Buf, &StringStorage[0], Len);
KeyStr = StringRef(Buf, Len);
}
HNode *ValueHNode = this->createHNodes(i->getValue());
if ( EC )
break;
mapHNode->Mapping[KeyStr] = ValueHNode;
}
return mapHNode;
}
else if ( llvm::isa<NullNode>(N) ) {
return new (Allocator) EmptyHNode(N);
}
else {
setError(N, "unknown node kind");
return NULL;
}
}
bool Input::MapHNode::isValidKey(StringRef Key) {
for (SmallVector<const char*, 6>::iterator i=ValidKeys.begin(),
End=ValidKeys.end(); i != End; ++i) {
if ( Key.equals(*i) )
return true;
}
return false;
}
void Input::setError(const Twine &Message) {
this->setError(CurrentNode, Message);
}
//===----------------------------------------------------------------------===//
// Output
//===----------------------------------------------------------------------===//
Output::Output(llvm::raw_ostream &yout, void *context)
: IO(context), Out(yout), Column(0), ColumnAtFlowStart(0),
NeedBitValueComma(false), NeedFlowSequenceComma(false),
EnumerationMatchFound(false), NeedsNewLine(false) {
}
Output::~Output() {
}
bool Output::outputting() {
return true;
}
void Output::beginMapping() {
StateStack.push_back(inMapFirstKey);
NeedsNewLine = true;
}
void Output::endMapping() {
StateStack.pop_back();
}
bool Output::preflightKey(const char *Key, bool Required, bool SameAsDefault,
bool &UseDefault, void *&) {
UseDefault = false;
if ( Required || !SameAsDefault ) {
this->newLineCheck();
this->paddedKey(Key);
return true;
}
return false;
}
void Output::postflightKey(void*) {
if ( StateStack.back() == inMapFirstKey ) {
StateStack.pop_back();
StateStack.push_back(inMapOtherKey);
}
}
void Output::beginDocuments() {
this->outputUpToEndOfLine("---");
}
bool Output::preflightDocument(unsigned index) {
if ( index > 0 )
this->outputUpToEndOfLine("\n---");
return true;
}
void Output::postflightDocument() {
}
void Output::endDocuments() {
output("\n...\n");
}
unsigned Output::beginSequence() {
StateStack.push_back(inSeq);
NeedsNewLine = true;
return 0;
}
void Output::endSequence() {
StateStack.pop_back();
}
bool Output::preflightElement(unsigned , void *&) {
return true;
}
void Output::postflightElement(void*) {
}
unsigned Output::beginFlowSequence() {
this->newLineCheck();
StateStack.push_back(inFlowSeq);
ColumnAtFlowStart = Column;
output("[ ");
NeedFlowSequenceComma = false;
return 0;
}
void Output::endFlowSequence() {
StateStack.pop_back();
this->outputUpToEndOfLine(" ]");
}
bool Output::preflightFlowElement(unsigned , void *&) {
if ( NeedFlowSequenceComma )
output(", ");
if ( Column > 70 ) {
output("\n");
for(int i=0; i < ColumnAtFlowStart; ++i)
output(" ");
Column = ColumnAtFlowStart;
output(" ");
}
return true;
}
void Output::postflightFlowElement(void*) {
NeedFlowSequenceComma = true;
}
void Output::beginEnumScalar() {
EnumerationMatchFound = false;
}
bool Output::matchEnumScalar(const char *Str, bool Match) {
if ( Match && !EnumerationMatchFound ) {
this->newLineCheck();
this->outputUpToEndOfLine(Str);
EnumerationMatchFound = true;
}
return false;
}
void Output::endEnumScalar() {
if ( !EnumerationMatchFound )
llvm_unreachable("bad runtime enum value");
}
bool Output::beginBitSetScalar(bool &DoClear) {
this->newLineCheck();
output("[ ");
NeedBitValueComma = false;
DoClear = false;
return true;
}
bool Output::bitSetMatch(const char *Str, bool Matches) {
if ( Matches ) {
if ( NeedBitValueComma )
output(", ");
this->output(Str);
NeedBitValueComma = true;
}
return false;
}
void Output::endBitSetScalar() {
this->outputUpToEndOfLine(" ]");
}
void Output::scalarString(StringRef &S) {
this->newLineCheck();
if (S.find('\n') == StringRef::npos) {
// No embedded new-line chars, just print string.
this->outputUpToEndOfLine(S);
return;
}
unsigned i = 0;
unsigned j = 0;
unsigned End = S.size();
output("'"); // Starting single quote.
const char *Base = S.data();
while (j < End) {
// Escape a single quote by doubling it.
if (S[j] == '\'') {
output(StringRef(&Base[i], j - i + 1));
output("'");
i = j + 1;
}
++j;
}
output(StringRef(&Base[i], j - i));
this->outputUpToEndOfLine("'"); // Ending single quote.
}
void Output::setError(const Twine &message) {
}
void Output::output(StringRef s) {
Column += s.size();
Out << s;
}
void Output::outputUpToEndOfLine(StringRef s) {
this->output(s);
if ( StateStack.back() != inFlowSeq )
NeedsNewLine = true;
}
void Output::outputNewLine() {
Out << "\n";
Column = 0;
}
// if seq at top, indent as if map, then add "- "
// if seq in middle, use "- " if firstKey, else use " "
//
void Output::newLineCheck() {
if ( ! NeedsNewLine )
return;
NeedsNewLine = false;
this->outputNewLine();
assert(StateStack.size() > 0);
unsigned Indent = StateStack.size() - 1;
bool OutputDash = false;
if ( StateStack.back() == inSeq ) {
OutputDash = true;
}
else if ( (StateStack.size() > 1)
&& (StateStack.back() == inMapFirstKey)
&& (StateStack[StateStack.size()-2] == inSeq) ) {
--Indent;
OutputDash = true;
}
for (unsigned i=0; i < Indent; ++i) {
output(" ");
}
if ( OutputDash ) {
output("- ");
}
}
void Output::paddedKey(StringRef key) {
output(key);
output(":");
const char *spaces = " ";
if ( key.size() < strlen(spaces) )
output(&spaces[key.size()]);
else
output(" ");
}
//===----------------------------------------------------------------------===//
// traits for built-in types
//===----------------------------------------------------------------------===//
template<>
struct ScalarTraits<bool> {
static void output(const bool &Val, void*, llvm::raw_ostream &Out) {
Out << ( Val ? "true" : "false");
}
static llvm::StringRef input(llvm::StringRef Scalar, void*, bool &Val) {
if ( Scalar.equals("true") ) {
Val = true;
return StringRef();
}
else if ( Scalar.equals("false") ) {
Val = false;
return StringRef();
}
return "invalid boolean";
}
};
template<>
struct ScalarTraits<StringRef> {
static void output(const StringRef &Val, void*, llvm::raw_ostream &Out) {
Out << Val;
}
static llvm::StringRef input(llvm::StringRef Scalar, void*, StringRef &Val){
Val = Scalar;
return StringRef();
}
};
template<>
struct ScalarTraits<uint8_t> {
static void output(const uint8_t &Val, void*, llvm::raw_ostream &Out) {
// use temp uin32_t because ostream thinks uint8_t is a character
uint32_t Num = Val;
Out << Num;
}
static llvm::StringRef input(llvm::StringRef Scalar, void*, uint8_t &Val) {
uint64_t n;
if ( getAsUnsignedInteger(Scalar, 0, n) )
return "invalid number";
if ( n > 0xFF )
return "out of range number";
Val = n;
return StringRef();
}
};
template<>
struct ScalarTraits<uint16_t> {
static void output(const uint16_t &Val, void*, llvm::raw_ostream &Out) {
Out << Val;
}
static llvm::StringRef input(llvm::StringRef Scalar, void*, uint16_t &Val) {
uint64_t n;
if ( getAsUnsignedInteger(Scalar, 0, n) )
return "invalid number";
if ( n > 0xFFFF )
return "out of range number";
Val = n;
return StringRef();
}
};
template<>
struct ScalarTraits<uint32_t> {
static void output(const uint32_t &Val, void*, llvm::raw_ostream &Out) {
Out << Val;
}
static llvm::StringRef input(llvm::StringRef Scalar, void*, uint32_t &Val) {
uint64_t n;
if ( getAsUnsignedInteger(Scalar, 0, n) )
return "invalid number";
if ( n > 0xFFFFFFFFUL )
return "out of range number";
Val = n;
return StringRef();
}
};
template<>
struct ScalarTraits<uint64_t> {
static void output(const uint64_t &Val, void*, llvm::raw_ostream &Out) {
Out << Val;
}
static llvm::StringRef input(llvm::StringRef Scalar, void*, uint64_t &Val) {
if ( getAsUnsignedInteger(Scalar, 0, Val) )
return "invalid number";
return StringRef();
}
};
template<>
struct ScalarTraits<int8_t> {
static void output(const int8_t &Val, void*, llvm::raw_ostream &Out) {
// use temp in32_t because ostream thinks int8_t is a character
int32_t Num = Val;
Out << Num;
}
static llvm::StringRef input(llvm::StringRef Scalar, void*, int8_t &Val) {
int64_t n;
if ( getAsSignedInteger(Scalar, 0, n) )
return "invalid number";
if ( (n > 127) || (n < -128) )
return "out of range number";
Val = n;
return StringRef();
}
};
template<>
struct ScalarTraits<int16_t> {
static void output(const int16_t &Val, void*, llvm::raw_ostream &Out) {
Out << Val;
}
static llvm::StringRef input(llvm::StringRef Scalar, void*, int16_t &Val) {
int64_t n;
if ( getAsSignedInteger(Scalar, 0, n) )
return "invalid number";
if ( (n > INT16_MAX) || (n < INT16_MIN) )
return "out of range number";
Val = n;
return StringRef();
}
};
template<>
struct ScalarTraits<int32_t> {
static void output(const int32_t &Val, void*, llvm::raw_ostream &Out) {
Out << Val;
}
static llvm::StringRef input(llvm::StringRef Scalar, void*, int32_t &Val) {
int64_t n;
if ( getAsSignedInteger(Scalar, 0, n) )
return "invalid number";
if ( (n > INT32_MAX) || (n < INT32_MIN) )
return "out of range number";
Val = n;
return StringRef();
}
};
template<>
struct ScalarTraits<int64_t> {
static void output(const int64_t &Val, void*, llvm::raw_ostream &Out) {
Out << Val;
}
static llvm::StringRef input(llvm::StringRef Scalar, void*, int64_t &Val) {
if ( getAsSignedInteger(Scalar, 0, Val) )
return "invalid number";
return StringRef();
}
};
template<>
struct ScalarTraits<double> {
static void output(const double &Val, void*, llvm::raw_ostream &Out) {
Out << format("%g", Val);
}
static llvm::StringRef input(llvm::StringRef Scalar, void*, double &Val) {
SmallString<32> buff(Scalar.begin(), Scalar.end());
char *end;
Val = strtod(buff.c_str(), &end);
if ( *end != '\0' )
return "invalid floating point number";
return StringRef();
}
};
template<>
struct ScalarTraits<float> {
static void output(const float &Val, void*, llvm::raw_ostream &Out) {
Out << format("%g", Val);
}
static llvm::StringRef input(llvm::StringRef Scalar, void*, float &Val) {
SmallString<32> buff(Scalar.begin(), Scalar.end());
char *end;
Val = strtod(buff.c_str(), &end);
if ( *end != '\0' )
return "invalid floating point number";
return StringRef();
}
};
template<>
struct ScalarTraits<Hex8> {
static void output(const Hex8 &Val, void*, llvm::raw_ostream &Out) {
uint8_t Num = Val;
Out << format("0x%02X", Num);
}
static llvm::StringRef input(llvm::StringRef Scalar, void*, Hex8 &Val) {
uint64_t n;
if ( getAsUnsignedInteger(Scalar, 0, n) )
return "invalid hex8 number";
if ( n > 0xFF )
return "out of range hex8 number";
Val = n;
return StringRef();
}
};
template<>
struct ScalarTraits<Hex16> {
static void output(const Hex16 &Val, void*, llvm::raw_ostream &Out) {
uint16_t Num = Val;
Out << format("0x%04X", Num);
}
static llvm::StringRef input(llvm::StringRef Scalar, void*, Hex16 &Val) {
uint64_t n;
if ( getAsUnsignedInteger(Scalar, 0, n) )
return "invalid hex16 number";
if ( n > 0xFFFF )
return "out of range hex16 number";
Val = n;
return StringRef();
}
};
template<>
struct ScalarTraits<Hex32> {
static void output(const Hex32 &Val, void*, llvm::raw_ostream &Out) {
uint32_t Num = Val;
Out << format("0x%08X", Num);
}
static llvm::StringRef input(llvm::StringRef Scalar, void*, Hex32 &Val) {
uint64_t n;
if ( getAsUnsignedInteger(Scalar, 0, n) )
return "invalid hex32 number";
if ( n > 0xFFFFFFFFUL )
return "out of range hex32 number";
Val = n;
return StringRef();
}
};
template<>
struct ScalarTraits<Hex64> {
static void output(const Hex64 &Val, void*, llvm::raw_ostream &Out) {
uint64_t Num = Val;
Out << format("0x%016llX", Num);
}
static llvm::StringRef input(llvm::StringRef Scalar, void*, Hex64 &Val) {
uint64_t Num;
if ( getAsUnsignedInteger(Scalar, 0, Num) )
return "invalid hex64 number";
Val = Num;
return StringRef();
}
};
// We want all the ScalarTrait specialized on built-in types
// to be instantiated here.
template <typename T>
struct ForceUse {
ForceUse() : oproc(ScalarTraits<T>::output), iproc(ScalarTraits<T>::input) {}
void (*oproc)(const T &, void*, llvm::raw_ostream &);
llvm::StringRef (*iproc)(llvm::StringRef, void*, T &);
};
static ForceUse<bool> Dummy1;
static ForceUse<llvm::StringRef> Dummy2;
static ForceUse<uint8_t> Dummy3;
static ForceUse<uint16_t> Dummy4;
static ForceUse<uint32_t> Dummy5;
static ForceUse<uint64_t> Dummy6;
static ForceUse<int8_t> Dummy7;
static ForceUse<int16_t> Dummy8;
static ForceUse<int32_t> Dummy9;
static ForceUse<int64_t> Dummy10;
static ForceUse<float> Dummy11;
static ForceUse<double> Dummy12;
static ForceUse<Hex8> Dummy13;
static ForceUse<Hex16> Dummy14;
static ForceUse<Hex32> Dummy15;
static ForceUse<Hex64> Dummy16;
} // namespace yaml
} // namespace llvm

1
llvm/unittests/Support/CMakeLists.txt
View File

@ -24,6 +24,7 @@ add_llvm_unittest(SupportTests
SwapByteOrderTest.cpp
TimeValue.cpp
ValueHandleTest.cpp
YAMLIOTest.cpp
YAMLParserTest.cpp
formatted_raw_ostream_test.cpp
raw_ostream_test.cpp

1288
llvm/unittests/Support/YAMLIOTest.cpp Normal file
View File

File diff suppressed because it is too large Load Diff