64 lines
2.9 KiB
Python
64 lines
2.9 KiB
Python
# processors/process_eq.py
|
|
# -*- coding: utf-8 -*-
|
|
import sympy
|
|
import traceback
|
|
# Usar las nuevas utilidades de SymPy
|
|
from .processor_utils import get_sympy_representation, format_variable_name
|
|
from .symbol_manager import SymbolManager
|
|
|
|
SCL_SUFFIX = "_sympy_processed" # Nuevo sufijo
|
|
|
|
def process_eq(instruction, network_id, sympy_map, symbol_manager: SymbolManager, data):
|
|
"""
|
|
Genera la expresión SymPy para el comparador de igualdad (EQ).
|
|
El resultado se propaga por sympy_map['out'].
|
|
"""
|
|
instr_uid = instruction["instruction_uid"]
|
|
instr_type_original = instruction.get("type", "Eq")
|
|
if instr_type_original.endswith(SCL_SUFFIX) or "_error" in instr_type_original:
|
|
return False
|
|
|
|
# Obtener operandos como expresiones SymPy o constantes/strings
|
|
in1_info = instruction["inputs"].get("in1")
|
|
in2_info = instruction["inputs"].get("in2")
|
|
op1_sympy = get_sympy_representation(in1_info, network_id, sympy_map, symbol_manager)
|
|
op2_sympy = get_sympy_representation(in2_info, network_id, sympy_map, symbol_manager)
|
|
|
|
# Obtener 'pre' (RLO anterior) como expresión SymPy
|
|
pre_input = instruction["inputs"].get("pre") # Asumir 'pre' como entrada RLO estándar
|
|
sympy_pre_rlo = get_sympy_representation(pre_input, network_id, sympy_map, symbol_manager) if pre_input else sympy.true
|
|
|
|
# Verificar dependencias
|
|
if op1_sympy is None or op2_sympy is None or sympy_pre_rlo is None:
|
|
# print(f"DEBUG EQ {instr_uid}: Dependency not ready")
|
|
return False
|
|
|
|
# Crear la expresión de igualdad SymPy
|
|
try:
|
|
# sympify puede ser necesario si los operandos son strings de constantes
|
|
op1_eval = sympy.sympify(op1_sympy) if isinstance(op1_sympy, str) else op1_sympy
|
|
op2_eval = sympy.sympify(op2_sympy) if isinstance(op2_sympy, str) else op2_sympy
|
|
comparison_expr = sympy.Eq(op1_eval, op2_eval) # Eq para igualdad
|
|
except (SyntaxError, TypeError, ValueError) as e:
|
|
print(f"Error creating SymPy equality for {instr_uid}: {e}")
|
|
instruction["scl"] = f"// ERROR creando expr SymPy EQ {instr_uid}: {e}"
|
|
instruction["type"] = instr_type_original + "_error"
|
|
return True
|
|
|
|
# Guardar resultado (Expresión SymPy booleana) en el mapa para 'out'
|
|
map_key_out = (network_id, instr_uid, "out")
|
|
sympy_map[map_key_out] = comparison_expr
|
|
|
|
# Guardar el RLO de entrada ('pre') como ENO en el mapa SymPy
|
|
map_key_eno = (network_id, instr_uid, "eno")
|
|
sympy_map[map_key_eno] = sympy_pre_rlo
|
|
|
|
# Marcar como procesado, SCL principal es solo comentario
|
|
instruction["scl"] = f"// SymPy EQ: {comparison_expr}" # Comentario opcional
|
|
instruction["type"] = instr_type_original + SCL_SUFFIX
|
|
return True
|
|
|
|
# --- Processor Information Function ---
|
|
def get_processor_info():
|
|
"""Devuelve la información para el comparador de igualdad (EQ)."""
|
|
return {'type_name': 'eq', 'processor_func': process_eq, 'priority': 2} |