X-Git-Url: http://source.jalview.org/gitweb/?a=blobdiff_plain;f=src%2Fjalview%2Fschemes%2FAnnotationColourGradient.java;h=e265598f6da0177e0b7a7a5c05dd4bc04535ad7a;hb=a45774ee31d9f35d4eff46d54d7deab719afb092;hp=58256122b6f8cfabdc6c279c933c85636173a2eb;hpb=df227ed4042d4c9fa2affb391fa09b662f3b8a2a;p=jalview.git diff --git a/src/jalview/schemes/AnnotationColourGradient.java b/src/jalview/schemes/AnnotationColourGradient.java index 5825612..e265598 100755 --- a/src/jalview/schemes/AnnotationColourGradient.java +++ b/src/jalview/schemes/AnnotationColourGradient.java @@ -1,133 +1,226 @@ -/* -* Jalview - A Sequence Alignment Editor and Viewer -* Copyright (C) 2005 AM Waterhouse, J Procter, G Barton, M Clamp, S Searle -* -* This program 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 2 -* of the License, or (at your option) any later version. -* -* This program 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 this program; if not, write to the Free Software -* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA -*/ -package jalview.schemes; - -import java.awt.*; - -import jalview.datamodel.*; - -public class AnnotationColourGradient extends ResidueColourScheme -{ - public static int NO_THRESHOLD = -1; - public static int BELOW_THRESHOLD = 0; - public static int ABOVE_THRESHOLD = 1; - - AlignmentAnnotation annotation; - int aboveAnnotationThreshold = -1; - - GraphLine annotationThreshold; - - float r1, g1, b1, rr, gg, bb, dr, dg, db; - float range; - - /** - * Creates a new AnnotationColourGradient object. - */ - public AnnotationColourGradient(AlignmentAnnotation annotation, - Color minColour, Color maxColour, int aboveThreshold) - { - this.annotation = annotation; - - aboveAnnotationThreshold = aboveThreshold; - - if(aboveThreshold!=NO_THRESHOLD && annotation.graphLines!=null) - annotationThreshold = annotation.getGraphLine(0); - - r1 = minColour.getRed(); - g1 = minColour.getGreen(); - b1 = minColour.getBlue(); - - rr = maxColour.getRed() - r1; - gg = maxColour.getGreen() - g1; - bb = maxColour.getBlue() - b1; - - range = annotation.graphMax - annotation.graphMin; - - } - - public Color getMinColour() - { - return new Color( (int) r1, (int) g1, (int) b1); - } - - public Color getMaxColour() - { - return new Color( (int) (r1 + rr), (int) (g1 + gg), (int) (b1 + bb)); - } - - /** - * DOCUMENT ME! - * - * @param n DOCUMENT ME! - * - * @return DOCUMENT ME! - */ - public Color findColour(String n) - { - System.out.println("AnnotationColourGradient findColour(string)"); - return Color.red; - } - - /** - * DOCUMENT ME! - * - * @param n DOCUMENT ME! - * @param j DOCUMENT ME! - * - * @return DOCUMENT ME! - */ - public Color findColour(String n, int j) - { - if ((threshold == 0) || aboveThreshold(n, j)) - { - if( j+1>annotation.annotations.length || annotation.annotations[j]==null) - currentColour = Color.white; - else - { - if( aboveAnnotationThreshold==NO_THRESHOLD - || (annotationThreshold!=null && aboveAnnotationThreshold==ABOVE_THRESHOLD && annotation.annotations[j].value>=annotationThreshold.value) - || (annotationThreshold!=null && aboveAnnotationThreshold==BELOW_THRESHOLD && annotation.annotations[j].value<=annotationThreshold.value)) - { - dr = rr * - ((annotation.annotations[j].value-annotation.graphMin) / range ) - +r1; - dg = gg * - ((annotation.annotations[j].value-annotation.graphMin) / range ) - +g1; - db = bb * - ((annotation.annotations[j].value-annotation.graphMin) / range ) - +b1; - - currentColour = new Color( (int) dr, (int) dg, (int) db); - } - else - currentColour = Color.white; - } - } - else - { - return Color.white; - } - - if(conservationColouring) - applyConservation(j); - - return currentColour; - } -} +/* + * Jalview - A Sequence Alignment Editor and Viewer (Version 2.7) + * Copyright (C) 2011 J Procter, AM Waterhouse, G Barton, M Clamp, S Searle + * + * 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 . + */ +package jalview.schemes; + +import java.awt.*; + +import jalview.datamodel.*; + +public class AnnotationColourGradient extends ResidueColourScheme +{ + public static final int NO_THRESHOLD = -1; + + public static final int BELOW_THRESHOLD = 0; + + public static final int ABOVE_THRESHOLD = 1; + + public AlignmentAnnotation annotation; + + int aboveAnnotationThreshold = -1; + + public boolean thresholdIsMinMax = false; + + GraphLine annotationThreshold; + + float r1, g1, b1, rr, gg, bb, dr, dg, db; + + ColourSchemeI colourScheme; + + public boolean predefinedColours = false; + + /** + * Creates a new AnnotationColourGradient object. + */ + public AnnotationColourGradient(AlignmentAnnotation annotation, + ColourSchemeI originalColour, int aboveThreshold) + { + if (originalColour instanceof AnnotationColourGradient) + { + colourScheme = ((AnnotationColourGradient) originalColour).colourScheme; + } + else + { + colourScheme = originalColour; + } + + this.annotation = annotation; + + aboveAnnotationThreshold = aboveThreshold; + + if (aboveThreshold != NO_THRESHOLD && annotation.threshold != null) + { + annotationThreshold = annotation.threshold; + } + } + + /** + * Creates a new AnnotationColourGradient object. + */ + public AnnotationColourGradient(AlignmentAnnotation annotation, + Color minColour, Color maxColour, int aboveThreshold) + { + this.annotation = annotation; + + aboveAnnotationThreshold = aboveThreshold; + + if (aboveThreshold != NO_THRESHOLD && annotation.threshold != null) + { + annotationThreshold = annotation.threshold; + } + + r1 = minColour.getRed(); + g1 = minColour.getGreen(); + b1 = minColour.getBlue(); + + rr = maxColour.getRed() - r1; + gg = maxColour.getGreen() - g1; + bb = maxColour.getBlue() - b1; + } + + public String getAnnotation() + { + return annotation.label; + } + + public int getAboveThreshold() + { + return aboveAnnotationThreshold; + } + + public float getAnnotationThreshold() + { + if (annotationThreshold == null) + { + return 0; + } + else + { + return annotationThreshold.value; + } + } + + public ColourSchemeI getBaseColour() + { + return colourScheme; + } + + public Color getMinColour() + { + return new Color((int) r1, (int) g1, (int) b1); + } + + public Color getMaxColour() + { + return new Color((int) (r1 + rr), (int) (g1 + gg), (int) (b1 + bb)); + } + + /** + * DOCUMENT ME! + * + * @param n + * DOCUMENT ME! + * + * @return DOCUMENT ME! + */ + public Color findColour(char c) + { + return Color.red; + } + + /** + * DOCUMENT ME! + * + * @param n + * DOCUMENT ME! + * @param j + * DOCUMENT ME! + * + * @return DOCUMENT ME! + */ + public Color findColour(char c, int j) + { + Color currentColour = Color.white; + + if ((threshold == 0) || aboveThreshold(c, j)) + { + if (j < annotation.annotations.length + && annotation.annotations[j] != null + && !jalview.util.Comparison.isGap(c)) + { + + if (predefinedColours) + { + if (annotation.annotations[j].colour != null) + return annotation.annotations[j].colour; + else + return currentColour; + } + + if (aboveAnnotationThreshold == NO_THRESHOLD + || (annotationThreshold != null + && aboveAnnotationThreshold == ABOVE_THRESHOLD && annotation.annotations[j].value >= annotationThreshold.value) + || (annotationThreshold != null + && aboveAnnotationThreshold == BELOW_THRESHOLD && annotation.annotations[j].value <= annotationThreshold.value)) + { + + float range = 1f; + if (thresholdIsMinMax + && annotation.threshold != null + && aboveAnnotationThreshold == ABOVE_THRESHOLD + && annotation.annotations[j].value > annotation.threshold.value) + { + range = (annotation.annotations[j].value - annotation.threshold.value) + / (annotation.graphMax - annotation.threshold.value); + } + else if (thresholdIsMinMax && annotation.threshold != null + && aboveAnnotationThreshold == BELOW_THRESHOLD + && annotation.annotations[j].value > annotation.graphMin) + { + range = (annotation.annotations[j].value - annotation.graphMin) + / (annotation.threshold.value - annotation.graphMin); + } + else + { + range = (annotation.annotations[j].value - annotation.graphMin) + / (annotation.graphMax - annotation.graphMin); + } + + if (colourScheme != null) + { + currentColour = colourScheme.findColour(c, j); + } + else if (range != 0) + { + dr = rr * range + r1; + dg = gg * range + g1; + db = bb * range + b1; + + currentColour = new Color((int) dr, (int) dg, (int) db); + } + } + } + } + + if (conservationColouring) + { + currentColour = applyConservation(currentColour, j); + } + + return currentColour; + } +}