Charts As Modules

1 files changed, 23 insertions, 2 deletions

diff --git a/app/Module/InteractiveTreeModule.php b/app/Module/InteractiveTreeModule.php
index dc461efb16..c7de450c6b 100644
--- a/app/Module/InteractiveTreeModule.php
+++ b/app/Module/InteractiveTreeModule.php
@@ -21,12 +21,13 @@ use Fisharebest\Webtrees\Filter;
 use Fisharebest\Webtrees\I18N;
 use Fisharebest\Webtrees\Individual;
 use Fisharebest\Webtrees\Module\InteractiveTree\TreeView;
+use Fisharebest\Webtrees\Menu;
 
 /**
  * Class InteractiveTreeModule
  * Tip : you could change the number of generations loaded before ajax calls both in individual page and in treeview page to optimize speed and server load
  */
-class InteractiveTreeModule extends AbstractModule implements ModuleTabInterface {
+class InteractiveTreeModule extends AbstractModule implements ModuleTabInterface,ModuleChartInterface {
 	/** {@inheritdoc} */
 	public function getTitle() {
 		return /* I18N: Name of a module */ I18N::translate('Interactive tree');
@@ -70,7 +71,27 @@ class InteractiveTreeModule extends AbstractModule implements ModuleTabInterface
 	public function canLoadAjax() {
 		return true;
 	}
-
+	
+	/**
+	 * Return a menu item for this chart.
+	 *
+	 * @return Menu
+	 */
+	public function getChartMenu(Individual $individual) {
+		global $controller, $WT_TREE;
+		
+		return new Menu(
+			$this->getTitle(), 
+			'module.php?mod=tree&mod_action=treeview&rootid=' . $individual->getXref() . '&ged=' . $WT_TREE->getNameUrl(), 
+			'menu-chart-tree', 
+			array('rel' => 'nofollow')
+		);
+	}
+	
+	public function getBoxChartMenu(Individual $individual) {
+		return $this->getChartMenu($individual);
+	}
+	
 	/** {@inheritdoc} */
 	public function getPreLoadContent() {
 		// We cannot use jQuery("head").append(<link rel="stylesheet" ...as jQuery is not loaded at this time