X-Git-Url: http://source.jalview.org/gitweb/?a=blobdiff_plain;f=src%2Fjalview%2Fschemes%2FResidueColourScheme.java;fp=src%2Fjalview%2Fschemes%2FResidueColourScheme.java;h=03fc129a2e99b8747c35a3709ad5fae405b7c37f;hb=2595e9d4ee0dbbd3406a98c4e49a61ccde806479;hp=bca98cfd34a93eff76fe0562b20c16a777a2dde5;hpb=e20075ba805d744d7cc4976e2b8d5e5840fb0a8d;p=jalview.git diff --git a/src/jalview/schemes/ResidueColourScheme.java b/src/jalview/schemes/ResidueColourScheme.java index bca98cf..03fc129 100755 --- a/src/jalview/schemes/ResidueColourScheme.java +++ b/src/jalview/schemes/ResidueColourScheme.java @@ -20,46 +20,37 @@ */ package jalview.schemes; -import jalview.analysis.AAFrequency; -import jalview.analysis.Conservation; import jalview.datamodel.AnnotatedCollectionI; import jalview.datamodel.SequenceCollectionI; +import jalview.datamodel.SequenceGroup; import jalview.datamodel.SequenceI; -import jalview.util.MessageManager; +import jalview.util.Comparison; import java.awt.Color; -import java.util.Hashtable; import java.util.Map; /** - * DOCUMENT ME! - * - * @author $author$ - * @version $Revision$ + * Base class for residue-based colour schemes */ -public class ResidueColourScheme implements ColourSchemeI +public abstract class ResidueColourScheme implements ColourSchemeI { - final int[] symbolIndex; + public static final String NONE = "None"; - boolean conservationColouring = false; + public static final String USER_DEFINED = "User Defined"; - Color[] colors = null; + /* + * lookup up by character value e.g. 'G' to the colors array index + * e.g. if symbolIndex['K'] = 11 then colors[11] is the colour for K + */ + final int[] symbolIndex; - int threshold = 0; + /* + * colour for residue characters as indexed by symbolIndex + */ + Color[] colors = null; /* Set when threshold colouring to either pid_gaps or pid_nogaps */ - protected String ignoreGaps = AAFrequency.PID_GAPS; - - /** Consenus as a hashtable array */ - Hashtable[] consensus; - - /** Conservation string as a char array */ - char[] conservation; - - int conservationLength = 0; - - /** DOCUMENT ME!! */ - int inc = 30; + protected boolean ignoreGaps = false; /** * Creates a new ResidueColourScheme object. @@ -68,15 +59,11 @@ public class ResidueColourScheme implements ColourSchemeI * ResidueProperties.aaIndex) * @param colors * colours for symbols in sequences - * @param threshold - * threshold for conservation shading */ - public ResidueColourScheme(int[] aaOrnaIndex, Color[] colours, - int threshold) + public ResidueColourScheme(int[] aaOrnaIndex, Color[] colours) { symbolIndex = aaOrnaIndex; this.colors = colours; - this.threshold = threshold; } /** @@ -102,231 +89,116 @@ public class ResidueColourScheme implements ColourSchemeI */ public Color findColour(char c) { - return colors == null ? Color.white : colors[symbolIndex[c]]; - } - - @Override - public Color findColour(char c, int j, SequenceI seq) - { - Color currentColour; - - if (colors != null && symbolIndex != null && (threshold == 0) - || aboveThreshold(c, j)) - { - currentColour = colors[symbolIndex[c]]; - } - else - { - currentColour = Color.white; - } + Color colour = Color.white; - if (conservationColouring) + if (!Comparison.isGap(c) && colors != null && symbolIndex != null + && c < symbolIndex.length + && symbolIndex[c] < colors.length) { - currentColour = applyConservation(currentColour, j); + colour = colors[symbolIndex[c]]; } - return currentColour; - } - - /** - * Get the percentage threshold for this colour scheme - * - * @return Returns the percentage threshold - */ - public int getThreshold() - { - return threshold; + return colour; } /** - * DOCUMENT ME! - * - * @param ct - * DOCUMENT ME! + * Default is to call the overloaded method that ignores consensus. A colour + * scheme that depends on consensus (for example, Blosum62), should override + * this method instead. */ - public void setThreshold(int ct, boolean ignoreGaps) + @Override + public Color findColour(char c, int j, SequenceI seq, + String consensusResidue, float pid) { - threshold = ct; - if (ignoreGaps) - { - this.ignoreGaps = AAFrequency.PID_NOGAPS; - } - else - { - this.ignoreGaps = AAFrequency.PID_GAPS; - } + return findColour(c, j, seq); } /** - * DOCUMENT ME! + * Default implementation looks up the residue colour in a fixed scheme, or + * returns White if not found. Override this method for a colour scheme that + * depends on the column position or sequence. * - * @param s - * DOCUMENT ME! + * @param c * @param j - * DOCUMENT ME! - * - * @return DOCUMENT ME! + * @param seq + * @return */ - public boolean aboveThreshold(char c, int j) - { - if ('a' <= c && c <= 'z') - { - // TO UPPERCASE !!! - // Faster than toUpperCase - c -= ('a' - 'A'); - } - - if (consensus == null || consensus.length < j || consensus[j] == null) - { - return false; - } - - if ((((Integer) consensus[j].get(AAFrequency.MAXCOUNT)).intValue() != -1) - && consensus[j].contains(String.valueOf(c))) - { - if (((Float) consensus[j].get(ignoreGaps)).floatValue() >= threshold) - { - return true; - } - } - - return false; - } - - public boolean conservationApplied() + protected Color findColour(char c, int j, SequenceI seq) { - return conservationColouring; + return findColour(c); } @Override - public void setConservationApplied(boolean conservationApplied) - { - conservationColouring = conservationApplied; - } - - public void setConservationInc(int i) - { - inc = i; - } - - public int getConservationInc() + public void alignmentChanged(AnnotatedCollectionI alignment, + Map hiddenReps) { - return inc; } /** - * DOCUMENT ME! - * - * @param consensus - * DOCUMENT ME! + * Answers false if the colour scheme is nucleotide or peptide specific, and + * the data does not match, else true. Override to modify or extend this test + * as required. */ - public void setConsensus(Hashtable[] consensus) + @Override + public boolean isApplicableTo(AnnotatedCollectionI ac) { - if (consensus == null) + if (!isPeptideSpecific() && !isNucleotideSpecific()) { - return; + return true; } - - this.consensus = consensus; - } - - public void setConservation(Conservation cons) - { - if (cons == null) + if (ac == null) { - conservationColouring = false; - conservation = null; + return true; } - else + /* + * pop-up menu on selection group before group created + * (no alignment context) + */ + // TODO: add nucleotide flag to SequenceGroup? + if (ac instanceof SequenceGroup && ac.getContext() == null) { - conservationColouring = true; - int i, iSize = cons.getConsSequence().getLength(); - conservation = new char[iSize]; - for (i = 0; i < iSize; i++) - { - conservation[i] = cons.getConsSequence().getCharAt(i); - } - conservationLength = conservation.length; + return true; } + /* + * inspect the data context (alignment) for residue type + */ + boolean nucleotide = ac.isNucleotide(); + + /* + * does data type match colour scheme type? + */ + return (nucleotide && isNucleotideSpecific()) + || (!nucleotide && isPeptideSpecific()); } /** - * DOCUMENT ME! + * Answers true if the colour scheme is normally only for peptide data * - * @param s - * DOCUMENT ME! - * @param i - * DOCUMENT ME! - * - * @return DOCUMENT ME! + * @return */ - - Color applyConservation(Color currentColour, int i) + public boolean isPeptideSpecific() { - - if ((conservationLength > i) && (conservation[i] != '*') - && (conservation[i] != '+')) - { - if (jalview.util.Comparison.isGap(conservation[i])) - { - currentColour = Color.white; - } - else - { - float t = 11 - (conservation[i] - '0'); - if (t == 0) - { - return Color.white; - } - - int red = currentColour.getRed(); - int green = currentColour.getGreen(); - int blue = currentColour.getBlue(); - - int dr = 255 - red; - int dg = 255 - green; - int db = 255 - blue; - - dr *= t / 10f; - dg *= t / 10f; - db *= t / 10f; - - red += (inc / 20f) * dr; - green += (inc / 20f) * dg; - blue += (inc / 20f) * db; - - if (red > 255 || green > 255 || blue > 255) - { - currentColour = Color.white; - } - else - { - currentColour = new Color(red, green, blue); - } - } - } - return currentColour; + return false; } - @Override - public void alignmentChanged(AnnotatedCollectionI alignment, - Map hiddenReps) + /** + * Answers true if the colour scheme is normally only for nucleotide data + * + * @return + */ + public boolean isNucleotideSpecific() { + return false; } + /** + * Default method returns true. Override this to return false in colour + * schemes that are not determined solely by the sequence symbol. + */ @Override - public ColourSchemeI applyTo(AnnotatedCollectionI sg, - Map hiddenRepSequences) + public boolean isSimple() { - try - { - return getClass().newInstance(); - } catch (Exception q) - { - throw new Error(MessageManager.formatMessage( - "error.implementation_error_cannot_duplicate_colour_scheme", - new String[] { getClass().getName() }), q); - } + return true; } }