#pragma once
#include <QStringList>
#include <QWidget>
#include <vector>
#include "interact/SlicePlaneMath.hpp"  // geopro::render::interact::SliceAxis
#include "repo/CategoryConfig.hpp"
#include "repo/RepoTypes.hpp"

class QTreeWidget;
class QComboBox;
class QDateEdit;
class QLabel;
class QToolButton;
class QWidget;

namespace geopro::data {
class DatasetFieldDictionary;
}

namespace geopro::app {

class DateRangeEdit;

// 单个数据类型大类段（spec §7）：段头（标题/折叠 + 装置类型/日期筛选 + 「+新增三维体」）+ 段体（可勾选数据树）。
// 勾选数据行 = 渲染（帘面/体素/切片）；段头生成按钮据当前勾选源发 generateVolumeRequested。
class CategorySection : public QWidget {
    Q_OBJECT
public:
    CategorySection(const CategorySpec& spec, geopro::data::DatasetFieldDictionary* dict,
                    QWidget* parent = nullptr);

    // 对象树同源的扁平 GS/TM 节点（段体容器分层用；Task 12 接入真实结构，当前仅存储）。
    void setStructure(const std::vector<geopro::data::StructNode>& nodes);
    void setDatasets(const std::vector<geopro::data::DsRow>& rows);
    void setChecked(const QString& dsId, bool on);  // 按 dsId 勾选/取消(新建切片自动勾选等场景)
    QStringList checkedIds() const { return checkedDsIds(); }  // 当前勾选 ds（异常显隐同步用）
    const CategorySpec& spec() const { return spec_; }

signals:
    void checkedDatasetsChanged(const QStringList& dsIds);                                  // 数据行勾选=渲染
    void generateVolumeRequested(const QString& dsTypeCode, const QStringList& sourceDsIds);  // 段头「+新增三维体」
    void detailRequested(const QString& dsId, const QString& ddCode, const QString& name);  // 双击/右键=详情
    void deleteDatasetRequested(const QString& dsId, const QString& ddCode);                // 右键删除（切片/异常）
    // ── 三维体段右键操作（迁自旧 Column3DAnalysis，全接）──
    void sliceRequested(geopro::render::interact::SliceAxis axis);  // 体→生成切片（轴）
    void colorScaleRequested(const QString& dsId);                  // 体/切片→色阶
    void sliceSaveRequested(const QString& dsId);                   // 切片→保存位姿
    void sliceSaveAsRequested(const QString& dsId);                 // 切片→另存
    void sliceExportImageRequested(const QString& dsId);           // 切片→导出图片
    void sliceExportDatRequested(const QString& dsId);             // 切片→导出 dat
    void anomalyVisibilityChanged(const QString& dsId, bool vis);  // 异常→显示/隐藏

private:
    void showContextMenu(const QPoint& pos);  // 段体树右键菜单（详情 + 删除）
    void rebuildList();   // 据 rows_（经装置/日期筛选）重建段体树并复原勾选
    void refreshArrayCombo();  // 据当前 rows_ 重填装置类型下拉项（经字典 value→中文）
    void emitChecked();   // 收集勾选 → checkedDatasetsChanged
    QStringList checkedDsIds() const;
    bool passesFilters(const geopro::data::DsRow& row) const;  // 装置类型 + 采集时间范围

    CategorySpec spec_;
    geopro::data::DatasetFieldDictionary* dict_ = nullptr;
    std::vector<geopro::data::DsRow> rows_;
    std::vector<geopro::data::StructNode> structure_;

    QToolButton* header_ = nullptr;   // 折叠头（标题 + 箭头）
    QWidget* body_ = nullptr;         // 段体容器（折叠时隐藏）
    QComboBox* arrayCombo_ = nullptr; // 装置类型筛选（仅 hasArrayTypeFilter）
    DateRangeEdit* dateRange_ = nullptr;  // 采集时间范围筛选（起止双日历，可清空）
    QTreeWidget* list_ = nullptr;
};

}  // namespace geopro::app