VX5_KRO_GearPump/scene/ScheduleShiftsEditor.cpp

#include "ScheduleShiftsEditor.h"
#include "ui_ScheduleShiftsEditor.h"
#include <QMouseEvent>
#include <QTimer>
#include <QModelIndex>
#include "PicScheduleShiftItem.h"
#include <QGridLayout>
#include <QDebug>
#include <QTableWidget>
#include <algorithm>
#include <QScrollBar>

ScheduleShiftsEditor::ScheduleShiftsEditor(QWidget *parent) :
    QWidget(parent),
    ui(new Ui::ScheduleShiftsEditor)
{
    ui->setupUi(this);

    m_generalSettings = new ItemGeneralSettings();
    QGridLayout* layout = new QGridLayout();
    layout->setContentsMargins(0, 0, 0, 0);
    layout->setMargin(0);
    layout->addWidget(m_generalSettings);
    ui->generalSettingsTab->setLayout(layout);

    // 禁用编辑
    ui->tableSchedules->setEditTriggers(QTableWidget::NoEditTriggers);

    ui->tableSchedules->setColumnWidth(0, 50);
    ui->tableSchedules->setColumnWidth(1, 100);
    ui->tableSchedules->setColumnWidth(2, 340);
    ui->tableSchedules->horizontalHeader()->setDefaultAlignment(Qt::AlignLeft);
    ui->tableSchedules->horizontalHeader()->setVisible(true);
    ui->tableSchedules->setHorizontalScrollBarPolicy(Qt::ScrollBarAlwaysOff);
    ui->tableSchedules->setVerticalScrollBarPolicy(Qt::ScrollBarAsNeeded);
    ui->tableSchedules->verticalScrollBar()->scroll(0, 0);
    // ui->tableSchedules->setModel(m_viewModel);

    connect(ui->tableSchedules, &QTableWidget::itemSelectionChanged, this, &ScheduleShiftsEditor::onTableSelectionChanged);
    connect(ui->tableSchedules, &QTableWidget::currentItemChanged, this, [this](QTableWidgetItem*, QTableWidgetItem*){onTableSelectionChanged();});
    connect(ui->tableSchedules, &QTableWidget::currentCellChanged, this, [this](int, int, int, int){onTableSelectionChanged();});
    connect(ui->LineEditScheduleName, &QLineEdit::textChanged, this, &ScheduleShiftsEditor::btnAddEnbCheck);
    connect(ui->timeEdit, &QTimeEdit::timeChanged, this, [this](const QTime&) {btnAddEnbCheck("");});

    ui->btnAdd->setEnabled(false);
    ui->btnRemove->setEnabled(false);
}

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

void ScheduleShiftsEditor::edit(PicScheduleShiftItem* item)
{
    m_backup.clear();
    m_generalSettings->edit(item);
    if(item) {
        item->saveAttrs(m_backup);
        m_dataModel = item->schedules();
    }
    else {
        m_dataModel.clear();
    }

    applyDataModel();
}

const QVector<ScheduleModel>& ScheduleShiftsEditor::dataModel() const
{
    return m_dataModel;
}

PicScheduleShiftItem *ScheduleShiftsEditor::currEditItem()
{
    return dynamic_cast<PicScheduleShiftItem*>(m_generalSettings->currentItem());
}

void ScheduleShiftsEditor::onBtnCloseClicked()
{
    if(currEditItem()) {
        currEditItem()->setAttrs(m_backup);
    }

    emit rejected();
}

void ScheduleShiftsEditor::onBtnAddClicked()
{
    if(m_dataModel.size() >= 24) {
        qWarning() << "The schedule shift has already reached the maximum number of 24 schedules.";
        return;
    }

    auto t = ui->timeEdit->time();
    auto b = std::binary_search(m_dataModel.begin(), m_dataModel.end(), ScheduleModel{t, QString()});
    if(b) {
        qWarning() << "The timestamp has already exist. " << t.toString("hh:mm");
        return;
    }

    if(ui->LineEditScheduleName->text().isEmpty()) {
        qWarning() << "The schedule name is empty.";
        return;
    }

    ScheduleModel model;
    model.startTime = t;
    model.name = ui->LineEditScheduleName->text();

    auto it = std::upper_bound(m_dataModel.begin(), m_dataModel.end(), model);
    m_dataModel.insert(it, model);
    qDebug() << "Schedule time: " << model.startTime.toString("hh:mm") << ", name: " << model.name << ", has added.";

    applyDataModel();
    ui->LineEditScheduleName->clear();
    ui->btnRemove->setEnabled(false);
}

void ScheduleShiftsEditor::onBtnRemoveClicked()
{
    if(ui->tableSchedules->currentRow() >= 0 && ui->tableSchedules->currentRow() < m_dataModel.size()) {
        m_dataModel.remove(ui->tableSchedules->currentRow());
        applyDataModel();
    }
}

void ScheduleShiftsEditor::onTableSelectionChanged()
{
    ui->btnRemove->setEnabled(ui->tableSchedules->currentRow() != -1);
}

void ScheduleShiftsEditor::btnAddEnbCheck(const QString &txt)
{
    Q_UNUSED(txt);
    ScheduleModel model {ui->timeEdit->time(), ""};
    ui->btnAdd->setEnabled(!std::binary_search(m_dataModel.begin(), m_dataModel.end(), model) && !ui->LineEditScheduleName->text().isEmpty());
}

void ScheduleShiftsEditor::onBtnOkClicked()
{
    emit accepted();
}

void ScheduleShiftsEditor::applyDataModel()
{
    auto b = ui->tableSchedules->blockSignals(true);
    ui->tableSchedules->clearContents();
    ui->tableSchedules->setRowCount(m_dataModel.size());
    for(int i = 0; i < m_dataModel.size(); ++i) {
        QString time = m_dataModel[i].startTime.toString("hh:mm");
        ui->tableSchedules->setItem(i, 0, new QTableWidgetItem(QString::number(i + 1)));
        ui->tableSchedules->setItem(i, 1, new QTableWidgetItem(time));
        ui->tableSchedules->setItem(i, 2, new QTableWidgetItem(m_dataModel[i].name));
    }

    ui->tableSchedules->blockSignals(b);
    
    auto txt = QString::number(24 - m_dataModel.size());
    ui->LabNbRemains->setText(QString::number(24 - m_dataModel.size()));
}

void ScheduleShiftsEditor::applyStyleSheet()
{
    // TODO:: 应用样式表
}

SelectedStyleTimeEdit::SelectedStyleTimeEdit(QWidget *parent)
    :QTimeEdit(parent)
{
    lineEdit()->installEventFilter(this);
}

SelectedStyleTimeEdit::~SelectedStyleTimeEdit()
{  
    lineEdit()->removeEventFilter(this);
}

bool SelectedStyleTimeEdit::eventFilter(QObject *obj, QEvent *event)
{
    if (obj == lineEdit() && event->type() == QEvent::MouseButtonPress) {
        // 先让父类处理（这会让内部状态更新）
        bool result = QTimeEdit::eventFilter(obj, event);

        // 延迟执行选中，确保currentSection已经更新
        QTimer::singleShot(0, this, [this]() {
            QDateTimeEdit::Section section = currentSection();
            setSelectedSection(section);
        });

        return result;
    }

    return QTimeEdit::eventFilter(obj, event);
}