Merge pull request #1524 from Holt59/iorganizer-virtualfiletree

Add IOrganizer::virtualFileTree().

1 files changed, 59 insertions, 0 deletions

diff --git a/src/virtualfiletree.cpp b/src/virtualfiletree.cpp
new file mode 100644
index 00000000..983523e6
--- /dev/null
+++ b/src/virtualfiletree.cpp
@@ -0,0 +1,59 @@
+#include "virtualfiletree.h"
+
+#include "shared/directoryentry.h"
+#include "shared/filesorigin.h"
+#include "shared/fileentry.h"
+// #include "shared/util.h"
+
+using namespace MOBase;
+using namespace MOShared;
+
+class VirtualFileTreeImpl : public VirtualFileTree {
+public:
+
+
+  /**
+   *
+   */
+  VirtualFileTreeImpl(std::shared_ptr<const IFileTree> parent, const DirectoryEntry* dir) :
+    FileTreeEntry(parent, parent ? QString::fromStdWString(dir->getName()) : ""),
+    VirtualFileTree(),
+    m_dirEntry(dir) { }
+
+protected:
+
+  /**
+   * No mutable operations allowed.
+   */
+  bool beforeReplace(IFileTree const* dstTree, FileTreeEntry const* destination, FileTreeEntry const* source) override { return false; }
+  bool beforeInsert(IFileTree const* entry, FileTreeEntry const* name) override { return false; }
+  bool beforeRemove(IFileTree const* entry, FileTreeEntry const* name) override { return false; }
+  std::shared_ptr<FileTreeEntry> makeFile(std::shared_ptr<const IFileTree> parent, QString name) const override { return nullptr; }
+  std::shared_ptr<IFileTree> makeDirectory(std::shared_ptr<const IFileTree> parent, QString name) const override { return nullptr; }
+
+  bool doPopulate(std::shared_ptr<const IFileTree> parent, std::vector<std::shared_ptr<FileTreeEntry>>& entries) const override {
+    for (auto* subdirEntry : m_dirEntry->getSubDirectories()) {
+      entries.push_back(std::make_shared<VirtualFileTreeImpl>(parent, subdirEntry));
+    }
+    for (auto& file : m_dirEntry->getFiles()) {
+      entries.push_back(createFileEntry(parent, QString::fromStdWString(file->getName())));
+    }
+
+    // Vector is already sorted:
+    return true;
+  }
+
+  std::shared_ptr<IFileTree> doClone() const {
+    return std::make_shared<VirtualFileTreeImpl>(nullptr, m_dirEntry);
+  }
+
+private:
+  const DirectoryEntry* m_dirEntry;
+};
+
+/**
+ *
+ */
+std::shared_ptr<const VirtualFileTree> VirtualFileTree::makeTree(const DirectoryEntry* rootEntry) {
+  return std::make_shared<VirtualFileTreeImpl>(nullptr, rootEntry);
+}