mirror of
https://github.com/ThePhD/sol2.git
synced 2024-03-22 13:10:44 +08:00
121 lines
3.2 KiB
C++
121 lines
3.2 KiB
C++
// sol2
|
|
|
|
// The MIT License (MIT)
|
|
|
|
// Copyright (c) 2013-2018 Rapptz, ThePhD and contributors
|
|
|
|
// Permission is hereby granted, free of charge, to any person obtaining a copy of
|
|
// this software and associated documentation files (the "Software"), to deal in
|
|
// the Software without restriction, including without limitation the rights to
|
|
// use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of
|
|
// the Software, and to permit persons to whom the Software is furnished to do so,
|
|
// subject to the following conditions:
|
|
|
|
// The above copyright notice and this permission notice shall be included in all
|
|
// copies or substantial portions of the Software.
|
|
|
|
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS
|
|
// FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR
|
|
// COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER
|
|
// IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
|
|
// CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
|
|
|
|
#ifndef SOL_TABLE_ITERATOR_HPP
|
|
#define SOL_TABLE_ITERATOR_HPP
|
|
|
|
#include "object.hpp"
|
|
#include <iterator>
|
|
|
|
namespace sol {
|
|
|
|
template <typename reference_type>
|
|
class basic_table_iterator : public std::iterator<std::input_iterator_tag, std::pair<object, object>> {
|
|
public:
|
|
typedef object key_type;
|
|
typedef object mapped_type;
|
|
typedef std::pair<object, object> value_type;
|
|
typedef std::input_iterator_tag iterator_category;
|
|
typedef std::ptrdiff_t difference_type;
|
|
typedef value_type* pointer;
|
|
typedef value_type& reference;
|
|
typedef const value_type& const_reference;
|
|
|
|
private:
|
|
std::pair<object, object> kvp;
|
|
reference_type ref;
|
|
int tableidx = 0;
|
|
int keyidx = 0;
|
|
std::ptrdiff_t idx = 0;
|
|
|
|
public:
|
|
basic_table_iterator()
|
|
: keyidx(-1), idx(-1) {
|
|
}
|
|
|
|
basic_table_iterator(reference_type x)
|
|
: ref(std::move(x)) {
|
|
ref.push();
|
|
tableidx = lua_gettop(ref.lua_state());
|
|
stack::push(ref.lua_state(), lua_nil);
|
|
this->operator++();
|
|
if (idx == -1) {
|
|
return;
|
|
}
|
|
--idx;
|
|
}
|
|
|
|
basic_table_iterator& operator++() {
|
|
if (idx == -1)
|
|
return *this;
|
|
|
|
if (lua_next(ref.lua_state(), tableidx) == 0) {
|
|
idx = -1;
|
|
keyidx = -1;
|
|
return *this;
|
|
}
|
|
++idx;
|
|
kvp.first = object(ref.lua_state(), -2);
|
|
kvp.second = object(ref.lua_state(), -1);
|
|
lua_pop(ref.lua_state(), 1);
|
|
// leave key on the stack
|
|
keyidx = lua_gettop(ref.lua_state());
|
|
return *this;
|
|
}
|
|
|
|
basic_table_iterator operator++(int) {
|
|
auto saved = *this;
|
|
this->operator++();
|
|
return saved;
|
|
}
|
|
|
|
reference operator*() {
|
|
return kvp;
|
|
}
|
|
|
|
const_reference operator*() const {
|
|
return kvp;
|
|
}
|
|
|
|
bool operator==(const basic_table_iterator& right) const {
|
|
return idx == right.idx;
|
|
}
|
|
|
|
bool operator!=(const basic_table_iterator& right) const {
|
|
return idx != right.idx;
|
|
}
|
|
|
|
~basic_table_iterator() {
|
|
if (keyidx != -1) {
|
|
stack::remove(ref.lua_state(), keyidx, 1);
|
|
}
|
|
if (ref.valid()) {
|
|
stack::remove(ref.lua_state(), tableidx, 1);
|
|
}
|
|
}
|
|
};
|
|
|
|
} // namespace sol
|
|
|
|
#endif // SOL_TABLE_ITERATOR_HPP
|