From ad5f39db56b5b93ffb7cb39c7dc1f62237d7e164 Mon Sep 17 00:00:00 2001
From: Sahil Jain <jsahil730@gmail.com>
Date: Mon, 30 Jun 2025 23:57:31 +0530
Subject: [PATCH] Support SLICE opcodes

---
 ASTree.cpp                           |  67 +++++++++++++++++++++++++++
 tests/compiled/binary_slice.3.12.pyc | Bin 0 -> 412 bytes
 tests/compiled/store_slice.3.12.pyc  | Bin 0 -> 358 bytes
 tests/input/binary_slice.py          |   7 +++
 tests/input/store_slice.py           |   8 ++++
 tests/tokenized/binary_slice.txt     |   7 +++
 tests/tokenized/store_slice.txt      |   7 +++
 7 files changed, 96 insertions(+)
 create mode 100644 tests/compiled/binary_slice.3.12.pyc
 create mode 100644 tests/compiled/store_slice.3.12.pyc
 create mode 100644 tests/input/binary_slice.py
 create mode 100644 tests/input/store_slice.py
 create mode 100644 tests/tokenized/binary_slice.txt
 create mode 100644 tests/tokenized/store_slice.txt

diff --git a/ASTree.cpp b/ASTree.cpp
index 1f419d0..281d224 100644
--- a/ASTree.cpp
+++ b/ASTree.cpp
@@ -2473,6 +2473,73 @@ PycRef<ASTNode> BuildFromCode(PycRef<PycCode> code, PycModule* mod)
                 stack.push(next_tup);
             }
             break;
+        case Pyc::BINARY_SLICE:
+            {
+                PycRef<ASTNode> end = stack.top();
+                stack.pop();
+                PycRef<ASTNode> start = stack.top();
+                stack.pop();
+                PycRef<ASTNode> dest = stack.top();
+                stack.pop();
+
+                if (start.type() == ASTNode::NODE_OBJECT
+                        && start.cast<ASTObject>()->object() == Pyc_None) {
+                    start = NULL;
+                }
+
+                if (end.type() == ASTNode::NODE_OBJECT
+                        && end.cast<ASTObject>()->object() == Pyc_None) {
+                    end = NULL;
+                }
+
+                PycRef<ASTNode> slice;
+                if (start == NULL && end == NULL) {
+                    slice = new ASTSlice(ASTSlice::SLICE0);
+                } else if (start == NULL) {
+                    slice = new ASTSlice(ASTSlice::SLICE2, start, end);
+                } else if (end == NULL) {
+                    slice = new ASTSlice(ASTSlice::SLICE1, start, end);
+                } else {
+                    slice = new ASTSlice(ASTSlice::SLICE3, start, end);
+                }
+                stack.push(new ASTSubscr(dest, slice));
+            }
+            break;
+        case Pyc::STORE_SLICE:
+            {
+                PycRef<ASTNode> end = stack.top();
+                stack.pop();
+                PycRef<ASTNode> start = stack.top();
+                stack.pop();
+                PycRef<ASTNode> dest = stack.top();
+                stack.pop();
+                PycRef<ASTNode> values = stack.top();
+                stack.pop();
+
+                if (start.type() == ASTNode::NODE_OBJECT
+                        && start.cast<ASTObject>()->object() == Pyc_None) {
+                    start = NULL;
+                }
+
+                if (end.type() == ASTNode::NODE_OBJECT
+                        && end.cast<ASTObject>()->object() == Pyc_None) {
+                    end = NULL;
+                }
+
+                PycRef<ASTNode> slice;
+                if (start == NULL && end == NULL) {
+                    slice = new ASTSlice(ASTSlice::SLICE0);
+                } else if (start == NULL) {
+                    slice = new ASTSlice(ASTSlice::SLICE2, start, end);
+                } else if (end == NULL) {
+                    slice = new ASTSlice(ASTSlice::SLICE1, start, end);
+                } else {
+                    slice = new ASTSlice(ASTSlice::SLICE3, start, end);
+                }
+
+                curblock->append(new ASTStore(values, new ASTSubscr(dest, slice)));
+            }
+            break;
         default:
             fprintf(stderr, "Unsupported opcode: %s (%d)\n", Pyc::OpcodeName(opcode), opcode);
             cleanBuild = false;
diff --git a/tests/compiled/binary_slice.3.12.pyc b/tests/compiled/binary_slice.3.12.pyc
new file mode 100644
index 0000000000000000000000000000000000000000..24cc0f2f81a267dfc0a819000ef4f8d51af39dcd
GIT binary patch
literal 412
zcmX@j%ge<81R|G{G7^FGV-N=hSfPy16M&5A4CxFh42u|}7?>DR8B-Zj7*m*}7*;bv
z)G)$CnNwJ>h%%)xV-aP=D$1O~hD9wis@h6sO?FMTmmu3;GJyzY5WxZ>K(2nt1|*6=
z5=9^}zap^opa1{=zx)RzG?{KO=G<Z}D9X$$S;_DjsDj~_oSvS3NosLPv3_P=L1~G8
zQf6LaQDuB_PG)kdUP0w84x8Nkl+v73yCQy|a*)NvVL;*oGb1D8Ef&EB#+wYR4T&9$
s9f}QUExZs8Q-@$fY6}lch6%x8>R?7FW9|@wb2=FtQd+r-c!5R(0QW~*_W%F@

literal 0
HcmV?d00001

diff --git a/tests/compiled/store_slice.3.12.pyc b/tests/compiled/store_slice.3.12.pyc
new file mode 100644
index 0000000000000000000000000000000000000000..252f5af5f78ffd0479cbdcf5adf17aca3ee23abc
GIT binary patch
literal 358
zcmXwzF;BxV5QXm&yG;raFjovMQ<g;CD;5?8#K6GN1q!x8q>x5VI)up)Ldrj2h7Ivk
zn2|y@#FP!G8x!Y3Z+QOh`RVT7JZ}J0Pw!VZ=k$J=W>f4RR+j{iARt14cN|~_5eVxU
zXDCsEumu}};6%MCB}HF2kwqO7&JZk+mcUp1o3(Z+O|C6r<06#IcRLg5c1e7@Gx^Rk
zVjJ;DPCIjINBcpU)082fwK#7vX}OxGk!*RU;<R{y75%}_eh`F(%!@pX(`;FUd9hIP
zB2VIL8DwSaOm7!rnaD%6M`^P%KOy<H5wYPteN>ODV~s7_(%85Ije4vPbg${Vgrka)
fS5Iob_9vQu^tbD?`%+^y((Dtso<BHhhX(!vO^Q&)

literal 0
HcmV?d00001

diff --git a/tests/input/binary_slice.py b/tests/input/binary_slice.py
new file mode 100644
index 0000000..fb47b61
--- /dev/null
+++ b/tests/input/binary_slice.py
@@ -0,0 +1,7 @@
+l = [1,2,3,4,5,6]
+print(l[1:3])
+print(l[:2])
+print(l[3:])
+print(l[-4:])
+print(l[:-2])
+print(l[:])
\ No newline at end of file
diff --git a/tests/input/store_slice.py b/tests/input/store_slice.py
new file mode 100644
index 0000000..1e04a3a
--- /dev/null
+++ b/tests/input/store_slice.py
@@ -0,0 +1,8 @@
+a = [0] * 16
+l = [1,2,3]
+a[13:] = l
+a[5] = 10
+a[:2] = [1,2]
+a[:] = range(16)
+
+print(a)
\ No newline at end of file
diff --git a/tests/tokenized/binary_slice.txt b/tests/tokenized/binary_slice.txt
new file mode 100644
index 0000000..bb3d1a0
--- /dev/null
+++ b/tests/tokenized/binary_slice.txt
@@ -0,0 +1,7 @@
+l = [ 1 , 2 , 3 , 4 , 5 , 6 ] <EOL>
+print ( l [ 1 : 3 ] ) <EOL>
+print ( l [ : 2 ] ) <EOL>
+print ( l [ 3 : ] ) <EOL>
+print ( l [ - 4 : ] ) <EOL>
+print ( l [ : - 2 ] ) <EOL>
+print ( l [ : ] ) <EOL>
diff --git a/tests/tokenized/store_slice.txt b/tests/tokenized/store_slice.txt
new file mode 100644
index 0000000..2842070
--- /dev/null
+++ b/tests/tokenized/store_slice.txt
@@ -0,0 +1,7 @@
+a = [ 0 ] * 16 <EOL>
+l = [ 1 , 2 , 3 ] <EOL>
+a [ 13 : ] = l <EOL>
+a [ 5 ] = 10 <EOL>
+a [ : 2 ] = [ 1 , 2 ] <EOL>
+a [ : ] = range ( 16 ) <EOL>
+print ( a ) <EOL>