110 lines
3.5 KiB
C++
110 lines
3.5 KiB
C++
#include <string>
|
|
#include <iostream>
|
|
#include <stack>
|
|
|
|
#include <QStandardItemModel>
|
|
|
|
#include "recoll.h"
|
|
#include "rclconfig.h"
|
|
#include "fstreewalk.h"
|
|
#include "idxmodel.h"
|
|
|
|
|
|
class WalkerCB : public FsTreeWalkerCB {
|
|
public:
|
|
WalkerCB(RclConfig *config, const std::string& topstring, FsTreeWalker& walker,
|
|
IdxTreeModel *model, const QModelIndex& index)
|
|
: m_config(config), m_topstring(topstring), m_walker(walker), m_model(model)
|
|
{
|
|
m_indexes.push(index);
|
|
m_rows.push(0);
|
|
}
|
|
virtual FsTreeWalker::Status processone(
|
|
const std::string& path, const struct PathStat *, FsTreeWalker::CbFlag flg) override;
|
|
|
|
RclConfig *m_config;
|
|
std::string m_topstring;
|
|
FsTreeWalker& m_walker;
|
|
IdxTreeModel *m_model;
|
|
std::stack<QModelIndex> m_indexes;
|
|
std::stack<int> m_rows;
|
|
};
|
|
|
|
FsTreeWalker::Status WalkerCB::processone(
|
|
const std::string& path, const struct PathStat *, FsTreeWalker::CbFlag flg)
|
|
{
|
|
if (flg == FsTreeWalker::FtwDirEnter || flg == FsTreeWalker::FtwDirReturn) {
|
|
m_config->setKeyDir(path);
|
|
// Set up filter/skipped patterns for this subtree.
|
|
m_walker.setOnlyNames(m_config->getOnlyNames());
|
|
m_walker.setSkippedNames(m_config->getSkippedNames());
|
|
}
|
|
if (flg == FsTreeWalker::FtwDirReturn) {
|
|
m_indexes.pop();
|
|
m_rows.pop();
|
|
return FsTreeWalker::FtwOk;
|
|
}
|
|
if (flg == FsTreeWalker::FtwDirEnter) {
|
|
if (m_model->columnCount(m_indexes.top()) == 0) {
|
|
if (!m_model->insertColumn(0, m_indexes.top()))
|
|
return FsTreeWalker::FtwError;
|
|
}
|
|
if (!m_model->insertRow(m_rows.top(), m_indexes.top()))
|
|
return FsTreeWalker::FtwError;
|
|
const QModelIndex child = m_model->index(m_rows.top(), 0, m_indexes.top());
|
|
// Setting the short path in DisplayRole and the real one in EditRole does not seem to work,
|
|
// the treeview shows the EditRole?? So use the ToolTip to store the full value
|
|
std::string disp;
|
|
if (m_topstring.empty()) {
|
|
disp = path_getsimple(path);
|
|
} else {
|
|
disp = m_topstring;
|
|
m_topstring.clear();
|
|
}
|
|
m_model->setData(child, QVariant(path2qs(disp)), Qt::DisplayRole);
|
|
m_model->setData(child, QVariant(path2qs(path)), Qt::ToolTipRole);
|
|
++m_rows.top();
|
|
m_indexes.push(child);
|
|
m_rows.push(0);
|
|
}
|
|
return FsTreeWalker::FtwOk;
|
|
}
|
|
|
|
static void populateDir(RclConfig *config, const std::string& topstr, IdxTreeModel *model,
|
|
const QModelIndex& index, const std::string& path, int depth)
|
|
{
|
|
FsTreeWalker walker;
|
|
walker.setSkippedPaths(config->getSkippedPaths());
|
|
// walker.setOpts(walker.getOpts() | FsTreeWalker::FtwSkipDotFiles);
|
|
walker.setMaxDepth(depth);
|
|
WalkerCB cb(config, topstr, walker, model, index);
|
|
walker.walk(path, cb);
|
|
}
|
|
|
|
void IdxTreeModel::populate()
|
|
{
|
|
QModelIndex index = this->index(0,0);
|
|
auto topdirs = m_config->getTopdirs();
|
|
|
|
auto prefix = commonprefix(topdirs);
|
|
|
|
if (this->columnCount(index) == 0) {
|
|
if (!this->insertColumn(0, index))
|
|
return;
|
|
}
|
|
|
|
int row = 0;
|
|
for (const auto& topdir : topdirs) {
|
|
const QModelIndex child = this->index(row, 0, index);
|
|
std::string topdisp;
|
|
if (prefix.size() > 1) {
|
|
topdisp = topdir.substr(prefix.size());
|
|
} else {
|
|
topdisp = topdir;
|
|
}
|
|
populateDir(m_config, topdisp, this, child, topdir, m_depth);
|
|
++row;
|
|
}
|
|
sort(0, Qt::AscendingOrder);
|
|
}
|