ASTNodes.java

import java.util.List;

/**
 * Абстрактный базовый класс для всех узлов абстрактного синтаксического дерева (AST).
 * AST представляет структуру программы в виде дерева.
 */
abstract class ASTNode {}

/**
 * Узел, представляющий всю программу.
 * Содержит объявления функций и главный вызов.
 */
class ProgNode extends ASTNode {
  List<FuncDeclNode> funcs;  // список объявлений функций
  CallNode mainCall;         // главный вызов функции (точка входа)

  /**
   * Конструктор узла программы.
   * @param funcs список объявлений функций
   * @param mainCall главный вызов функции
   */
  public ProgNode(List<FuncDeclNode> funcs, CallNode mainCall) {
    this.funcs = funcs;
    this.mainCall = mainCall;
  }
}

/**
 * Узел объявления функции.
 * Содержит информацию о типе возвращаемого значения, имени, параметрах и теле функции.
 */
class FuncDeclNode extends ASTNode {
  String retType;          // тип возвращаемого значения ("void" или "int")
  String name;             // имя функции
  List<Param> params;      // список параметров функции
  List<ASTNode> body;      // тело функции (список инструкций)

  /**
   * Конструктор узла объявления функции.
   * @param retType тип возвращаемого значения
   * @param name имя функции
   * @param params список параметров
   * @param body тело функции
   */
  public FuncDeclNode(String retType, String name, List<Param> params, List<ASTNode> body) {
    this.retType = retType;
    this.name = name;
    this.params = params;
    this.body = body;
  }
}

/**
 * Класс, представляющий параметр функции.
 * Содержит тип и имя параметра.
 */
class Param {
  String type;  // тип параметра (в данном языке всегда "int")
  String name;  // имя параметра

  /**
   * Конструктор параметра.
   * @param type тип параметра
   * @param name имя параметра
   */
  public Param(String type, String name) {
    this.type = type;
    this.name = name;
  }
}

/**
 * Узел вызова функции.
 * Содержит имя вызываемой функции и список аргументов.
 */
class CallNode extends ASTNode {
  String name;           // имя вызываемой функции
  List<ASTNode> args;    // список аргументов (выражений)

  /**
   * Конструктор узла вызова функции.
   * @param name имя функции
   * @param args список аргументов
   */
  public CallNode(String name, List<ASTNode> args) {
    this.name = name;
    this.args = args;
  }
}

/**
 * Узел, представляющий переменную.
 * Содержит имя переменной.
 */
class VarNode extends ASTNode {
  String name;  // имя переменной

  /**
   * Конструктор узла переменной.
   * @param name имя переменной
   */
  public VarNode(String name) {
    this.name = name;
  }
}

/**
 * Узел, представляющий числовую константу.
 * Содержит целочисленное значение.
 */
class NumNode extends ASTNode {
  int value;  // числовое значение

  /**
   * Конструктор узла числовой константы.
   * @param value числовое значение
   */
  public NumNode(int value) {
    this.value = value;
  }
}

/**
 * Узел бинарной операции.
 * Содержит оператор и левый/правый операнды.
 */
class BinNode extends ASTNode {
  char op;        // оператор ('+' или '-')
  ASTNode left;   // левый операнд
  ASTNode right;  // правый операнд

  /**
   * Конструктор узла бинарной операции.
   * @param op оператор
   * @param left левый операнд
   * @param right правый операнд
   */
  public BinNode(char op, ASTNode left, ASTNode right) {
    this.op = op;
    this.left = left;
    this.right = right;
  }
}

/**
 * Узел инструкции return.
 * Содержит возвращаемое выражение (может быть null для void функций).
 */
class ReturnNode extends ASTNode {
  ASTNode expr;  // возвращаемое выражение (может быть null)

  /**
   * Конструктор узла return.
   * @param expr возвращаемое выражение
   */
  public ReturnNode(ASTNode expr) {
    this.expr = expr;
  }
}

/**
 * Узел объявления переменной.
 * Содержит имя объявляемой переменной.
 */
class VarDeclNode extends ASTNode {
  String name;  // имя переменной

  /**
   * Конструктор узла объявления переменной.
   * @param name имя переменной
   */
  public VarDeclNode(String name) {
    this.name = name;
  }
}