package com.algo;

import com.algo.model.AGVStatus;
import com.algo.model.TaskAssignment;
import com.algo.model.TaskData;
import com.algo.service.PathPlanningService;
import com.algo.service.TaskAllocationService;
import com.algo.util.AgvTaskUtils;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.test.context.junit4.SpringRunner;

import java.util.List;

@SpringBootTest
@RunWith(SpringRunner.class)
public class AlgorithmMain {


    @Autowired
    private TaskAllocationService taskAllocationService;

    @Autowired
    private PathPlanningService pathPlanningService;


    /**
     * 任务分配测试
     */
    @Test
    public void taskAllocationService() {


        List<AGVStatus> agvStatusList = AgvTaskUtils.loadAgvStatus("ctu_agv_status.json");
        System.out.println(agvStatusList);
        List<TaskData> taskList = AgvTaskUtils.loadTaskList("ctu_task_data.json");
        System.out.println(taskList);

        System.out.println("执行任务分配方法：");
        List<TaskAssignment> taskAssignments = taskAllocationService.allocateTasks(agvStatusList, taskList);
        System.out.println(taskAssignments);
        System.out.println("验证任务分配结果：");
        boolean validateAssignments = taskAllocationService.validateAssignments(taskAssignments);
        System.out.println(validateAssignments);
    }

    /**
     * 路径优化测试
     */
    @Test
    public void pathPlanningService() {

        List<TaskData> taskList = AgvTaskUtils.loadTaskList("ctu_task_data.json");
        System.out.println(taskList);

        List<AGVStatus> agvStatusList = AgvTaskUtils.loadAgvStatus("ctu_agv_status.json");
        System.out.println(agvStatusList);
        PathPlanningService.PathPlanningResult planningResult = pathPlanningService.planAllAgvPaths(taskList, agvStatusList, true, null);
        System.out.println(planningResult);
    }

}