using System;
using TheSingularityWorkshop.FSM_API;

// 1. Định Nghĩa Lớp Ngữ Cảnh
// Lớp này chứa tất cả thông tin thời gian thực mà FSM cần để đưa ra quyết định.
public class CarContext : IStateContext
{
    public float Speed { get; set; } = 0.0f;
    public bool BrakePedalPressed { get; set; } = false;
    public bool ThrottleDown { get; set; } = false;
    public bool EngineFault { get; set; } = false;
}

// 2. Định Nghĩa FSM
// Lớp này chứa bản thiết kế cho toàn bộ hệ thống hộp số ô tô.
public static class CarTransmissionFSM
{
    public static void CreateComplexFSM()
    {
        FSM_API.Create.CreateFiniteStateMachine("CarFSM", -1, "VehicleUpdate")
            // --- Định Nghĩa Các Trạng Thái Cấp Cao ---
            .State("Park",
                onEnter: (ctx) => Console.WriteLine("Hộp số đang ở chế độ Đỗ xe."),
                onUpdate: (ctx) => { /* Kiểm tra đầu vào */ },
                onExit: (ctx) => Console.WriteLine("Rời chế độ Đỗ xe."))
            .State("Reverse",
                onEnter: (ctx) => Console.WriteLine("Hộp số đang ở chế độ Lùi."),
                onUpdate: (ctx) => { /* Logic lùi */ },
                onExit: (ctx) => Console.WriteLine("Rời chế độ Lùi."))
            .State("Neutral",
                onEnter: (ctx) => Console.WriteLine("Hộp số đang ở chế độ Trung lập."),
                onUpdate: (ctx) => { /* Logic động cơ ở chế độ nhàn rỗi */ },
                onExit: (ctx) => Console.WriteLine("Rời chế độ Trung lập."))
            .State("Fault",
                onEnter: (ctx) =>
                {
                    Console.WriteLine("LỖI NGHIÊM TRỌNG: Chuyển sang trạng thái Lỗi.");
                    // Mô phỏng hành động an toàn nghiêm trọng: vô hiệu hóa động cơ
                    ((CarContext)ctx).EngineFault = true;
                },
                onUpdate: (ctx) => Console.WriteLine("Đang ở chế độ Lỗi. Không thể tiếp tục."),
                onExit: (ctx) => Console.WriteLine("Lỗi đã được đặt lại thủ công."))

            // --- Định Nghĩa Các Trạng Thái Phụ Khi Lái (Số) ---
            .State("First_Gear",
                onEnter: (ctx) => Console.WriteLine("Chuyển vào Số Một."),
                onUpdate: (ctx) => { /* Áp dụng mô-men xoắn cho số một */ },
                onExit: (ctx) => { /* Chuẩn bị chuyển lên */ })
            .State("Second_Gear",
                onEnter: (ctx) => Console.WriteLine("Chuyển vào Số Hai."),
                onUpdate: (ctx) => { /* Áp dụng mô-men xoắn cho số hai */ },
                onExit: (ctx) => { /* Chuẩn bị chuyển số */ })
            .State("Third_Gear",
                onEnter: (ctx) => Console.WriteLine("Chuyển vào Số Ba."),
                onUpdate: (ctx) => { /* Áp dụng mô-men xoắn cho số ba */ },
                onExit: (ctx) => { /* Chuẩn bị chuyển số */ })
            .State("Fourth_Gear",
                onEnter: (ctx) => Console.WriteLine("Chuyển vào Số Bốn."),
                onUpdate: (ctx) => { /* Áp dụng mô-men xoắn cho số bốn */ },
                onExit: (ctx) => { /* Chuẩn bị chuyển số */ })

            // --- Đặt Trạng Thái Ban Đầu ---
            .WithInitialState("Park")

            // --- Định Nghĩa Các Chuyển Tiếp ---
            // Các chuyển tiếp cấp cao (P, R, N, D)
            .Transition("Park", "Reverse", (ctx) => ((CarContext)ctx).BrakePedalPressed)
            .Transition("Park", "First_Gear", (ctx) => ((CarContext)ctx).BrakePedalPressed)
            .Transition("Reverse", "Park", (ctx) => !((CarContext)ctx).BrakePedalPressed)
            .Transition("Reverse", "Neutral", (ctx) => true) // Luôn có thể chuyển từ R sang N
            .Transition("Neutral", "Park", (ctx) => true)
            .Transition("Neutral", "Reverse", (ctx) => ((CarContext)ctx).BrakePedalPressed)
            .Transition("Neutral", "First_Gear", (ctx) => ((CarContext)ctx).BrakePedalPressed)
            // Một chuyển tiếp từ bất kỳ số lái nào về Trung lập hoặc Đỗ xe
            .AnyTransition("Neutral", (ctx) => !((CarContext)ctx).BrakePedalPressed && ((CarContext)ctx).Speed < 1.0f)
            .AnyTransition("Park", (ctx) => !((CarContext)ctx).BrakePedalPressed && ((CarContext)ctx).Speed < 1.0f)

            // Chuyển tiếp trạng thái phụ khi lái (Chuyển số)
            .Transition("First_Gear", "Second_Gear", (ctx) => ((CarContext)ctx).Speed > 15.0f && !((CarContext)ctx).ThrottleDown)
            .Transition("Second_Gear", "Third_Gear", (ctx) => ((CarContext)ctx).Speed > 30.0f)
            .Transition("Third_Gear", "Fourth_Gear", (ctx) => ((CarContext)ctx).Speed > 45.0f)

            // Chuyển tiếp xuống số
            .Transition("Fourth_Gear", "Third_Gear", (ctx) => ((CarContext)ctx).Speed < 40.0f)
            .Transition("Third_Gear", "Second_Gear", (ctx) => ((CarContext)ctx).Speed < 25.0f)
            .Transition("Second_Gear", "First_Gear", (ctx) => ((CarContext)ctx).Speed < 10.0f || ((CarContext)ctx).ThrottleDown)

            // Chuyển tiếp "Bất kỳ trạng thái" để xử lý an toàn và lỗi
            .AnyTransition("Fault", (ctx) => ((CarContext)ctx).EngineFault)
            .AnyTransition("Park", (ctx) => ((CarContext)ctx).BrakePedalPressed && ((CarContext)ctx).Speed < 1.0f)

            // Hoàn tất bản thiết kế FSM
            .BuildDefinition();
    }
}