make sure TipDatesRandomWalker picks up correct taxon set when used i…

…n BEAuti. CompEvol/BeastFX#45
CompEvol · Oct 1, 2023 · 8d560c9 · 8d560c9
1 parent 2ef96e8
commit 8d560c9
Showing 1 changed file with 11 additions and 3 deletions.
diff --git a/src/beast/base/evolution/operator/TipDatesRandomWalker.java b/src/beast/base/evolution/operator/TipDatesRandomWalker.java
@@ -43,9 +43,17 @@ public void initAndValidate() {
 
         // determine taxon set to choose from
         if (m_taxonsetInput.get() != null) {
-            List<String> taxaNames = new ArrayList<>();
-            for (String taxon : treeInput.get().getTaxaNames()) {
-                taxaNames.add(taxon);
+            List<String> taxaNames;
+            if (treeInput.get().m_taxonset.get() != null) {
+            	// taxonset may have changed in BEAUti, and
+            	// internal tree state not updated yet, so use
+            	// taxonset input of tree directly
+            	taxaNames = treeInput.get().m_taxonset.get().asStringList();
+            } else {
+            	taxaNames = new ArrayList<>();
+            	for (String taxon : treeInput.get().getTaxaNames()) {
+            		taxaNames.add(taxon);
+            	}
             }
 
             List<String> set = m_taxonsetInput.get().asStringList();