ctkPluginResourcesTreeModel.cxx 6.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266
  1. /*=============================================================================
  2. Library: CTK
  3. Copyright (c) 2010 German Cancer Research Center,
  4. Division of Medical and Biological Informatics
  5. Licensed under the Apache License, Version 2.0 (the "License");
  6. you may not use this file except in compliance with the License.
  7. You may obtain a copy of the License at
  8. http://www.apache.org/licenses/LICENSE-2.0
  9. Unless required by applicable law or agreed to in writing, software
  10. distributed under the License is distributed on an "AS IS" BASIS,
  11. WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  12. See the License for the specific language governing permissions and
  13. limitations under the License.
  14. =============================================================================*/
  15. #include "ctkPluginResourcesTreeModel.h"
  16. #include <PluginFramework/ctkPlugin.h>
  17. namespace ctk {
  18. class PluginResourceTreeNode;
  19. class PluginResourceTreeItem
  20. {
  21. public:
  22. PluginResourceTreeItem(const QString& path, PluginResourceTreeNode* parent = 0);
  23. virtual ~PluginResourceTreeItem();
  24. virtual PluginResourceTreeItem* child(int row);
  25. virtual int childCount() const;
  26. int row();
  27. QVariant data(int role) const;
  28. PluginResourceTreeNode* parent() const;
  29. protected:
  30. const QString path;
  31. PluginResourceTreeNode* parentNode;
  32. };
  33. class PluginResourceTreeNode : public PluginResourceTreeItem
  34. {
  35. public:
  36. PluginResourceTreeNode(const Plugin* plugin, const QString& path, PluginResourceTreeNode* parent = 0);
  37. ~PluginResourceTreeNode();
  38. PluginResourceTreeItem* child(int row);
  39. int childCount() const;
  40. int indexOf(PluginResourceTreeItem* child) const;
  41. private:
  42. const Plugin* plugin;
  43. QList<PluginResourceTreeItem*> children;
  44. };
  45. PluginResourceTreeItem::PluginResourceTreeItem(const QString& path,
  46. PluginResourceTreeNode* parent)
  47. : path(path), parentNode(parent)
  48. {
  49. }
  50. PluginResourceTreeItem::~PluginResourceTreeItem()
  51. {
  52. }
  53. PluginResourceTreeItem* PluginResourceTreeItem::child(int row)
  54. {
  55. return 0;
  56. }
  57. int PluginResourceTreeItem::childCount() const
  58. {
  59. return 0;
  60. }
  61. int PluginResourceTreeItem::row()
  62. {
  63. if (parentNode)
  64. {
  65. return parentNode->indexOf(this);
  66. }
  67. return 0;
  68. }
  69. QVariant PluginResourceTreeItem::data(int role) const
  70. {
  71. if (role == Qt::DisplayRole)
  72. {
  73. QString p = path;
  74. if (p.endsWith('/')) p = p.left(p.size()-1);
  75. int i = p.lastIndexOf('/');
  76. return p.mid(i+1);
  77. }
  78. return QVariant();
  79. }
  80. PluginResourceTreeNode* PluginResourceTreeItem::parent() const
  81. {
  82. return parentNode;
  83. }
  84. PluginResourceTreeNode::PluginResourceTreeNode(const Plugin* plugin, const QString& path, PluginResourceTreeNode* parent)
  85. : PluginResourceTreeItem(path, parent), plugin(plugin)
  86. {
  87. QStringList pathEntries = plugin->getResourceList(path);
  88. QStringListIterator it(pathEntries);
  89. while (it.hasNext())
  90. {
  91. const QString& entry = it.next();
  92. if (entry.endsWith('/'))
  93. {
  94. children.push_back(new PluginResourceTreeNode(plugin, entry, this));
  95. }
  96. else
  97. {
  98. children.push_back(new PluginResourceTreeItem(entry, this));
  99. }
  100. }
  101. }
  102. PluginResourceTreeNode::~PluginResourceTreeNode()
  103. {
  104. qDeleteAll(children);
  105. }
  106. PluginResourceTreeItem* PluginResourceTreeNode::child(int row)
  107. {
  108. return children.value(row);
  109. }
  110. int PluginResourceTreeNode::childCount() const
  111. {
  112. return children.size();
  113. }
  114. int PluginResourceTreeNode::indexOf(PluginResourceTreeItem* child) const
  115. {
  116. return children.indexOf(child);
  117. }
  118. PluginResourcesTreeModel::PluginResourcesTreeModel(const Plugin* plugin, QObject* parent)
  119. : QAbstractItemModel(parent), plugin(plugin)
  120. {
  121. rootItem = new PluginResourceTreeNode(plugin, "/");
  122. }
  123. PluginResourcesTreeModel::~PluginResourcesTreeModel()
  124. {
  125. delete rootItem;
  126. }
  127. QVariant PluginResourcesTreeModel::data(const QModelIndex &index, int role) const
  128. {
  129. if (!index.isValid())
  130. return QVariant();
  131. if (role == Qt::DisplayRole)
  132. {
  133. PluginResourceTreeItem* item = static_cast<PluginResourceTreeItem*>(index.internalPointer());
  134. return item->data(role);
  135. }
  136. return QVariant();
  137. }
  138. Qt::ItemFlags PluginResourcesTreeModel::flags(const QModelIndex &index) const
  139. {
  140. if (!index.isValid())
  141. return 0;
  142. return Qt::ItemIsEnabled | Qt::ItemIsSelectable;
  143. }
  144. QVariant PluginResourcesTreeModel::headerData(int section, Qt::Orientation orientation,
  145. int role) const
  146. {
  147. return QVariant();
  148. }
  149. QModelIndex PluginResourcesTreeModel::index(int row, int column,
  150. const QModelIndex &parent) const
  151. {
  152. if (!hasIndex(row, column, parent))
  153. return QModelIndex();
  154. PluginResourceTreeItem* parentItem;
  155. if (!parent.isValid())
  156. parentItem = rootItem;
  157. else
  158. parentItem = static_cast<PluginResourceTreeItem*>(parent.internalPointer());
  159. PluginResourceTreeItem* childItem = parentItem->child(row);
  160. if (childItem)
  161. return createIndex(row, column, childItem);
  162. else
  163. return QModelIndex();
  164. }
  165. QModelIndex PluginResourcesTreeModel::parent(const QModelIndex &index) const
  166. {
  167. if (!index.isValid())
  168. return QModelIndex();
  169. PluginResourceTreeItem* childItem = static_cast<PluginResourceTreeItem*>(index.internalPointer());
  170. PluginResourceTreeItem* parentItem = childItem->parent();
  171. if (parentItem == rootItem)
  172. return QModelIndex();
  173. return createIndex(parentItem->row(), 0, parentItem);
  174. }
  175. int PluginResourcesTreeModel::rowCount(const QModelIndex &parent) const
  176. {
  177. if (parent.column() > 0) return 0;
  178. PluginResourceTreeItem* parentItem;
  179. if (!parent.isValid())
  180. {
  181. parentItem = rootItem;
  182. }
  183. else
  184. {
  185. parentItem = static_cast<PluginResourceTreeItem*>(parent.internalPointer());
  186. }
  187. return parentItem->childCount();
  188. }
  189. int PluginResourcesTreeModel::columnCount(const QModelIndex &parent) const
  190. {
  191. return 1;
  192. }
  193. }