peisongxiao/yaml-cpp
1
0
Fork 0
mirror of https://github.com/jbeder/yaml-cpp.git synced 2025-09-09 20:51:16 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fixed node iterator

Browse Source
This commit is contained in:
beder
2011-09-10 14:36:10 -05:00
parent c12a03473e
commit b3086ac260
4 changed files with 22 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
include/yaml-cpp/value/detail/iterator.h
View File

@@ -47,9 +47,9 @@ namespace YAML
value_type dereference() const {
const typename base_type::value_type& v = *this->base();
if(v.pNode)
return value_type(Value(*v.pNode, m_pMemory));
if(v.pKey && v.pValue)
return value_type(Value(*v.pKey, m_pMemory), Value(*v.pValue, m_pMemory));
return value_type(Value(*v, m_pMemory));
if(v.first && v.second)
return value_type(Value(*v.first, m_pMemory), Value(*v.second, m_pMemory));
return value_type();
}

14
include/yaml-cpp/value/detail/node_iterator.h
View File

@@ -21,13 +21,17 @@ namespace YAML
struct iterator_type { enum value { None, Sequence, Map }; };
template<typename V>
struct node_iterator_value {
node_iterator_value(): pNode(0), pKey(0), pValue(0) {}
explicit node_iterator_value(V& rhs): pNode(&rhs), pKey(0), pValue(0) {}
explicit node_iterator_value(V& key, V& value): pNode(0), pKey(&key), pValue(&value) {}
struct node_iterator_value: public std::pair<V*, V*> {
typedef std::pair<V*, V*> kv;
node_iterator_value(): kv(), pNode(0) {}
explicit node_iterator_value(V& rhs): kv(), pNode(&rhs) {}
explicit node_iterator_value(V& key, V& value): kv(&key, &value), pNode(0) {}
V& operator *() const { return *pNode; }
V& operator ->() const { return *pNode; }
V *pNode;
V *pKey, *pValue;
};
typedef std::vector<node *> node_seq;

10
src/value/valueevents.cpp
View File

@@ -1,4 +1,5 @@
#include "valueevents.h"
#include "yaml-cpp/value.h"
namespace YAML
{
@@ -7,7 +8,7 @@ namespace YAML
Visit(m_root);
}
void Visit(detail::node& node)
void ValueEvents::Visit(const detail::node& node)
{
int& refCount = m_refCount[node.ref()];
refCount++;
@@ -15,8 +16,13 @@ namespace YAML
return;
if(node.type() == ValueType::Sequence) {
for(detail::const_node_iterator it=node.begin();it!=node.end();++it)
Visit(**it);
} else if(node.type() == ValueType::Map) {
for(detail::const_node_iterator it=node.begin();it!=node.end();++it) {
Visit(*it->first);
Visit(*it->second);
}
}
}
}

4
src/value/valueevents.h
View File

@@ -7,7 +7,7 @@
#include "yaml-cpp/anchor.h"
#include "yaml-cpp/value/ptr.h"
#include <multiset>
#include <map>
#include <vector>
namespace YAML
@@ -20,7 +20,7 @@ namespace YAML
ValueEvents(const Value& value);
private:
void Visit(detail::node& node);
void Visit(const detail::node& node);
private:
detail::shared_memory_holder m_pMemory;