mirror of
https://github.com/AxioDL/PrimeWorldEditor.git
synced 2025-12-17 08:57:09 +00:00
CResourceProxyModel: Move implementation to cpp file
Allows modifying how the resource model works without rebuilding half of the codebase.
This commit is contained in:
Lioncache
152
src/Editor/ResourceBrowser/CResourceProxyModel.cpp
Normal file
152
src/Editor/ResourceBrowser/CResourceProxyModel.cpp
Normal file
@@ -0,0 +1,152 @@
|
||||
#include "Editor/ResourceBrowser/CResourceProxyModel.h"
|
||||
|
||||
CResourceProxyModel::CResourceProxyModel(QObject* parent)
|
||||
: QSortFilterProxyModel(parent)
|
||||
{
|
||||
SetSortMode(ESortMode::ByName);
|
||||
}
|
||||
|
||||
CResourceProxyModel::~CResourceProxyModel() = default;
|
||||
|
||||
void CResourceProxyModel::setSourceModel(QAbstractItemModel* sourceModel)
|
||||
{
|
||||
auto* resModel = qobject_cast<CResourceTableModel*>(sourceModel);
|
||||
mpModel = resModel;
|
||||
QSortFilterProxyModel::setSourceModel(resModel);
|
||||
sort(0);
|
||||
}
|
||||
|
||||
bool CResourceProxyModel::lessThan(const QModelIndex& rkLeft, const QModelIndex& rkRight) const
|
||||
{
|
||||
// Parent directory is always row 0 and should always be at the top
|
||||
if (mpModel->HasParentDirectoryEntry())
|
||||
{
|
||||
if (rkLeft.row() == 0)
|
||||
return true;
|
||||
|
||||
if (rkRight.row() == 0)
|
||||
return false;
|
||||
}
|
||||
|
||||
// Fetch directories and compare them
|
||||
const CVirtualDirectory* pLeftDir = mpModel->IndexDirectory(rkLeft);
|
||||
const CVirtualDirectory* pRightDir = mpModel->IndexDirectory(rkRight);
|
||||
const CResourceEntry* pLeftRes = mpModel->IndexEntry(rkLeft);
|
||||
const CResourceEntry* pRightRes = mpModel->IndexEntry(rkRight);
|
||||
|
||||
if (pLeftDir && !pRightDir)
|
||||
return true;
|
||||
|
||||
if (pRightDir && !pLeftDir)
|
||||
return false;
|
||||
|
||||
if (pLeftDir && pRightDir)
|
||||
return pLeftDir->Name().ToUpper() < pRightDir->Name().ToUpper();
|
||||
|
||||
if (mSortMode == ESortMode::ByName)
|
||||
return pLeftRes->UppercaseName() < pRightRes->UppercaseName();
|
||||
|
||||
return pLeftRes->Size() < pRightRes->Size();
|
||||
}
|
||||
|
||||
bool CResourceProxyModel::filterAcceptsRow(int SourceRow, const QModelIndex& rkSourceParent) const
|
||||
{
|
||||
QModelIndex Index = mpModel->index(SourceRow, 0, rkSourceParent);
|
||||
CResourceEntry *pEntry = mpModel->IndexEntry(Index);
|
||||
|
||||
if (pEntry && !IsTypeAccepted(pEntry->TypeInfo()))
|
||||
return false;
|
||||
|
||||
// Compare search results
|
||||
if (!mSearchString.IsEmpty())
|
||||
{
|
||||
if (!pEntry)
|
||||
return false;
|
||||
|
||||
bool HasNameMatch = pEntry->UppercaseName().Contains(mSearchString);
|
||||
|
||||
if (!HasNameMatch)
|
||||
{
|
||||
bool HasIDMatch = false;
|
||||
|
||||
if (mCompareBitLength > 0)
|
||||
{
|
||||
const auto IDBitLength = static_cast<uint32>(pEntry->ID().Length()) * 8;
|
||||
|
||||
if (mCompareBitLength <= IDBitLength)
|
||||
{
|
||||
uint64 ID = pEntry->ID().ToLongLong();
|
||||
uint32 MaxShift = IDBitLength - mCompareBitLength;
|
||||
|
||||
for (uint32 Shift = 0; Shift <= MaxShift; Shift += 4)
|
||||
{
|
||||
uint64 ShiftCompare = mCompareID << Shift;
|
||||
uint64 Mask = mCompareMask << Shift;
|
||||
|
||||
if ((ID & Mask) == ShiftCompare)
|
||||
{
|
||||
HasIDMatch = true;
|
||||
break;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
if (!HasIDMatch)
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
void CResourceProxyModel::SetTypeFilter(CResTypeInfo *pInfo, bool Allow)
|
||||
{
|
||||
if (Allow)
|
||||
mTypeFilter.insert(pInfo);
|
||||
else
|
||||
mTypeFilter.remove(pInfo);
|
||||
}
|
||||
|
||||
bool CResourceProxyModel::IsTypeAccepted(CResTypeInfo* pTypeInfo) const
|
||||
{
|
||||
return mTypeFilter.isEmpty() || mTypeFilter.contains(pTypeInfo);
|
||||
}
|
||||
|
||||
void CResourceProxyModel::SetSortMode(ESortMode Mode)
|
||||
{
|
||||
if (mSortMode != Mode)
|
||||
{
|
||||
mSortMode = Mode;
|
||||
sort(0, (mSortMode == ESortMode::ByName ? Qt::AscendingOrder : Qt::DescendingOrder));
|
||||
}
|
||||
}
|
||||
|
||||
// ************ SLOTS ************
|
||||
void CResourceProxyModel::SetSearchString(const TString& rkString)
|
||||
{
|
||||
mSearchString = rkString.ToUpper();
|
||||
|
||||
// Check if this is an asset ID
|
||||
TString IDString = rkString;
|
||||
IDString.RemoveWhitespace();
|
||||
|
||||
if (IDString.StartsWith("0x"))
|
||||
IDString = IDString.ChopFront(2);
|
||||
|
||||
if (IDString.Size() <= 16 && IDString.IsHexString())
|
||||
{
|
||||
mCompareBitLength = IDString.Size() * 4;
|
||||
mCompareMask = ((uint64) 1 << mCompareBitLength) - 1;
|
||||
mCompareID = IDString.ToInt64(16);
|
||||
|
||||
if (mCompareMask == 0)
|
||||
mCompareMask = -1;
|
||||
}
|
||||
else
|
||||
{
|
||||
mCompareID = -1;
|
||||
mCompareMask = 0;
|
||||
mCompareBitLength = 0;
|
||||
}
|
||||
}
|
||||
@@ -19,6 +19,23 @@ public:
|
||||
ByName, BySize
|
||||
};
|
||||
|
||||
explicit CResourceProxyModel(QObject* parent = nullptr);
|
||||
~CResourceProxyModel() override;
|
||||
|
||||
void setSourceModel(QAbstractItemModel* pSourceModel) override;
|
||||
bool lessThan(const QModelIndex& rkLeft, const QModelIndex& rkRight) const override;
|
||||
bool filterAcceptsRow(int SourceRow, const QModelIndex& rkSourceParent) const override;
|
||||
|
||||
void SetTypeFilter(CResTypeInfo *pInfo, bool Allow);
|
||||
void ClearTypeFilter() { mTypeFilter.clear(); }
|
||||
bool HasTypeFilter() const { return !mTypeFilter.isEmpty(); }
|
||||
bool IsTypeAccepted(CResTypeInfo* pTypeInfo) const;
|
||||
|
||||
void SetSortMode(ESortMode Mode);
|
||||
|
||||
public slots:
|
||||
void SetSearchString(const TString& rkString);
|
||||
|
||||
private:
|
||||
CResourceTableModel *mpModel = nullptr;
|
||||
TString mSearchString;
|
||||
@@ -28,167 +45,6 @@ private:
|
||||
uint64 mCompareID = 0;
|
||||
uint64 mCompareMask = 0;
|
||||
uint32 mCompareBitLength = 0;
|
||||
|
||||
public:
|
||||
explicit CResourceProxyModel(QObject *pParent = nullptr)
|
||||
: QSortFilterProxyModel(pParent)
|
||||
{
|
||||
SetSortMode(ESortMode::ByName);
|
||||
}
|
||||
|
||||
void setSourceModel(QAbstractItemModel* pSourceModel) override
|
||||
{
|
||||
CResourceTableModel *pResModel = qobject_cast<CResourceTableModel*>(pSourceModel);
|
||||
mpModel = pResModel;
|
||||
QSortFilterProxyModel::setSourceModel(pResModel);
|
||||
sort(0);
|
||||
}
|
||||
|
||||
bool lessThan(const QModelIndex& rkLeft, const QModelIndex& rkRight) const override
|
||||
{
|
||||
// Parent directory is always row 0 and should always be at the top
|
||||
if (mpModel->HasParentDirectoryEntry())
|
||||
{
|
||||
if (rkLeft.row() == 0)
|
||||
return true;
|
||||
|
||||
if (rkRight.row() == 0)
|
||||
return false;
|
||||
}
|
||||
|
||||
// Fetch directories and compare them
|
||||
CVirtualDirectory *pLeftDir = mpModel->IndexDirectory(rkLeft);
|
||||
CVirtualDirectory *pRightDir = mpModel->IndexDirectory(rkRight);
|
||||
CResourceEntry *pLeftRes = mpModel->IndexEntry(rkLeft);
|
||||
CResourceEntry *pRightRes = mpModel->IndexEntry(rkRight);
|
||||
|
||||
if (pLeftDir && !pRightDir)
|
||||
return true;
|
||||
|
||||
else if (pRightDir && !pLeftDir)
|
||||
return false;
|
||||
|
||||
else if (pLeftDir && pRightDir)
|
||||
return pLeftDir->Name().ToUpper() < pRightDir->Name().ToUpper();
|
||||
|
||||
else if (mSortMode == ESortMode::ByName)
|
||||
return pLeftRes->UppercaseName() < pRightRes->UppercaseName();
|
||||
|
||||
else
|
||||
return pLeftRes->Size() < pRightRes->Size();
|
||||
}
|
||||
|
||||
bool filterAcceptsRow(int SourceRow, const QModelIndex& rkSourceParent) const override
|
||||
{
|
||||
QModelIndex Index = mpModel->index(SourceRow, 0, rkSourceParent);
|
||||
CResourceEntry *pEntry = mpModel->IndexEntry(Index);
|
||||
|
||||
if (pEntry && !IsTypeAccepted(pEntry->TypeInfo()))
|
||||
return false;
|
||||
|
||||
// Compare search results
|
||||
if (!mSearchString.IsEmpty())
|
||||
{
|
||||
if (!pEntry)
|
||||
return false;
|
||||
|
||||
bool HasNameMatch = pEntry->UppercaseName().Contains(mSearchString);
|
||||
|
||||
if (!HasNameMatch)
|
||||
{
|
||||
bool HasIDMatch = false;
|
||||
|
||||
if (mCompareBitLength > 0)
|
||||
{
|
||||
const auto IDBitLength = static_cast<uint32>(pEntry->ID().Length()) * 8;
|
||||
|
||||
if (mCompareBitLength <= IDBitLength)
|
||||
{
|
||||
uint64 ID = pEntry->ID().ToLongLong();
|
||||
uint32 MaxShift = IDBitLength - mCompareBitLength;
|
||||
|
||||
for (uint32 Shift = 0; Shift <= MaxShift; Shift += 4)
|
||||
{
|
||||
uint64 ShiftCompare = mCompareID << Shift;
|
||||
uint64 Mask = mCompareMask << Shift;
|
||||
|
||||
if ((ID & Mask) == ShiftCompare)
|
||||
{
|
||||
HasIDMatch = true;
|
||||
break;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
if (!HasIDMatch)
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
void SetTypeFilter(CResTypeInfo *pInfo, bool Allow)
|
||||
{
|
||||
if (Allow)
|
||||
mTypeFilter.insert(pInfo);
|
||||
else
|
||||
mTypeFilter.remove(pInfo);
|
||||
}
|
||||
|
||||
void ClearTypeFilter()
|
||||
{
|
||||
mTypeFilter.clear();
|
||||
}
|
||||
|
||||
bool HasTypeFilter() const
|
||||
{
|
||||
return !mTypeFilter.isEmpty();
|
||||
}
|
||||
|
||||
bool IsTypeAccepted(CResTypeInfo *pTypeInfo) const
|
||||
{
|
||||
return mTypeFilter.isEmpty() || mTypeFilter.contains(pTypeInfo);
|
||||
}
|
||||
|
||||
void SetSortMode(ESortMode Mode)
|
||||
{
|
||||
if (mSortMode != Mode)
|
||||
{
|
||||
mSortMode = Mode;
|
||||
sort(0, (mSortMode == ESortMode::ByName ? Qt::AscendingOrder : Qt::DescendingOrder));
|
||||
}
|
||||
}
|
||||
|
||||
public slots:
|
||||
void SetSearchString(const TString& rkString)
|
||||
{
|
||||
mSearchString = rkString.ToUpper();
|
||||
|
||||
// Check if this is an asset ID
|
||||
TString IDString = rkString;
|
||||
IDString.RemoveWhitespace();
|
||||
|
||||
if (IDString.StartsWith("0x"))
|
||||
IDString = IDString.ChopFront(2);
|
||||
|
||||
if (IDString.Size() <= 16 && IDString.IsHexString())
|
||||
{
|
||||
mCompareBitLength = IDString.Size() * 4;
|
||||
mCompareMask = ((uint64) 1 << mCompareBitLength) - 1;
|
||||
mCompareID = IDString.ToInt64(16);
|
||||
|
||||
if (mCompareMask == 0)
|
||||
mCompareMask = -1;
|
||||
}
|
||||
else
|
||||
{
|
||||
mCompareID = -1;
|
||||
mCompareMask = 0;
|
||||
mCompareBitLength = 0;
|
||||
}
|
||||
}
|
||||
};
|
||||
|
||||
#endif // CRESOURCEPROXYMODEL
|
||||
|
||||
Reference in New Issue
Block a user