Finally nailing down SFT

Author: dluman

pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta:__legacy__"
 
 [project]
 name = "little-python-computer"
-version = "0.2.0"
+version = "0.2.1"
 dependencies = [
     "arglite"
 ]

src/lpc/cmd.py

@@ -128,22 +128,21 @@ def __brp(self, acc, storage):
 
     @manipulate
     def __sft(self, acc, storage):
-        acc.value = str(acc.value).zfill(3)
         self._val = str(self._val).zfill(2)
         # Left shift first
         if int(self._val[0]) > 0:
             shifts = int(self._val[0])
-            tmp = str(acc.value)
-            for _ in range(shifts):
-                tmp = f"{tmp}0"
-            acc.value = tmp[-3:]
+            tmp = str(acc.value).zfill(4)  + ('0' * shifts)
+            acc.value = tmp[-4:]
+            if shifts >= 4:
+                acc.value = 0
         # Right shift second
         if int(self._val[1]) > 0:
             shifts = int(self._val[1])
-            tmp = str(acc.value)
-            for _ in range(shifts):
-                tmp = f"0{tmp}"
-            acc.value = tmp[0:3]
+            tmp = ('0' * shifts) + str(acc.value).zfill(4)
+            acc.value = tmp[0:4]
+            if shifts >= 4:
+                acc.value = 0
         acc.value = int(acc.value)
 
     @inputs