Audio plugin host https://kx.studio/carla
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

juce_TreeView.cpp 53KB

10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
10 years ago
1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840184118421843184418451846184718481849185018511852185318541855185618571858185918601861186218631864186518661867186818691870187118721873187418751876187718781879188018811882188318841885188618871888188918901891189218931894189518961897
  1. /*
  2. ==============================================================================
  3. This file is part of the JUCE library.
  4. Copyright (c) 2013 - Raw Material Software Ltd.
  5. Permission is granted to use this software under the terms of either:
  6. a) the GPL v2 (or any later version)
  7. b) the Affero GPL v3
  8. Details of these licenses can be found at: www.gnu.org/licenses
  9. JUCE is distributed in the hope that it will be useful, but WITHOUT ANY
  10. WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR
  11. A PARTICULAR PURPOSE. See the GNU General Public License for more details.
  12. ------------------------------------------------------------------------------
  13. To release a closed-source product which uses JUCE, commercial licenses are
  14. available: visit www.juce.com for more information.
  15. ==============================================================================
  16. */
  17. class TreeView::ContentComponent : public Component,
  18. public TooltipClient,
  19. public AsyncUpdater
  20. {
  21. public:
  22. ContentComponent (TreeView& tree)
  23. : owner (tree),
  24. buttonUnderMouse (nullptr),
  25. isDragging (false),
  26. needSelectionOnMouseUp (false)
  27. {
  28. }
  29. void mouseDown (const MouseEvent& e) override
  30. {
  31. updateButtonUnderMouse (e);
  32. isDragging = false;
  33. needSelectionOnMouseUp = false;
  34. Rectangle<int> pos;
  35. if (TreeViewItem* const item = findItemAt (e.y, pos))
  36. {
  37. if (isEnabled())
  38. {
  39. // (if the open/close buttons are hidden, we'll treat clicks to the left of the item
  40. // as selection clicks)
  41. if (e.x < pos.getX() && owner.openCloseButtonsVisible)
  42. {
  43. if (e.x >= pos.getX() - owner.getIndentSize())
  44. item->setOpen (! item->isOpen());
  45. // (clicks to the left of an open/close button are ignored)
  46. }
  47. else
  48. {
  49. // mouse-down inside the body of the item..
  50. if (! owner.isMultiSelectEnabled())
  51. item->setSelected (true, true);
  52. else if (item->isSelected())
  53. needSelectionOnMouseUp = ! e.mods.isPopupMenu();
  54. else
  55. selectBasedOnModifiers (item, e.mods);
  56. if (e.x >= pos.getX())
  57. item->itemClicked (e.withNewPosition (e.position - pos.getPosition().toFloat()));
  58. }
  59. }
  60. }
  61. }
  62. void mouseUp (const MouseEvent& e) override
  63. {
  64. updateButtonUnderMouse (e);
  65. if (needSelectionOnMouseUp && e.mouseWasClicked() && isEnabled())
  66. {
  67. Rectangle<int> pos;
  68. if (TreeViewItem* const item = findItemAt (e.y, pos))
  69. selectBasedOnModifiers (item, e.mods);
  70. }
  71. }
  72. void mouseDoubleClick (const MouseEvent& e) override
  73. {
  74. if (e.getNumberOfClicks() != 3 && isEnabled()) // ignore triple clicks
  75. {
  76. Rectangle<int> pos;
  77. if (TreeViewItem* const item = findItemAt (e.y, pos))
  78. if (e.x >= pos.getX() || ! owner.openCloseButtonsVisible)
  79. item->itemDoubleClicked (e.withNewPosition (e.position - pos.getPosition().toFloat()));
  80. }
  81. }
  82. void mouseDrag (const MouseEvent& e) override
  83. {
  84. if (isEnabled()
  85. && ! (isDragging || e.mouseWasClicked()
  86. || e.getDistanceFromDragStart() < 5
  87. || e.mods.isPopupMenu()))
  88. {
  89. isDragging = true;
  90. Rectangle<int> pos;
  91. TreeViewItem* const item = findItemAt (e.getMouseDownY(), pos);
  92. if (item != nullptr && e.getMouseDownX() >= pos.getX())
  93. {
  94. const var dragDescription (item->getDragSourceDescription());
  95. if (! (dragDescription.isVoid() || (dragDescription.isString() && dragDescription.toString().isEmpty())))
  96. {
  97. if (DragAndDropContainer* const dragContainer = DragAndDropContainer::findParentDragContainerFor (this))
  98. {
  99. pos.setSize (pos.getWidth(), item->itemHeight);
  100. Image dragImage (Component::createComponentSnapshot (pos, true));
  101. dragImage.multiplyAllAlphas (0.6f);
  102. Point<int> imageOffset (pos.getPosition() - e.getPosition());
  103. dragContainer->startDragging (dragDescription, &owner, dragImage, true, &imageOffset);
  104. }
  105. else
  106. {
  107. // to be able to do a drag-and-drop operation, the treeview needs to
  108. // be inside a component which is also a DragAndDropContainer.
  109. jassertfalse;
  110. }
  111. }
  112. }
  113. }
  114. }
  115. void mouseMove (const MouseEvent& e) override { updateButtonUnderMouse (e); }
  116. void mouseExit (const MouseEvent& e) override { updateButtonUnderMouse (e); }
  117. void paint (Graphics& g) override
  118. {
  119. if (owner.rootItem != nullptr)
  120. {
  121. owner.recalculateIfNeeded();
  122. if (! owner.rootItemVisible)
  123. g.setOrigin (0, -owner.rootItem->itemHeight);
  124. owner.rootItem->paintRecursively (g, getWidth());
  125. }
  126. }
  127. TreeViewItem* findItemAt (int y, Rectangle<int>& itemPosition) const
  128. {
  129. if (owner.rootItem != nullptr)
  130. {
  131. owner.recalculateIfNeeded();
  132. if (! owner.rootItemVisible)
  133. y += owner.rootItem->itemHeight;
  134. if (TreeViewItem* const ti = owner.rootItem->findItemRecursively (y))
  135. {
  136. itemPosition = ti->getItemPosition (false);
  137. return ti;
  138. }
  139. }
  140. return nullptr;
  141. }
  142. void updateComponents()
  143. {
  144. const int visibleTop = -getY();
  145. const int visibleBottom = visibleTop + getParentHeight();
  146. for (int i = items.size(); --i >= 0;)
  147. items.getUnchecked(i)->shouldKeep = false;
  148. {
  149. TreeViewItem* item = owner.rootItem;
  150. int y = (item != nullptr && ! owner.rootItemVisible) ? -item->itemHeight : 0;
  151. while (item != nullptr && y < visibleBottom)
  152. {
  153. y += item->itemHeight;
  154. if (y >= visibleTop)
  155. {
  156. if (RowItem* const ri = findItem (item->uid))
  157. {
  158. ri->shouldKeep = true;
  159. }
  160. else if (Component* const comp = item->createItemComponent())
  161. {
  162. items.add (new RowItem (item, comp, item->uid));
  163. addAndMakeVisible (comp);
  164. }
  165. }
  166. item = item->getNextVisibleItem (true);
  167. }
  168. }
  169. for (int i = items.size(); --i >= 0;)
  170. {
  171. RowItem* const ri = items.getUnchecked(i);
  172. bool keep = false;
  173. if (isParentOf (ri->component))
  174. {
  175. if (ri->shouldKeep)
  176. {
  177. Rectangle<int> pos (ri->item->getItemPosition (false));
  178. pos.setSize (pos.getWidth(), ri->item->itemHeight);
  179. if (pos.getBottom() >= visibleTop && pos.getY() < visibleBottom)
  180. {
  181. keep = true;
  182. ri->component->setBounds (pos);
  183. }
  184. }
  185. if ((! keep) && isMouseDraggingInChildCompOf (ri->component))
  186. {
  187. keep = true;
  188. ri->component->setSize (0, 0);
  189. }
  190. }
  191. if (! keep)
  192. items.remove (i);
  193. }
  194. }
  195. bool isMouseOverButton (TreeViewItem* const item) const noexcept
  196. {
  197. return item == buttonUnderMouse;
  198. }
  199. void resized() override
  200. {
  201. owner.itemsChanged();
  202. }
  203. String getTooltip() override
  204. {
  205. Rectangle<int> pos;
  206. if (TreeViewItem* const item = findItemAt (getMouseXYRelative().y, pos))
  207. return item->getTooltip();
  208. return owner.getTooltip();
  209. }
  210. private:
  211. //==============================================================================
  212. TreeView& owner;
  213. struct RowItem
  214. {
  215. RowItem (TreeViewItem* const it, Component* const c, const int itemUID)
  216. : component (c), item (it), uid (itemUID), shouldKeep (true)
  217. {
  218. }
  219. ~RowItem()
  220. {
  221. delete component.get();
  222. }
  223. WeakReference<Component> component;
  224. TreeViewItem* item;
  225. int uid;
  226. bool shouldKeep;
  227. };
  228. OwnedArray <RowItem> items;
  229. TreeViewItem* buttonUnderMouse;
  230. bool isDragging, needSelectionOnMouseUp;
  231. void selectBasedOnModifiers (TreeViewItem* const item, const ModifierKeys modifiers)
  232. {
  233. TreeViewItem* firstSelected = nullptr;
  234. if (modifiers.isShiftDown() && ((firstSelected = owner.getSelectedItem (0)) != nullptr))
  235. {
  236. TreeViewItem* const lastSelected = owner.getSelectedItem (owner.getNumSelectedItems() - 1);
  237. jassert (lastSelected != nullptr);
  238. int rowStart = firstSelected->getRowNumberInTree();
  239. int rowEnd = lastSelected->getRowNumberInTree();
  240. if (rowStart > rowEnd)
  241. std::swap (rowStart, rowEnd);
  242. int ourRow = item->getRowNumberInTree();
  243. int otherEnd = ourRow < rowEnd ? rowStart : rowEnd;
  244. if (ourRow > otherEnd)
  245. std::swap (ourRow, otherEnd);
  246. for (int i = ourRow; i <= otherEnd; ++i)
  247. owner.getItemOnRow (i)->setSelected (true, false);
  248. }
  249. else
  250. {
  251. const bool cmd = modifiers.isCommandDown();
  252. item->setSelected ((! cmd) || ! item->isSelected(), ! cmd);
  253. }
  254. }
  255. bool containsItem (TreeViewItem* const item) const noexcept
  256. {
  257. for (int i = items.size(); --i >= 0;)
  258. if (items.getUnchecked(i)->item == item)
  259. return true;
  260. return false;
  261. }
  262. RowItem* findItem (const int uid) const noexcept
  263. {
  264. for (int i = items.size(); --i >= 0;)
  265. {
  266. RowItem* const ri = items.getUnchecked(i);
  267. if (ri->uid == uid)
  268. return ri;
  269. }
  270. return nullptr;
  271. }
  272. void updateButtonUnderMouse (const MouseEvent& e)
  273. {
  274. TreeViewItem* newItem = nullptr;
  275. if (owner.openCloseButtonsVisible)
  276. {
  277. Rectangle<int> pos;
  278. TreeViewItem* item = findItemAt (e.y, pos);
  279. if (item != nullptr && e.x < pos.getX() && e.x >= pos.getX() - owner.getIndentSize())
  280. {
  281. newItem = item;
  282. if (! newItem->mightContainSubItems())
  283. newItem = nullptr;
  284. }
  285. }
  286. if (buttonUnderMouse != newItem)
  287. {
  288. repaintButtonUnderMouse();
  289. buttonUnderMouse = newItem;
  290. repaintButtonUnderMouse();
  291. }
  292. }
  293. void repaintButtonUnderMouse()
  294. {
  295. if (buttonUnderMouse != nullptr && containsItem (buttonUnderMouse))
  296. {
  297. const Rectangle<int> r (buttonUnderMouse->getItemPosition (false));
  298. repaint (0, r.getY(), r.getX(), buttonUnderMouse->getItemHeight());
  299. }
  300. }
  301. static bool isMouseDraggingInChildCompOf (Component* const comp)
  302. {
  303. const Array<MouseInputSource>& mouseSources = Desktop::getInstance().getMouseSources();
  304. for (MouseInputSource* mi = mouseSources.begin(), * const e = mouseSources.end(); mi != e; ++mi)
  305. {
  306. if (mi->isDragging())
  307. if (Component* const underMouse = mi->getComponentUnderMouse())
  308. if (comp == underMouse || comp->isParentOf (underMouse))
  309. return true;
  310. }
  311. return false;
  312. }
  313. void handleAsyncUpdate() override
  314. {
  315. owner.recalculateIfNeeded();
  316. }
  317. JUCE_DECLARE_NON_COPYABLE_WITH_LEAK_DETECTOR (ContentComponent)
  318. };
  319. //==============================================================================
  320. class TreeView::TreeViewport : public Viewport
  321. {
  322. public:
  323. TreeViewport() noexcept : lastX (-1) {}
  324. void updateComponents (const bool triggerResize)
  325. {
  326. if (ContentComponent* const tvc = getContentComp())
  327. {
  328. if (triggerResize)
  329. tvc->resized();
  330. else
  331. tvc->updateComponents();
  332. }
  333. repaint();
  334. }
  335. void visibleAreaChanged (const Rectangle<int>& newVisibleArea) override
  336. {
  337. const bool hasScrolledSideways = (newVisibleArea.getX() != lastX);
  338. lastX = newVisibleArea.getX();
  339. updateComponents (hasScrolledSideways);
  340. }
  341. ContentComponent* getContentComp() const noexcept
  342. {
  343. return static_cast <ContentComponent*> (getViewedComponent());
  344. }
  345. bool keyPressed (const KeyPress& key) override
  346. {
  347. Component* const tree = getParentComponent();
  348. return (tree != nullptr && tree->keyPressed (key))
  349. || Viewport::keyPressed (key);
  350. }
  351. private:
  352. int lastX;
  353. JUCE_DECLARE_NON_COPYABLE_WITH_LEAK_DETECTOR (TreeViewport)
  354. };
  355. //==============================================================================
  356. TreeView::TreeView (const String& name)
  357. : Component (name),
  358. viewport (new TreeViewport()),
  359. rootItem (nullptr),
  360. indentSize (-1),
  361. defaultOpenness (false),
  362. needsRecalculating (true),
  363. rootItemVisible (true),
  364. multiSelectEnabled (false),
  365. openCloseButtonsVisible (true)
  366. {
  367. addAndMakeVisible (viewport);
  368. viewport->setViewedComponent (new ContentComponent (*this));
  369. setWantsKeyboardFocus (true);
  370. }
  371. TreeView::~TreeView()
  372. {
  373. if (rootItem != nullptr)
  374. rootItem->setOwnerView (nullptr);
  375. }
  376. void TreeView::setRootItem (TreeViewItem* const newRootItem)
  377. {
  378. if (rootItem != newRootItem)
  379. {
  380. if (newRootItem != nullptr)
  381. {
  382. jassert (newRootItem->ownerView == nullptr); // can't use a tree item in more than one tree at once..
  383. if (newRootItem->ownerView != nullptr)
  384. newRootItem->ownerView->setRootItem (nullptr);
  385. }
  386. if (rootItem != nullptr)
  387. rootItem->setOwnerView (nullptr);
  388. rootItem = newRootItem;
  389. if (newRootItem != nullptr)
  390. newRootItem->setOwnerView (this);
  391. needsRecalculating = true;
  392. recalculateIfNeeded();
  393. if (rootItem != nullptr && (defaultOpenness || ! rootItemVisible))
  394. {
  395. rootItem->setOpen (false); // force a re-open
  396. rootItem->setOpen (true);
  397. }
  398. }
  399. }
  400. void TreeView::deleteRootItem()
  401. {
  402. const ScopedPointer <TreeViewItem> deleter (rootItem);
  403. setRootItem (nullptr);
  404. }
  405. void TreeView::setRootItemVisible (const bool shouldBeVisible)
  406. {
  407. rootItemVisible = shouldBeVisible;
  408. if (rootItem != nullptr && (defaultOpenness || ! rootItemVisible))
  409. {
  410. rootItem->setOpen (false); // force a re-open
  411. rootItem->setOpen (true);
  412. }
  413. itemsChanged();
  414. }
  415. void TreeView::colourChanged()
  416. {
  417. setOpaque (findColour (backgroundColourId).isOpaque());
  418. repaint();
  419. }
  420. void TreeView::setIndentSize (const int newIndentSize)
  421. {
  422. if (indentSize != newIndentSize)
  423. {
  424. indentSize = newIndentSize;
  425. resized();
  426. }
  427. }
  428. int TreeView::getIndentSize() noexcept
  429. {
  430. return indentSize >= 0 ? indentSize
  431. : getLookAndFeel().getTreeViewIndentSize (*this);
  432. }
  433. void TreeView::setDefaultOpenness (const bool isOpenByDefault)
  434. {
  435. if (defaultOpenness != isOpenByDefault)
  436. {
  437. defaultOpenness = isOpenByDefault;
  438. itemsChanged();
  439. }
  440. }
  441. void TreeView::setMultiSelectEnabled (const bool canMultiSelect)
  442. {
  443. multiSelectEnabled = canMultiSelect;
  444. }
  445. void TreeView::setOpenCloseButtonsVisible (const bool shouldBeVisible)
  446. {
  447. if (openCloseButtonsVisible != shouldBeVisible)
  448. {
  449. openCloseButtonsVisible = shouldBeVisible;
  450. itemsChanged();
  451. }
  452. }
  453. Viewport* TreeView::getViewport() const noexcept
  454. {
  455. return viewport;
  456. }
  457. //==============================================================================
  458. void TreeView::clearSelectedItems()
  459. {
  460. if (rootItem != nullptr)
  461. rootItem->deselectAllRecursively (nullptr);
  462. }
  463. int TreeView::getNumSelectedItems (int maximumDepthToSearchTo) const noexcept
  464. {
  465. return rootItem != nullptr ? rootItem->countSelectedItemsRecursively (maximumDepthToSearchTo) : 0;
  466. }
  467. TreeViewItem* TreeView::getSelectedItem (const int index) const noexcept
  468. {
  469. return rootItem != nullptr ? rootItem->getSelectedItemWithIndex (index) : 0;
  470. }
  471. int TreeView::getNumRowsInTree() const
  472. {
  473. return rootItem != nullptr ? (rootItem->getNumRows() - (rootItemVisible ? 0 : 1)) : 0;
  474. }
  475. TreeViewItem* TreeView::getItemOnRow (int index) const
  476. {
  477. if (! rootItemVisible)
  478. ++index;
  479. if (rootItem != nullptr && index >= 0)
  480. return rootItem->getItemOnRow (index);
  481. return nullptr;
  482. }
  483. TreeViewItem* TreeView::getItemAt (int y) const noexcept
  484. {
  485. ContentComponent* const tc = viewport->getContentComp();
  486. Rectangle<int> pos;
  487. return tc->findItemAt (tc->getLocalPoint (this, Point<int> (0, y)).y, pos);
  488. }
  489. TreeViewItem* TreeView::findItemFromIdentifierString (const String& identifierString) const
  490. {
  491. if (rootItem == nullptr)
  492. return nullptr;
  493. return rootItem->findItemFromIdentifierString (identifierString);
  494. }
  495. //==============================================================================
  496. static void addAllSelectedItemIds (TreeViewItem* item, XmlElement& parent)
  497. {
  498. if (item->isSelected())
  499. parent.createNewChildElement ("SELECTED")->setAttribute ("id", item->getItemIdentifierString());
  500. const int numSubItems = item->getNumSubItems();
  501. for (int i = 0; i < numSubItems; ++i)
  502. addAllSelectedItemIds (item->getSubItem(i), parent);
  503. }
  504. XmlElement* TreeView::getOpennessState (const bool alsoIncludeScrollPosition) const
  505. {
  506. XmlElement* e = nullptr;
  507. if (rootItem != nullptr)
  508. {
  509. e = rootItem->getOpennessState (false);
  510. if (e != nullptr)
  511. {
  512. if (alsoIncludeScrollPosition)
  513. e->setAttribute ("scrollPos", viewport->getViewPositionY());
  514. addAllSelectedItemIds (rootItem, *e);
  515. }
  516. }
  517. return e;
  518. }
  519. void TreeView::restoreOpennessState (const XmlElement& newState, const bool restoreStoredSelection)
  520. {
  521. if (rootItem != nullptr)
  522. {
  523. rootItem->restoreOpennessState (newState);
  524. if (newState.hasAttribute ("scrollPos"))
  525. viewport->setViewPosition (viewport->getViewPositionX(),
  526. newState.getIntAttribute ("scrollPos"));
  527. if (restoreStoredSelection)
  528. {
  529. clearSelectedItems();
  530. forEachXmlChildElementWithTagName (newState, e, "SELECTED")
  531. {
  532. if (TreeViewItem* const item = rootItem->findItemFromIdentifierString (e->getStringAttribute ("id")))
  533. item->setSelected (true, false);
  534. }
  535. }
  536. }
  537. }
  538. //==============================================================================
  539. void TreeView::paint (Graphics& g)
  540. {
  541. g.fillAll (findColour (backgroundColourId));
  542. }
  543. void TreeView::resized()
  544. {
  545. viewport->setBounds (getLocalBounds());
  546. itemsChanged();
  547. recalculateIfNeeded();
  548. }
  549. void TreeView::enablementChanged()
  550. {
  551. repaint();
  552. }
  553. void TreeView::moveSelectedRow (const int delta)
  554. {
  555. const int numRowsInTree = getNumRowsInTree();
  556. if (numRowsInTree > 0)
  557. {
  558. int rowSelected = 0;
  559. if (TreeViewItem* const firstSelected = getSelectedItem (0))
  560. rowSelected = firstSelected->getRowNumberInTree();
  561. rowSelected = jlimit (0, numRowsInTree - 1, rowSelected + delta);
  562. for (;;)
  563. {
  564. if (TreeViewItem* const item = getItemOnRow (rowSelected))
  565. {
  566. if (! item->canBeSelected())
  567. {
  568. // if the row we want to highlight doesn't allow it, try skipping
  569. // to the next item..
  570. const int nextRowToTry = jlimit (0, numRowsInTree - 1, rowSelected + (delta < 0 ? -1 : 1));
  571. if (rowSelected != nextRowToTry)
  572. {
  573. rowSelected = nextRowToTry;
  574. continue;
  575. }
  576. break;
  577. }
  578. item->setSelected (true, true);
  579. scrollToKeepItemVisible (item);
  580. }
  581. break;
  582. }
  583. }
  584. }
  585. void TreeView::scrollToKeepItemVisible (TreeViewItem* item)
  586. {
  587. if (item != nullptr && item->ownerView == this)
  588. {
  589. recalculateIfNeeded();
  590. item = item->getDeepestOpenParentItem();
  591. const int y = item->y;
  592. const int viewTop = viewport->getViewPositionY();
  593. if (y < viewTop)
  594. {
  595. viewport->setViewPosition (viewport->getViewPositionX(), y);
  596. }
  597. else if (y + item->itemHeight > viewTop + viewport->getViewHeight())
  598. {
  599. viewport->setViewPosition (viewport->getViewPositionX(),
  600. (y + item->itemHeight) - viewport->getViewHeight());
  601. }
  602. }
  603. }
  604. bool TreeView::toggleOpenSelectedItem()
  605. {
  606. if (TreeViewItem* const firstSelected = getSelectedItem (0))
  607. {
  608. if (firstSelected->mightContainSubItems())
  609. {
  610. firstSelected->setOpen (! firstSelected->isOpen());
  611. return true;
  612. }
  613. }
  614. return false;
  615. }
  616. void TreeView::moveOutOfSelectedItem()
  617. {
  618. if (TreeViewItem* const firstSelected = getSelectedItem (0))
  619. {
  620. if (firstSelected->isOpen())
  621. {
  622. firstSelected->setOpen (false);
  623. }
  624. else
  625. {
  626. TreeViewItem* parent = firstSelected->parentItem;
  627. if ((! rootItemVisible) && parent == rootItem)
  628. parent = nullptr;
  629. if (parent != nullptr)
  630. {
  631. parent->setSelected (true, true);
  632. scrollToKeepItemVisible (parent);
  633. }
  634. }
  635. }
  636. }
  637. void TreeView::moveIntoSelectedItem()
  638. {
  639. if (TreeViewItem* const firstSelected = getSelectedItem (0))
  640. {
  641. if (firstSelected->isOpen() || ! firstSelected->mightContainSubItems())
  642. moveSelectedRow (1);
  643. else
  644. firstSelected->setOpen (true);
  645. }
  646. }
  647. void TreeView::moveByPages (int numPages)
  648. {
  649. if (TreeViewItem* currentItem = getSelectedItem (0))
  650. {
  651. const Rectangle<int> pos (currentItem->getItemPosition (false));
  652. const int targetY = pos.getY() + numPages * (getHeight() - pos.getHeight());
  653. int currentRow = currentItem->getRowNumberInTree();
  654. for (;;)
  655. {
  656. moveSelectedRow (numPages);
  657. currentItem = getSelectedItem (0);
  658. if (currentItem == nullptr)
  659. break;
  660. const int y = currentItem->getItemPosition (false).getY();
  661. if ((numPages < 0 && y <= targetY) || (numPages > 0 && y >= targetY))
  662. break;
  663. const int newRow = currentItem->getRowNumberInTree();
  664. if (newRow == currentRow)
  665. break;
  666. currentRow = newRow;
  667. }
  668. }
  669. }
  670. bool TreeView::keyPressed (const KeyPress& key)
  671. {
  672. if (rootItem != nullptr)
  673. {
  674. if (key == KeyPress::upKey) { moveSelectedRow (-1); return true; }
  675. if (key == KeyPress::downKey) { moveSelectedRow (1); return true; }
  676. if (key == KeyPress::homeKey) { moveSelectedRow (-0x3fffffff); return true; }
  677. if (key == KeyPress::endKey) { moveSelectedRow (0x3fffffff); return true; }
  678. if (key == KeyPress::pageUpKey) { moveByPages (-1); return true; }
  679. if (key == KeyPress::pageDownKey) { moveByPages (1); return true; }
  680. if (key == KeyPress::returnKey) { return toggleOpenSelectedItem(); }
  681. if (key == KeyPress::leftKey) { moveOutOfSelectedItem(); return true; }
  682. if (key == KeyPress::rightKey) { moveIntoSelectedItem(); return true; }
  683. }
  684. return false;
  685. }
  686. void TreeView::itemsChanged() noexcept
  687. {
  688. needsRecalculating = true;
  689. repaint();
  690. viewport->getContentComp()->triggerAsyncUpdate();
  691. }
  692. void TreeView::recalculateIfNeeded()
  693. {
  694. if (needsRecalculating)
  695. {
  696. needsRecalculating = false;
  697. const ScopedLock sl (nodeAlterationLock);
  698. if (rootItem != nullptr)
  699. rootItem->updatePositions (rootItemVisible ? 0 : -rootItem->itemHeight);
  700. viewport->updateComponents (false);
  701. if (rootItem != nullptr)
  702. {
  703. viewport->getViewedComponent()
  704. ->setSize (jmax (viewport->getMaximumVisibleWidth(), rootItem->totalWidth),
  705. rootItem->totalHeight - (rootItemVisible ? 0 : rootItem->itemHeight));
  706. }
  707. else
  708. {
  709. viewport->getViewedComponent()->setSize (0, 0);
  710. }
  711. }
  712. }
  713. //==============================================================================
  714. struct TreeView::InsertPoint
  715. {
  716. InsertPoint (TreeView& view, const StringArray& files,
  717. const DragAndDropTarget::SourceDetails& dragSourceDetails)
  718. : pos (dragSourceDetails.localPosition),
  719. item (view.getItemAt (dragSourceDetails.localPosition.y)),
  720. insertIndex (0)
  721. {
  722. if (item != nullptr)
  723. {
  724. Rectangle<int> itemPos (item->getItemPosition (true));
  725. insertIndex = item->getIndexInParent();
  726. const int oldY = pos.y;
  727. pos.y = itemPos.getY();
  728. if (item->getNumSubItems() == 0 || ! item->isOpen())
  729. {
  730. if (files.size() > 0 ? item->isInterestedInFileDrag (files)
  731. : item->isInterestedInDragSource (dragSourceDetails))
  732. {
  733. // Check if we're trying to drag into an empty group item..
  734. if (oldY > itemPos.getY() + itemPos.getHeight() / 4
  735. && oldY < itemPos.getBottom() - itemPos.getHeight() / 4)
  736. {
  737. insertIndex = 0;
  738. pos.x = itemPos.getX() + view.getIndentSize();
  739. pos.y = itemPos.getBottom();
  740. return;
  741. }
  742. }
  743. }
  744. if (oldY > itemPos.getCentreY())
  745. {
  746. pos.y += item->getItemHeight();
  747. while (item->isLastOfSiblings() && item->getParentItem() != nullptr
  748. && item->getParentItem()->getParentItem() != nullptr)
  749. {
  750. if (pos.x > itemPos.getX())
  751. break;
  752. item = item->getParentItem();
  753. itemPos = item->getItemPosition (true);
  754. insertIndex = item->getIndexInParent();
  755. }
  756. ++insertIndex;
  757. }
  758. pos.x = itemPos.getX();
  759. item = item->getParentItem();
  760. }
  761. }
  762. Point<int> pos;
  763. TreeViewItem* item;
  764. int insertIndex;
  765. };
  766. //==============================================================================
  767. class TreeView::InsertPointHighlight : public Component
  768. {
  769. public:
  770. InsertPointHighlight()
  771. : lastItem (nullptr), lastIndex (0)
  772. {
  773. setSize (100, 12);
  774. setAlwaysOnTop (true);
  775. setInterceptsMouseClicks (false, false);
  776. }
  777. void setTargetPosition (const InsertPoint& insertPos, const int width) noexcept
  778. {
  779. lastItem = insertPos.item;
  780. lastIndex = insertPos.insertIndex;
  781. const int offset = getHeight() / 2;
  782. setBounds (insertPos.pos.x - offset, insertPos.pos.y - offset,
  783. width - (insertPos.pos.x - offset), getHeight());
  784. }
  785. void paint (Graphics& g) override
  786. {
  787. Path p;
  788. const float h = (float) getHeight();
  789. p.addEllipse (2.0f, 2.0f, h - 4.0f, h - 4.0f);
  790. p.startNewSubPath (h - 2.0f, h / 2.0f);
  791. p.lineTo ((float) getWidth(), h / 2.0f);
  792. g.setColour (findColour (TreeView::dragAndDropIndicatorColourId, true));
  793. g.strokePath (p, PathStrokeType (2.0f));
  794. }
  795. TreeViewItem* lastItem;
  796. int lastIndex;
  797. private:
  798. JUCE_DECLARE_NON_COPYABLE (InsertPointHighlight)
  799. };
  800. //==============================================================================
  801. class TreeView::TargetGroupHighlight : public Component
  802. {
  803. public:
  804. TargetGroupHighlight()
  805. {
  806. setAlwaysOnTop (true);
  807. setInterceptsMouseClicks (false, false);
  808. }
  809. void setTargetPosition (TreeViewItem* const item) noexcept
  810. {
  811. Rectangle<int> r (item->getItemPosition (true));
  812. r.setHeight (item->getItemHeight());
  813. setBounds (r);
  814. }
  815. void paint (Graphics& g) override
  816. {
  817. g.setColour (findColour (TreeView::dragAndDropIndicatorColourId, true));
  818. g.drawRoundedRectangle (1.0f, 1.0f, getWidth() - 2.0f, getHeight() - 2.0f, 3.0f, 2.0f);
  819. }
  820. private:
  821. JUCE_DECLARE_NON_COPYABLE (TargetGroupHighlight)
  822. };
  823. //==============================================================================
  824. void TreeView::showDragHighlight (const InsertPoint& insertPos) noexcept
  825. {
  826. beginDragAutoRepeat (100);
  827. if (dragInsertPointHighlight == nullptr)
  828. {
  829. addAndMakeVisible (dragInsertPointHighlight = new InsertPointHighlight());
  830. addAndMakeVisible (dragTargetGroupHighlight = new TargetGroupHighlight());
  831. }
  832. dragInsertPointHighlight->setTargetPosition (insertPos, viewport->getViewWidth());
  833. dragTargetGroupHighlight->setTargetPosition (insertPos.item);
  834. }
  835. void TreeView::hideDragHighlight() noexcept
  836. {
  837. dragInsertPointHighlight = nullptr;
  838. dragTargetGroupHighlight = nullptr;
  839. }
  840. void TreeView::handleDrag (const StringArray& files, const SourceDetails& dragSourceDetails)
  841. {
  842. const bool scrolled = viewport->autoScroll (dragSourceDetails.localPosition.x,
  843. dragSourceDetails.localPosition.y, 20, 10);
  844. InsertPoint insertPos (*this, files, dragSourceDetails);
  845. if (insertPos.item != nullptr)
  846. {
  847. if (scrolled || dragInsertPointHighlight == nullptr
  848. || dragInsertPointHighlight->lastItem != insertPos.item
  849. || dragInsertPointHighlight->lastIndex != insertPos.insertIndex)
  850. {
  851. if (files.size() > 0 ? insertPos.item->isInterestedInFileDrag (files)
  852. : insertPos.item->isInterestedInDragSource (dragSourceDetails))
  853. showDragHighlight (insertPos);
  854. else
  855. hideDragHighlight();
  856. }
  857. }
  858. else
  859. {
  860. hideDragHighlight();
  861. }
  862. }
  863. void TreeView::handleDrop (const StringArray& files, const SourceDetails& dragSourceDetails)
  864. {
  865. hideDragHighlight();
  866. InsertPoint insertPos (*this, files, dragSourceDetails);
  867. if (insertPos.item == nullptr)
  868. insertPos.item = rootItem;
  869. if (insertPos.item != nullptr)
  870. {
  871. if (files.size() > 0)
  872. {
  873. if (insertPos.item->isInterestedInFileDrag (files))
  874. insertPos.item->filesDropped (files, insertPos.insertIndex);
  875. }
  876. else
  877. {
  878. if (insertPos.item->isInterestedInDragSource (dragSourceDetails))
  879. insertPos.item->itemDropped (dragSourceDetails, insertPos.insertIndex);
  880. }
  881. }
  882. }
  883. //==============================================================================
  884. bool TreeView::isInterestedInFileDrag (const StringArray&)
  885. {
  886. return true;
  887. }
  888. void TreeView::fileDragEnter (const StringArray& files, int x, int y)
  889. {
  890. fileDragMove (files, x, y);
  891. }
  892. void TreeView::fileDragMove (const StringArray& files, int x, int y)
  893. {
  894. handleDrag (files, SourceDetails (String::empty, this, Point<int> (x, y)));
  895. }
  896. void TreeView::fileDragExit (const StringArray&)
  897. {
  898. hideDragHighlight();
  899. }
  900. void TreeView::filesDropped (const StringArray& files, int x, int y)
  901. {
  902. handleDrop (files, SourceDetails (String::empty, this, Point<int> (x, y)));
  903. }
  904. bool TreeView::isInterestedInDragSource (const SourceDetails& /*dragSourceDetails*/)
  905. {
  906. return true;
  907. }
  908. void TreeView::itemDragEnter (const SourceDetails& dragSourceDetails)
  909. {
  910. itemDragMove (dragSourceDetails);
  911. }
  912. void TreeView::itemDragMove (const SourceDetails& dragSourceDetails)
  913. {
  914. handleDrag (StringArray(), dragSourceDetails);
  915. }
  916. void TreeView::itemDragExit (const SourceDetails& /*dragSourceDetails*/)
  917. {
  918. hideDragHighlight();
  919. }
  920. void TreeView::itemDropped (const SourceDetails& dragSourceDetails)
  921. {
  922. handleDrop (StringArray(), dragSourceDetails);
  923. }
  924. //==============================================================================
  925. enum TreeViewOpenness
  926. {
  927. opennessDefault = 0,
  928. opennessClosed = 1,
  929. opennessOpen = 2
  930. };
  931. TreeViewItem::TreeViewItem()
  932. : ownerView (nullptr),
  933. parentItem (nullptr),
  934. y (0),
  935. itemHeight (0),
  936. totalHeight (0),
  937. itemWidth (0),
  938. totalWidth (0),
  939. selected (false),
  940. redrawNeeded (true),
  941. drawLinesInside (false),
  942. drawLinesSet (false),
  943. drawsInLeftMargin (false),
  944. openness (opennessDefault)
  945. {
  946. static int nextUID = 0;
  947. uid = nextUID++;
  948. }
  949. TreeViewItem::~TreeViewItem()
  950. {
  951. }
  952. String TreeViewItem::getUniqueName() const
  953. {
  954. return String::empty;
  955. }
  956. void TreeViewItem::itemOpennessChanged (bool)
  957. {
  958. }
  959. int TreeViewItem::getNumSubItems() const noexcept
  960. {
  961. return subItems.size();
  962. }
  963. TreeViewItem* TreeViewItem::getSubItem (const int index) const noexcept
  964. {
  965. return subItems [index];
  966. }
  967. void TreeViewItem::clearSubItems()
  968. {
  969. if (ownerView != nullptr)
  970. {
  971. const ScopedLock sl (ownerView->nodeAlterationLock);
  972. if (subItems.size() > 0)
  973. {
  974. removeAllSubItemsFromList();
  975. treeHasChanged();
  976. }
  977. }
  978. else
  979. {
  980. removeAllSubItemsFromList();
  981. }
  982. }
  983. void TreeViewItem::removeAllSubItemsFromList()
  984. {
  985. for (int i = subItems.size(); --i >= 0;)
  986. removeSubItemFromList (i, true);
  987. }
  988. void TreeViewItem::addSubItem (TreeViewItem* const newItem, const int insertPosition)
  989. {
  990. if (newItem != nullptr)
  991. {
  992. newItem->parentItem = this;
  993. newItem->setOwnerView (ownerView);
  994. newItem->y = 0;
  995. newItem->itemHeight = newItem->getItemHeight();
  996. newItem->totalHeight = 0;
  997. newItem->itemWidth = newItem->getItemWidth();
  998. newItem->totalWidth = 0;
  999. if (ownerView != nullptr)
  1000. {
  1001. const ScopedLock sl (ownerView->nodeAlterationLock);
  1002. subItems.insert (insertPosition, newItem);
  1003. treeHasChanged();
  1004. if (newItem->isOpen())
  1005. newItem->itemOpennessChanged (true);
  1006. }
  1007. else
  1008. {
  1009. subItems.insert (insertPosition, newItem);
  1010. if (newItem->isOpen())
  1011. newItem->itemOpennessChanged (true);
  1012. }
  1013. }
  1014. }
  1015. void TreeViewItem::removeSubItem (int index, bool deleteItem)
  1016. {
  1017. if (ownerView != nullptr)
  1018. {
  1019. const ScopedLock sl (ownerView->nodeAlterationLock);
  1020. if (removeSubItemFromList (index, deleteItem))
  1021. treeHasChanged();
  1022. }
  1023. else
  1024. {
  1025. removeSubItemFromList (index, deleteItem);
  1026. }
  1027. }
  1028. bool TreeViewItem::removeSubItemFromList (int index, bool deleteItem)
  1029. {
  1030. if (TreeViewItem* child = subItems [index])
  1031. {
  1032. child->parentItem = nullptr;
  1033. subItems.remove (index, deleteItem);
  1034. return true;
  1035. }
  1036. return false;
  1037. }
  1038. bool TreeViewItem::isOpen() const noexcept
  1039. {
  1040. if (openness == opennessDefault)
  1041. return ownerView != nullptr && ownerView->defaultOpenness;
  1042. return openness == opennessOpen;
  1043. }
  1044. void TreeViewItem::setOpen (const bool shouldBeOpen)
  1045. {
  1046. if (isOpen() != shouldBeOpen)
  1047. {
  1048. openness = shouldBeOpen ? opennessOpen
  1049. : opennessClosed;
  1050. treeHasChanged();
  1051. itemOpennessChanged (isOpen());
  1052. }
  1053. }
  1054. bool TreeViewItem::isFullyOpen() const noexcept
  1055. {
  1056. if (! isOpen())
  1057. return false;
  1058. for (int i = 0; i < subItems.size(); ++i)
  1059. if (! subItems.getUnchecked(i)->isFullyOpen())
  1060. return false;
  1061. return true;
  1062. }
  1063. void TreeViewItem::restoreToDefaultOpenness()
  1064. {
  1065. if (openness != opennessDefault && ownerView != nullptr)
  1066. {
  1067. setOpen (ownerView->defaultOpenness);
  1068. openness = opennessDefault;
  1069. }
  1070. }
  1071. bool TreeViewItem::isSelected() const noexcept
  1072. {
  1073. return selected;
  1074. }
  1075. void TreeViewItem::deselectAllRecursively (TreeViewItem* itemToIgnore)
  1076. {
  1077. if (this != itemToIgnore)
  1078. setSelected (false, false);
  1079. for (int i = 0; i < subItems.size(); ++i)
  1080. subItems.getUnchecked(i)->deselectAllRecursively (itemToIgnore);
  1081. }
  1082. void TreeViewItem::setSelected (const bool shouldBeSelected,
  1083. const bool deselectOtherItemsFirst,
  1084. const NotificationType notify)
  1085. {
  1086. if (shouldBeSelected && ! canBeSelected())
  1087. return;
  1088. if (deselectOtherItemsFirst)
  1089. getTopLevelItem()->deselectAllRecursively (this);
  1090. if (shouldBeSelected != selected)
  1091. {
  1092. selected = shouldBeSelected;
  1093. if (ownerView != nullptr)
  1094. ownerView->repaint();
  1095. if (notify != dontSendNotification)
  1096. itemSelectionChanged (shouldBeSelected);
  1097. }
  1098. }
  1099. void TreeViewItem::paintItem (Graphics&, int, int)
  1100. {
  1101. }
  1102. void TreeViewItem::paintOpenCloseButton (Graphics& g, const Rectangle<float>& area, Colour backgroundColour, bool isMouseOver)
  1103. {
  1104. getOwnerView()->getLookAndFeel()
  1105. .drawTreeviewPlusMinusBox (g, area, backgroundColour, isOpen(), isMouseOver);
  1106. }
  1107. void TreeViewItem::paintHorizontalConnectingLine (Graphics& g, const Line<float>& line)
  1108. {
  1109. g.setColour (ownerView->findColour (TreeView::linesColourId));
  1110. g.drawLine (line);
  1111. }
  1112. void TreeViewItem::paintVerticalConnectingLine (Graphics& g, const Line<float>& line)
  1113. {
  1114. g.setColour (ownerView->findColour (TreeView::linesColourId));
  1115. g.drawLine (line);
  1116. }
  1117. void TreeViewItem::itemClicked (const MouseEvent&)
  1118. {
  1119. }
  1120. void TreeViewItem::itemDoubleClicked (const MouseEvent&)
  1121. {
  1122. if (mightContainSubItems())
  1123. setOpen (! isOpen());
  1124. }
  1125. void TreeViewItem::itemSelectionChanged (bool)
  1126. {
  1127. }
  1128. String TreeViewItem::getTooltip()
  1129. {
  1130. return String::empty;
  1131. }
  1132. var TreeViewItem::getDragSourceDescription()
  1133. {
  1134. return var();
  1135. }
  1136. bool TreeViewItem::isInterestedInFileDrag (const StringArray&)
  1137. {
  1138. return false;
  1139. }
  1140. void TreeViewItem::filesDropped (const StringArray& /*files*/, int /*insertIndex*/)
  1141. {
  1142. }
  1143. bool TreeViewItem::isInterestedInDragSource (const DragAndDropTarget::SourceDetails& /*dragSourceDetails*/)
  1144. {
  1145. return false;
  1146. }
  1147. void TreeViewItem::itemDropped (const DragAndDropTarget::SourceDetails& /*dragSourceDetails*/, int /*insertIndex*/)
  1148. {
  1149. }
  1150. Rectangle<int> TreeViewItem::getItemPosition (const bool relativeToTreeViewTopLeft) const noexcept
  1151. {
  1152. const int indentX = getIndentX();
  1153. int width = itemWidth;
  1154. if (ownerView != nullptr && width < 0)
  1155. width = ownerView->viewport->getViewWidth() - indentX;
  1156. Rectangle<int> r (indentX, y, jmax (0, width), totalHeight);
  1157. if (relativeToTreeViewTopLeft && ownerView != nullptr)
  1158. r -= ownerView->viewport->getViewPosition();
  1159. return r;
  1160. }
  1161. void TreeViewItem::treeHasChanged() const noexcept
  1162. {
  1163. if (ownerView != nullptr)
  1164. ownerView->itemsChanged();
  1165. }
  1166. void TreeViewItem::repaintItem() const
  1167. {
  1168. if (ownerView != nullptr && areAllParentsOpen())
  1169. {
  1170. Rectangle<int> r (getItemPosition (true));
  1171. r.setLeft (0);
  1172. ownerView->viewport->repaint (r);
  1173. }
  1174. }
  1175. bool TreeViewItem::areAllParentsOpen() const noexcept
  1176. {
  1177. return parentItem == nullptr
  1178. || (parentItem->isOpen() && parentItem->areAllParentsOpen());
  1179. }
  1180. void TreeViewItem::updatePositions (int newY)
  1181. {
  1182. y = newY;
  1183. itemHeight = getItemHeight();
  1184. totalHeight = itemHeight;
  1185. itemWidth = getItemWidth();
  1186. totalWidth = jmax (itemWidth, 0) + getIndentX();
  1187. if (isOpen())
  1188. {
  1189. newY += totalHeight;
  1190. for (int i = 0; i < subItems.size(); ++i)
  1191. {
  1192. TreeViewItem* const ti = subItems.getUnchecked(i);
  1193. ti->updatePositions (newY);
  1194. newY += ti->totalHeight;
  1195. totalHeight += ti->totalHeight;
  1196. totalWidth = jmax (totalWidth, ti->totalWidth);
  1197. }
  1198. }
  1199. }
  1200. TreeViewItem* TreeViewItem::getDeepestOpenParentItem() noexcept
  1201. {
  1202. TreeViewItem* result = this;
  1203. TreeViewItem* item = this;
  1204. while (item->parentItem != nullptr)
  1205. {
  1206. item = item->parentItem;
  1207. if (! item->isOpen())
  1208. result = item;
  1209. }
  1210. return result;
  1211. }
  1212. void TreeViewItem::setOwnerView (TreeView* const newOwner) noexcept
  1213. {
  1214. ownerView = newOwner;
  1215. for (int i = subItems.size(); --i >= 0;)
  1216. subItems.getUnchecked(i)->setOwnerView (newOwner);
  1217. }
  1218. int TreeViewItem::getIndentX() const noexcept
  1219. {
  1220. int x = ownerView->rootItemVisible ? 1 : 0;
  1221. if (! ownerView->openCloseButtonsVisible)
  1222. --x;
  1223. for (TreeViewItem* p = parentItem; p != nullptr; p = p->parentItem)
  1224. ++x;
  1225. return x * ownerView->getIndentSize();
  1226. }
  1227. void TreeViewItem::setDrawsInLeftMargin (bool canDrawInLeftMargin) noexcept
  1228. {
  1229. drawsInLeftMargin = canDrawInLeftMargin;
  1230. }
  1231. namespace TreeViewHelpers
  1232. {
  1233. static int calculateDepth (const TreeViewItem* item, const bool rootIsVisible) noexcept
  1234. {
  1235. jassert (item != nullptr);
  1236. int depth = rootIsVisible ? 0 : -1;
  1237. for (const TreeViewItem* p = item->getParentItem(); p != nullptr; p = p->getParentItem())
  1238. ++depth;
  1239. return depth;
  1240. }
  1241. }
  1242. bool TreeViewItem::areLinesDrawn() const
  1243. {
  1244. return drawLinesSet ? drawLinesInside
  1245. : (ownerView != nullptr && ownerView->getLookAndFeel().areLinesDrawnForTreeView (*ownerView));
  1246. }
  1247. void TreeViewItem::paintRecursively (Graphics& g, int width)
  1248. {
  1249. jassert (ownerView != nullptr);
  1250. if (ownerView == nullptr)
  1251. return;
  1252. const int indent = getIndentX();
  1253. const int itemW = itemWidth < 0 ? width - indent : itemWidth;
  1254. {
  1255. Graphics::ScopedSaveState ss (g);
  1256. g.setOrigin (indent, 0);
  1257. if (g.reduceClipRegion (drawsInLeftMargin ? -indent : 0, 0,
  1258. drawsInLeftMargin ? itemW + indent : itemW, itemHeight))
  1259. {
  1260. if (isSelected())
  1261. g.fillAll (ownerView->findColour (TreeView::selectedItemBackgroundColourId));
  1262. paintItem (g, itemW, itemHeight);
  1263. }
  1264. }
  1265. const float halfH = itemHeight * 0.5f;
  1266. const int indentWidth = ownerView->getIndentSize();
  1267. const int depth = TreeViewHelpers::calculateDepth (this, ownerView->rootItemVisible);
  1268. if (depth >= 0 && ownerView->openCloseButtonsVisible)
  1269. {
  1270. float x = (depth + 0.5f) * indentWidth;
  1271. const bool parentLinesDrawn = parentItem != nullptr && parentItem->areLinesDrawn();
  1272. if (parentLinesDrawn)
  1273. paintVerticalConnectingLine (g, Line<float> (x, 0, x, isLastOfSiblings() ? halfH : (float) itemHeight));
  1274. if (parentLinesDrawn || (parentItem == nullptr && areLinesDrawn()))
  1275. paintHorizontalConnectingLine (g, Line<float> (x, halfH, x + indentWidth / 2, halfH));
  1276. {
  1277. TreeViewItem* p = parentItem;
  1278. int d = depth;
  1279. while (p != nullptr && --d >= 0)
  1280. {
  1281. x -= (float) indentWidth;
  1282. if ((p->parentItem == nullptr || p->parentItem->areLinesDrawn()) && ! p->isLastOfSiblings())
  1283. p->paintVerticalConnectingLine (g, Line<float> (x, 0, x, (float) itemHeight));
  1284. p = p->parentItem;
  1285. }
  1286. }
  1287. if (mightContainSubItems())
  1288. paintOpenCloseButton (g, Rectangle<float> ((float) (depth * indentWidth), 0, (float) indentWidth, (float) itemHeight),
  1289. Colours::white,
  1290. ownerView->viewport->getContentComp()->isMouseOverButton (this));
  1291. }
  1292. if (isOpen())
  1293. {
  1294. const Rectangle<int> clip (g.getClipBounds());
  1295. for (int i = 0; i < subItems.size(); ++i)
  1296. {
  1297. TreeViewItem* const ti = subItems.getUnchecked(i);
  1298. const int relY = ti->y - y;
  1299. if (relY >= clip.getBottom())
  1300. break;
  1301. if (relY + ti->totalHeight >= clip.getY())
  1302. {
  1303. Graphics::ScopedSaveState ss (g);
  1304. g.setOrigin (0, relY);
  1305. if (g.reduceClipRegion (0, 0, width, ti->totalHeight))
  1306. ti->paintRecursively (g, width);
  1307. }
  1308. }
  1309. }
  1310. }
  1311. bool TreeViewItem::isLastOfSiblings() const noexcept
  1312. {
  1313. return parentItem == nullptr
  1314. || parentItem->subItems.getLast() == this;
  1315. }
  1316. int TreeViewItem::getIndexInParent() const noexcept
  1317. {
  1318. return parentItem == nullptr ? 0
  1319. : parentItem->subItems.indexOf (this);
  1320. }
  1321. TreeViewItem* TreeViewItem::getTopLevelItem() noexcept
  1322. {
  1323. return parentItem == nullptr ? this
  1324. : parentItem->getTopLevelItem();
  1325. }
  1326. int TreeViewItem::getNumRows() const noexcept
  1327. {
  1328. int num = 1;
  1329. if (isOpen())
  1330. {
  1331. for (int i = subItems.size(); --i >= 0;)
  1332. num += subItems.getUnchecked(i)->getNumRows();
  1333. }
  1334. return num;
  1335. }
  1336. TreeViewItem* TreeViewItem::getItemOnRow (int index) noexcept
  1337. {
  1338. if (index == 0)
  1339. return this;
  1340. if (index > 0 && isOpen())
  1341. {
  1342. --index;
  1343. for (int i = 0; i < subItems.size(); ++i)
  1344. {
  1345. TreeViewItem* const item = subItems.getUnchecked(i);
  1346. if (index == 0)
  1347. return item;
  1348. const int numRows = item->getNumRows();
  1349. if (numRows > index)
  1350. return item->getItemOnRow (index);
  1351. index -= numRows;
  1352. }
  1353. }
  1354. return nullptr;
  1355. }
  1356. TreeViewItem* TreeViewItem::findItemRecursively (int targetY) noexcept
  1357. {
  1358. if (isPositiveAndBelow (targetY, totalHeight))
  1359. {
  1360. const int h = itemHeight;
  1361. if (targetY < h)
  1362. return this;
  1363. if (isOpen())
  1364. {
  1365. targetY -= h;
  1366. for (int i = 0; i < subItems.size(); ++i)
  1367. {
  1368. TreeViewItem* const ti = subItems.getUnchecked(i);
  1369. if (targetY < ti->totalHeight)
  1370. return ti->findItemRecursively (targetY);
  1371. targetY -= ti->totalHeight;
  1372. }
  1373. }
  1374. }
  1375. return nullptr;
  1376. }
  1377. int TreeViewItem::countSelectedItemsRecursively (int depth) const noexcept
  1378. {
  1379. int total = isSelected() ? 1 : 0;
  1380. if (depth != 0)
  1381. for (int i = subItems.size(); --i >= 0;)
  1382. total += subItems.getUnchecked(i)->countSelectedItemsRecursively (depth - 1);
  1383. return total;
  1384. }
  1385. TreeViewItem* TreeViewItem::getSelectedItemWithIndex (int index) noexcept
  1386. {
  1387. if (isSelected())
  1388. {
  1389. if (index == 0)
  1390. return this;
  1391. --index;
  1392. }
  1393. if (index >= 0)
  1394. {
  1395. for (int i = 0; i < subItems.size(); ++i)
  1396. {
  1397. TreeViewItem* const item = subItems.getUnchecked(i);
  1398. if (TreeViewItem* const found = item->getSelectedItemWithIndex (index))
  1399. return found;
  1400. index -= item->countSelectedItemsRecursively (-1);
  1401. }
  1402. }
  1403. return nullptr;
  1404. }
  1405. int TreeViewItem::getRowNumberInTree() const noexcept
  1406. {
  1407. if (parentItem != nullptr && ownerView != nullptr)
  1408. {
  1409. int n = 1 + parentItem->getRowNumberInTree();
  1410. int ourIndex = parentItem->subItems.indexOf (this);
  1411. jassert (ourIndex >= 0);
  1412. while (--ourIndex >= 0)
  1413. n += parentItem->subItems [ourIndex]->getNumRows();
  1414. if (parentItem->parentItem == nullptr
  1415. && ! ownerView->rootItemVisible)
  1416. --n;
  1417. return n;
  1418. }
  1419. return 0;
  1420. }
  1421. void TreeViewItem::setLinesDrawnForSubItems (const bool drawLines) noexcept
  1422. {
  1423. drawLinesInside = drawLines;
  1424. drawLinesSet = true;
  1425. }
  1426. TreeViewItem* TreeViewItem::getNextVisibleItem (const bool recurse) const noexcept
  1427. {
  1428. if (recurse && isOpen() && subItems.size() > 0)
  1429. return subItems [0];
  1430. if (parentItem != nullptr)
  1431. {
  1432. const int nextIndex = parentItem->subItems.indexOf (this) + 1;
  1433. if (nextIndex >= parentItem->subItems.size())
  1434. return parentItem->getNextVisibleItem (false);
  1435. return parentItem->subItems [nextIndex];
  1436. }
  1437. return nullptr;
  1438. }
  1439. String TreeViewItem::getItemIdentifierString() const
  1440. {
  1441. String s;
  1442. if (parentItem != nullptr)
  1443. s = parentItem->getItemIdentifierString();
  1444. return s + "/" + getUniqueName().replaceCharacter ('/', '\\');
  1445. }
  1446. TreeViewItem* TreeViewItem::findItemFromIdentifierString (const String& identifierString)
  1447. {
  1448. const String thisId ("/" + getUniqueName());
  1449. if (thisId == identifierString)
  1450. return this;
  1451. if (identifierString.startsWith (thisId + "/"))
  1452. {
  1453. const String remainingPath (identifierString.substring (thisId.length()));
  1454. const bool wasOpen = isOpen();
  1455. setOpen (true);
  1456. for (int i = subItems.size(); --i >= 0;)
  1457. if (TreeViewItem* item = subItems.getUnchecked(i)->findItemFromIdentifierString (remainingPath))
  1458. return item;
  1459. setOpen (wasOpen);
  1460. }
  1461. return nullptr;
  1462. }
  1463. void TreeViewItem::restoreOpennessState (const XmlElement& e)
  1464. {
  1465. if (e.hasTagName ("CLOSED"))
  1466. {
  1467. setOpen (false);
  1468. }
  1469. else if (e.hasTagName ("OPEN"))
  1470. {
  1471. setOpen (true);
  1472. Array<TreeViewItem*> items;
  1473. items.addArray (subItems);
  1474. forEachXmlChildElement (e, n)
  1475. {
  1476. const String id (n->getStringAttribute ("id"));
  1477. for (int i = 0; i < items.size(); ++i)
  1478. {
  1479. TreeViewItem* const ti = items.getUnchecked(i);
  1480. if (ti->getUniqueName() == id)
  1481. {
  1482. ti->restoreOpennessState (*n);
  1483. items.remove (i);
  1484. break;
  1485. }
  1486. }
  1487. }
  1488. for (int i = 0; i < items.size(); ++i)
  1489. items.getUnchecked(i)->restoreToDefaultOpenness();
  1490. }
  1491. }
  1492. XmlElement* TreeViewItem::getOpennessState() const
  1493. {
  1494. return getOpennessState (true);
  1495. }
  1496. XmlElement* TreeViewItem::getOpennessState (const bool canReturnNull) const
  1497. {
  1498. const String name (getUniqueName());
  1499. if (name.isNotEmpty())
  1500. {
  1501. XmlElement* e;
  1502. if (isOpen())
  1503. {
  1504. if (canReturnNull && ownerView != nullptr && ownerView->defaultOpenness && isFullyOpen())
  1505. return nullptr;
  1506. e = new XmlElement ("OPEN");
  1507. for (int i = subItems.size(); --i >= 0;)
  1508. e->prependChildElement (subItems.getUnchecked(i)->getOpennessState (true));
  1509. }
  1510. else
  1511. {
  1512. if (canReturnNull && ownerView != nullptr && ! ownerView->defaultOpenness)
  1513. return nullptr;
  1514. e = new XmlElement ("CLOSED");
  1515. }
  1516. e->setAttribute ("id", name);
  1517. return e;
  1518. }
  1519. // trying to save the openness for an element that has no name - this won't
  1520. // work because it needs the names to identify what to open.
  1521. jassertfalse;
  1522. return nullptr;
  1523. }
  1524. //==============================================================================
  1525. TreeViewItem::OpennessRestorer::OpennessRestorer (TreeViewItem& item)
  1526. : treeViewItem (item),
  1527. oldOpenness (item.getOpennessState())
  1528. {
  1529. }
  1530. TreeViewItem::OpennessRestorer::~OpennessRestorer()
  1531. {
  1532. if (oldOpenness != nullptr)
  1533. treeViewItem.restoreOpennessState (*oldOpenness);
  1534. }