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

add PK_SLICE_LOOP

Browse Source
This commit is contained in:
blueloveTH 2024-02-20 02:13:33 +08:00
parent 3534492bb6
commit 38dceac61b
3 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
include/pocketpy/common.h
View File

@ -234,4 +234,6 @@ inline const char* kPlatformStrings[] = {
"unknown" // 6 "unknown" // 6
}; };
#define PK_SLICE_LOOP(i, start, stop, step) for(int i=start; step>0?i<stop:i>stop; i+=step)
} // namespace pkpy } // namespace pkpy

2
src/pocketpy.cpp
View File

@ -20,7 +20,7 @@ PyObject* PyArrayGetItem(VM* vm, PyObject* _0, PyObject* _1){
int start, stop, step; int start, stop, step;
vm->parse_int_slice(s, self.size(), start, stop, step); vm->parse_int_slice(s, self.size(), start, stop, step);
List new_list; List new_list;
for(int i=start; step>0?i<stop:i>stop; i+=step) new_list.push_back(self[i]); PK_SLICE_LOOP(i, start, stop, step) new_list.push_back(self[i]);
return VAR(T(std::move(new_list))); return VAR(T(std::move(new_list)));
} }
vm->TypeError("indices must be integers or slices"); vm->TypeError("indices must be integers or slices");

4
src/str.cpp
View File

@ -334,9 +334,9 @@ int utf8len(unsigned char c, bool suppress){
Str Str::u8_slice(int start, int stop, int step) const{ Str Str::u8_slice(int start, int stop, int step) const{
SStream ss; SStream ss;
if(is_ascii){ if(is_ascii){
for(int i=start; step>0?i<stop:i>stop; i+=step) ss << data[i]; PK_SLICE_LOOP(i, start, stop, step) ss << data[i];
}else{ }else{
for(int i=start; step>0?i<stop:i>stop; i+=step) ss << u8_getitem(i); PK_SLICE_LOOP(i, start, stop, step) ss << u8_getitem(i);
} }
return ss.str(); return ss.str();
} }