/*................................................................................................*/
public void setPreferredTaxa(Taxa taxa) { // Copied from SampleTreesMultSources
if (taxa != currentTaxa) {
if (currentTaxa != null) currentTaxa.removeListener(this);
currentTaxa = taxa;
currentTaxa.addListener(this);
}
}
/**
* A request for the MesquiteModule to perform a command. It is passed two strings, the name of
* the command and the arguments. This should be overridden by any module that wants to respond to
* a command.
*/
public Object doCommand(String commandName, String arguments, CommandChecker checker) {
if (checker.compare(MesquiteModule.class, null, null, commandName, "paint")) {
MesquiteInteger io = new MesquiteInteger(0);
int column = MesquiteInteger.fromString(arguments, io);
int row = MesquiteInteger.fromString(arguments, io);
if (MesquiteInteger.isCombinable(row)) {
if (!MesquiteLong.isCombinable(currentColor)) removeColor(row, true);
else setColor(row, (int) currentColor);
}
} else if (checker.compare(
this.getClass(),
"Sets the color to be used to paint cells",
"[name of color]",
commandName,
"setColor")) {
int bc = ColorDistribution.standardColorNames.indexOf(parser.getFirstToken(arguments));
if (bc >= 0 && MesquiteLong.isCombinable(bc)) {
removeColor.setValue(false);
currentColor = bc;
savedColor = bc;
colorString = "Color " + ColorDistribution.standardColorNames.getValue(bc);
}
} else if (checker.compare(
this.getClass(),
"Sets the color of selected taxa",
"[name of color]",
commandName,
"setColorSelected")) {
int bc = ColorDistribution.standardColorNames.indexOf(parser.getFirstToken(arguments));
if (bc >= 0 && MesquiteLong.isCombinable(bc)) {
for (int it = 0; it < taxa.getNumTaxa(); it++) if (taxa.getSelected(it)) setColor(it, bc);
}
} else if (checker.compare(
this.getClass(), "Removes color from all the cells", null, commandName, "removeAllColor")) {
removeAllColor(true);
} else if (checker.compare(
this.getClass(),
"Sets the paint brush so that it removes colors from any cells touched",
null,
commandName,
"removeColor")) {
if (StringUtil.blank(arguments)) removeColor.setValue(!removeColor.getValue());
else removeColor.toggleValue(parser.getFirstToken(arguments));
if (removeColor.getValue()) {
colorString = "Remove color";
currentColor = MesquiteLong.unassigned;
} else {
colorString = "Color " + ColorDistribution.standardColorNames.getValue((int) currentColor);
currentColor = savedColor;
}
} else return super.doCommand(commandName, arguments, checker);
return null;
}
/*.................................................................................................................*/
void zapData(CharacterData data, int it) {
Taxa taxa = data.getTaxa();
if (it < 0 || it >= taxa.getNumTaxa()) return;
Associable tInfo = data.getTaxaInfo(false);
int myColumn = -1;
if (getEmployer() instanceof ListModule) {
myColumn = ((ListModule) getEmployer()).getMyColumn(this);
}
if (tInfo != null) tInfo.deassignAssociated(it);
for (int ic = 0; ic < data.getNumChars(); ic++) data.deassign(ic, it);
data.notifyListeners(this, new Notification(MesquiteListener.DATA_CHANGED));
outputInvalid();
parametersChanged();
}
/*.................................................................................................................*/
public void setTaxonTranslation(Taxa taxa) {
int countTaxa = 0;
for (int it = 0; it < data.getNumTaxa(); it++)
if ((!selectedTaxaOnly || taxa.getSelected(it)) && (data.hasDataForTaxon(it, false))) {
setTaxonNumberInTree(taxa, it, countTaxa);
countTaxa++;
} else setTaxonNumberInTree(taxa, it, -1);
}
/*...............................................................................................................*/
boolean selected(Taxa taxa, int it, int myColumn) {
if (taxa.getSelected(it)) {
return true;
}
if (table != null && myColumn >= 0) {
if (table.isCellSelectedAnyWay(myColumn, it)) return true;
}
return false;
}
public void doCalcs() {
if (bits != null) bits.clearAllBits();
if (taxa == null) return;
if (bits == null) bits = new Bits(taxa.getNumTaxa());
else bits.resetSize(taxa.getNumTaxa());
if (observedStates == null) {
tInfo = null;
observedStates = matrixSourceTask.getCurrentMatrix(taxa);
if (observedStates != null) {
captureCharacterDataFromObservedStates();
if (data != null) tInfo = data.getTaxaInfo(true);
}
}
if (observedStates == null) return;
for (int it = 0; it < taxa.getNumTaxa(); it++) {
if (hasData(it)) bits.setBit(it);
}
}
/*.................................................................................................................*/
public boolean transformTree(AdjustableTree tree, MesquiteString resultString, boolean notify) {
Taxa taxa = tree.getTaxa();
// counting how many excluded
int count = 0;
for (int i = 0; i < taxa.getNumTaxa(); i++) {
if (!tree.taxonInTree(i)) count++;
}
if (count == 0) {
discreetAlert(
"The tree contains all of the taxa available in the block of taxa. There are none available to be added");
return true;
}
// getting list of excluded
ListableVector excluded = new ListableVector(count);
count = 0;
for (int i = 0; i < taxa.getNumTaxa(); i++) {
if (!tree.taxonInTree(i)) {
excluded.addElement(taxa.getTaxon(i), false);
}
}
// presenting choice
Listable[] toInclude =
ListDialog.queryListMultiple(
containerOfModule(),
"Add taxa to tree",
"Select taxa to be added to the tree",
MesquiteString.helpString,
"Add",
false,
excluded,
null);
if (toInclude == null || toInclude.length == 0) return true;
for (int i = 0; i < toInclude.length; i++) {
int taxon = taxa.whichTaxonNumber((Taxon) toInclude[i]);
tree.graftTaxon(taxon, tree.getRoot(), false);
}
if (notify && tree instanceof Listened)
((Listened) tree)
.notifyListeners(this, new Notification(MesquiteListener.BRANCHES_REARRANGED));
return true;
}
public void calculateNumber(Taxon taxon, MesquiteNumber result, MesquiteString resultString) {
if (result == null) return;
clearResultAndLastResult(result);
Taxa taxa = taxon.getTaxa();
int it = taxa.whichTaxonNumber(taxon);
if (taxa != currentTaxa || observedStates == null) {
observedStates = matrixSourceTask.getCurrentMatrix(taxa);
currentTaxa = taxa;
}
if (observedStates == null || !(observedStates.getParentData() instanceof DNAData)) return;
DNAData data = (DNAData) observedStates.getParentData();
int count = data.getAminoAcidNumbers(it, ProteinData.TER, countEvenIfOthers.getValue());
if (result != null) result.setValue(count);
if (resultString != null)
resultString.setValue(
"Number of stop codons in taxon " + observedStates.getName() + ": " + count);
saveLastResult(result);
saveLastResultString(resultString);
}
/*.................................................................................................................*/
private void recordSourceProject() {
MesquiteString s = new MesquiteString();
s.setValue(getProject().getHomeFileName());
if (!QueryDialogs.queryString(
containerOfModule(),
"Name to Stamp",
"Indicate source file name to stamp on matrix rows",
s)) return;
int numMatrices = getProject().getNumberCharMatrices();
NameReference sourceRef = NameReference.getNameReference("SourceFile");
for (int im = 0; im < numMatrices; im++) {
CharacterData data = getProject().getCharacterMatrix(im);
Taxa taxa = data.getTaxa();
Associable tInfo = data.getTaxaInfo(true);
boolean anySelected = taxa.anySelected();
for (int it = 0; it < taxa.getNumTaxa(); it++) {
if (data.hasDataForTaxon(it) && (!anySelected || taxa.getSelected(it)))
tInfo.setAssociatedObject(sourceRef, it, s.getValue());
}
}
}
/*.................................................................................................................*/
public int[] getTaxonNumberTranslation(Taxa taxa) {
int max = -1;
for (int it = 0; it < taxa.getNumTaxa(); it++) {
long translateNumber = getTaxonNumberInTree(taxa, it);
if (MesquiteLong.isCombinable(translateNumber) && translateNumber > max) {
max = (int) translateNumber;
}
}
int[] translate = new int[max + 1];
for (int it = 0; it < data.getNumTaxa(); it++) {
long translateNumber = getTaxonNumberInTree(taxa, it);
if (MesquiteLong.isCombinable(translateNumber) && translateNumber >= 0) {
translate[(int) translateNumber] = it;
}
}
return translate;
}
private void setColor(int it, long c) {
if (it < 0) return;
taxa.setAssociatedLong(colorNameRef, it, c);
table.redrawCell(-1, it);
}
private void removeAllColor(boolean notify) {
for (int it = 0; it < taxa.getNumTaxa(); it++) setColor(it, MesquiteLong.unassigned);
if (notify) table.repaintAll();
}
/*.................................................................................................................*/
public Object doCommand(String commandName, String arguments, CommandChecker checker) {
if (checker.compare(
this.getClass(), "Returns the matrix source", null, commandName, "getMatrixSource")) {
return matrixSourceTask;
} else if (checker.compare(
this.getClass(), "Copies the data for selected taxon", null, commandName, "copyData")) {
if (observedStates == null) return null;
CharacterData data = observedStates.getParentData();
if (data == null) return null;
int it = MesquiteInteger.fromString(parser.getFirstToken(arguments));
if (MesquiteInteger.isCombinable(it)) {
StringBuffer sb = new StringBuffer();
data.copyDataFromRowIntoBuffer(it, sb);
if (StringUtil.notEmpty(sb.toString())) {
localCopyDataClipboard = sb.toString();
localCopyData = data;
localCopyDataTaxon = data.getTaxa().getTaxonName(it);
} else {
localCopyDataClipboard = null;
localCopyData = null;
localCopyDataTaxon = null;
}
}
return null;
} else if (checker.compare(
this.getClass(), "Pastes the data for selected taxon", null, commandName, "pasteData")) {
if (observedStates == null) return null;
CharacterData data = observedStates.getParentData();
if (data == null) return null;
int it = MesquiteInteger.fromString(parser.getFirstToken(arguments));
if (MesquiteInteger.isCombinable(it) && StringUtil.notEmpty(localCopyDataClipboard)) {
data.pasteDataFromStringIntoTaxon(it, localCopyDataClipboard);
}
return null;
} else if (checker.compare(
this.getClass(),
"Pastes the data for selected taxon",
null,
commandName,
"deleteDataTouched")) {
if (observedStates == null) return null;
CharacterData data = observedStates.getParentData();
if (data == null) return null;
int it = MesquiteInteger.fromString(parser.getFirstToken(arguments));
Debugg.println("prepare to delete row: " + it);
if (MesquiteInteger.isCombinable(it)) {
if (!AlertDialog.query(
containerOfModule(),
"Delete Data?",
"Are you sure you want to delete the data for taxon "
+ data.getTaxa().getTaxonName(it)
+ " in the matrix \""
+ data.getName()
+ "\"",
"No",
"Yes")) {
zapData(data, it);
}
}
return null;
} else if (checker.compare(
this.getClass(), "Deletes the data for selected taxa", null, commandName, "deleteData")) {
if (observedStates == null) return null;
captureCharacterDataFromObservedStates();
if (data == null) return null;
if (!AlertDialog.query(
containerOfModule(),
"Delete Data?",
"Are you sure you want to delete the data for these taxa in the matrix \""
+ data.getName()
+ "\"",
"No",
"Yes")) zapData(data);
return null;
} else if (checker.compare(
this.getClass(),
"deleteds () and anything between",
null,
commandName,
"deletePrepended")) {
if (observedStates == null || taxa == null) return null;
boolean anySelected = taxa.anySelected();
int myColumn = -1;
if (getEmployer() instanceof ListModule) {
myColumn = ((ListModule) getEmployer()).getMyColumn(this);
if (table != null)
anySelected = anySelected || table.anyCellSelectedInColumnAnyWay(myColumn);
}
for (int it = 0; it < taxa.getNumTaxa(); it++) {
if ((!anySelected || selected(taxa, it, myColumn))) {
String note = getNote(it);
while (!StringUtil.blank(note) && note.indexOf("(") >= 0) {
int start = note.indexOf("(");
int end = note.indexOf(")");
String firstBit = "";
if (start > 0) firstBit = note.substring(0, start);
note = firstBit + note.substring(end + 1, note.length());
}
setNote(it, note);
}
}
outputInvalid();
parametersChanged();
return null;
} else if (checker.compare(this.getClass(), "deletes *", null, commandName, "deleteStar")) {
if (observedStates == null || taxa == null) return null;
boolean anySelected = taxa.anySelected();
int myColumn = -1;
if (getEmployer() instanceof ListModule) {
myColumn = ((ListModule) getEmployer()).getMyColumn(this);
if (table != null)
anySelected = anySelected || table.anyCellSelectedInColumnAnyWay(myColumn);
}
for (int it = 0; it < taxa.getNumTaxa(); it++) {
if ((!anySelected || selected(taxa, it, myColumn))) {
String note = getNote(it);
while (!StringUtil.blank(note) && note.indexOf("*") >= 0) {
int start = note.indexOf("*");
String firstBit = "";
if (start > 0) firstBit = note.substring(0, start);
note = firstBit + note.substring(start + 1, note.length());
}
setNote(it, note);
}
}
outputInvalid();
parametersChanged();
return null;
} else if (checker.compare(
this.getClass(),
"Prepends to the note the sequence length (including N\'s and ?\'s) for the selected taxa",
null,
commandName,
"prependLength")) {
if (observedStates == null || taxa == null) return null;
boolean anySelected = taxa.anySelected();
int myColumn = -1;
if (getEmployer() instanceof ListModule) {
myColumn = ((ListModule) getEmployer()).getMyColumn(this);
if (table != null)
anySelected = anySelected || table.anyCellSelectedInColumnAnyWay(myColumn);
}
for (int it = 0; it < taxa.getNumTaxa(); it++) {
if (hasData(it) && (!anySelected || selected(taxa, it, myColumn))) {
String note = getNote(it);
if (StringUtil.blank(note)) note = "(" + sequenceLength(it) + ")";
else note = "(" + sequenceLength(it) + ") " + note;
setNote(it, note);
}
}
outputInvalid();
parametersChanged();
return null;
} else if (checker.compare(
this.getClass(),
"Prepends to the note the number of non-missing sites (not including N\'s and ?\'s) for the selected taxa",
null,
commandName,
"prependNumSites")) {
if (observedStates == null || taxa == null) return null;
boolean anySelected = taxa.anySelected();
int myColumn = -1;
if (getEmployer() instanceof ListModule) {
myColumn = ((ListModule) getEmployer()).getMyColumn(this);
if (table != null)
anySelected = anySelected || table.anyCellSelectedInColumnAnyWay(myColumn);
}
for (int it = 0; it < taxa.getNumTaxa(); it++) {
if (hasData(it) && (!anySelected || selected(taxa, it, myColumn))) {
String note = getNote(it);
if (StringUtil.blank(note)) note = "(" + numSites(it) + ")";
else note = "(" + numSites(it) + ") " + note;
setNote(it, note);
}
}
outputInvalid();
parametersChanged();
return null;
} else if (checker.compare(
this.getClass(),
"Deletes the notes for the selected taxa",
null,
commandName,
"deleteAnnotation")) {
if (observedStates == null || taxa == null) return null;
boolean anySelected = taxa.anySelected();
int myColumn = -1;
if (getEmployer() instanceof ListModule) {
myColumn = ((ListModule) getEmployer()).getMyColumn(this);
if (table != null)
anySelected = anySelected || table.anyCellSelectedInColumnAnyWay(myColumn);
}
for (int it = 0; it < taxa.getNumTaxa(); it++) {
if (hasData(it) && (!anySelected || selected(taxa, it, myColumn))) {
setNote(it, null);
}
}
outputInvalid();
parametersChanged();
return null;
} else return super.doCommand(commandName, arguments, checker);
}
public CharacterData makeCharacterData(CharMatrixManager manager, Taxa taxa) {
GeographicData data = new GeographicData(getMatrixManager(), taxa.getNumTaxa(), 0, taxa);
data.setItemsAs(this);
return data;
}
/*.................................................................................................................*/
public Tree getTrees(
TreeVector trees,
Taxa taxa,
MCharactersDistribution matrix,
long seed,
MesquiteDouble finalScore) {
if (!initializeGetTrees(CategoricalData.class, taxa, matrix)) return null;
setTNTSeed(seed);
isProtein = data instanceof ProteinData;
// David: if isDoomed() then module is closing down; abort somehow
// write data file
String tempDir =
MesquiteFileUtil.createDirectoryForFiles(
this, MesquiteFileUtil.IN_SUPPORT_DIR, "TNT", "-Run.");
if (tempDir == null) return null;
String dataFileName = "data.ss"; // replace this with actual file name?
String dataFilePath = tempDir + dataFileName;
FileInterpreterI exporter = ZephyrUtil.getFileInterpreter(this, "#InterpretTNT");
if (exporter == null) return null;
boolean fileSaved = false;
String translationTable = namer.getTranslationTable(taxa);
((InterpretHennig86Base) exporter).setTaxonNamer(namer);
fileSaved = ZephyrUtil.saveExportFile(this, exporter, dataFilePath, data, selectedTaxaOnly);
if (!fileSaved) return null;
String translationFileName = IOUtil.translationTableFileName;
setTaxonTranslation(taxa);
taxonNumberTranslation = getTaxonNumberTranslation(taxa);
namer.setNumberTranslationTable(taxonNumberTranslation);
setFileNames();
TaxaSelectionSet outgroupSet =
(TaxaSelectionSet) taxa.getSpecsSet(outgroupTaxSetString, TaxaSelectionSet.class);
int firstOutgroup = MesquiteInteger.unassigned;
if (outgroupSet != null) firstOutgroup = outgroupSet.firstBitOn();
formCommandFile(dataFileName, firstOutgroup);
logln("\n\nCommands given to TNT:");
logln(commands);
logln("");
MesquiteString arguments = new MesquiteString();
arguments.setValue(" proc " + commandsFileName);
String programCommand = externalProcRunner.getExecutableCommand();
int numInputFiles = 3;
String[] fileContents = new String[numInputFiles];
String[] fileNames = new String[numInputFiles];
for (int i = 0; i < numInputFiles; i++) {
fileContents[i] = "";
fileNames[i] = "";
}
fileContents[0] = MesquiteFile.getFileContentsAsString(dataFilePath);
fileNames[0] = dataFileName;
fileContents[1] = commands;
fileNames[1] = commandsFileName;
fileContents[2] = translationTable;
fileNames[2] = translationFileName;
// ----------//
boolean success =
runProgramOnExternalProcess(
programCommand, arguments, fileContents, fileNames, ownerModule.getName());
if (!isDoomed()) {
if (success) {
desuppressProjectPanelReset();
return retrieveTreeBlock(trees, finalScore); // here's where we actually process everything.
} else {
if (!beanWritten) postBean("unsuccessful [1]", false);
beanWritten = true;
}
}
desuppressProjectPanelReset();
if (data == null) data.decrementEditInhibition();
externalProcRunner.finalCleanup();
return null;
}
