feat(agent): Added agent property value update functionality

TomasBoda · Jan 17, 2024 · b4805f3 · b4805f3
1 parent f05c4ff
commit b4805f3
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 1 deletion.
diff --git a/src/interpreter/interpreter.ts b/src/interpreter/interpreter.ts
@@ -180,4 +180,8 @@ export class Interpreter {
 
         return heightFunction;
     }
+
+    public updateAgentValue(agentIndex: number, propertyIdentifier: string, value: number): void {
+        this.runtime?.updateAgentValue(agentIndex, propertyIdentifier, value);
+    }
 }
diff --git a/src/parser/parser-util.ts b/src/parser/parser-util.ts
@@ -19,7 +19,7 @@ export class ParserUtil {
                 const objectDeclaration = ast as ObjectDeclaration;
 
                 const identifier = objectDeclaration.identifier;
-                const count = objectDeclaration.count;
+                const count = ParserUtil.astToCode(objectDeclaration.count);
 
                 code += `agent ${identifier} ${count} {\n`;
                 code += objectDeclaration.body.map(declaration => `\t${ParserUtil.astToCode(declaration)}`).join("\n");

diff --git a/src/runtime/runtime.ts b/src/runtime/runtime.ts
@@ -53,6 +53,11 @@ export class Runtime {
         this.output = { type: ValueType.Output, step: 0, agents: [] };
     }
 
+    // update specific agent variable value
+    public updateAgentValue(agentIndex: number, propertyIdentifier: string, value: number): void {
+        this.previousAgents[agentIndex].variables.set(propertyIdentifier, { type: ValueType.Number, value } as NumberValue);
+    }
+
     private evaluateProgram(program: Program): RuntimeValue {
         for (const statement of program.body) {
             switch (statement.type) {