Zxc200611/pocketpy
1
0
Fork 0
mirror of https://github.com/pocketpy/pocketpy synced 2025-11-10 05:30:16 +00:00
Code Issues Packages Projects Releases Wiki Activity

WIP: added more functionalities

Browse Source
This commit is contained in:
S. Mahmudul Hasan 2023-10-14 19:14:51 -04:00
parent fd938a7c29
commit 6681fb21af
1 changed files with 97 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

105
src/collections.cpp
View File

@ -13,6 +13,91 @@ namespace pkpy
return vm->heap.gcnew<PyDeque>(cls_t, vm, iterable, maxlen);
});
vm->bind(type, "__add__(self, other) -> deque",
[](VM *vm, ArgsView args)
{
auto _lock = vm->heap.gc_scope_lock();
PyObject *newDequeObj = vm->heap.gcnew<PyDeque>(PyDeque::_type(vm), vm, vm->None, vm->None); // create the new deque
PyDeque &newDeque = _CAST(PyDeque &, newDequeObj);
PyDeque &self = _CAST(PyDeque &, args[0]);
PyDeque &other = _CAST(PyDeque &, args[1]);
for (auto it = self.dequeItems.begin(); it != self.dequeItems.end(); ++it)
{
newDeque.append(*it);
}
for (auto it = other.dequeItems.begin(); it != other.dequeItems.end(); ++it)
{
newDeque.append(*it);
}
return newDequeObj;
});
vm->bind(type, "__bool__(self) -> bool",
[](VM *vm, ArgsView args)
{
PyDeque &self = _CAST(PyDeque &, args[0]);
return VAR(!self.dequeItems.empty());
});
vm->bind(type, "__class__(self) -> deque()", // creates a new empty deque
[](VM *vm, ArgsView args)
{
return vm->heap.gcnew<PyDeque>(PyDeque::_type(vm), vm, vm->None, vm->None);
});
vm->bind(type, "__contains__(self, obj) -> bool",
[](VM *vm, ArgsView args)
{
PyDeque &self = _CAST(PyDeque &, args[0]);
PyObject *obj = args[1];
bool found = self.findIndex(vm, obj, -1, -1) != -1;
return VAR(found);
});
vm->bind(type, "__copy__(self)", // shallow copy
[](VM *vm, ArgsView args)
{
auto _lock = vm->heap.gc_scope_lock(); // locking the heap
PyDeque &self = _CAST(PyDeque &, args[0]);
PyObject *newDequeObj = vm->heap.gcnew<PyDeque>(PyDeque::_type(vm), vm, vm->None, vm->None); // create the empty deque
PyDeque &newDeque = _CAST(PyDeque &, newDequeObj);
for (auto it = self.dequeItems.begin(); it != self.dequeItems.end(); ++it)
{
newDeque.append(*it);
}
return newDequeObj;
});
vm->bind(type, "__delitem__(self, index) -> None",
[](VM *vm, ArgsView args)
{
// TODO
return vm->None;
});
vm->bind(type, "__eq__(self, other) -> bool",
[](VM *vm, ArgsView args)
{
//TODO
return vm->None;
});
vm->bind(type, "__getitem__(self, idx) -> PyObject*",
[](VM *vm, ArgsView args)
{
//TODO
return VM->None;
}
);
// TODO: __iadd__, __imul__, __reversed__, __str__, __mul__, __rmul__, __setitem__,
vm->bind(type, "__len__(self) -> int",
[](VM *vm, ArgsView args)
{
@ -35,6 +120,7 @@ namespace pkpy
return vm->heap.gcnew<PyDequeIter>(PyDequeIter::_type(vm), args[0], self.dequeItems.begin(), self.dequeItems.end());
});
vm->bind(type, "append(self, item) -> None",
[](VM *vm, ArgsView args)
{
@ -64,14 +150,17 @@ namespace pkpy
vm->bind(type, "copy(self) -> deque",
[](VM *vm, ArgsView args)
{
// TODO: FIX after implementing the __iter__ method
auto _lock = vm->heap.gc_scope_lock(); // locking the heap
PyDeque &self = _CAST(PyDeque &, args[0]);
PyDeque *newDeque = new PyDeque(vm, vm->None, vm->None);
// shallow copy
PyObject *newDequeObj = vm->heap.gcnew<PyDeque>(PyDeque::_type(vm), vm, vm->None, vm->None); // create the empty deque
PyDeque &newDeque = _CAST(PyDeque &, newDequeObj);
for (auto it = self.dequeItems.begin(); it != self.dequeItems.end(); ++it)
{
newDeque->append(*it);
newDeque.append(*it);
}
return vm->heap.gcnew<PyDeque>(PyDeque::_type(vm), *newDeque);
return newDequeObj;
});
vm->bind(type, "count(self, obj) -> int",
@ -202,6 +291,7 @@ namespace pkpy
vm->AttributeError("attribute 'maxlen' of 'collections.deque' objects is not writable");
return vm->None;
});
}
PyDeque::PyDeque(VM *vm, PyObject *iterable, PyObject *maxlen)
@ -302,10 +392,9 @@ namespace pkpy
}
}
if (this->bounded)
{
ss << ", maxlen=" << this->maxlen;
}
ss << "])";
ss << "], maxlen=" << this->maxlen << ")";
else
ss << "])";
return ss;
}