src/jalview/appletgui/RotatableCanvas.java

   1 /*
   2  * Jalview - A Sequence Alignment Editor and Viewer ($$Version-Rel$$)
   3  * Copyright (C) $$Year-Rel$$ The Jalview Authors
   4  *
   5  * This file is part of Jalview.
   6  *
   7  * Jalview is free software: you can redistribute it and/or
   8  * modify it under the terms of the GNU General Public License
   9  * as published by the Free Software Foundation, either version 3
  10  * of the License, or (at your option) any later version.
  11  *
  12  * Jalview is distributed in the hope that it will be useful, but
  13  * WITHOUT ANY WARRANTY; without even the implied warranty
  14  * of MERCHANTABILITY or FITNESS FOR A PARTICULAR
  15  * PURPOSE.  See the GNU General Public License for more details.
  16  *
  17  * You should have received a copy of the GNU General Public License
  18  * along with Jalview.  If not, see <http://www.gnu.org/licenses/>.
  19  * The Jalview Authors are detailed in the 'AUTHORS' file.
  20  */
  21 package jalview.appletgui;
  22
  23 import jalview.api.RotatableCanvasI;
  24 import jalview.datamodel.Point;
  25 import jalview.datamodel.SequenceGroup;
  26 import jalview.datamodel.SequenceI;
  27 import jalview.datamodel.SequencePoint;
  28 import jalview.math.RotatableMatrix;
  29 import jalview.math.RotatableMatrix.Axis;
  30 import jalview.util.MessageManager;
  31 import jalview.viewmodel.AlignmentViewport;
  32
  33 import java.awt.Color;
  34 import java.awt.Dimension;
  35 import java.awt.Font;
  36 import java.awt.Graphics;
  37 import java.awt.Image;
  38 import java.awt.Panel;
  39 import java.awt.event.KeyEvent;
  40 import java.awt.event.KeyListener;
  41 import java.awt.event.MouseEvent;
  42 import java.awt.event.MouseListener;
  43 import java.awt.event.MouseMotionListener;
  44 import java.util.Vector;
  45
  46 public class RotatableCanvas extends Panel implements MouseListener,
  47         MouseMotionListener, KeyListener, RotatableCanvasI
  48 {
  49   private static final int DIMS = 3;
  50
  51   String tooltip;
  52
  53   int toolx;
  54
  55   int tooly;
  56
  57   // RubberbandRectangle rubberband;
  58
  59   boolean drawAxes = true;
  60
  61   int mouseX = 0;
  62
  63   int mouseY = 0;
  64
  65   Image img;
  66
  67   Graphics ig;
  68
  69   Dimension prefsize;
  70
  71   Point centre;
  72
  73   float[] width = new float[DIMS];
  74
  75   float[] max = new float[DIMS];
  76
  77   float[] min = new float[DIMS];
  78
  79   float maxwidth;
  80
  81   float scale;
  82
  83   int npoint;
  84
  85   Vector<SequencePoint> points;
  86
  87   Point[] orig;
  88
  89   Point[] axisEndPoints;
  90
  91   int startx;
  92
  93   int starty;
  94
  95   int lastx;
  96
  97   int lasty;
  98
  99   int rectx1;
 100
 101   int recty1;
 102
 103   int rectx2;
 104
 105   int recty2;
 106
 107   float scalefactor = 1;
 108
 109   AlignmentViewport av;
 110
 111   boolean showLabels = false;
 112
 113   public RotatableCanvas(AlignmentViewport viewport)
 114   {
 115     this.av = viewport;
 116     axisEndPoints = new Point[DIMS];
 117   }
 118
 119   public void showLabels(boolean b)
 120   {
 121     showLabels = b;
 122     repaint();
 123   }
 124
 125   @Override
 126   public void setPoints(Vector<SequencePoint> points, int npoint)
 127   {
 128     this.points = points;
 129     this.npoint = npoint;
 130     PaintRefresher.Register(this, av.getSequenceSetId());
 131
 132     prefsize = getPreferredSize();
 133     orig = new Point[npoint];
 134
 135     for (int i = 0; i < npoint; i++)
 136     {
 137       SequencePoint sp = points.elementAt(i);
 138       orig[i] = sp.coord;
 139     }
 140
 141     resetAxes();
 142
 143     findCentre();
 144     findWidth();
 145
 146     scale = findScale();
 147
 148     // System.out.println("Scale factor = " + scale);
 149
 150     addMouseListener(this);
 151     addKeyListener(this);
 152     // if (getParent() != null) {
 153     // getParent().addKeyListener(this);
 154     // }
 155     addMouseMotionListener(this);
 156
 157     // Add rubberband
 158     // rubberband = new RubberbandRectangle(this);
 159     // rubberband.setActive(true);
 160     // rubberband.addListener(this);
 161   }
 162
 163   /*
 164    * public boolean handleSequenceSelectionEvent(SequenceSelectionEvent evt) {
 165    * redrawneeded = true; repaint(); return true; }
 166    *
 167    * public void removeNotify() { controller.removeListener(this);
 168    * super.removeNotify(); }
 169    */
 170
 171   /**
 172    * Resets axes to the initial state: x-axis to the right, y-axis up, z-axis to
 173    * back (so obscured in a 2-D display)
 174    */
 175   public void resetAxes()
 176   {
 177     axisEndPoints[0] = new Point(1f, 0f, 0f);
 178     axisEndPoints[1] = new Point(0f, 1f, 0f);
 179     axisEndPoints[2] = new Point(0f, 0f, 1f);
 180   }
 181
 182   /**
 183    * Computes and saves the maximum and minimum (x, y, z) positions of any
 184    * sequence point, and also the min-max range (width) for each dimension, and
 185    * the maximum width for all dimensions
 186    */
 187   public void findWidth()
 188   {
 189     max = new float[3];
 190     min = new float[3];
 191
 192     max[0] = Float.MIN_VALUE;
 193     max[1] = Float.MIN_VALUE;
 194     max[2] = Float.MIN_VALUE;
 195
 196     min[0] = Float.MAX_VALUE;
 197     min[1] = Float.MAX_VALUE;
 198     min[2] = Float.MAX_VALUE;
 199
 200     for (SequencePoint sp : points)
 201     {
 202       max[0] = Math.max(max[0], sp.coord.x);
 203       max[1] = Math.max(max[1], sp.coord.y);
 204       max[2] = Math.max(max[2], sp.coord.z);
 205       min[0] = Math.min(min[0], sp.coord.x);
 206       min[1] = Math.min(min[1], sp.coord.y);
 207       min[2] = Math.min(min[2], sp.coord.z);
 208     }
 209
 210     width[0] = Math.abs(max[0] - min[0]);
 211     width[1] = Math.abs(max[1] - min[1]);
 212     width[2] = Math.abs(max[2] - min[2]);
 213
 214     maxwidth = Math.max(width[0], Math.max(width[1], width[2]));
 215   }
 216
 217   public float findScale()
 218   {
 219     int dim, w, height;
 220     if (getSize().width != 0)
 221     {
 222       w = getSize().width;
 223       height = getSize().height;
 224     }
 225     else
 226     {
 227       w = prefsize.width;
 228       height = prefsize.height;
 229     }
 230
 231     if (w < height)
 232     {
 233       dim = w;
 234     }
 235     else
 236     {
 237       dim = height;
 238     }
 239
 240     return dim * scalefactor / (2 * maxwidth);
 241   }
 242
 243   /**
 244    * Computes and saves the position of the centre of the view
 245    */
 246   public void findCentre()
 247   {
 248     findWidth();
 249
 250     float x = (max[0] + min[0]) / 2;
 251     float y = (max[1] + min[1]) / 2;
 252     float z = (max[2] + min[2]) / 2;
 253
 254     centre = new Point(x, y, z);
 255   }
 256
 257   @Override
 258   public Dimension getPreferredSize()
 259   {
 260     if (prefsize != null)
 261     {
 262       return prefsize;
 263     }
 264     else
 265     {
 266       return new Dimension(400, 400);
 267     }
 268   }
 269
 270   @Override
 271   public Dimension getMinimumSize()
 272   {
 273     return getPreferredSize();
 274   }
 275
 276   @Override
 277   public void update(Graphics g)
 278   {
 279     paint(g);
 280   }
 281
 282   @Override
 283   public void paint(Graphics g)
 284   {
 285     if (points == null)
 286     {
 287       g.setFont(new Font("Verdana", Font.PLAIN, 18));
 288       g.drawString(
 289               MessageManager.getString("label.calculating_pca") + "....",
 290               20, getSize().height / 2);
 291     }
 292     else
 293     {
 294
 295       // Only create the image at the beginning -
 296       if ((img == null) || (prefsize.width != getSize().width)
 297               || (prefsize.height != getSize().height))
 298       {
 299         prefsize.width = getSize().width;
 300         prefsize.height = getSize().height;
 301
 302         scale = findScale();
 303
 304         // System.out.println("New scale = " + scale);
 305         img = createImage(getSize().width, getSize().height);
 306         ig = img.getGraphics();
 307
 308       }
 309
 310       drawBackground(ig, Color.black);
 311       drawScene(ig);
 312       if (drawAxes)
 313       {
 314         drawAxes(ig);
 315       }
 316
 317       if (tooltip != null)
 318       {
 319         ig.setColor(Color.red);
 320         ig.drawString(tooltip, toolx, tooly);
 321       }
 322
 323       g.drawImage(img, 0, 0, this);
 324     }
 325   }
 326
 327   public void drawAxes(Graphics g)
 328   {
 329
 330     g.setColor(Color.yellow);
 331     for (int i = 0; i < 3; i++)
 332     {
 333       g.drawLine(getSize().width / 2, getSize().height / 2,
 334               (int) (axisEndPoints[i].x * scale * max[0] + getSize().width / 2),
 335               (int) (axisEndPoints[i].y * scale * max[1] + getSize().height / 2));
 336     }
 337   }
 338
 339   public void drawBackground(Graphics g, Color col)
 340   {
 341     g.setColor(col);
 342     g.fillRect(0, 0, prefsize.width, prefsize.height);
 343   }
 344
 345   public void drawScene(Graphics g)
 346   {
 347     for (int i = 0; i < npoint; i++)
 348     {
 349       SequencePoint sp = points.elementAt(i);
 350       SequenceI sequence = sp.getSequence();
 351       Color sequenceColour = av.getSequenceColour(sequence);
 352       g.setColor(
 353               sequenceColour == Color.black ? Color.white : sequenceColour);
 354       if (av.getSelectionGroup() != null)
 355       {
 356         if (av.getSelectionGroup().getSequences(null)
 357                 .contains(sequence))
 358         {
 359           g.setColor(Color.gray);
 360         }
 361       }
 362
 363       if (sp.coord.z < centre.z)
 364       {
 365         g.setColor(g.getColor().darker());
 366       }
 367
 368       int halfwidth = getSize().width / 2;
 369       int halfheight = getSize().height / 2;
 370       int x = (int) ((sp.coord.x - centre.x) * scale) + halfwidth;
 371       int y = (int) ((sp.coord.y - centre.y) * scale) + halfheight;
 372       g.fillRect(x - 3, y - 3, 6, 6);
 373
 374       if (showLabels)
 375       {
 376         g.setColor(Color.red);
 377         g.drawString(sequence.getName(), x - 3, y - 4);
 378       }
 379     }
 380   }
 381
 382   @Override
 383   public void keyTyped(KeyEvent evt)
 384   {
 385   }
 386
 387   @Override
 388   public void keyReleased(KeyEvent evt)
 389   {
 390   }
 391
 392   @Override
 393   public void keyPressed(KeyEvent evt)
 394   {
 395     if (evt.getKeyCode() == KeyEvent.VK_UP)
 396     {
 397       scalefactor = (float) (scalefactor * 1.1);
 398       scale = findScale();
 399     }
 400     else if (evt.getKeyCode() == KeyEvent.VK_DOWN)
 401     {
 402       scalefactor = (float) (scalefactor * 0.9);
 403       scale = findScale();
 404     }
 405     else if (evt.getKeyChar() == 's')
 406     {
 407       System.err.println("DEBUG: Rectangle selection"); // log.debug
 408       if (rectx2 != -1 && recty2 != -1)
 409       {
 410         rectSelect(rectx1, recty1, rectx2, recty2);
 411
 412       }
 413     }
 414     repaint();
 415   }
 416
 417   @Override
 418   public void mouseClicked(MouseEvent evt)
 419   {
 420   }
 421
 422   @Override
 423   public void mouseEntered(MouseEvent evt)
 424   {
 425   }
 426
 427   @Override
 428   public void mouseExited(MouseEvent evt)
 429   {
 430   }
 431
 432   @Override
 433   public void mouseReleased(MouseEvent evt)
 434   {
 435   }
 436
 437   @Override
 438   public void mousePressed(MouseEvent evt)
 439   {
 440     int x = evt.getX();
 441     int y = evt.getY();
 442
 443     mouseX = x;
 444     mouseY = y;
 445
 446     startx = x;
 447     starty = y;
 448
 449     rectx1 = x;
 450     recty1 = y;
 451
 452     rectx2 = -1;
 453     recty2 = -1;
 454
 455     SequenceI found = findSequenceAtPoint(x, y);
 456
 457     if (found != null)
 458     {
 459       // TODO: applet PCA is not associatable with multi-panels - only parent
 460       // view
 461       if (av.getSelectionGroup() != null)
 462       {
 463         av.getSelectionGroup().addOrRemove(found, true);
 464         av.getSelectionGroup().setEndRes(av.getAlignment().getWidth() - 1);
 465       }
 466       else
 467       {
 468         av.setSelectionGroup(new SequenceGroup());
 469         av.getSelectionGroup().addOrRemove(found, true);
 470         av.getSelectionGroup().setEndRes(av.getAlignment().getWidth() - 1);
 471
 472       }
 473       PaintRefresher.Refresh(this, av.getSequenceSetId());
 474       av.sendSelection();
 475     }
 476     repaint();
 477   }
 478
 479   @Override
 480   public void mouseMoved(MouseEvent evt)
 481   {
 482     SequenceI found = findSequenceAtPoint(evt.getX(), evt.getY());
 483     if (found == null)
 484     {
 485       tooltip = null;
 486     }
 487     else
 488     {
 489       tooltip = found.getName();
 490       toolx = evt.getX();
 491       tooly = evt.getY();
 492     }
 493     repaint();
 494   }
 495
 496   @Override
 497   public void mouseDragged(MouseEvent evt)
 498   {
 499     int xPos = evt.getX();
 500     int yPos = evt.getY();
 501
 502     RotatableMatrix rotmat = new RotatableMatrix();
 503
 504     rotmat.rotate(yPos - mouseY, Axis.X);
 505     rotmat.rotate(xPos - mouseX, Axis.Y);
 506
 507     for (int i = 0; i < npoint; i++)
 508     {
 509       SequencePoint sp = points.elementAt(i);
 510       sp.translateBack(centre);
 511
 512       // Now apply the rotation matrix
 513       sp.coord = rotmat.vectorMultiply(sp.coord);
 514
 515       // Now translate back again
 516       sp.translate(centre);
 517     }
 518
 519     for (int i = 0; i < 3; i++)
 520     {
 521       axisEndPoints[i] = rotmat.vectorMultiply(axisEndPoints[i]);
 522     }
 523     mouseX = xPos;
 524     mouseY = yPos;
 525
 526     paint(this.getGraphics());
 527   }
 528
 529   public void rectSelect(int x1, int y1, int x2, int y2)
 530   {
 531     // boolean changedSel = false;
 532     for (int i = 0; i < npoint; i++)
 533     {
 534       SequencePoint sp = points.elementAt(i);
 535       int tmp1 = (int) ((sp.coord.x - centre.x) * scale
 536               + getSize().width / 2.0);
 537       int tmp2 = (int) ((sp.coord.y - centre.y) * scale
 538               + getSize().height / 2.0);
 539
 540       SequenceI sequence = sp.getSequence();
 541       if (tmp1 > x1 && tmp1 < x2 && tmp2 > y1 && tmp2 < y2)
 542       {
 543         if (av != null)
 544         {
 545           if (!av.getSelectionGroup().getSequences(null)
 546                   .contains(sequence))
 547           {
 548             av.getSelectionGroup().addSequence(sequence, true);
 549           }
 550         }
 551       }
 552     }
 553   }
 554
 555   /**
 556    * Answers the first sequence found whose point on the display is within 2
 557    * pixels of the given coordinates, or null if none is found
 558    *
 559    * @param x
 560    * @param y
 561    *
 562    * @return
 563    */
 564   public SequenceI findSequenceAtPoint(int x, int y)
 565   {
 566     int halfwidth = getSize().width / 2;
 567     int halfheight = getSize().height / 2;
 568
 569     int found = -1;
 570
 571     for (int i = 0; i < npoint; i++)
 572     {
 573
 574       SequencePoint sp = points.elementAt(i);
 575       int px = (int) ((sp.coord.x - centre.x) * scale)
 576               + halfwidth;
 577       int py = (int) ((sp.coord.y - centre.y) * scale)
 578               + halfheight;
 579
 580       if (Math.abs(px - x) < 3 && Math.abs(py - y) < 3)
 581       {
 582         found = i;
 583         break;
 584       }
 585     }
 586
 587     if (found != -1)
 588     {
 589       return points.elementAt(found).getSequence();
 590     }
 591     else
 592     {
 593       return null;
 594     }
 595   }
 596
 597   /**
 598    * Resets the view to initial state (no rotation)
 599    */
 600   public void resetView()
 601   {
 602     img = null;
 603     resetAxes();
 604   }
 605
 606 }