← Back to branch summary

~ubuntu-branches/ubuntu/utopic/figtree/utopic

« back to all changes in this revision

Viewing changes to src/figtree/treeviewer/DefaultTreeViewer.java

  • Committer: Bazaar Package Importer
  • Author(s): Andreas Tille
  • Date: 2011-02-21 08:17:38 UTC
  • Revision ID: james.westby@ubuntu.com-20110221081738-80pe2ulo8rg7up10
Tags: upstream-1.3.1
Import upstream version 1.3.1

Show diffs side-by-side

added added

removed removed

Lines of Context:
src/figtree/treeviewer/DefaultTreeViewer.java
 
1
/*
 
2
 * AlignmentPanel.java
 
3
 *
 
4
 * (c) 2002-2005 BEAST Development Core Team
 
5
 *
 
6
 * This package may be distributed under the
 
7
 * Lesser Gnu Public Licence (LGPL)
 
8
 */
 
9
 
 
10
package figtree.treeviewer;
 
11
 
 
12
import jebl.evolution.graphs.Node;
 
13
import jebl.evolution.taxa.Taxon;
 
14
import jebl.evolution.trees.*;
 
15
import figtree.treeviewer.decorators.Decorator;
 
16
import figtree.treeviewer.painters.LabelPainter;
 
17
import figtree.treeviewer.painters.ScalePainter;
 
18
import figtree.treeviewer.painters.ScaleGridPainter;
 
19
import figtree.treeviewer.painters.NodeBarPainter;
 
20
import figtree.treeviewer.treelayouts.TreeLayout;
 
21
import jam.panels.StatusProvider;
 
22
 
 
23
import javax.swing.*;
 
24
import java.awt.*;
 
25
import java.awt.print.PageFormat;
 
26
import java.awt.print.PrinterException;
 
27
import java.util.*;
 
28
import java.util.List;
 
29
import java.util.regex.PatternSyntaxException;
 
30
 
 
31
/**
 
32
 * @author Andrew Rambaut
 
33
 * @version $Id: DefaultTreeViewer.java 822 2007-10-26 13:50:26Z rambaut $
 
34
 */
 
35
public class DefaultTreeViewer extends TreeViewer {
 
36
    private final static double MAX_ZOOM = 20;
 
37
    private final static double MAX_VERTICAL_EXPANSION = 20;
 
38
 
 
39
    /**
 
40
     * Creates new TreeViewer
 
41
     */
 
42
    public DefaultTreeViewer() {
 
43
        setLayout(new BorderLayout());
 
44
 
 
45
        this.treePane = new TreePane();
 
46
        treePane.setAutoscrolls(true); //enable synthetic drag events
 
47
 
 
48
        treePane.addTreePaneListener(new TreePaneListener() {
 
49
            public void treePaneSettingsChanged() {
 
50
                fireTreeSettingsChanged();
 
51
            }
 
52
        });
 
53
 
 
54
        JScrollPane scrollPane = new JScrollPane(treePane, JScrollPane.VERTICAL_SCROLLBAR_ALWAYS, JScrollPane.HORIZONTAL_SCROLLBAR_ALWAYS);
 
55
        scrollPane.setMinimumSize(new Dimension(150, 150));
 
56
 
 
57
        scrollPane.setBorder(null);
 
58
        viewport = scrollPane.getViewport();
 
59
 
 
60
        add(scrollPane, BorderLayout.CENTER);
 
61
 
 
62
        // This overrides MouseListener and MouseMotionListener to allow selection in the TreePane -
 
63
        // It installs itself within the constructor.
 
64
        treePaneSelector = new TreePaneSelector(treePane);
 
65
 
 
66
        treePaneRollOver = new TreePaneRollOver(treePane);
 
67
 
 
68
        setFocusable(true);
 
69
 
 
70
    }
 
71
 
 
72
    public void setTree(Tree tree) {
 
73
        trees.clear();
 
74
        addTree(tree);
 
75
        showTree(0);
 
76
    }
 
77
 
 
78
    public void setTrees(Collection<? extends Tree> trees) {
 
79
        this.trees.clear();
 
80
        for (Tree tree : trees) {
 
81
            addTree(tree);
 
82
        }
 
83
        showTree(0);
 
84
    }
 
85
 
 
86
    public void addTree(Tree tree) {
 
87
        this.trees.add(tree);
 
88
 
 
89
        if (treePane.getTipLabelPainter() != null) {
 
90
            treePane.getTipLabelPainter().setupAttributes(trees);
 
91
        }
 
92
 
 
93
        if (treePane.getBranchLabelPainter() != null) {
 
94
            treePane.getBranchLabelPainter().setupAttributes(trees);
 
95
        }
 
96
 
 
97
        if (treePane.getNodeLabelPainter() != null) {
 
98
            treePane.getNodeLabelPainter().setupAttributes(trees);
 
99
        }
 
100
 
 
101
        if (treePane.getNodeBarPainter() != null) {
 
102
            treePane.getNodeBarPainter().setupAttributes(trees);
 
103
        }
 
104
    }
 
105
 
 
106
    public void addTrees(Collection<? extends Tree> trees) {
 
107
        int count = getTreeCount();
 
108
        for (Tree tree : trees) {
 
109
            addTree(tree);
 
110
        }
 
111
        showTree(count);
 
112
    }
 
113
 
 
114
    public List<Tree> getTrees() {
 
115
        return trees;
 
116
    }
 
117
 
 
118
    public Tree getCurrentTree() {
 
119
        return trees.get(currentTreeIndex);
 
120
    }
 
121
 
 
122
    public List<Tree> getTreesAsViewed() {
 
123
        List<Tree> treeAsDisplayed = new ArrayList<Tree>();
 
124
        for (Tree originalTree : trees) {
 
125
            treeAsDisplayed.add(treePane.constructTransformedTree((RootedTree)originalTree));
 
126
        }
 
127
        return treeAsDisplayed;
 
128
    }
 
129
 
 
130
    public int getCurrentTreeIndex() {
 
131
        return currentTreeIndex;
 
132
    }
 
133
 
 
134
    public int getTreeCount() {
 
135
        if (trees == null) return 0;
 
136
        return trees.size();
 
137
    }
 
138
 
 
139
    public StatusProvider getStatusProvider() {
 
140
        return treePaneRollOver;
 
141
    }
 
142
 
 
143
    public void showTree(int index) {
 
144
        Tree tree = trees.get(index);
 
145
        if (tree instanceof RootedTree) {
 
146
            treePane.setTree((RootedTree)tree);
 
147
        } else {
 
148
            treePane.setTree(Utils.rootTheTree(tree));
 
149
        }
 
150
 
 
151
        currentTreeIndex = index;
 
152
        fireTreeChanged();
 
153
    }
 
154
 
 
155
    public void showNextTree() {
 
156
        if (currentTreeIndex < trees.size() - 1) {
 
157
            showTree(currentTreeIndex + 1);
 
158
        }
 
159
    }
 
160
 
 
161
    public void showPreviousTree() {
 
162
        if (currentTreeIndex > 0) {
 
163
            showTree(currentTreeIndex - 1);
 
164
        }
 
165
    }
 
166
 
 
167
    public void setTreeLayout(TreeLayout treeLayout) {
 
168
        treePane.setTreeLayout(treeLayout);
 
169
        fireTreeSettingsChanged();
 
170
    }
 
171
 
 
172
    private boolean zoomPending = false;
 
173
    private double zoom = 0.0, verticalExpansion = 0.0;
 
174
 
 
175
    public void setZoom(double zoom) {
 
176
        this.zoom = zoom * MAX_ZOOM;
 
177
        refreshZoom();
 
178
    }
 
179
 
 
180
    public void setVerticalExpansion(double verticalExpansion) {
 
181
        this.verticalExpansion = verticalExpansion * MAX_VERTICAL_EXPANSION;
 
182
        refreshZoom();
 
183
    }
 
184
 
 
185
    public boolean verticalExpansionAllowed() {
 
186
        return !treePane.maintainAspectRatio();
 
187
    }
 
188
 
 
189
    public void setTimeScale(TimeScale timeScale) {
 
190
        treePane.setTimeScale(timeScale);
 
191
    }
 
192
 
 
193
    private void refreshZoom() {
 
194
        setZoom(zoom, zoom + verticalExpansion);
 
195
    }
 
196
 
 
197
    private void setZoom(double xZoom, double yZoom) {
 
198
 
 
199
        Dimension viewportSize = viewport.getViewSize();
 
200
        Point position = viewport.getViewPosition();
 
201
 
 
202
        Dimension extentSize = viewport.getExtentSize();
 
203
        double w = extentSize.getWidth() * (1.0 + xZoom);
 
204
        double h = extentSize.getHeight() * (1.0 + yZoom);
 
205
 
 
206
        Dimension newSize = new Dimension((int) w, (int) h);
 
207
        treePane.setPreferredSize(newSize);
 
208
 
 
209
        double cx = position.getX() + (0.5 * extentSize.getWidth());
 
210
        double cy = position.getY() + (0.5 * extentSize.getHeight());
 
211
 
 
212
        double rx = ((double) newSize.getWidth()) / viewportSize.getWidth();
 
213
        double ry = ((double) newSize.getHeight()) / viewportSize.getHeight();
 
214
 
 
215
        double px = (cx * rx) - (extentSize.getWidth() / 2.0);
 
216
        double py = (cy * ry) - (extentSize.getHeight() / 2.0);
 
217
 
 
218
        Point newPosition = new Point((int) px, (int) py);
 
219
        viewport.setViewPosition(newPosition);
 
220
        treePane.revalidate();
 
221
    }
 
222
 
 
223
    public boolean hasSelection() {
 
224
        return treePane.hasSelection();
 
225
    }
 
226
 
 
227
    public Set<Node> getSelectedNodes() {
 
228
        return treePane.getSelectedNodes();
 
229
    }
 
230
 
 
231
    public Set<Node> getSelectedTips() {
 
232
        return treePane.getSelectedTips();
 
233
    }
 
234
 
 
235
    public void selectTaxa(String attributeName, TextSearchType searchType, String searchString, boolean caseSensitive) {
 
236
        if (treePane.getTree() == null) {
 
237
            return;
 
238
        }
 
239
 
 
240
        treePane.clearSelection();
 
241
 
 
242
        String query = searchString;
 
243
 
 
244
        if (searchType != TextSearchType.REG_EX) {
 
245
            query = (caseSensitive ? searchString : searchString.toUpperCase());
 
246
            query = query.trim();
 
247
        }
 
248
 
 
249
        Tree tree = treePane.getTree();
 
250
 
 
251
        for (Node node : tree.getExternalNodes()) {
 
252
            Taxon taxon = tree.getTaxon(node);
 
253
 
 
254
            if (attributeName == null) {
 
255
                Object target = taxon.getName();
 
256
                if (matchesItem(searchType, target, query, caseSensitive)) {
 
257
                    treePane.addSelectedTip(node);
 
258
                    break;
 
259
                }
 
260
                for (String name : taxon.getAttributeNames()) {
 
261
                    target = taxon.getAttribute(name);
 
262
                    if (matchesItem(searchType, target, query, caseSensitive)) {
 
263
                        treePane.addSelectedTip(node);
 
264
                        break;
 
265
                    }
 
266
                }
 
267
            } else {
 
268
                Object target;
 
269
                if (attributeName.equals("!name")) {
 
270
                    target = taxon.getName();
 
271
                } else {
 
272
                    target = taxon.getAttribute(attributeName);
 
273
                }
 
274
                if (matchesItem(searchType, target, query, caseSensitive)) {
 
275
                    treePane.addSelectedTip(node);
 
276
                }
 
277
            }
 
278
        }
 
279
    }
 
280
 
 
281
    public void selectNodes(String attributeName, TextSearchType searchType, String searchString, boolean caseSensitive) {
 
282
        treePane.clearSelection();
 
283
 
 
284
        String query = searchString;
 
285
 
 
286
        if (searchType != TextSearchType.REG_EX) {
 
287
            query = (caseSensitive ? searchString : searchString.toUpperCase());
 
288
            query = query.trim();
 
289
        }
 
290
 
 
291
        Tree tree = treePane.getTree();
 
292
 
 
293
        for (Node node : tree.getNodes()) {
 
294
            if (attributeName == null) {
 
295
                for (String name : node.getAttributeNames()) {
 
296
                    Object target = node.getAttribute(name);
 
297
                    if (matchesItem(searchType, target, query, caseSensitive)) {
 
298
                        treePane.addSelectedNode(node);
 
299
                        break;
 
300
                    }
 
301
                }
 
302
            } else {
 
303
                Object target = node.getAttribute(attributeName);
 
304
                if (matchesItem(searchType, target, query, caseSensitive)) {
 
305
                    treePane.addSelectedNode(node);
 
306
                }
 
307
            }
 
308
        }
 
309
    }
 
310
 
 
311
    private boolean matchesItem(TextSearchType searchType, Object object, String query, boolean caseSensitive) {
 
312
 
 
313
        if (object != null) {
 
314
            String target = (caseSensitive ? object.toString() : object.toString().toUpperCase());
 
315
            switch (searchType) {
 
316
                case CONTAINS:
 
317
                    if (target.contains(query)) {
 
318
                        return true;
 
319
                    }
 
320
                    break;
 
321
                case STARTS_WITH:
 
322
                    if (target.startsWith(query)) {
 
323
                        return true;
 
324
                    }
 
325
                    break;
 
326
                case ENDS_WITH:
 
327
                    if (target.endsWith(query)) {
 
328
                        return true;
 
329
                    }
 
330
                    break;
 
331
                case MATCHES:
 
332
                    if (target.equals(query)) {
 
333
                        return true;
 
334
                    }
 
335
                    break;
 
336
                case REG_EX:
 
337
                    try {
 
338
                        if (target.matches(query)) {
 
339
                            return true;
 
340
                        }
 
341
                    } catch (PatternSyntaxException pse) {
 
342
                        // ignore
 
343
                    }
 
344
                    break;
 
345
            }
 
346
        }
 
347
        return false;
 
348
    }
 
349
 
 
350
    public void selectTaxa(String attributeName, NumberSearchType searchType, Number searchValue) {
 
351
        treePane.clearSelection();
 
352
 
 
353
        RootedTree tree = treePane.getTree();
 
354
 
 
355
        for (Node node : tree.getExternalNodes()) {
 
356
            Object value = null;
 
357
            if (attributeName.equals("!length")) {
 
358
                value = tree.getLength(node);
 
359
            } else if (attributeName.equals("!height")) {
 
360
                value = tree.getHeight(node);
 
361
            } else {
 
362
                Taxon taxon = tree.getTaxon(node);
 
363
                value = taxon.getAttribute(attributeName);
 
364
            }
 
365
            if (matchesItem(value, searchType, searchValue)) {
 
366
                treePane.addSelectedTip(node);
 
367
            }
 
368
        }
 
369
    }
 
370
 
 
371
    public void selectNodes(String attributeName, NumberSearchType searchType, Number searchValue) {
 
372
        treePane.clearSelection();
 
373
 
 
374
        RootedTree tree = treePane.getTree();
 
375
 
 
376
        for (Node node : tree.getNodes()) {
 
377
            Object value = null;
 
378
            if (attributeName.equals("!length")) {
 
379
                value = tree.getLength(node);
 
380
            } else if (attributeName.equals("!height")) {
 
381
                value = tree.getHeight(node);
 
382
            } else {
 
383
                value = node.getAttribute(attributeName);
 
384
            }
 
385
            if (matchesItem(value, searchType, searchValue)) {
 
386
                treePane.addSelectedNode(node);
 
387
            }
 
388
        }
 
389
    }
 
390
 
 
391
    public void selectTaxa(Collection<String> taxonNames) {
 
392
        treePane.clearSelection();
 
393
 
 
394
        RootedTree tree = treePane.getTree();
 
395
 
 
396
        for (Node node : tree.getExternalNodes()) {
 
397
            Object value = null;
 
398
            if (taxonNames.contains(tree.getTaxon(node).getName())) {
 
399
                treePane.addSelectedTip(node);
 
400
            }
 
401
        }
 
402
    }
 
403
 
 
404
    private boolean matchesItem(Object item, NumberSearchType searchType, Number searchValue) {
 
405
        if (item != null && item instanceof Number) {
 
406
 
 
407
            Number value = (Number)item;
 
408
 
 
409
            switch (searchType) {
 
410
                case EQUALS:
 
411
                    if (value.equals(searchValue)) {
 
412
                        return true;
 
413
                    }
 
414
                    break;
 
415
                case EQUALS_OR_GREATER_THAN:
 
416
                    if (value.doubleValue() >= searchValue.doubleValue()) {
 
417
                        return true;
 
418
                    }
 
419
                    break;
 
420
                case EQUALS_OR_LESS_THAN:
 
421
                    if (value.doubleValue() <= searchValue.doubleValue()) {
 
422
                        return true;
 
423
                    }
 
424
                    break;
 
425
                case GREATER_THAN:
 
426
                    if (value.doubleValue() > searchValue.doubleValue()) {
 
427
                        return true;
 
428
                    }
 
429
                    break;
 
430
                case LESS_THAN:
 
431
                    if (value.doubleValue() < searchValue.doubleValue()) {
 
432
                        return true;
 
433
                    }
 
434
                    break;
 
435
                case NOT_EQUALS:
 
436
                    if (!searchValue.equals(value)) {
 
437
                        return true;
 
438
                    }
 
439
                    break;
 
440
            }
 
441
        }
 
442
        return false;
 
443
    }
 
444
 
 
445
    public void cartoonSelectedNodes() {
 
446
        treePane.cartoonSelectedNodes();
 
447
        fireTreeSettingsChanged();
 
448
    }
 
449
 
 
450
    public void collapseSelectedNodes() {
 
451
        treePane.collapseSelectedNodes();
 
452
        fireTreeSettingsChanged();
 
453
    }
 
454
 
 
455
    public void clearCollapsedNodes() {
 
456
        treePane.clearCollapsedNodes();
 
457
        fireTreeSettingsChanged();
 
458
    }
 
459
 
 
460
    public void hilightSelectedNodes(Color color) {
 
461
        treePane.hilightSelectedNodes(color);
 
462
        fireTreeSettingsChanged();
 
463
    }
 
464
 
 
465
    public void clearHilighting() {
 
466
        treePane.clearHilightedNodes();
 
467
        fireTreeSettingsChanged();
 
468
    }
 
469
 
 
470
    public void rerootOnSelectedBranch() {
 
471
        treePane.rerootOnSelectedBranch();
 
472
        fireTreeSettingsChanged();
 
473
    }
 
474
 
 
475
    public void clearRooting() {
 
476
        treePane.clearRooting();
 
477
        fireTreeSettingsChanged();
 
478
    }
 
479
 
 
480
    public void rotateSelectedNode() {
 
481
        treePane.rotateSelectedNode();
 
482
        fireTreeSettingsChanged();
 
483
    }
 
484
 
 
485
    public void clearRotations() {
 
486
        treePane.clearSelectedNodeRotations();
 
487
        fireTreeSettingsChanged();
 
488
    }
 
489
 
 
490
    public void annotateSelectedNodes(String name, Object value) {
 
491
        treePane.annotateSelectedNodes(name, value);
 
492
        fireTreeSettingsChanged();
 
493
    }
 
494
 
 
495
    public void annotateSelectedTips(String name, Object value) {
 
496
        treePane.annotateSelectedTips(name, value);
 
497
        fireTreeSettingsChanged();
 
498
    }
 
499
 
 
500
    public void clearAnnotation(String name) {
 
501
        treePane.clearSelectedNodeAnnotation(name);
 
502
        treePane.clearSelectedTipAnnotation(name);
 
503
        fireTreeSettingsChanged();
 
504
    }
 
505
 
 
506
    public void clearColouring() {
 
507
        treePane.clearSelectedNodeAnnotation("!color");
 
508
        treePane.clearSelectedTipAnnotation("!color");
 
509
        fireTreeSettingsChanged();
 
510
    }
 
511
 
 
512
    public void selectAll() {
 
513
        if (treePaneSelector.getSelectionMode() == TreePaneSelector.SelectionMode.TAXA) {
 
514
            treePane.selectAllTaxa();
 
515
        } else {
 
516
            treePane.selectAllNodes();
 
517
        }
 
518
    }
 
519
 
 
520
    public void clearSelectedTaxa() {
 
521
        treePane.clearSelection();
 
522
    }
 
523
 
 
524
    public void addTreeSelectionListener(TreeSelectionListener treeSelectionListener) {
 
525
        treePane.addTreeSelectionListener(treeSelectionListener);
 
526
    }
 
527
 
 
528
    public void removeTreeSelectionListener(TreeSelectionListener treeSelectionListener) {
 
529
        treePane.removeTreeSelectionListener(treeSelectionListener);
 
530
    }
 
531
 
 
532
    public void setSelectionMode(TreePaneSelector.SelectionMode selectionMode) {
 
533
        TreePaneSelector.SelectionMode oldSelectionMode = treePaneSelector.getSelectionMode();
 
534
 
 
535
        if (selectionMode == oldSelectionMode) {
 
536
            return;
 
537
        }
 
538
 
 
539
        if (oldSelectionMode == TreePaneSelector.SelectionMode.TAXA) {
 
540
            treePane.selectNodesFromSelectedTips();
 
541
        } else if (selectionMode == TreePaneSelector.SelectionMode.TAXA) {
 
542
            treePane.selectTipsFromSelectedNodes();
 
543
        } else if (selectionMode == TreePaneSelector.SelectionMode.CLADE) {
 
544
            treePane.selectCladesFromSelectedNodes();
 
545
        }
 
546
        treePaneSelector.setSelectionMode(selectionMode);
 
547
    }
 
548
 
 
549
    public void setDragMode(TreePaneSelector.DragMode dragMode) {
 
550
        treePaneSelector.setDragMode(dragMode);
 
551
    }
 
552
 
 
553
    // A load of deligated method calls through to treePane (which is now hidden outside the package).
 
554
    public void setTipLabelPainter(LabelPainter<Node> tipLabelPainter) {
 
555
        treePane.setTipLabelPainter(tipLabelPainter);
 
556
        tipLabelPainter.setupAttributes(trees);
 
557
 
 
558
    }
 
559
 
 
560
    public void setNodeLabelPainter(LabelPainter<Node> nodeLabelPainter) {
 
561
        treePane.setNodeLabelPainter(nodeLabelPainter);
 
562
        nodeLabelPainter.setupAttributes(trees);
 
563
    }
 
564
 
 
565
    public void setNodeBarPainter(NodeBarPainter nodeBarPainter) {
 
566
        treePane.setNodeBarPainter(nodeBarPainter);
 
567
        nodeBarPainter.setupAttributes(trees);
 
568
    }
 
569
 
 
570
    public void setBranchLabelPainter(LabelPainter<Node> branchLabelPainter) {
 
571
        treePane.setBranchLabelPainter(branchLabelPainter);
 
572
        branchLabelPainter.setupAttributes(trees);
 
573
    }
 
574
 
 
575
    public void addScalePainter(ScalePainter scalePainter) {
 
576
        treePane.addScalePainter(scalePainter);
 
577
    }
 
578
 
 
579
    public void removeScalePainter(ScalePainter scalePainter) {
 
580
        treePane.removeScalePainter(scalePainter);
 
581
    }
 
582
 
 
583
    public void setScaleGridPainter(ScaleGridPainter scaleGridPainter) {
 
584
        treePane.setScaleGridPainter(scaleGridPainter);
 
585
    }
 
586
 
 
587
    public void setBranchDecorator(Decorator branchDecorator) {
 
588
        treePane.setBranchDecorator(branchDecorator);
 
589
    }
 
590
 
 
591
    public void setBranchColouringDecorator(String branchColouringAttribute, Decorator branchColouringDecorator) {
 
592
        treePane.setBranchColouringDecorator(branchColouringAttribute, branchColouringDecorator);
 
593
    }
 
594
 
 
595
    public void setNodeBackgroundDecorator(Decorator nodeBackgroundDecorator) {
 
596
        treePane.setNodeBackgroundDecorator(nodeBackgroundDecorator);
 
597
    }
 
598
    
 
599
    public void setSelectionColor(Color selectionColor) {
 
600
        treePane.setSelectionColor(selectionColor);
 
601
    }
 
602
 
 
603
    public Paint getSelectionPaint() {
 
604
        return treePane.getSelectionPaint();
 
605
    }
 
606
 
 
607
    public void setBranchStroke(BasicStroke branchStroke) {
 
608
        treePane.setBranchStroke(branchStroke);
 
609
    }
 
610
 
 
611
    public boolean isTransformBranchesOn() {
 
612
        return treePane.isTransformBranchesOn();
 
613
    }
 
614
 
 
615
    public TransformedRootedTree.Transform getBranchTransform() {
 
616
        return treePane.getBranchTransform();
 
617
    }
 
618
 
 
619
    public void setTransformBranchesOn(boolean transformBranchesOn) {
 
620
        treePane.setTransformBranchesOn(transformBranchesOn);
 
621
    }
 
622
 
 
623
    public void setBranchTransform(TransformedRootedTree.Transform transform) {
 
624
        treePane.setBranchTransform(transform);
 
625
    }
 
626
 
 
627
    public boolean isOrderBranchesOn() {
 
628
        return treePane.isOrderBranchesOn();
 
629
    }
 
630
 
 
631
    public SortedRootedTree.BranchOrdering getBranchOrdering() {
 
632
        return treePane.getBranchOrdering();
 
633
    }
 
634
 
 
635
    public void setOrderBranchesOn(boolean orderBranchesOn) {
 
636
        treePane.setOrderBranchesOn(orderBranchesOn);
 
637
    }
 
638
 
 
639
    public void setBranchOrdering(SortedRootedTree.BranchOrdering branchOrdering) {
 
640
        treePane.setBranchOrdering(branchOrdering);
 
641
    }
 
642
 
 
643
    public boolean isRootingOn() {
 
644
        return treePane.isRootingOn();
 
645
    }
 
646
 
 
647
    public TreePane.RootingType getRootingType() {
 
648
        return treePane.getRootingType();
 
649
    }
 
650
 
 
651
    public void setRootingOn(boolean rootingOn) {
 
652
        treePane.setRootingOn(rootingOn);
 
653
    }
 
654
 
 
655
    public void setRootingType(TreePane.RootingType rootingType) {
 
656
        treePane.setRootingType(rootingType);
 
657
    }
 
658
 
 
659
    public void setToolMode(TreePaneSelector.ToolMode toolMode) {
 
660
        treePaneSelector.setToolMode(toolMode);
 
661
    }
 
662
 
 
663
    public JComponent getContentPane() {
 
664
        return treePane;
 
665
    }
 
666
 
 
667
    public void paint(Graphics g) {
 
668
        if( zoomPending  ) {
 
669
            refreshZoom();
 
670
            zoomPending = false;
 
671
        }
 
672
        super.paint(g);
 
673
    }
 
674
 
 
675
    public int print(Graphics g, PageFormat pageFormat, int pageIndex) throws PrinterException {
 
676
        return treePane.print(g, pageFormat, pageIndex);
 
677
    }
 
678
 
 
679
    public void addTreeViewerListener(TreeViewerListener listener) {
 
680
        listeners.add(listener);
 
681
    }
 
682
 
 
683
    public void removeTreeViewerListener(TreeViewerListener listener) {
 
684
        listeners.remove(listener);
 
685
    }
 
686
 
 
687
    public void fireTreeChanged() {
 
688
        for (TreeViewerListener listener : listeners) {
 
689
            listener.treeChanged();
 
690
        }
 
691
    }
 
692
 
 
693
    public void fireTreeSettingsChanged() {
 
694
        for (TreeViewerListener listener : listeners) {
 
695
            listener.treeSettingsChanged();
 
696
        }
 
697
    }
 
698
 
 
699
    private java.util.List<TreeViewerListener> listeners = new ArrayList<TreeViewerListener>();
 
700
 
 
701
    private List<Tree> trees = new ArrayList<Tree>();
 
702
    private int currentTreeIndex = 0;
 
703
 
 
704
    protected TreePane treePane;
 
705
    protected TreePaneSelector treePaneSelector;
 
706
    protected TreePaneRollOver treePaneRollOver;
 
707
 
 
708
    protected JViewport viewport;
 
709
 
 
710
}
 
 
b'\\ No newline at end of file'