mirror of
https://github.com/rizinorg/cutter.git
synced 2024-12-22 12:56:11 +00:00
132 lines
3.4 KiB
C++
132 lines
3.4 KiB
C++
#ifndef TYPESWIDGET_H
|
|
#define TYPESWIDGET_H
|
|
|
|
#include <memory>
|
|
|
|
#include "core/Cutter.h"
|
|
#include "CutterDockWidget.h"
|
|
#include "CutterTreeWidget.h"
|
|
|
|
#include <QAbstractListModel>
|
|
#include <QSortFilterProxyModel>
|
|
|
|
class MainWindow;
|
|
class QTreeWidget;
|
|
class TypesWidget;
|
|
|
|
namespace Ui {
|
|
class TypesWidget;
|
|
}
|
|
|
|
|
|
class MainWindow;
|
|
class QTreeWidgetItem;
|
|
|
|
|
|
class TypesModel: public QAbstractListModel
|
|
{
|
|
Q_OBJECT
|
|
|
|
friend TypesWidget;
|
|
|
|
private:
|
|
QList<TypeDescription> *types;
|
|
|
|
public:
|
|
enum Columns { TYPE = 0, SIZE, FORMAT, CATEGORY, COUNT };
|
|
static const int TypeDescriptionRole = Qt::UserRole;
|
|
|
|
TypesModel(QList<TypeDescription> *types, QObject *parent = nullptr);
|
|
|
|
int rowCount(const QModelIndex &parent = QModelIndex()) const override;
|
|
int columnCount(const QModelIndex &parent = QModelIndex()) const override;
|
|
|
|
QVariant data(const QModelIndex &index, int role) const override;
|
|
QVariant headerData(int section, Qt::Orientation orientation, int role = Qt::DisplayRole) const override;
|
|
|
|
bool removeRows(int row, int count, const QModelIndex &parent = QModelIndex()) override;
|
|
};
|
|
|
|
|
|
|
|
class TypesSortFilterProxyModel : public QSortFilterProxyModel
|
|
{
|
|
Q_OBJECT
|
|
|
|
friend TypesWidget;
|
|
|
|
public:
|
|
TypesSortFilterProxyModel(TypesModel *source_model, QObject *parent = nullptr);
|
|
|
|
protected:
|
|
bool filterAcceptsRow(int row, const QModelIndex &parent) const override;
|
|
bool lessThan(const QModelIndex &left, const QModelIndex &right) const override;
|
|
|
|
QString selectedCategory;
|
|
};
|
|
|
|
|
|
|
|
class TypesWidget : public CutterDockWidget
|
|
{
|
|
Q_OBJECT
|
|
|
|
public:
|
|
explicit TypesWidget(MainWindow *main, QAction *action = nullptr);
|
|
~TypesWidget();
|
|
|
|
private slots:
|
|
void refreshTypes();
|
|
|
|
/**
|
|
* @brief Show custom context menu
|
|
* @param pt Position of the place where the right mouse button was clicked
|
|
*/
|
|
void showTypesContextMenu(const QPoint &pt);
|
|
|
|
/**
|
|
* @brief Executed on clicking the Export Types option in the context menu
|
|
* It shows the user a file dialog box to select a file where the types
|
|
* will be exported. It uses the "tc" command of radare2 to export the types.
|
|
*/
|
|
void on_actionExport_Types_triggered();
|
|
|
|
/**
|
|
* @brief Executed on clicking the Load New types option in the context menu
|
|
* It will open the LoadNewTypesDialog where the user can either enter the
|
|
* types manually, or can select a file from where the types will be loaded
|
|
*/
|
|
void on_actionLoad_New_Types_triggered();
|
|
|
|
/**
|
|
* @brief Executed on clicking the Delete Type option in the context menu
|
|
* Upon confirmation from the user, it will delete the selected type.
|
|
*/
|
|
void on_actionDelete_Type_triggered();
|
|
|
|
/**
|
|
* @brief Executed on clicking the Link To Address option in the context menu
|
|
* Opens the LinkTypeDialog box from where the user can link a address to a type
|
|
*/
|
|
void on_actionLink_Type_To_Address_triggered();
|
|
|
|
private:
|
|
std::unique_ptr<Ui::TypesWidget> ui;
|
|
|
|
TypesModel *types_model;
|
|
TypesSortFilterProxyModel *types_proxy_model;
|
|
QList<TypeDescription> types;
|
|
CutterTreeWidget *tree;
|
|
|
|
void setScrollMode();
|
|
|
|
/**
|
|
* @brief Sets the contents of the ComboBox to the supplied contents
|
|
* @param categories The list of categories which has to be added to the ComboBox
|
|
*/
|
|
void refreshCategoryCombo(const QStringList &categories);
|
|
};
|
|
|
|
|
|
#endif // TYPESWIDGET_H
|