DYTSrouce/src/ui/ModelBrowser/PresetModelPanel.cpp

#include "ui/ModelBrowser/PresetModelPanel.h"

#include <QApplication>
#include <QDebug>
#include <QListWidgetItem>
#include <QMimeData>
#include <QDrag>
#include <QPixmap>
#include <QPainter>
#include <QVBoxLayout>
#include <QMetaType>
#include <QHBoxLayout>
#include <QLabel>
#include <QIcon>
#include <QMouseEvent>
#include <QFontMetrics>

#include "common/RecourceHelper.h"
#include "common/SpdLogger.h"
#include "ui/DockTitleBar.h"
#include "ui/DockWidget.h"

#include "ui_PresetModelPanel.h"


PresetModelPanel::PresetModelPanel(QWidget *parent)
    : QWidget(parent)
    , ui(new Ui::PresetModelPanel)
    , dragSourceWidget_(nullptr)
{
    ui->setupUi(this);
    InitUI();
}

PresetModelPanel::~PresetModelPanel()
{
    delete ui;
}

void PresetModelPanel::AttachDock(DockWidget* dockWidget)
{
    if (nullptr == dockWidget) {
        qDebug() << __FUNCTION__ << "dockwidget is nullptr";
        return;
    }

    dockWidget->SetDockWidgetTitleBar(nullptr);
    dockWidget->setWidget(this);

    DockTitleBar* dockTitleBar = new DockTitleBar;
    dockWidget->SetDockWidgetTitleBar(dockTitleBar);
}

void PresetModelPanel::InitUI()
{
    // Try to load model data from configuration files
    QString presetsPath = QString("%1/presets").arg(RecourceHelper::Get().GetResourcesPath());
    if (PresetModelConfigParser::Get().LoadAllConfigs(presetsPath)) {
        // Successfully loaded from configuration files
        LoadModelsFromConfig();
    } else {
        // Failed to load configuration files, use hardcoded defaults as fallback
        LOG_WARN("Failed to load preset models from config files, using hardcoded defaults. Error: {}", 
                 PresetModelConfigParser::Get().GetLastError().toStdString());
        LoadDefaultModels();
    }
}

void PresetModelPanel::LoadModelsFromConfig() {
    // Load ship models
    ModelCategory shipCategory = PresetModelConfigParser::Get().GetCategory("Ship");
    if (!shipCategory.models.isEmpty()) {
        ui->shipList->setModelType("Ship");
        ui->shipList->setModelList(shipCategory.models);
    }
    
    // Load satellite models
    ModelCategory satelliteCategory = PresetModelConfigParser::Get().GetCategory("Satellite");
    if (!satelliteCategory.models.isEmpty()) {
        ui->satelliteList->setModelType("Satellite");
        ui->satelliteList->setModelList(satelliteCategory.models);
    }
    
    // Load missile models
    ModelCategory missileCategory = PresetModelConfigParser::Get().GetCategory("Missile");
    if (!missileCategory.models.isEmpty()) {
        ui->missileList->setModelType("Missile");
        ui->missileList->setModelList(missileCategory.models);
    }
    
    // Load jammer models
    ModelCategory jammerCategory = PresetModelConfigParser::Get().GetCategory("Jammer");
    if (!jammerCategory.models.isEmpty()) {
        ui->jammerList->setModelType("Jammer");
        ui->jammerList->setModelList(jammerCategory.models);
    }
}

void PresetModelPanel::LoadDefaultModels() {
    // Use hardcoded default models as fallback mechanism
    QVector<ModelInfo> shipModels = {
        {"destroyer", "Destroyer", "Naval destroyer vessel", "", ":/icons/ship.png", true},
        {"aircraft_carrier", "Aircraft Carrier", "Large naval vessel", "", ":/icons/ship.png", true},
        {"frigate", "Frigate", "Fast naval vessel", "", ":/icons/ship.png", true},
        {"submarine", "Submarine", "Underwater vessel", "", ":/icons/ship.png", true}
    };
    ui->shipList->setModelType("Ship");
    ui->shipList->setModelList(shipModels);
    
    QVector<ModelInfo> satelliteModels = {
        {"comm_satellite", "Communication Satellite", "Communication satellite", "", ":/icons/satellite.png", true},
        {"weather_satellite", "Weather Satellite", "Weather monitoring satellite", "", ":/icons/satellite.png", true},
        {"gps_satellite", "GPS Satellite", "GPS navigation satellite", "", ":/icons/satellite.png", true},
        {"spy_satellite", "Spy Satellite", "Reconnaissance satellite", "", ":/icons/satellite.png", true}
    };
    ui->satelliteList->setModelType("Satellite");
    ui->satelliteList->setModelList(satelliteModels);
    
    QVector<ModelInfo> missileModels = {
        {"cruise_missile", "Cruise Missile", "Long-range cruise missile", "", ":/icons/missile.png", true},
        {"ballistic_missile", "Ballistic Missile", "Ballistic missile", "", ":/icons/missile.png", true},
        {"anti_ship_missile", "Anti-Ship Missile", "Anti-ship missile", "", ":/icons/missile.png", true},
        {"sam_missile", "Surface-to-Air Missile", "Surface-to-air missile", "", ":/icons/missile.png", true}
    };
    ui->missileList->setModelType("Missile");
    ui->missileList->setModelList(missileModels);
    
    QVector<ModelInfo> jammerModels = {
        {"electronic_jammer", "Electronic Jammer", "Electronic warfare jammer", "", ":/icons/jammer.png", true},
        {"comm_jammer", "Communication Jammer", "Communication jammer", "", ":/icons/jammer.png", true},
        {"radar_jammer", "Radar Jammer", "Radar jamming device", "", ":/icons/jammer.png", true},
        {"gps_jammer", "GPS Jammer", "GPS jamming device", "", ":/icons/jammer.png", true}
    };
    ui->jammerList->setModelType("Jammer");
    ui->jammerList->setModelList(jammerModels);
}