1 package jalview.schemes;
3 import jalview.datamodel.AnnotatedCollectionI;
4 import jalview.datamodel.HiddenMarkovModel;
5 import jalview.datamodel.SequenceCollectionI;
6 import jalview.datamodel.SequenceI;
7 import jalview.util.ColorUtils;
8 import jalview.util.Comparison;
10 import java.awt.Color;
11 import java.util.List;
14 public class HMMERColourScheme extends ResidueColourScheme
17 AnnotatedCollectionI alignment;
19 HiddenMarkovModel hmm;
21 boolean peptideSpecific;
23 boolean nucleotideSpecific;
25 public HMMERColourScheme(HiddenMarkovModel markov)
29 public HMMERColourScheme()
35 public Color findColour(char symbol, int position, SequenceI seq,
36 String consensusResidue, float pid)
42 return findColour(symbol, position);
45 public Color findColour(char symbol, int position)
48 if (Comparison.isGap(symbol))
53 prob = hmm.getMatchEmissionProbability(position, symbol);
55 if ("amino".equals(hmm.getAlphabetType()))
57 if (!ResidueProperties.aminoBackgroundFrequencies.containsKey(symbol))
61 freq = ResidueProperties.aminoBackgroundFrequencies.get(symbol);
63 else if ("DNA".equals(hmm.getAlphabetType()))
65 if (!ResidueProperties.dnaBackgroundFrequencies.containsKey(symbol))
69 freq = ResidueProperties.dnaBackgroundFrequencies.get(symbol);
71 else if ("RNA".equals(hmm.getAlphabetType()))
73 if (!ResidueProperties.rnaBackgroundFrequencies.containsKey(symbol))
77 freq = ResidueProperties.rnaBackgroundFrequencies.get(symbol);
83 Double value = Math.log(prob / freq);
88 colour = ColorUtils.getGraduatedColour(value.floatValue(), 0,
89 Color.WHITE, 4.52f, Color.blue);
106 public void alignmentChanged(AnnotatedCollectionI collection,
107 Map<SequenceI, SequenceCollectionI> hiddenReps)
109 List<SequenceI> seqs = collection.getSequences();
110 for (SequenceI seq : seqs)
112 if (seq.getHMM() != null)
124 public ColourSchemeI getInstance(AnnotatedCollectionI sg,
125 Map<SequenceI, SequenceCollectionI> hiddenRepSequences)
127 HiddenMarkovModel markov = null;
128 List<SequenceI> seqs = sg.getSequences();
129 for (SequenceI seq : seqs)
131 if (seq.getHMM() != null)
133 markov = seq.getHMM();
137 HMMERColourScheme colour = new HMMERColourScheme(markov);
143 public boolean isApplicableTo(AnnotatedCollectionI ac)
150 public String getSchemeName()
153 return JalviewColourScheme.HMMER.name();
157 public boolean isSimple()
159 // TODO Auto-generated method stub