circuitscript 0.1.15 → 0.1.17

package/dist/cjs/visitor.js

@@ -22,19 +22,23 @@ class ParserVisitor extends BaseVisitor_js_1.BaseVisitor {
             this.setResult(ctx, [id, value]);
         };
         this.visitPin_select_expr = (ctx) => {
-            let value = null;
+            let pinId = null;
             const ctxData = ctx.data_expr();
             const result = this.visitResult(ctxData);
+            let pinValue;
             if (result instanceof ParamDefinition_js_1.NumericValue) {
-                value = result.toNumber();
+                pinValue = result.toNumber();
             }
             else if (typeof result === 'string') {
-                value = result;
+                pinValue = result;
             }
             else {
-                throw new utils_js_2.RuntimeExecutionError("Invalid value for pin", ctx);
+                throw new utils_js_2.RuntimeExecutionError("Invalid select pin: " + result, ctx);
             }
-            this.setResult(ctx, value);
+            if (pinValue !== undefined) {
+                pinId = new PinDefinition_js_1.PinId(pinValue);
+            }
+            this.setResult(ctx, pinId);
         };
         this.visitAdd_component_expr = (ctx) => {
             const [component, pinValue] = this.visitResult(ctx.data_expr_with_assignment());
@@ -221,11 +225,18 @@ class ParserVisitor extends BaseVisitor_js_1.BaseVisitor {
                                     }
                                 }
                                 else {
-                                    if (!(value instanceof ParamDefinition_js_1.NumericValue)) {
+                                    if (!(value instanceof ParamDefinition_js_1.NumericValue) && !(typeof value === 'string')) {
                                         throw new utils_js_2.RuntimeExecutionError(`Invalid numeric value for arrange.${sideKeyName}`, ctx);
                                     }
                                     else {
-                                        checkPinExistsAndNotDuplicated(value.toNumber(), ctx);
+                                        let useValue;
+                                        if (value instanceof ParamDefinition_js_1.NumericValue) {
+                                            useValue = value.toNumber();
+                                        }
+                                        else if (typeof value === 'string') {
+                                            useValue = value;
+                                        }
+                                        value && checkPinExistsAndNotDuplicated(useValue, ctx);
                                     }
                                 }
                             }
@@ -265,7 +276,6 @@ class ParserVisitor extends BaseVisitor_js_1.BaseVisitor {
             scope.exitContext();
             scope.popOnPropertyHandler();
             const properties = this.getPropertyExprList(ctx.property_expr());
-            const pins = this.parseCreateComponentPins(properties.get('pins'));
             let instanceName = this.getExecutor().getUniqueInstanceName();
             const propParams = properties.get('params');
             const params = this.parseCreateComponentParams(propParams);
@@ -278,11 +288,11 @@ class ParserVisitor extends BaseVisitor_js_1.BaseVisitor {
                 }
                 instanceName += `${globals_js_1.Delimiter1}${appendValue}`;
             }
-            const arrange = properties.has('arrange') ?
+            const arrangeProp = properties.has('arrange') ?
                 properties.get('arrange') : null;
-            const display = properties.has('display') ?
+            const displayProp = properties.has('display') ?
                 properties.get('display') : null;
-            const type = properties.has('type') ?
+            const typeProp = properties.has('type') ?
                 properties.get('type') : null;
             const copy = properties.has('copy') ?
                 properties.get('copy') : false;
@@ -294,8 +304,29 @@ class ParserVisitor extends BaseVisitor_js_1.BaseVisitor {
                 properties.get(globals_js_1.ParamKeys.angle) : null;
             const followWireOrientation = properties.has('followWireOrientation') ?
                 properties.get('followWireOrientation') : true;
+            let pins = [];
+            if (displayProp !== null && arrangeProp === null
+                && typeProp !== types_js_1.TypeProps.Graphic) {
+                const drawCommands = displayProp.getCommands();
+                drawCommands.forEach(command => {
+                    const [commandValue,] = command;
+                    if (commandValue === draw_symbols_js_1.PlaceHolderCommands.vpin
+                        || commandValue === draw_symbols_js_1.PlaceHolderCommands.hpin
+                        || commandValue === draw_symbols_js_1.PlaceHolderCommands.pin) {
+                        const id = PinDefinition_js_1.PinId.from(command[1][0]);
+                        const pinType = id.getType();
+                        const pinName = id.toString();
+                        pins.push(new PinDefinition_js_1.PinDefinition(id, pinType, pinName, PinTypes_js_1.PinTypes.Any));
+                    }
+                });
+            }
+            else {
+                pins = this.parseCreateComponentPins(properties.get('pins'));
+            }
             const props = {
-                arrange, display, type, width, height, copy,
+                arrange: arrangeProp,
+                display: displayProp,
+                type: typeProp, width, height, copy,
                 angle, followWireOrientation
             };
             try {
@@ -381,7 +412,7 @@ class ParserVisitor extends BaseVisitor_js_1.BaseVisitor {
             if (ctxNestedProperties) {
                 const nestedKeyValues = this.visitResult(ctxNestedProperties);
                 nestedKeyValues.forEach((value, key) => {
-                    parameters.push(['keyword', key, value]);
+                    parameters.push(['keyword', key, (0, utils_js_1.unwrapValue)(value)]);
                 });
             }
             else {
@@ -539,36 +570,36 @@ class ParserVisitor extends BaseVisitor_js_1.BaseVisitor {
             this.setResult(ctx, result);
         };
         this.visitData_expr_with_assignment = (ctx) => {
-            let component = null;
-            let componentCtx = null;
+            let dataResult = null;
+            let componentCtx;
             const ctxDataExpr = ctx.data_expr();
             const ctxAssignmentExpr = ctx.assignment_expr();
             if (ctxDataExpr) {
-                component = this.visitResult(ctxDataExpr);
-                component = (0, utils_js_1.unwrapValue)(component);
+                dataResult = this.visitResult(ctxDataExpr);
                 componentCtx = ctxDataExpr;
-                if (component === null || component === undefined) {
-                    this.throwWithContext(ctxDataExpr, "Could not find component: " + ctxDataExpr.getText());
-                }
             }
             else if (ctxAssignmentExpr) {
-                component = this.visitResult(ctxAssignmentExpr);
+                dataResult = this.visitResult(ctxAssignmentExpr);
                 componentCtx = ctxAssignmentExpr;
             }
-            if (component instanceof ClassComponent_js_1.ClassComponent
-                && component.copyProp) {
-                component = this.getExecutor().copyComponent(component);
-            }
-            if (component instanceof types_js_1.UndeclaredReference) {
-                const { reference: { trailers = [], parentValue = null } } = component;
+            if (dataResult instanceof types_js_1.AnyReference) {
+                const { trailers = [], parentValue = null } = dataResult;
                 if (parentValue instanceof ClassComponent_js_1.ClassComponent
                     && trailers.length > 0
                     && trailers[0] === globals_js_1.ModuleContainsKeyword) {
-                    component = parentValue;
-                    this.placeModuleContains(component);
+                    dataResult = parentValue;
+                    this.placeModuleContains(dataResult);
                 }
             }
-            if (component && component instanceof ClassComponent_js_1.ClassComponent) {
+            dataResult = (0, utils_js_1.unwrapValue)(dataResult);
+            if (dataResult === null || dataResult === undefined) {
+                this.throwWithContext(componentCtx, "Could not find component: " + componentCtx.getText());
+            }
+            if (dataResult instanceof ClassComponent_js_1.ClassComponent
+                && dataResult.copyProp) {
+                dataResult = this.getExecutor().copyComponent(dataResult);
+            }
+            if (dataResult && dataResult instanceof ClassComponent_js_1.ClassComponent) {
                 const modifiers = ctx.component_modifier_expr();
                 modifiers.forEach(modifier => {
                     const modifierText = modifier.ID(0).getText();
@@ -585,23 +616,23 @@ class ParserVisitor extends BaseVisitor_js_1.BaseVisitor {
                     if (modifierText === globals_js_1.ParamKeys.flip) {
                         const flipValue = result;
                         if (flipValue.indexOf('x') !== -1) {
-                            component.setParam(globals_js_1.ParamKeys.flipX, 1);
+                            dataResult.setParam(globals_js_1.ParamKeys.flipX, 1);
                             shouldIgnoreWireOrientation = true;
                         }
                         if (flipValue.indexOf('y') !== -1) {
-                            component.setParam(globals_js_1.ParamKeys.flipY, 1);
+                            dataResult.setParam(globals_js_1.ParamKeys.flipY, 1);
                             shouldIgnoreWireOrientation = true;
                         }
                     }
                     else if (modifierText === globals_js_1.ParamKeys.angle) {
-                        component.setParam(globals_js_1.ParamKeys.angle, result);
+                        dataResult.setParam(globals_js_1.ParamKeys.angle, result);
                         shouldIgnoreWireOrientation = true;
                     }
                     else if (modifierText === 'anchor') {
-                        component.setParam('anchor', result);
+                        dataResult.setParam('anchor', result);
                     }
                     if (shouldIgnoreWireOrientation) {
-                        component.useWireOrientationAngle = false;
+                        dataResult.useWireOrientationAngle = false;
                     }
                 });
             }
@@ -611,15 +642,15 @@ class ParserVisitor extends BaseVisitor_js_1.BaseVisitor {
                 pinValue = this.visitResult(ctxPinSelectExpr);
             }
             else {
-                if (component instanceof ClassComponent_js_1.ClassComponent) {
-                    pinValue = component.getDefaultPin();
+                if (dataResult instanceof ClassComponent_js_1.ClassComponent) {
+                    pinValue = dataResult.getDefaultPin();
                 }
                 else {
-                    const undeclaredRef = component;
+                    const undeclaredRef = dataResult;
                     this.throwWithContext(componentCtx, 'Invalid component: ' + undeclaredRef.reference.name);
                 }
             }
-            this.setResult(ctx, [component, pinValue]);
+            this.setResult(ctx, [dataResult, pinValue]);
         };
         this.visitUnaryOperatorExpr = (ctx) => {
             let value = this.visitResult(ctx.data_expr());
@@ -749,8 +780,8 @@ class ParserVisitor extends BaseVisitor_js_1.BaseVisitor {
             this.setResult(ctx, result);
         };
         this.visitAdditionExpr = (ctx) => {
-            const value1 = this.resolveDataExpr(ctx.data_expr(0));
-            const value2 = this.resolveDataExpr(ctx.data_expr(1));
+            const value1 = (0, utils_js_1.unwrapValue)(this.resolveDataExpr(ctx.data_expr(0)));
+            const value2 = (0, utils_js_1.unwrapValue)(this.resolveDataExpr(ctx.data_expr(1)));
             if (ctx.Addition() && (typeof value1 === 'string' || typeof value2 === 'string')) {
                 let tmpValue1 = value1;
                 if (value1 instanceof ParamDefinition_js_1.NumericValue) {
@@ -779,6 +810,8 @@ class ParserVisitor extends BaseVisitor_js_1.BaseVisitor {
         };
         this.visitFunction_def_expr = (ctx) => {
             const functionName = ctx.ID().getText();
+            const uniqueFunctionID = '__._' + ctx.start.line + '_'
+                + ctx.start.column + '_' + functionName + '_' + ctx.getText();
             let funcDefinedParameters = [];
             const ctxFunctionArgsExpr = ctx.function_args_expr();
             if (ctxFunctionArgsExpr) {
@@ -800,19 +833,26 @@ class ParserVisitor extends BaseVisitor_js_1.BaseVisitor {
                 nextLastExecution.mergeScope(lastExecution.scope, executionContextName);
                 return [lastExecution, returnValue];
             };
-            this.getExecutor().createFunction(functionName, __runFunc);
+            this.getExecutor().createFunction(functionName, __runFunc, ctx, uniqueFunctionID);
         };
         this.visitPin_select_expr2 = (ctx) => {
             const ctxStringValue = ctx.STRING_VALUE();
             const ctxIntegerValue = ctx.INTEGER_VALUE();
-            let result = null;
+            let pinIdValue;
+            let pinId = null;
             if (ctxStringValue) {
-                result = this.prepareStringValue(ctxStringValue.getText());
+                pinIdValue = this.prepareStringValue(ctxStringValue.getText());
             }
             else if (ctxIntegerValue) {
-                result = Number(ctxIntegerValue.getText());
+                pinIdValue = Number(ctxIntegerValue.getText());
             }
-            this.setResult(ctx, result);
+            if (pinIdValue !== undefined) {
+                pinId = new PinDefinition_js_1.PinId(pinIdValue);
+            }
+            else {
+                throw new utils_js_2.RuntimeExecutionError("Invalid select pin", ctx);
+            }
+            this.setResult(ctx, pinId);
         };
         this.visitAt_block_pin_expr = (ctx) => {
             const executor = this.getExecutor();
@@ -1204,7 +1244,7 @@ class ParserVisitor extends BaseVisitor_js_1.BaseVisitor {
                         return item;
                     }
                     else {
-                        return (0, ParamDefinition_js_1.numeric)(nameToPinId.get(item));
+                        return new PinDefinition_js_1.PinId(nameToPinId.get(item));
                     }
                 });
                 if (items.length > 0) {
@@ -1295,7 +1335,7 @@ class ParserVisitor extends BaseVisitor_js_1.BaseVisitor {
     getGraph() {
         const executor = this.getExecutor();
         const fullSequence = executor.scope.sequence;
-        const tmpNet = executor.scope.getNet(executor.scope.componentRoot, 1);
+        const tmpNet = executor.scope.getNet(executor.scope.componentRoot, new PinDefinition_js_1.PinId(1));
         const sequence = (tmpNet === null)
             ? fullSequence.slice(1) : fullSequence;
         const nets = executor.scope.getNets();

package/dist/esm/BaseVisitor.js

@@ -1,5 +1,4 @@
 import { Big } from 'big.js';
-import { ExpressionContext } from "./antlr/CircuitScriptParser.js";
 import { CircuitScriptVisitor } from "./antlr/CircuitScriptVisitor.js";
 import { ExecutionContext } from "./execute.js";
 import { Logger } from "./logger.js";
@@ -7,11 +6,12 @@ import { ClassComponent } from "./objects/ClassComponent.js";
 import { NumberOperator, NumericValue, PercentageValue } from "./objects/ParamDefinition.js";
 import { PinTypes } from "./objects/PinTypes.js";
 import { Direction, AnyReference, UndeclaredReference } from "./objects/types.js";
-import { ComponentTypes, DoubleDelimiter1, GlobalDocumentName, ReferenceTypes } from './globals.js';
+import { ComponentTypes, DoubleDelimiter1, GlobalDocumentName, ReferenceTypes, TrailerArrayIndex } from './globals.js';
 import { isReference, unwrapValue as unwrapValue } from "./utils.js";
 import { linkBuiltInMethods } from './builtinMethods.js';
 import { resolveToNumericValue, RuntimeExecutionError, throwWithContext } from './utils.js';
 import { SequenceAction } from './objects/ExecutionScope.js';
+import { PinId } from './objects/PinDefinition.js';
 export class BaseVisitor extends CircuitScriptVisitor {
     indentLevel = 0;
     startingContext;
@@ -134,28 +134,37 @@ export class BaseVisitor extends CircuitScriptVisitor {
         const ctxAtom = ctx.atom_expr();
         const ctxFuncCallRef = ctx.function_call_expr();
         let leftSideReference;
+        let lhsCtx;
         if (ctxAtom) {
             leftSideReference = this.getReference(ctx.atom_expr());
+            lhsCtx = ctxAtom;
         }
         else if (ctxFuncCallRef) {
             this.setResult(ctxFuncCallRef, { keepReference: true });
             leftSideReference = this.visitResult(ctxFuncCallRef);
+            lhsCtx = ctxFuncCallRef;
+        }
+        const rhsCtx = ctx.data_expr();
+        const rhsCtxResult = this.visitResult(rhsCtx);
+        if (isReference(rhsCtxResult) && !rhsCtxResult.found) {
+            this.throwWithContext(rhsCtx, rhsCtx.getText() + ' is not defined');
         }
-        const rhsCtxResult = this.visitResult(ctx.data_expr());
         const rhsValue = unwrapValue(rhsCtxResult);
         const trailers = leftSideReference.trailers ?? [];
         const sequenceParts = [];
         if (trailers.length === 0) {
+            this.getScope().setVariable(leftSideReference.name, rhsValue);
+            let itemType = '';
             if (rhsValue instanceof ClassComponent) {
-                this.getScope().setVariable(leftSideReference.name, rhsValue);
-                sequenceParts.push(...['instance', leftSideReference.name, rhsValue]);
+                itemType = ReferenceTypes.instance;
                 this.log2(`assigned '${leftSideReference.name}' to ClassComponent`);
             }
             else {
+                itemType = ReferenceTypes.variable;
                 this.getScope().setVariable(leftSideReference.name, rhsValue);
                 this.log2(`assigned variable ${leftSideReference.name} to ${rhsValue}`);
-                sequenceParts.push(...['variable', leftSideReference.name, rhsValue]);
             }
+            sequenceParts.push(...[itemType, leftSideReference.name, rhsValue]);
         }
         else {
             if (leftSideReference.parentValue instanceof ClassComponent) {
@@ -163,7 +172,7 @@ export class BaseVisitor extends CircuitScriptVisitor {
                 this.log2(`assigned component param ${leftSideReference.parentValue} trailers: ${trailers} value: ${rhsValue}`);
                 sequenceParts.push(...['instance', [leftSideReference.parentValue, trailers], rhsValue]);
                 if (leftSideReference.parentValue.typeProp === ComponentTypes.net) {
-                    const net = this.getScope().getNet(leftSideReference.parentValue, 1);
+                    const net = this.getScope().getNet(leftSideReference.parentValue, new PinId(1));
                     if (net) {
                         const trailerValue = trailers.join(".");
                         net.params.set(trailerValue, rhsValue);
@@ -171,8 +180,20 @@ export class BaseVisitor extends CircuitScriptVisitor {
                 }
             }
             else if (leftSideReference.parentValue instanceof Object) {
-                leftSideReference.parentValue[trailers.join('.')] = rhsValue;
-                this.log2(`assigned object ${leftSideReference.parentValue} trailers: ${trailers} value: ${rhsValue}`);
+                if (Array.isArray(trailers[0]) && trailers[0][0] === TrailerArrayIndex) {
+                    if (Array.isArray(leftSideReference.parentValue)) {
+                        const arrayIndexValue = trailers[0][1];
+                        leftSideReference.parentValue[arrayIndexValue] = rhsValue;
+                        this.log2(`assigned array index ${leftSideReference.parentValue} index: ${arrayIndexValue} value: ${rhsValue}`);
+                    }
+                    else {
+                        this.throwWithContext(lhsCtx, "Invalid array");
+                    }
+                }
+                else {
+                    leftSideReference.parentValue[trailers.join('.')] = rhsValue;
+                    this.log2(`assigned object ${leftSideReference.parentValue} trailers: ${trailers} value: ${rhsValue}`);
+                }
                 sequenceParts.push(...['variable', [leftSideReference.parentValue, trailers], rhsValue]);
             }
         }
@@ -252,19 +273,39 @@ export class BaseVisitor extends CircuitScriptVisitor {
         }
         return reference;
     }
+    visitTrailer_expr2 = (ctx) => {
+        const reference = this.getResult(ctx);
+        const ctxID = ctx.ID();
+        const ctxDataExpr = ctx.data_expr();
+        const useValue = reference.value;
+        let nextReference;
+        if (ctxID) {
+            reference.trailers.push(ctxID.getText());
+            nextReference = this.getExecutor().resolveTrailers(reference.type, useValue, reference.trailers);
+        }
+        else if (ctxDataExpr) {
+            const arrayIndex = this.visitResult(ctxDataExpr);
+            if (arrayIndex instanceof NumericValue) {
+                const arrayIndexValue = arrayIndex.toNumber();
+                const foundValue = useValue[arrayIndexValue];
+                const refType = foundValue instanceof ClassComponent
+                    ? ReferenceTypes.instance : ReferenceTypes.variable;
+                nextReference = new AnyReference({
+                    found: true,
+                    type: refType,
+                    value: foundValue,
+                    trailers: [[TrailerArrayIndex, arrayIndexValue]],
+                    parentValue: useValue
+                });
+            }
+        }
+        this.setResult(ctx, nextReference);
+    };
     visitAtom_expr = (ctx) => {
         const executor = this.getExecutor();
         const firstId = ctx.ID(0);
         const atomId = firstId.getText();
         let currentReference;
-        const idTrailers = [];
-        if (ctx.ID().length > 1) {
-            const idLength = ctx.ID().length;
-            for (let i = 1; i < idLength; i++) {
-                const tmpCtx = ctx.ID(i);
-                idTrailers.push(tmpCtx.getText());
-            }
-        }
         if (this.pinTypesList.indexOf(atomId) !== -1) {
             currentReference = new AnyReference({
                 found: true,
@@ -273,18 +314,18 @@ export class BaseVisitor extends CircuitScriptVisitor {
             });
         }
         else {
-            currentReference = executor.resolveVariable(this.executionStack, atomId, idTrailers);
+            currentReference = executor.resolveVariable(this.executionStack, atomId);
         }
-        if (currentReference.found && currentReference.type === 'instance' && idTrailers.length === 0) {
-            const tmpComponent = currentReference.value;
-            for (const [pinId, net] of tmpComponent.pinNets) {
-                executor.scope.setNet(tmpComponent, pinId, net);
+        if (currentReference !== undefined && currentReference.found) {
+            const trailersLength = ctx.trailer_expr2().length;
+            for (let i = 0; i < trailersLength; i++) {
+                const trailerCtx = ctx.trailer_expr2(i);
+                this.setResult(trailerCtx, currentReference);
+                currentReference = this.visitResult(trailerCtx);
             }
         }
-        if (ctx.parent instanceof ExpressionContext && !currentReference.found) {
-            this.throwWithContext(ctx, "Unknown symbol: " + atomId);
-        }
         this.setResult(ctx, currentReference);
+        this.log2('atom resolved: ' + ctx.getText() + ' -> ' + currentReference);
     };
     visitFunctionCallExpr = (ctx) => {
         const result = this.visitResult(ctx.function_call_expr());
@@ -315,6 +356,11 @@ export class BaseVisitor extends CircuitScriptVisitor {
             currentReference.trailers = [];
             ctx.trailer_expr().forEach(item => {
                 if (item.OPEN_PAREN() && item.CLOSE_PAREN()) {
+                    if (currentReference.type === ReferenceTypes.variable) {
+                        if (currentReference.value instanceof AnyReference && currentReference.value.type === ReferenceTypes.function) {
+                            currentReference = currentReference.value;
+                        }
+                    }
                     let parameters = [];
                     const ctxParameters = item.parameters();
                     if (ctxParameters) {
@@ -341,10 +387,9 @@ export class BaseVisitor extends CircuitScriptVisitor {
                     }
                 }
                 else {
-                    currentReference.trailers.push(item.ID().getText());
-                    currentReference = this.getExecutor().resolveTrailers(currentReference.type, (currentReference.parentValue !== undefined)
-                        ? currentReference.parentValue
-                        : currentReference.value, currentReference.trailers);
+                    const ctxTrailer = item.trailer_expr2();
+                    this.setResult(ctxTrailer, currentReference);
+                    currentReference = this.visitResult(ctxTrailer);
                 }
             });
         }
@@ -410,7 +455,8 @@ export class BaseVisitor extends CircuitScriptVisitor {
                     value = reference;
                 }
                 else {
-                    if (reference.trailers && reference.trailers.length > 0) {
+                    if ((reference.trailers && reference.trailers.length > 0)
+                        || reference.type === ReferenceTypes.function) {
                         value = reference;
                     }
                     else {
@@ -445,11 +491,11 @@ export class BaseVisitor extends CircuitScriptVisitor {
         const returnList = [];
         dataExpressions.forEach((item, index) => {
             const value = this.visitResult(item);
-            returnList.push(['position', index, value]);
+            returnList.push(['position', index, unwrapValue(value)]);
         });
         keywordAssignmentExpressions.forEach((item) => {
             const [key, value] = this.visitResult(item);
-            returnList.push(['keyword', key, value]);
+            returnList.push(['keyword', key, unwrapValue(value)]);
         });
         this.setResult(ctx, returnList);
     };
@@ -459,9 +505,7 @@ export class BaseVisitor extends CircuitScriptVisitor {
     visitFunction_return_expr = (ctx) => {
         const executor = this.getExecutor();
         executor.log('return from function');
-        const ctxDataExpr = ctx.data_expr();
-        this.visit(ctxDataExpr);
-        const returnValue = this.getResult(ctxDataExpr);
+        const returnValue = this.visitResult(ctx.data_expr());
         executor.stopFurtherExpressions = true;
         executor.returnValue = returnValue;
         this.setResult(ctx, returnValue);
@@ -495,6 +539,25 @@ export class BaseVisitor extends CircuitScriptVisitor {
     visitArrayExpr = (ctx) => {
         this.setResult(ctx, this.visitResult(ctx.array_expr()));
     };
+    visitArrayIndexExpr = (ctx) => {
+        const ctxArray = ctx.data_expr(0);
+        const ctxArrayIndex = ctx.data_expr(1);
+        const arrayItem = this.visitResult(ctxArray);
+        const indexValue = this.visitResult(ctxArrayIndex);
+        if (!Array.isArray(arrayItem)) {
+            throw new RuntimeExecutionError("Invalid array", ctxArray);
+        }
+        if (!(indexValue instanceof NumericValue)) {
+            throw new RuntimeExecutionError("Invalid index value", ctxArrayIndex);
+        }
+        const indexValueNumber = indexValue.toNumber();
+        if (isNaN(indexValueNumber)) {
+            throw new RuntimeExecutionError("Invalid index value", ctxArrayIndex);
+        }
+        if (Array.isArray(arrayItem)) {
+            this.setResult(ctx, arrayItem[indexValueNumber]);
+        }
+    };
     setResult(ctx, value) {
         this.resultData.set(ctx, value);
     }

package/dist/esm/antlr/CircuitScriptLexer.js

@@ -69,7 +69,7 @@ export class CircuitScriptLexer extends antlr.Lexer {
         "DEFAULT_TOKEN_CHANNEL", "HIDDEN"
     ];
     static literalNames = [
-        null, "':'", "','", "'='", "'..'", "'.'", "'['", "']'", "'break'",
+        null, "':'", "','", "'='", "'..'", "'['", "']'", "'.'", "'break'",
         "'branch'", "'create'", "'component'", "'graphic'", "'module'",
         "'wire'", "'pin'", "'add'", "'at'", "'to'", "'point'", "'join'",
         "'parallel'", "'return'", "'def'", "'import'", "'for'", "'in'",
@@ -235,8 +235,8 @@ export class CircuitScriptLexer extends antlr.Lexer {
         0, 131, 473, 1, 0, 0, 0, 133, 477, 1, 0, 0, 0, 135, 486, 1, 0, 0, 0, 137, 138, 5, 58, 0,
         0, 138, 2, 1, 0, 0, 0, 139, 140, 5, 44, 0, 0, 140, 4, 1, 0, 0, 0, 141, 142, 5, 61, 0, 0,
         142, 6, 1, 0, 0, 0, 143, 144, 5, 46, 0, 0, 144, 145, 5, 46, 0, 0, 145, 8, 1, 0, 0, 0, 146,
-        147, 5, 46, 0, 0, 147, 10, 1, 0, 0, 0, 148, 149, 5, 91, 0, 0, 149, 12, 1, 0, 0, 0, 150,
-        151, 5, 93, 0, 0, 151, 14, 1, 0, 0, 0, 152, 153, 5, 98, 0, 0, 153, 154, 5, 114, 0, 0, 154,
+        147, 5, 91, 0, 0, 147, 10, 1, 0, 0, 0, 148, 149, 5, 93, 0, 0, 149, 12, 1, 0, 0, 0, 150,
+        151, 5, 46, 0, 0, 151, 14, 1, 0, 0, 0, 152, 153, 5, 98, 0, 0, 153, 154, 5, 114, 0, 0, 154,
         155, 5, 101, 0, 0, 155, 156, 5, 97, 0, 0, 156, 157, 5, 107, 0, 0, 157, 16, 1, 0, 0, 0,
         158, 159, 5, 98, 0, 0, 159, 160, 5, 114, 0, 0, 160, 161, 5, 97, 0, 0, 161, 162, 5, 110,
         0, 0, 162, 163, 5, 99, 0, 0, 163, 164, 5, 104, 0, 0, 164, 18, 1, 0, 0, 0, 165, 166, 5,