From b88279c8414bb865073ee795b11a96b42c19f007 Mon Sep 17 00:00:00 2001
From: mmichlol <michal.lewandowski.133@gmail.com>
Date: Fri, 6 Feb 2026 16:51:47 +0100
Subject: [PATCH] fixed IF statment

---
 PCCcompiler/codegen.cpp | 72 +++++++++++++++++++++++++++++------------
 1 file changed, 51 insertions(+), 21 deletions(-)

diff --git a/PCCcompiler/codegen.cpp b/PCCcompiler/codegen.cpp
index b193802..841cf96 100644
--- a/PCCcompiler/codegen.cpp
+++ b/PCCcompiler/codegen.cpp
@@ -1,39 +1,71 @@
 #include "codegen.h"
 #include <string>
 #include <vector>
+#include <iostream>
 
 std::string generateAssembly(const CompilerState& state) {
     std::string result;
 
-    // --- NAGŁÓWEK I SEKCJA DATA ---
+    // --- NAGŁÓWEK ---
     result += "global main\n";
     result += "extern printf\n";
     result += "extern GetAsyncKeyState\n\n";
 
     result += "section .data\n";
-
-    // Generowanie zmiennych
     for (const auto& v : state.variables) {
         result += "    " + v.first + " dd " + std::to_string(v.second) + "\n";
     }
-
-    // Stałe stringi
     result += "    fmt db '%d', 10, 0\n";
     result += "    pause_msg db 'Nacisnij ESC aby zamknac...', 10, 0\n\n";
 
     result += "section .text\n";
 
-    // --- DEFINICJE FUNKCJI UŻYTKOWNIKA ---
+    // --- GENEROWANIE FUNKCJI ---
     for (const auto& func : state.functions) {
         result += func.first + ":\n";
-        result += "    sub rsp, 40\n"; // Shadow space
+        result += "    sub rsp, 40\n";
 
-        // Printy wewnątrz funkcji
-        for (const std::string& var : func.second) {
-            if (state.variables.count(var)) {
-                result += "    mov edx, [" + var + "]\n";
-                result += "    lea rcx, [rel fmt]\n";
-                result += "    call printf\n";
+        // Iterujemy przez LISTĘ ROZKAZÓW danej funkcji
+        // Rozkazy mogą być typu "PRINT:x" albo "IF:index"
+        for (const std::string& cmd : func.second) {
+
+            // 1. ZWYKŁY PRINT (format: "PRINT:nazwa_zmiennej")
+            if (cmd.find("PRINT:") == 0) {
+                std::string varName = cmd.substr(6); // utnij "PRINT:"
+                if (state.variables.count(varName)) {
+                    result += "    mov edx, [" + varName + "]\n";
+                    result += "    lea rcx, [rel fmt]\n";
+                    result += "    call printf\n";
+                }
+            }
+            // 2. BLOK IF (format: "IF:index_w_tablicy")
+            else if (cmd.find("IF:") == 0) {
+                int ifIndex = std::stoi(cmd.substr(3)); // pobierz numer IFa
+
+                if (ifIndex >= 0 && ifIndex < state.ifBlocks.size()) {
+                    const auto& ifBlock = state.ifBlocks[ifIndex];
+
+                    if (state.variables.count(ifBlock.conditionVar)) {
+                        std::string labelSkip = "skip_if_" + std::to_string(ifIndex);
+
+                        result += "    ; --- IF START [" + ifBlock.conditionVar + "] ---\n";
+                        result += "    mov eax, [" + ifBlock.conditionVar + "]\n";
+                        result += "    cmp eax, 1\n";
+                        result += "    jne " + labelSkip + "\n"; // Skocz jeśli FAŁSZ (0)
+
+                        // Generuj zawartość środka IFa
+                        for (const std::string& innerCmd : ifBlock.prints) {
+                            if (state.variables.count(innerCmd)) {
+                                result += "    mov edx, [" + innerCmd + "]\n";
+                                result += "    lea rcx, [rel fmt]\n";
+                                result += "    call printf\n";
+                            }
+                        }
+
+                        result += labelSkip + ":\n";
+                        result += "    ; --- IF END ---\n";
+                    }
+                }
             }
         }
 
@@ -41,11 +73,11 @@ std::string generateAssembly(const CompilerState& state) {
         result += "    ret\n\n";
     }
 
-    // --- FUNKCJA MAIN ---
+    // --- MAIN ---
     result += "main:\n";
     result += "    sub rsp, 40\n";
 
-    // Globalne printy (poza funkcjami)
+    // Globalne printy (bez zmian)
     for (const std::string& var : state.globalPrints) {
         if (state.variables.count(var)) {
             result += "    mov edx, [" + var + "]\n";
@@ -54,26 +86,24 @@ std::string generateAssembly(const CompilerState& state) {
         }
     }
 
-    // Wywołania funkcji zdefiniowanych wcześniej
+    // Wywołania funkcji
     for (const std::string& call : state.printCalls) {
         if (call.find("CALL_") == 0) {
-            std::string funcName = call.substr(5); // Usuń prefiks "CALL_"
+            std::string funcName = call.substr(5);
             if (state.functions.count(funcName)) {
                 result += "    call " + funcName + "\n";
             }
         }
     }
 
-    // --- PAUSE LOOP (Czekanie na ESC) ---
+    // Pauza
     result += "    lea rcx, [rel pause_msg]\n";
     result += "    call printf\n";
     result += "pause_loop:\n";
-    result += "    mov ecx, 27\n"; // VK_ESCAPE
+    result += "    mov ecx, 27\n";
     result += "    call GetAsyncKeyState\n";
     result += "    test ax, 8000h\n";
     result += "    jz pause_loop\n";
-
-    // Wyjście z programu
     result += "    add rsp, 40\n";
     result += "    ret\n";