DADataOperateOfDataFrameWidget.h

#ifndef DADATAOPERATEOFDATAFRAMEWIDGET_H
#define DADATAOPERATEOFDATAFRAMEWIDGET_H
#include <QtWidgets/QWidget>
#include <QUndoStack>
#include <optional>
#include "DAGuiAPI.h"
#include "DAData.h"
#include "DADataPyDataFrame.h"
#include "numpy/DAPyDType.h"
#include "DADataOperatePageWidget.h"
namespace Ui
{
class DADataOperateOfDataFrameWidget;
}
 
namespace DA
{
class DADataTableModel;
class DAAppRibbonArea;
class DADialogDataframeColumnCastToNumeric;
class DADialogDataframeColumnCastToDatetime;
class DADialogDataFrameDataSelect;
class DADialogDataFrameEvalDatas;
class DADialogDataFrameQueryDatas;
class DADialogDataFrameDataSearch;
class DADialogDataFrameSort;
class DADialogCreatePivotTable;
 
class DAGUI_API DADataOperateOfDataFrameWidget : public DADataOperatePageWidget
{
    Q_OBJECT
public:
    virtual int getDataOperatePageType() const;
 
public:
    explicit DADataOperateOfDataFrameWidget(const DAData& d, QWidget* parent = nullptr);
    ~DADataOperateOfDataFrameWidget();
    // 是否存在data
    bool haveData() const;
    // 获取dataframe
    DAPyDataFrame getDataframe() const;
    // 获取Data的引用
    const DAData& data() const;
    // 创建dataframe的数据描述
    DAPyDataFrame createDataDescribe();
    // dataframe表格是否有选中项
    bool isDataframeTableHaveSelection() const;
    // 返回选中的列数，列数不会重复
    QList< int > getSelectedDataframeCoumns(bool ensureInDataframe = true) const;
    QList< int > getSelectedDataframeRows(bool ensureInDataframe = true) const;
    // 返回选中的列/行
    QList< int > getFullySelectedDataframeColumns(bool ensureInDataframe = true) const;
    QList< int > getFullySelectedDataframeRows(bool ensureInDataframe = true) const;
 
    QList< QPoint > getSelectedDataframeCells(bool ensureInDataframe = true) const;
    int getSelectedOneDataframeRow(bool ensureInDataframe = true) const;
    int getSelectedOneDataframeColumn(bool ensureInDataframe = true) const;
 
    // 获取选中的序列，如果用户打开一个表格，选中了其中一列，那么将返回那一列pd.Series作为数据，如果用户选中了多列，那么每列作为一个DAData并组成list返回
    QList< DAData > getSlectedSeries() const;
    // 刷新表格
    void refreshTable();
    // 确保列可见
    void ensureColumnVisible(const QString& colName, bool selectCol = true);
public Q_SLOTS:
    void setDAData(const DA::DAData& d);
    // 在选中行后面插入行
    void insertRowAboveBySelect();
    void insertRowBelowBySelect();
    void insertRowAt(int row);
    void insertColumnRightBySelect();
    void insertColumnLeftBySelect();
    void insertColumnAt(int col);
    // 移除选中的行
    int removeSelectRow();
    // 移除选择的列
    int removeSelectColumn();
    // 设置选中单元格为nan,返回设置成功的个数
    int removeSelectCell();
    // 列更名
    void renameColumns();
    // 设置选择列的数据类型，发射信号columnTypeChanged
    bool changeSelectColumnType(const DAPyDType& dt);
    // 把选择的行转换为数值，带交互
    void castSelectToNum();
    void castSelectToDatetime();
    // 把选择的列转换为索引
    bool changeSelectColumnToIndex();
    // 过滤给定条件外的数据
    bool queryDatas();
    bool queryDatas(const DAPyDataFrame& df, const QString& exper);
    // 检索给定的数据
    bool searchData();
    QList< QPair< int, int > > searchData(const DAPyDataFrame& df, const QString& exper) const;
    // 列运算
    bool evalDatas();
    bool evalDatas(const DAPyDataFrame& df, const QString& exper);
    // 过滤给定条件外的数据
    bool filterByColumn();
    bool filterByColumn(const DAPyDataFrame& df, double lower = 0.0, double upper = 0.0, const QString& index = QString());
    // 数据排序
    bool sortDatas();
    bool sortDatas(const DAPyDataFrame& df, const QString& by, const bool ascending);
    // 创建数据透视表
    DAPyDataFrame createPivotTable();
    DAPyDataFrame createPivotTable(const DAPyDataFrame& df,
                                   const QStringList value    = QStringList(),
                                   const QStringList index    = QStringList(),
                                   const QStringList columns  = QStringList(),
                                   const QString& aggfunc     = QStringLiteral("mean"),
                                   bool margins               = false,
                                   const QString& marginsName = QStringLiteral("All"),
                                   bool sort                  = false);
Q_SIGNALS:
    void selectTypeChanged(const QList< int >& column, DA::DAPyDType dt);
private Q_SLOTS:
    // 表格点击
    void onTableViewClicked(const QModelIndex& index);
 
protected:
    void changeEvent(QEvent* e);
 
private:
    Ui::DADataOperateOfDataFrameWidget* ui;
    DAData mData;
    DADataTableModel* mModel { nullptr };
 
    DADialogDataframeColumnCastToNumeric* mDialogCastNumArgs { nullptr };
    DADialogDataframeColumnCastToDatetime* mDialogCastDatetimeArgs { nullptr };
    DADialogDataFrameDataSelect* mDialogDataFrameDataSelect { nullptr };
    DADialogCreatePivotTable* mDialogCreatePivotTable { nullptr };
    DADialogDataFrameEvalDatas* mDialogDataFrameEvalDatas { nullptr };
    DADialogDataFrameQueryDatas* mDialogDataFrameQueryDatas { nullptr };
    DADialogDataFrameDataSearch* mDialogDataFrameDataSearch { nullptr };
    DADialogDataFrameSort* mDADialogDataFrameSort { nullptr };
};
}  // end of namespace DA
#endif  // DADATAOPERATEOFDATAFRAMEWIDGET_H