+package MCview;\r
+\r
+import java.awt.*;\r
+import java.awt.event.*;\r
+import java.util.*;\r
+import java.io.*;\r
+import javax.swing.*;\r
+\r
+public class rotCanvas extends JPanel implements KeyListener,\r
+ MouseListener,\r
+ MouseMotionListener {\r
+ MCMatrix idmat = new MCMatrix(3,3);\r
+ MCMatrix objmat = new MCMatrix(3,3);\r
+\r
+ boolean redrawneeded =true;\r
+\r
+ int omx = 0;\r
+ int mx = 0;\r
+ int omy = 0;\r
+ int my = 0;\r
+\r
+ public PDBfile pdb;\r
+\r
+ int bsize;\r
+\r
+ Image img;\r
+ Graphics ig;\r
+\r
+ Dimension prefsize;\r
+\r
+ float centre[] = new float[3];\r
+ float width[] = new float[3];\r
+\r
+ float maxwidth;\r
+ float scale;\r
+\r
+ String inStr;\r
+ String inType;\r
+\r
+ boolean depthcue = true;\r
+ boolean wire = false;\r
+ boolean bymolecule = false;\r
+ boolean zbuffer = true;\r
+\r
+ boolean dragging;\r
+\r
+ int xstart;\r
+ int xend;\r
+ int ystart;\r
+ int yend;\r
+ int xmid;\r
+ int ymid;\r
+\r
+ Font font = new Font("Helvetica",Font.PLAIN,10);\r
+\r
+ public rotCanvas(PDBfile pdb) throws IOException {\r
+\r
+ this.pdb = pdb;\r
+ this.prefsize = new Dimension( getWidth(), getHeight());\r
+\r
+ //Initialize the matrices to identity\r
+\r
+ for (int i = 0; i < 3; i++) {\r
+ for (int j = 0; j < 3 ; j++) {\r
+ if (i != j) {\r
+ idmat.addElement(i,j,0);\r
+ objmat.addElement(i,j,0);\r
+ } else {\r
+ idmat.addElement(i,j,1);\r
+ objmat.addElement(i,j,1);\r
+ }\r
+ }\r
+ }\r
+\r
+ addMouseMotionListener(this);\r
+ addMouseListener(this);\r
+ addKeyListener(this);\r
+\r
+ addPDBfile();\r
+ ToolTipManager.sharedInstance().registerComponent(this);\r
+\r
+ }\r
+\r
+ public void addPDBfile() {\r
+ findCentre();\r
+ findWidth();\r
+\r
+ scale = findScale();\r
+\r
+ System.out.println("Scale factor = " + scale);\r
+ }\r
+\r
+ public void deleteBonds() {\r
+ scale = 0;\r
+ maxwidth = 0;\r
+\r
+ width[0] = 0;\r
+ width[1] = 0;\r
+ width[2] = 0;\r
+\r
+ centre[0] = 0;\r
+ centre[1] = 0;\r
+ centre[2] = 0;\r
+\r
+ for (int i=0; i < pdb.chains.size(); i++) {\r
+ ((PDBChain)pdb.chains.elementAt(i)).bonds = null;\r
+ }\r
+ }\r
+\r
+ public void findWidth() {\r
+ float max[] = new float[3];\r
+ float min[] = new float[3];\r
+\r
+ max[0] = (float)-1e30;\r
+ max[1] = (float)-1e30;\r
+ max[2] = (float)-1e30;\r
+\r
+ min[0] = (float)1e30;\r
+ min[1] = (float)1e30;\r
+ min[2] = (float)1e30;\r
+\r
+ for (int ii=0; ii < pdb.chains.size(); ii++) {\r
+ if (((PDBChain)pdb.chains.elementAt(ii)).isVisible) {\r
+\r
+ Vector bonds = ((PDBChain)pdb.chains.elementAt(ii)).bonds;\r
+\r
+ for (int i = 0; i < bonds.size(); i++) {\r
+ Bond tmp = (Bond)bonds.elementAt(i);\r
+\r
+ if (tmp.start[0] >= max[0])\r
+ max[0] = tmp.start[0];\r
+ if (tmp.start[1] >= max[1])\r
+ max[1] = tmp.start[1];\r
+ if (tmp.start[2] >= max[2])\r
+ max[2] = tmp.start[2];\r
+ if (tmp.start[0] <= min[0])\r
+ min[0] = tmp.start[0];\r
+ if (tmp.start[1] <= min[1])\r
+ min[1] = tmp.start[1];\r
+ if (tmp.start[2] <= min[2])\r
+ min[2] = tmp.start[2];\r
+\r
+ if (tmp.end[0] >= max[0])\r
+ max[0] = tmp.end[0];\r
+ if (tmp.end[1] >= max[1])\r
+ max[1] = tmp.end[1];\r
+ if (tmp.end[2] >= max[2])\r
+ max[2] = tmp.end[2];\r
+ if (tmp.end[0] <= min[0])\r
+ min[0] = tmp.end[0];\r
+ if (tmp.end[1] <= min[1])\r
+ min[1] = tmp.end[1];\r
+ if (tmp.end[2] <= min[2])\r
+ min[2] = tmp.end[2];\r
+ }\r
+ }\r
+ }\r
+\r
+ System.out.println("xmax " + max[0] + " min " + min[0]);\r
+ System.out.println("ymax " + max[1] + " min " + min[1]);\r
+ System.out.println("zmax " + max[2] + " min " + min[2]);\r
+\r
+ width[0] = (float)Math.abs(max[0] - min[0]);\r
+ width[1] = (float)Math.abs(max[1] - min[1]);\r
+ width[2] = (float)Math.abs(max[2] - min[2]);\r
+\r
+ maxwidth = width[0];\r
+\r
+ if (width[1] > width[0])\r
+ maxwidth = width[1];\r
+ if (width[2] > width[1])\r
+ maxwidth = width[2];\r
+\r
+ System.out.println("Maxwidth = " + maxwidth);\r
+ }\r
+\r
+ public float findScale() {\r
+ int dim, width, height;\r
+\r
+ if ( getWidth() != 0) {\r
+ width = getWidth();\r
+ height = getHeight();\r
+\r
+ } else {\r
+ width = prefsize.width;\r
+ height = prefsize.height;\r
+ }\r
+\r
+ if (width < height) {\r
+ dim = width;\r
+ } else {\r
+ dim = height;\r
+ }\r
+\r
+ return (float)(dim/(1.5d*maxwidth));\r
+ }\r
+\r
+\r
+\r
+ public void findCentre() {\r
+ float xtot = 0;\r
+ float ytot = 0;\r
+ float ztot = 0;\r
+\r
+ int bsize = 0;\r
+ //Find centre coordinate\r
+\r
+ for (int ii = 0; ii < pdb.chains.size() ; ii++) {\r
+ if (((PDBChain)pdb.chains.elementAt(ii)).isVisible) {\r
+\r
+ Vector bonds = ((PDBChain)pdb.chains.elementAt(ii)).bonds;\r
+\r
+ bsize += bonds.size();\r
+\r
+ for (int i = 0; i < bonds.size(); i++) {\r
+\r
+ xtot = xtot + ((Bond)bonds.elementAt(i)).start[0] +\r
+ ((Bond)bonds.elementAt(i)).end[0];\r
+\r
+ ytot = ytot + ((Bond)bonds.elementAt(i)).start[1] +\r
+ ((Bond)bonds.elementAt(i)).end[1];\r
+\r
+ ztot = ztot + ((Bond)bonds.elementAt(i)).start[2] +\r
+ ((Bond)bonds.elementAt(i)).end[2];\r
+ }\r
+ }\r
+ }\r
+\r
+ centre[0] = xtot / (2 * (float)bsize);\r
+ centre[1] = ytot / (2 * (float)bsize);\r
+ centre[2] = ztot / (2 * (float)bsize);\r
+ }\r
+\r
+ public void paintComponent(Graphics g) {\r
+ //Only create the image at the beginning -\r
+ //this saves much memory usage\r
+ if ((img == null) || (prefsize.width != getWidth()) || (prefsize.height != getHeight())) {\r
+ prefsize.width = getWidth();\r
+ prefsize.height = getHeight();\r
+\r
+ scale = findScale();\r
+ img = createImage(prefsize.width,prefsize.height);\r
+ ig = img.getGraphics();\r
+\r
+ redrawneeded = true;\r
+ }\r
+\r
+ if (redrawneeded == true) {\r
+ drawBackground(ig,Color.black);\r
+ drawScene(ig);\r
+ redrawneeded = false;\r
+ } else {\r
+ ig = img.getGraphics();\r
+ }\r
+\r
+ g.drawImage(img,0,0,this);\r
+ }\r
+\r
+ public void drawBackground(Graphics g, Color col) {\r
+ g.setColor(col);\r
+ g.fillRect(0,0,prefsize.width,prefsize.height);\r
+ }\r
+\r
+\r
+ public void drawScene(Graphics g) {\r
+ // Sort the bonds by z coord\r
+\r
+ Vector bonds = new Vector();\r
+\r
+\r
+ for (int ii = 0; ii < pdb.chains.size() ; ii++) {\r
+ if (((PDBChain)pdb.chains.elementAt(ii)).isVisible) {\r
+\r
+ Vector tmp = ((PDBChain)pdb.chains.elementAt(ii)).bonds;\r
+\r
+ for (int i=0; i < tmp.size(); i++) {\r
+ bonds.addElement(tmp.elementAt(i));\r
+ }\r
+ }\r
+ }\r
+\r
+ if (zbuffer) {\r
+ Zsort.Zsort(bonds);\r
+ }\r
+\r
+ for (int i = 0; i < bonds.size(); i++) {\r
+ Bond tmpBond = (Bond)bonds.elementAt(i);\r
+\r
+ xstart = (int)((tmpBond.start[0] - centre[0])*scale + getWidth()/2);\r
+ ystart = (int)((tmpBond.start[1] - centre[1])*scale + getHeight()/2);\r
+\r
+ xend = (int)((tmpBond.end[0] - centre[0])*scale + getWidth()/2);\r
+ yend = (int)((tmpBond.end[1] - centre[1])*scale + getHeight()/2);\r
+\r
+ xmid = (xend+xstart)/2;\r
+ ymid = (yend+ystart)/2;\r
+\r
+\r
+ if (depthcue && !bymolecule) {\r
+\r
+ if (tmpBond.start[2] < (centre[2] - maxwidth/6)) {\r
+ g.setColor(tmpBond.startCol.darker().darker());\r
+ drawLine(g,xstart,ystart,xmid,ymid);\r
+\r
+ g.setColor(tmpBond.endCol.darker().darker());\r
+ drawLine(g,xmid,ymid,xend,yend);\r
+ } else if (tmpBond.start[2] < (centre[2]+maxwidth/6)) {\r
+ g.setColor(tmpBond.startCol.darker());\r
+ drawLine(g,xstart,ystart,xmid,ymid);\r
+\r
+ g.setColor(tmpBond.endCol.darker());\r
+ drawLine(g,xmid,ymid,xend,yend);\r
+ } else {\r
+ g.setColor(tmpBond.startCol);\r
+ drawLine(g,xstart,ystart,xmid,ymid);\r
+\r
+ g.setColor(tmpBond.endCol);\r
+ drawLine(g,xmid,ymid,xend,yend);\r
+ }\r
+ } else if (depthcue && bymolecule) {\r
+ if (tmpBond.start[2] < (centre[2] - maxwidth/6)) {\r
+ g.setColor(Color.green.darker().darker());\r
+ drawLine(g,xstart,ystart,xend,yend);\r
+\r
+ } else if (tmpBond.start[2] < (centre[2]+maxwidth/6)) {\r
+ g.setColor(Color.green.darker());\r
+ drawLine(g,xstart,ystart,xend,yend);\r
+\r
+ } else {\r
+ g.setColor(Color.green);\r
+ drawLine(g,xstart,ystart,xend,yend);\r
+\r
+ }\r
+ } else if (!depthcue && !bymolecule) {\r
+ g.setColor(tmpBond.startCol);\r
+ drawLine(g,xstart,ystart,xmid,ymid);\r
+ g.setColor(tmpBond.endCol);\r
+ drawLine(g,xmid,ymid,xend,yend);\r
+\r
+ } else {\r
+ drawLine(g,xstart,ystart,xend,yend);\r
+ }\r
+ }\r
+ }\r
+\r
+ public void drawLine(Graphics g, int x1, int y1, int x2, int y2) {\r
+ if (!wire) {\r
+\r
+ if (((float)Math.abs(y2-y1)/(float)Math.abs(x2-x1)) < 0.5) {\r
+\r
+ g.drawLine(x1,y1,x2,y2);\r
+ g.drawLine(x1+1,y1+1,x2+1,y2+1);\r
+ g.drawLine(x1,y1-1,x2,y2-1);\r
+ } else {\r
+ g.setColor(g.getColor().brighter());\r
+ g.drawLine(x1,y1,x2,y2);\r
+ g.drawLine(x1+1,y1,x2+1,y2);\r
+ g.drawLine(x1-1,y1,x2-1,y2);\r
+\r
+ }\r
+ } else {\r
+ g.drawLine(x1,y1,x2,y2);\r
+ }\r
+ }\r
+\r
+ public Dimension minimumsize() {\r
+ return prefsize;\r
+ }\r
+ public Dimension preferredsize() {\r
+ return prefsize;\r
+ }\r
+\r
+ public void keyTyped(KeyEvent evt) { }\r
+ public void keyReleased(KeyEvent evt) { }\r
+ public void keyPressed(KeyEvent evt) {\r
+ int key = evt.getKeyChar();\r
+\r
+ if (evt.getKeyCode() == KeyEvent.VK_UP) {\r
+ scale = (float)(scale * 1.1);\r
+ redrawneeded = true;\r
+ repaint();\r
+\r
+ } else if (evt.getKeyCode() == KeyEvent.VK_DOWN) {\r
+ scale = (float)(scale * 0.9);\r
+ redrawneeded = true;\r
+ repaint();\r
+\r
+ } else if (key == 'w') {\r
+\r
+ wire = !wire;\r
+ System.out.println("wireframe " + wire);\r
+ redrawneeded = true;\r
+ repaint();\r
+\r
+ } else if (key == 'd') {\r
+ depthcue = !depthcue;\r
+ System.out.println("Depth cueing is " + depthcue);\r
+ redrawneeded = true;\r
+ repaint();\r
+\r
+ } else if (key == 'm') {\r
+ bymolecule = !bymolecule;\r
+ System.out.println("Bymolecule is " + bymolecule);\r
+ redrawneeded = true;\r
+ repaint();\r
+\r
+ } else if (key == 'z') {\r
+ zbuffer = !zbuffer;\r
+ System.out.println("Z buffering is " + zbuffer);\r
+ redrawneeded = true;\r
+ repaint();\r
+\r
+ } else if (key == 'c') {\r
+ bymolecule = false;\r
+ pdb.setChainColours();\r
+ System.out.println("Colouring by chain");\r
+ redrawneeded = true;\r
+ repaint();\r
+\r
+ } else if (key == 'h') {\r
+ bymolecule = false;\r
+ pdb.setHydrophobicityColours();\r
+ System.out.println("Colouring by hydrophobicity");\r
+ redrawneeded = true;\r
+ repaint();\r
+\r
+ } else if (key == 'q') {\r
+ bymolecule = false;\r
+ pdb.setChargeColours();\r
+ System.out.println("Colouring charges and cysteines");\r
+ redrawneeded = true;\r
+ repaint();\r
+\r
+ }\r
+ return;\r
+ }\r
+\r
+ public void mousePressed(MouseEvent e)\r
+ {\r
+\r
+ mx = e.getX();\r
+ my = e.getY();\r
+ omx = mx;\r
+ omy = my;\r
+ dragging = false;\r
+}\r
+\r
+ public void mouseMoved(MouseEvent e)\r
+ {\r
+ myAtom fatom = null;\r
+ for (int ii = 0; ii < pdb.chains.size(); ii++)\r
+ {\r
+\r
+ PDBChain chain = (PDBChain) pdb.chains.elementAt(ii);\r
+ if (chain.isVisible)\r
+ {\r
+ Vector bonds = ( (PDBChain) pdb.chains.elementAt(ii)).bonds;\r
+ for (int i = 0; i < bonds.size(); i++)\r
+ {\r
+ Bond tmpBond = (Bond) bonds.elementAt(i);\r
+ int truex = (int) ( (tmpBond.start[0] - centre[0]) * scale +\r
+ getWidth() / 2);\r
+ if (Math.abs(truex - e.getX()) <= 2)\r
+ {\r
+ int truey = (int) ( (tmpBond.start[1] - centre[1]) * scale +\r
+ getHeight() / 2);\r
+ if (Math.abs(truey - e.getY()) <= 2)\r
+ fatom = tmpBond.at1;\r
+ }\r
+ }\r
+ }\r
+ }\r
+ if(fatom!=null)\r
+ this.setToolTipText(fatom.resName);\r
+ else\r
+ this.setToolTipText(null);\r
+ }\r
+\r
+ public void mouseClicked(MouseEvent e) { }\r
+ public void mouseEntered(MouseEvent e) { }\r
+ public void mouseExited(MouseEvent e) { }\r
+\r
+ public void mouseDragged(MouseEvent evt) {\r
+ int x = evt.getX();\r
+ int y = evt.getY();\r
+ mx = x;\r
+ my = y;\r
+\r
+ MCMatrix objmat = new MCMatrix(3,3);\r
+ objmat.setIdentity();\r
+\r
+ if ((evt.getModifiers() & Event.META_MASK) != 0) {\r
+ objmat.rotatez((float)((mx-omx)));\r
+ } else {\r
+ objmat.rotatex((float)((my-omy)));\r
+ objmat.rotatey((float)((omx-mx)));\r
+ }\r
+\r
+ //Alter the bonds\r
+ for (int ii = 0; ii < pdb.chains.size() ; ii++) {\r
+ Vector bonds = ((PDBChain)pdb.chains.elementAt(ii)).bonds;\r
+\r
+ for (int i = 0; i < bonds.size(); i++) {\r
+\r
+ Bond tmpBond = (Bond)bonds.elementAt(i);\r
+\r
+ //Translate the bond so the centre is 0,0,0\r
+ tmpBond.translate(-centre[0],-centre[1],-centre[2]);\r
+\r
+ //Now apply the rotation matrix\r
+ tmpBond.start = objmat.vectorMultiply(tmpBond.start);\r
+ tmpBond.end = objmat.vectorMultiply(tmpBond.end);\r
+\r
+ //Now translate back again\r
+ tmpBond.translate(centre[0],centre[1],centre[2]);\r
+ }\r
+ }\r
+\r
+ objmat = null;\r
+\r
+ omx = mx;\r
+ omy = my;\r
+\r
+ redrawneeded = true;\r
+\r
+ paint(this.getGraphics());\r
+\r
+ dragging = true;\r
+ return;\r
+ }\r
+\r
+ public void mouseReleased(MouseEvent evt) {\r
+ int x = evt.getX();\r
+ int y = evt.getY();\r
+\r
+ if (!dragging) {\r
+ myAtom tmp = findAtom(x,y);\r
+ }\r
+ drawLabels();\r
+ return;\r
+ }\r
+\r
+ public void drawLabels() {\r
+ redrawneeded = true;\r
+ paint(this.getGraphics());\r
+\r
+ for (int ii = 0; ii < pdb.chains.size() ; ii++) {\r
+\r
+ PDBChain chain = (PDBChain)pdb.chains.elementAt(ii);\r
+\r
+ if (chain.isVisible) {\r
+\r
+ Vector bonds = ((PDBChain)pdb.chains.elementAt(ii)).bonds;\r
+\r
+ for (int i = 0; i < bonds.size(); i++) {\r
+ Bond tmpBond = (Bond)bonds.elementAt(i);\r
+\r
+ if (tmpBond.at1.isSelected) {\r
+ labelAtom(img.getGraphics(),tmpBond,1);\r
+ }\r
+\r
+ if (tmpBond.at2.isSelected) {\r
+ labelAtom(img.getGraphics(),tmpBond,2);\r
+ }\r
+\r
+ }\r
+ }\r
+ }\r
+\r
+ this.getGraphics().drawImage(img,0,0,this);\r
+\r
+ dragging = false;\r
+\r
+ }\r
+\r
+ public void labelAtom(Graphics g,Bond b, int n) {\r
+\r
+ g.setFont(font);\r
+\r
+ if (n ==1) {\r
+\r
+ int xstart = (int)((b.start[0] - centre[0])*scale + getWidth()/2);\r
+ int ystart = (int)((b.start[1] - centre[1])*scale + getHeight()/2);\r
+\r
+ g.setColor(Color.red);\r
+ g.drawString(b.at1.resName + "-" + b.at1.resNumber,xstart,ystart);\r
+ }\r
+\r
+ if (n ==2) {\r
+\r
+ int xstart = (int)((b.end[0] - centre[0])*scale + getWidth()/2);\r
+ int ystart = (int)((b.end[1] - centre[1])*scale + getHeight()/2);\r
+\r
+ g.setColor(Color.red);\r
+ g.drawString(b.at2.resName + "-" + b.at2.resNumber,xstart,ystart);\r
+ }\r
+ }\r
+\r
+\r
+ public myAtom findAtom(int x, int y) {\r
+ myAtom fatom = null;\r
+\r
+ int foundchain = -1;\r
+\r
+ for (int ii = 0; ii < pdb.chains.size() ; ii++) {\r
+\r
+ PDBChain chain = (PDBChain)pdb.chains.elementAt(ii);\r
+\r
+ if (chain.isVisible) {\r
+\r
+ Vector bonds = ((PDBChain)pdb.chains.elementAt(ii)).bonds;\r
+\r
+ for (int i = 0; i < bonds.size(); i++) {\r
+ Bond tmpBond = (Bond)bonds.elementAt(i);\r
+\r
+ int truex = (int)((tmpBond.start[0] - centre[0])*scale + getWidth()/2);\r
+\r
+ if (Math.abs(truex - x) <= 2) {\r
+\r
+ int truey = (int)((tmpBond.start[1] - centre[1])*scale + getHeight()/2);\r
+\r
+ if (Math.abs(truey - y) <= 2) {\r
+\r
+ System.out.println("Found match");\r
+ System.out.println(x + " " + y);\r
+ System.out.println(truex + " " + truey);\r
+ System.out.println(tmpBond.start[0] + " " + tmpBond.start[1]);\r
+ System.out.println("Atom 1 = " + tmpBond.at1.resName + " " +\r
+ tmpBond.at1.resNumber + " " + tmpBond.at1.chain);\r
+ fatom = tmpBond.at1;\r
+ fatom.isSelected = !fatom.isSelected;\r
+ foundchain = ii;\r
+ }\r
+ }\r
+ }\r
+ }\r
+\r
+ if (fatom != null) //)&& chain.ds != null)\r
+ {\r
+ chain = (PDBChain)pdb.chains.elementAt(foundchain);\r
+ // SMJS TODO\r
+ // int tmp = chain.ds.seqstart + fatom.resNumber - chain.offset;\r
+ // int pos = chain.ds.findIndex(tmp);\r
+ // System.out.println("Found seq " + chain.ds.name + " " + tmp + " " + pos);\r
+ }\r
+\r
+ }\r
+ return fatom;\r
+ }\r
+\r
+ public void update(Graphics g) {\r
+ paint(g);\r
+ }\r
+\r
+}\r