/*
* Jalview - A Sequence Alignment Editor and Viewer ($$Version-Rel$$)
* Copyright (C) $$Year-Rel$$ The Jalview Authors
*
* This file is part of Jalview.
*
* Jalview is free software: you can redistribute it and/or
* modify it under the terms of the GNU General Public License
* as published by the Free Software Foundation, either version 3
* of the License, or (at your option) any later version.
*
* Jalview is distributed in the hope that it will be useful, but
* WITHOUT ANY WARRANTY; without even the implied warranty
* of MERCHANTABILITY or FITNESS FOR A PARTICULAR
* PURPOSE. See the GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with Jalview. If not, see .
* The Jalview Authors are detailed in the 'AUTHORS' file.
*/
package jalview.io;
import jalview.analysis.Conservation;
import jalview.api.AlignViewportI;
import jalview.datamodel.AlignmentAnnotation;
import jalview.datamodel.AlignmentI;
import jalview.datamodel.Annotation;
import jalview.datamodel.ColumnSelection;
import jalview.datamodel.GraphLine;
import jalview.datamodel.HiddenColumns;
import jalview.datamodel.HiddenSequences;
import jalview.datamodel.SequenceGroup;
import jalview.datamodel.SequenceI;
import jalview.schemes.ColourSchemeI;
import jalview.schemes.ColourSchemeProperty;
import jalview.util.ColorUtils;
import java.awt.Color;
import java.io.BufferedReader;
import java.io.FileReader;
import java.io.InputStreamReader;
import java.io.StringReader;
import java.net.URL;
import java.util.ArrayList;
import java.util.BitSet;
import java.util.Enumeration;
import java.util.Hashtable;
import java.util.List;
import java.util.Map;
import java.util.StringTokenizer;
import java.util.Vector;
public class AnnotationFile
{
public AnnotationFile()
{
init();
}
/**
* character used to write newlines
*/
protected String newline = System.getProperty("line.separator");
/**
* set new line string and reset the output buffer
*
* @param nl
*/
public void setNewlineString(String nl)
{
newline = nl;
init();
}
public String getNewlineString()
{
return newline;
}
StringBuffer text;
private void init()
{
text = new StringBuffer("JALVIEW_ANNOTATION" + newline + "# Created: "
+ new java.util.Date() + newline + newline);
refSeq = null;
refSeqId = null;
}
/**
* convenience method for pre-2.9 annotation files which have no view, hidden
* columns or hidden row keywords.
*
* @param annotations
* @param list
* @param properties
* @return annotation file as a string.
*/
public String printAnnotations(AlignmentAnnotation[] annotations,
List list, Hashtable properties)
{
return printAnnotations(annotations, list, properties, null, null,
null);
}
/**
* hold all the information about a particular view definition read from or
* written out in an annotations file.
*/
public class ViewDef
{
// TODO this class is not used - remove?
public final String viewname;
public final HiddenSequences hidseqs;
public final HiddenColumns hiddencols;
public final Hashtable hiddenRepSeqs;
public ViewDef(String vname, HiddenSequences hseqs, HiddenColumns hcols,
Hashtable hRepSeqs)
{
this.viewname = vname;
this.hidseqs = hseqs;
this.hiddencols = hcols;
this.hiddenRepSeqs = hRepSeqs;
}
}
/**
* Prepare an annotation file given a set of annotations, groups, alignment
* properties and views.
*
* @param annotations
* @param list
* @param properties
* @param views
* @return annotation file
*/
public String printAnnotations(AlignmentAnnotation[] annotations,
List list, Hashtable properties, HiddenColumns cs,
AlignmentI al, ViewDef view)
{
if (view != null)
{
if (view.viewname != null)
{
text.append("VIEW_DEF\t" + view.viewname + "\n");
}
if (list == null)
{
// list = view.visibleGroups;
}
if (cs == null)
{
cs = view.hiddencols;
}
if (al == null)
{
// add hidden rep sequences.
}
}
// first target - store and restore all settings for a view.
if (al != null && al.hasSeqrep())
{
text.append("VIEW_SETREF\t" + al.getSeqrep().getName() + "\n");
}
if (cs != null && cs.hasHiddenColumns())
{
text.append("VIEW_HIDECOLS\t");
String regions = cs.regionsToString(",", "-");
text.append(regions);
text.append("\n");
}
// TODO: allow efficient recovery of annotation data shown in several
// different views
if (annotations != null)
{
boolean oneColour = true;
AlignmentAnnotation row;
String comma;
SequenceI refSeq = null;
SequenceGroup refGroup = null;
StringBuffer colours = new StringBuffer();
StringBuffer graphLine = new StringBuffer();
StringBuffer rowprops = new StringBuffer();
Hashtable graphGroup = new Hashtable<>();
Hashtable graphGroup_refs = new Hashtable<>();
BitSet graphGroupSeen = new BitSet();
java.awt.Color color;
for (int i = 0; i < annotations.length; i++)
{
row = annotations[i];
if (!row.visible && !row.hasScore() && !(row.graphGroup > -1
&& graphGroupSeen.get(row.graphGroup)))
{
continue;
}
color = null;
oneColour = true;
// mark any sequence references for the row
writeSequence_Ref(refSeq, row.sequenceRef);
refSeq = row.sequenceRef;
// mark any group references for the row
writeGroup_Ref(refGroup, row.groupRef);
refGroup = row.groupRef;
boolean hasGlyphs = row.hasIcons, hasLabels = row.hasText,
hasValues = row.hasScore, hasText = false;
// lookahead to check what the annotation row object actually contains.
for (int j = 0; row.annotations != null
&& j < row.annotations.length
&& (!hasGlyphs || !hasLabels || !hasValues); j++)
{
if (row.annotations[j] != null)
{
hasLabels |= (row.annotations[j].displayCharacter != null
&& row.annotations[j].displayCharacter.length() > 0
&& !row.annotations[j].displayCharacter.equals(" "));
hasGlyphs |= (row.annotations[j].secondaryStructure != 0
&& row.annotations[j].secondaryStructure != ' ');
hasValues |= (!Float.isNaN(row.annotations[j].value)); // NaNs can't
// be
// rendered..
hasText |= (row.annotations[j].description != null
&& row.annotations[j].description.length() > 0);
}
}
if (row.graph == AlignmentAnnotation.NO_GRAPH)
{
text.append("NO_GRAPH\t");
hasValues = false; // only secondary structure
// hasLabels = false; // and annotation description string.
}
else
{
if (row.graph == AlignmentAnnotation.BAR_GRAPH)
{
text.append("BAR_GRAPH\t");
hasGlyphs = false; // no secondary structure
}
else if (row.graph == AlignmentAnnotation.LINE_GRAPH)
{
hasGlyphs = false; // no secondary structure
text.append("LINE_GRAPH\t");
}
if (row.getThreshold() != null)
{
graphLine.append("GRAPHLINE\t");
graphLine.append(row.label);
graphLine.append("\t");
graphLine.append(row.getThreshold().value);
graphLine.append("\t");
graphLine.append(row.getThreshold().label);
graphLine.append("\t");
graphLine.append(jalview.util.Format
.getHexString(row.getThreshold().colour));
graphLine.append(newline);
}
if (row.graphGroup > -1)
{
graphGroupSeen.set(row.graphGroup);
Integer key = new Integer(row.graphGroup);
if (graphGroup.containsKey(key))
{
graphGroup.put(key, graphGroup.get(key) + "\t" + row.label);
}
else
{
graphGroup_refs.put(key, new Object[] { refSeq, refGroup });
graphGroup.put(key, row.label);
}
}
}
text.append(row.label + "\t");
if (row.description != null)
{
text.append(row.description + "\t");
}
for (int j = 0; row.annotations != null
&& j < row.annotations.length; j++)
{
if (refSeq != null
&& jalview.util.Comparison.isGap(refSeq.getCharAt(j)))
{
continue;
}
if (row.annotations[j] != null)
{
comma = "";
if (hasGlyphs) // could be also hasGlyphs || ...
{
text.append(comma);
if (row.annotations[j].secondaryStructure != ' ')
{
// only write out the field if its not whitespace.
text.append(row.annotations[j].secondaryStructure);
}
comma = ",";
}
if (hasValues)
{
if (!Float.isNaN(row.annotations[j].value))
{
text.append(comma + row.annotations[j].value);
}
else
{
// System.err.println("Skipping NaN - not valid value.");
text.append(comma + 0f);// row.annotations[j].value);
}
comma = ",";
}
if (hasLabels)
{
// TODO: labels are emitted after values for bar graphs.
if // empty labels are allowed, so
(row.annotations[j].displayCharacter != null
&& row.annotations[j].displayCharacter.length() > 0
&& !row.annotations[j].displayCharacter.equals(" "))
{
text.append(comma + row.annotations[j].displayCharacter);
comma = ",";
}
}
if (hasText)
{
if (row.annotations[j].description != null
&& row.annotations[j].description.length() > 0
&& !row.annotations[j].description
.equals(row.annotations[j].displayCharacter))
{
text.append(comma + row.annotations[j].description);
comma = ",";
}
}
if (color != null && !color.equals(row.annotations[j].colour))
{
oneColour = false;
}
color = row.annotations[j].colour;
if (row.annotations[j].colour != null
&& row.annotations[j].colour != java.awt.Color.black)
{
text.append(comma + "[" + jalview.util.Format
.getHexString(row.annotations[j].colour) + "]");
comma = ",";
}
}
text.append("|");
}
if (row.hasScore())
{
text.append("\t" + row.score);
}
text.append(newline);
if (color != null && color != java.awt.Color.black && oneColour)
{
colours.append("COLOUR\t");
colours.append(row.label);
colours.append("\t");
colours.append(jalview.util.Format.getHexString(color));
colours.append(newline);
}
if (row.scaleColLabel || row.showAllColLabels
|| row.centreColLabels)
{
rowprops.append("ROWPROPERTIES\t");
rowprops.append(row.label);
rowprops.append("\tscaletofit=");
rowprops.append(row.scaleColLabel);
rowprops.append("\tshowalllabs=");
rowprops.append(row.showAllColLabels);
rowprops.append("\tcentrelabs=");
rowprops.append(row.centreColLabels);
rowprops.append(newline);
}
if (graphLine.length() > 0)
{
text.append(graphLine.toString());
graphLine.setLength(0);
}
}
text.append(newline);
text.append(colours.toString());
if (graphGroup.size() > 0)
{
SequenceI oldRefSeq = refSeq;
SequenceGroup oldRefGroup = refGroup;
for (Map.Entry combine_statement : graphGroup
.entrySet())
{
Object[] seqRefAndGroup = graphGroup_refs
.get(combine_statement.getKey());
writeSequence_Ref(refSeq, (SequenceI) seqRefAndGroup[0]);
refSeq = (SequenceI) seqRefAndGroup[0];
writeGroup_Ref(refGroup, (SequenceGroup) seqRefAndGroup[1]);
refGroup = (SequenceGroup) seqRefAndGroup[1];
text.append("COMBINE\t");
text.append(combine_statement.getValue());
text.append(newline);
}
writeSequence_Ref(refSeq, oldRefSeq);
refSeq = oldRefSeq;
writeGroup_Ref(refGroup, oldRefGroup);
refGroup = oldRefGroup;
}
text.append(rowprops.toString());
}
if (list != null)
{
printGroups(list);
}
if (properties != null)
{
text.append(newline);
text.append(newline);
text.append("ALIGNMENT");
Enumeration en = properties.keys();
while (en.hasMoreElements())
{
String key = en.nextElement().toString();
text.append("\t");
text.append(key);
text.append("=");
text.append(properties.get(key));
}
// TODO: output alignment visualization settings here if required
// iterate through one or more views, defining, marking columns and rows
// as visible/hidden, and emmitting view properties.
// View specific annotation is
}
return text.toString();
}
private Object writeGroup_Ref(SequenceGroup refGroup,
SequenceGroup next_refGroup)
{
if (next_refGroup == null)
{
if (refGroup != null)
{
text.append(newline);
text.append("GROUP_REF\t");
text.append("ALIGNMENT");
text.append(newline);
}
return true;
}
else
{
if (refGroup == null || refGroup != next_refGroup)
{
text.append(newline);
text.append("GROUP_REF\t");
text.append(next_refGroup.getName());
text.append(newline);
return true;
}
}
return false;
}
private boolean writeSequence_Ref(SequenceI refSeq, SequenceI next_refSeq)
{
if (next_refSeq == null)
{
if (refSeq != null)
{
text.append(newline);
text.append("SEQUENCE_REF\t");
text.append("ALIGNMENT");
text.append(newline);
return true;
}
}
else
{
if (refSeq == null || refSeq != next_refSeq)
{
text.append(newline);
text.append("SEQUENCE_REF\t");
text.append(next_refSeq.getName());
text.append(newline);
return true;
}
}
return false;
}
protected void printGroups(List list)
{
SequenceI seqrep = null;
for (SequenceGroup sg : list)
{
if (!sg.hasSeqrep())
{
text.append("SEQUENCE_GROUP\t" + sg.getName() + "\t"
+ (sg.getStartRes() + 1) + "\t" + (sg.getEndRes() + 1)
+ "\t" + "-1\t");
seqrep = null;
}
else
{
seqrep = sg.getSeqrep();
text.append("SEQUENCE_REF\t");
text.append(seqrep.getName());
text.append(newline);
text.append("SEQUENCE_GROUP\t");
text.append(sg.getName());
text.append("\t");
text.append((seqrep.findPosition(sg.getStartRes())));
text.append("\t");
text.append((seqrep.findPosition(sg.getEndRes())));
text.append("\t");
text.append("-1\t");
}
for (int s = 0; s < sg.getSize(); s++)
{
text.append(sg.getSequenceAt(s).getName());
text.append("\t");
}
text.append(newline);
text.append("PROPERTIES\t");
text.append(sg.getName());
text.append("\t");
if (sg.getDescription() != null)
{
text.append("description=");
text.append(sg.getDescription());
text.append("\t");
}
if (sg.cs != null)
{
text.append("colour=");
text.append(ColourSchemeProperty
.getColourName(sg.cs.getColourScheme()));
text.append("\t");
if (sg.cs.getThreshold() != 0)
{
text.append("pidThreshold=");
text.append(sg.cs.getThreshold());
}
if (sg.cs.conservationApplied())
{
text.append("consThreshold=");
text.append(sg.cs.getConservationInc());
text.append("\t");
}
}
text.append("outlineColour=");
text.append(jalview.util.Format.getHexString(sg.getOutlineColour()));
text.append("\t");
text.append("displayBoxes=");
text.append(sg.getDisplayBoxes());
text.append("\t");
text.append("displayText=");
text.append(sg.getDisplayText());
text.append("\t");
text.append("colourText=");
text.append(sg.getColourText());
text.append("\t");
text.append("showUnconserved=");
text.append(sg.getShowNonconserved());
text.append("\t");
if (sg.textColour != java.awt.Color.black)
{
text.append("textCol1=");
text.append(jalview.util.Format.getHexString(sg.textColour));
text.append("\t");
}
if (sg.textColour2 != java.awt.Color.white)
{
text.append("textCol2=");
text.append(jalview.util.Format.getHexString(sg.textColour2));
text.append("\t");
}
if (sg.thresholdTextColour != 0)
{
text.append("textColThreshold=");
text.append(sg.thresholdTextColour);
text.append("\t");
}
if (sg.idColour != null)
{
text.append("idColour=");
text.append(jalview.util.Format.getHexString(sg.idColour));
text.append("\t");
}
if (sg.isHidereps())
{
text.append("hide=true\t");
}
if (sg.isHideCols())
{
text.append("hidecols=true\t");
}
if (seqrep != null)
{
// terminate the last line and clear the sequence ref for the group
text.append(newline);
text.append("SEQUENCE_REF");
}
text.append(newline);
text.append(newline);
}
}
SequenceI refSeq = null;
String refSeqId = null;
public boolean annotateAlignmentView(AlignViewportI viewport, String file,
DataSourceType protocol)
{
ColumnSelection colSel = viewport.getColumnSelection();
HiddenColumns hidden = viewport.getAlignment().getHiddenColumns();
if (colSel == null)
{
colSel = new ColumnSelection();
}
if (hidden == null)
{
hidden = new HiddenColumns();
}
boolean rslt = readAnnotationFile(viewport.getAlignment(), hidden, file,
protocol);
if (rslt && (colSel.hasSelectedColumns() || hidden.hasHiddenColumns()))
{
viewport.setColumnSelection(colSel);
viewport.getAlignment().setHiddenColumns(hidden);
}
return rslt;
}
public boolean readAnnotationFile(AlignmentI al, String file,
DataSourceType sourceType)
{
return readAnnotationFile(al, null, file, sourceType);
}
public boolean readAnnotationFile(AlignmentI al, HiddenColumns hidden,
String file, DataSourceType sourceType)
{
BufferedReader in = null;
try
{
if (sourceType == DataSourceType.FILE)
{
in = new BufferedReader(new FileReader(file));
}
else if (sourceType == DataSourceType.URL)
{
URL url = new URL(file);
in = new BufferedReader(new InputStreamReader(url.openStream()));
}
else if (sourceType == DataSourceType.PASTE)
{
in = new BufferedReader(new StringReader(file));
}
else if (sourceType == DataSourceType.CLASSLOADER)
{
java.io.InputStream is = getClass().getResourceAsStream("/" + file);
if (is != null)
{
in = new BufferedReader(new java.io.InputStreamReader(is));
}
}
if (in != null)
{
return parseAnnotationFrom(al, hidden, in);
}
} catch (Exception ex)
{
ex.printStackTrace();
System.out.println("Problem reading annotation file: " + ex);
if (nlinesread > 0)
{
System.out.println("Last read line " + nlinesread + ": '" + lastread
+ "' (first 80 chars) ...");
}
return false;
}
return false;
}
long nlinesread = 0;
String lastread = "";
private static String GRAPHLINE = "GRAPHLINE", COMBINE = "COMBINE";
public boolean parseAnnotationFrom(AlignmentI al, HiddenColumns hidden,
BufferedReader in) throws Exception
{
nlinesread = 0;
ArrayList