GEEKING

RobloxStudio/RobloxFindWidget.h

@@ -0,0 +1,184 @@
+/**
+ * FindWidget.h
+ * Copyright (c) 2014 ROBLOX Corp. All Rights Reserved.
+ */
+
+#pragma once
+
+#include <QObject>
+#include <QToolButton>
+#include <QListView>
+#include <QMutex>
+#include <QTextCursor>
+#include <QTextDocument>
+
+#include "LuaSourceBuffer.h"
+#include "Script/Script.h"
+
+class FindData
+{
+public:
+	FindData()
+		: m_line(0)
+		, m_lineText("")
+	{}
+	FindData(QString text)
+		: m_line(0)
+		, m_lineText(text)
+	{}
+	FindData(LuaSourceBuffer instance,
+			 int lineNumber,
+			 QString text,
+			 QString hierarchy = QString())
+		: m_instance(instance)
+		, m_line(lineNumber)
+		, m_lineText(text)
+		, m_hierarchy(hierarchy)
+	{}
+
+	void setPlaceData(int position, int length) { m_position = position; m_length = length; }
+
+	int getPosition() { return m_position; }
+	int getLength() { return m_length; }
+
+	Qt::ItemFlags flags() const;
+
+	QString toString() const;
+
+	LuaSourceBuffer getInstance() { return m_instance; }
+
+	int getLine() { return m_line; }
+
+private:
+	LuaSourceBuffer         m_instance;
+	int						m_line;
+	QString					m_lineText;
+	QString					m_hierarchy;
+
+	int						m_position;
+	int						m_length;
+
+};
+
+Q_DECLARE_METATYPE(FindData)
+
+class FindListModel : public QAbstractListModel
+{
+	Q_OBJECT
+public:
+	FindListModel();
+
+	void addItem(FindData item);
+	void addItems(QList<FindData> items) { m_FindList.append(items); }
+
+	void updateEntireView();
+
+	void clear();
+
+	virtual int rowCount( const QModelIndex& parent = QModelIndex() ) const;
+
+	virtual QVariant data(const QModelIndex& index, int role = Qt::DisplayRole) const;
+	Qt::ItemFlags flags(const QModelIndex& index) const;
+
+public Q_SLOTS:
+	void updateView();
+
+	int openScript(int row);
+
+private:
+
+	mutable QMutex				m_listMutex;
+
+	bool						m_stopUpdater;
+	bool						m_runningUpdater;
+	int							m_loadedIndex;
+
+	int							m_previousCount;
+
+	QList<FindData>				m_FindList;
+};
+
+class RobloxFindWidget : public QWidget
+{
+	Q_OBJECT
+public:
+	RobloxFindWidget(QWidget *pParent = 0);
+	~RobloxFindWidget();
+
+	static RobloxFindWidget& singleton();
+
+	enum eFindFlags
+	{
+		NO_FLAGS   = 1 << 0,
+		MATCH_CASE = 1 << 1,
+		MATCH_WORD = 1 << 2,
+		REG_EXP    = 1 << 3,
+	};
+	inline friend eFindFlags operator|(eFindFlags a, eFindFlags b) { return static_cast<eFindFlags>(static_cast<int>(a) | static_cast<int>(b)); }
+
+	void findAll(const QString& findStr, bool allScripts, eFindFlags flags);
+	void dataModelClosing();
+
+protected:
+	virtual bool event(QEvent* e);
+
+Q_SIGNALS:
+	void populatingWidgetFinished();
+
+protected Q_SLOTS:
+	void openItem(const QModelIndex & index);
+
+	void openCurrentItem();
+	void previousItem();
+	void nextItem();
+	void clearItems();
+
+	void resetItems();
+	void stopPopulatingItems();
+
+	void addItem(FindData item);
+
+private:
+	void initWidget();
+
+	void populateWidget(shared_ptr<RBX::DataModel> dm, const QString& findStr, eFindFlags flags, bool allScripts);
+
+	void populateScriptList(bool allScripts);
+
+	QTextCursor findNextInScript(const QTextDocument& doc, const QString& findStr, const QTextCursor& cursor, QTextDocument::FindFlags flags, bool regExp);
+
+	void getNextScript(int index, shared_ptr<RBX::DataModel> dm, const QString& findStr, eFindFlags flags);
+	void searchScriptChain(int index, const QString& sourceText, shared_ptr<RBX::DataModel> dm, const QString& findStr, eFindFlags flags, const QString& heirarchy);
+
+	QString getInstanceHierarchy(shared_ptr<RBX::Instance> script);
+	
+	enum eDataRole
+	{
+		SCRIPT_ROLE	= Qt::UserRole + 1,
+		POS_ROLE	= Qt::UserRole + 2,
+		LENGTH_ROLE = Qt::UserRole + 3
+	};
+
+	QToolButton *m_pFindInAllScriptsButton;
+	QToolButton *m_pGotoLocationButton;
+	QToolButton *m_pPreviousButton;
+	QToolButton *m_pNextButton;
+	QToolButton *m_pClearButton;
+	QToolButton *m_pStopButton;
+
+	int								m_findListIndex;
+
+	QListView*						m_pListView;
+
+	QList<LuaSourceBuffer>          m_scriptList;
+
+	int								m_matchesFound;
+	int								m_matchingScripts;
+	int								m_searchedScripts;
+
+	int								m_totalAttempted;
+
+	bool							m_stopPopulating;
+
+	FindListModel*					m_pFindListModel;
+};