RMG/Source/RMG/UserInterface/RomBrowserWidget.cpp

/*
 * Rosalie's Mupen GUI - https://github.com/Rosalie241/RMG
 *  Copyright (C) 2020 Rosalie Wanders <rosalie@mailbox.org>
 *
 *  This program is free software: you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License version 3.
 *  You should have received a copy of the GNU General Public License
 *  along with this program. If not, see <https://www.gnu.org/licenses/>.
*/
#include "RomBrowserWidget.hpp"
#include "../Globals.hpp"
#include "../Config.hpp"

#include <QDir>

using namespace UserInterface;

RomBrowserWidget::RomBrowserWidget(void) : QTableView(nullptr)
{
    this->model_Init();
    this->model_Setup();
    this->widget_Init();
}

RomBrowserWidget::~RomBrowserWidget()
{
}

void RomBrowserWidget::RefreshRomList(void)
{
    this->model_Setup();
}

void RomBrowserWidget::model_Init(void)
{
    this->model_Model = new QStandardItemModel(this);
}

void RomBrowserWidget::model_Setup(void)
{
    if (this->rom_List_Fill_Thread_Running)
        return;

    this->rom_List_Index = 0;
    this->rom_List_Recursive = false;
    this->model_Model->clear();

    this->rom_List_Fill("/home/rosalie/Downloads/n64_roms/");

    this->model_LabelList_Setup();

    this->model_Model->setColumnCount(this->model_LabelList.size());
    this->model_Model->setHorizontalHeaderLabels(this->model_LabelList);
}

void RomBrowserWidget::model_LabelList_Setup(void)
{
    this->model_LabelList.clear();
    this->model_LabelList.append("Name");
    this->model_LabelList.append("Internal Name");
    this->model_LabelList.append("MD5");
}

void RomBrowserWidget::widget_Init(void)
{
    this->widget_Delegate = new NoFocusDelegate();

    this->setModel(this->model_Model);
    this->setItemDelegate(this->widget_Delegate);
    this->setWordWrap(false);
    this->setShowGrid(false);
    this->setSortingEnabled(true);
    this->setEditTriggers(QAbstractItemView::NoEditTriggers);
    this->setSelectionBehavior(QTableView::SelectRows);

    this->verticalHeader()->hide();

    this->horizontalHeader()->setSectionResizeMode(QHeaderView::Interactive);
    this->horizontalHeader()->setDefaultAlignment(Qt::AlignLeft);
    this->horizontalHeader()->setSortIndicatorShown(false);
    this->horizontalHeader()->setHighlightSections(false);

    this->verticalHeader()->setSectionResizeMode(QHeaderView::ResizeToContents);
    
    this->column_SetSize();
}

void RomBrowserWidget::rom_Searcher_Init(void)
{
    this->rom_Searcher_Thread = new Thread::RomSearcherThread();
    this->rom_Searcher_Thread->SetMaximumFiles(APP_ROMSEARCHER_MAX);

    // TODO
    this->rom_Searcher_Thread->SetRecursive(true);

    connect(rom_Searcher_Thread, &Thread::RomSearcherThread::on_Rom_Found, this, &RomBrowserWidget::on_RomBrowserThread_Received);
    connect(rom_Searcher_Thread, &Thread::RomSearcherThread::finished, this, &RomBrowserWidget::on_RomBrowserThread_Finished);
}

bool init = false;

void RomBrowserWidget::rom_List_Fill(QString directory)
{
    if (!init)
    {
        this->rom_Searcher_Init();
        init = true;
    }

    if (this->rom_List_Fill_Thread_Running)
        return;

    this->rom_List_Fill_Thread_Running = true;
    this->rom_Searcher_Thread->SetDirectory(directory);
    this->rom_Searcher_Thread->start();
}

void RomBrowserWidget::column_SetSize(void)
{
    for (int i = 0; i < this->model_LabelList.size(); i++)
    {
        QString label = this->model_LabelList.at(i);

        int oldSize = this->columnWidth(i);
        int newSize = 0;
        if (label == "Name")
        {
            newSize = 250;
        } else if (label == "Internal Name")
        {
            newSize = 100;
        } else if (label == "MD5")
        {
            newSize = 100;
        }

        this->setColumnWidth(i, newSize);
    continue;
        if (oldSize != newSize)
        {
            this->setColumnWidth(i, oldSize);
        }
        else
        {
            this->setColumnWidth(i, newSize);
        }
    }
}

void RomBrowserWidget::on_RomBrowserThread_Received(M64P::Wrapper::RomInfo_t romInfo)
{
    QList<QStandardItem*> list;

    list.append(new QStandardItem(romInfo.Settings.goodname));
    list.append(new QStandardItem(QString((char*)romInfo.Header.Name)));
    list.append(new QStandardItem(romInfo.Settings.MD5));

    this->model_Model->appendRow(list);

    this->column_SetSize();

    this->horizontalHeader()->setStretchLastSection(true);
}

void RomBrowserWidget::on_RomBrowserThread_Finished(void)
{
    this->rom_List_Fill_Thread_Running = false;
}