collectionstack.h 1.06 KB
Newer Older
1
2
3
#ifndef COLLECTIONSTACK_H_62B23520_7C8E_11DE_8A39_0800200C9A66
#define COLLECTIONSTACK_H_62B23520_7C8E_11DE_8A39_0800200C9A66

Jesse Beder's avatar
Jesse Beder committed
4
5
6
#if defined(_MSC_VER) ||                                            \
    (defined(__GNUC__) && (__GNUC__ == 3 && __GNUC_MINOR__ >= 4) || \
     (__GNUC__ >= 4))  // GCC supports "pragma once" correctly since 3.4
7
8
9
#pragma once
#endif

10
11
12
#include <stack>
#include <cassert>

Jesse Beder's avatar
Jesse Beder committed
13
14
namespace YAML {
struct CollectionType {
15
  enum value { NoCollection, BlockMap, BlockSeq, FlowMap, FlowSeq, CompactMap };
Jesse Beder's avatar
Jesse Beder committed
16
17
18
19
20
21
};

class CollectionStack {
 public:
  CollectionType::value GetCurCollectionType() const {
    if (collectionStack.empty())
22
      return CollectionType::NoCollection;
Jesse Beder's avatar
Jesse Beder committed
23
24
25
26
27
28
29
30
31
32
    return collectionStack.top();
  }

  void PushCollectionType(CollectionType::value type) {
    collectionStack.push(type);
  }
  void PopCollectionType(CollectionType::value type) {
    assert(type == GetCurCollectionType());
    collectionStack.pop();
  }
33

Jesse Beder's avatar
Jesse Beder committed
34
35
36
 private:
  std::stack<CollectionType::value> collectionStack;
};
37
38
}

Jesse Beder's avatar
Jesse Beder committed
39
#endif  // COLLECTIONSTACK_H_62B23520_7C8E_11DE_8A39_0800200C9A66