package jalview.gui;


import jalview.analysis.*;
import jalview.datamodel.*;
import jalview.util.*;
import javax.swing.*;
import java.awt.*;
import java.awt.event.*;
import java.util.*;
import java.io.*;

public class TreeCanvas extends JPanel implements MouseListener
{
  NJTree tree;
  JScrollPane scrollPane;
  AlignViewport av;

  Font font;
  int  fontSize = 12;

  boolean showDistances = false;
  boolean showBootstrap = false;

  int offx = 20;
  int offy = 20;

  int threshold;

  String longestName;
  int labelLength=-1;

  //RubberbandRectangle rubberband;

  Selection selected;
  Vector    listeners;

  Hashtable nameHash = new Hashtable();
  Hashtable nodeHash = new Hashtable();

  public TreeCanvas(AlignViewport av, NJTree tree, JScrollPane scroller, String label)
  {
    this.av = av;
    this.tree     = tree;
    selected = av.getSelection();
    scrollPane = scroller;
    addMouseListener(this);
    tree.findHeight(tree.getTopNode());
    longestName = label;

    PaintRefresher.Register(this);
  }
  public void TreeSelectionChanged(Sequence sequence)
 {
    selected = av.getSelection();

    if (selected.contains(sequence))
        selected.removeElement(sequence);
    else
        selected.addElement(sequence);

    setSelected(selected);
    PaintRefresher.Refresh(this);
    repaint();
 }


  public void setSelected(Selection selected)
  {
        this.selected = selected;
  }

  public void setTree(NJTree tree) {
    this.tree = tree;
    tree.findHeight(tree.getTopNode());
  }

    public void drawNode(Graphics g,SequenceNode node, float chunk, float scale, int width,int offx, int offy) {
    if (node == null) {
      return;
    }

    if (node.left() == null && node.right() == null) {
      // Drawing leaf node

      float height = node.height;
      float dist   = node.dist;

      int xstart = (int)((height-dist)*scale) + offx;
      int xend =   (int)(height*scale)        + offx;

      int ypos = (int)(node.ycount * chunk) + offy;

      if (node.element() instanceof SequenceI) {
          g.setColor(((SequenceI)((SequenceNode)node).element()).getColor().darker());
      } else {
          g.setColor(Color.black);
      }

      // Draw horizontal line
      g.drawLine(xstart,ypos,xend,ypos);

      String nodeLabel = "";
      if (showDistances && node.dist > 0) {
        nodeLabel = new Format("%5.2f").form(node.dist);
      }
      if (showBootstrap) {
        if (showDistances) {
          nodeLabel = nodeLabel + " : ";
        }
        nodeLabel = nodeLabel + String.valueOf(node.getBootstrap());
      }
      if (! nodeLabel.equals("")) {
      g.drawString(nodeLabel,xstart,ypos - 10);
      }

      // Colour selected leaves differently
      String name    = node.getName();
      FontMetrics fm = g.getFontMetrics(font);
      int charWidth  = fm.stringWidth(node.getName()) + 3;
      int charHeight = fm.getHeight();

      Rectangle rect = new Rectangle(xend+20,ypos-charHeight,
                                     charWidth,charHeight);

      nameHash.put((SequenceI)node.element(),rect);

      if (selected.contains((SequenceI)node.element())) {
        g.setColor(Color.gray);

        g.fillRect(xend + 10, ypos - charHeight + 3,charWidth,charHeight);
        g.setColor(Color.white);
      }
      g.drawString(node.getName(),xend+10,ypos);
      g.setColor(Color.black);
    } else {
      drawNode(g,(SequenceNode)node.left(), chunk,scale,width,offx,offy);
      drawNode(g,(SequenceNode)node.right(),chunk,scale,width,offx,offy);

      float height = node.height;
      float dist   = node.dist;

      int xstart = (int)((height-dist)*scale) + offx;
      int xend   = (int)(height       *scale) + offx;
      int ypos   = (int)(node.ycount  *chunk) + offy;

      g.setColor(((SequenceNode)node).color.darker());

      // Draw horizontal line
      g.drawLine(xstart,ypos,xend,ypos);
      g.fillRect(xend-2, ypos-2, 4,4);

      int ystart = (int)(((SequenceNode)node.left()) .ycount * chunk) + offy;
      int yend   = (int)(((SequenceNode)node.right()).ycount * chunk) + offy;

      Rectangle pos = new Rectangle(xend-2,ypos-2,5,5);
      nodeHash.put(node,pos);

      g.drawLine((int)(height*scale) + offx, ystart,
                 (int)(height*scale) + offx, yend);

      if (showDistances && node.dist > 0) {
        g.drawString(new Format("%5.2f").form(node.dist),xstart,ypos - 5);
      }

    }
  }
  public Object findElement(int x, int y) {
       Enumeration keys = nameHash.keys();

    while (keys.hasMoreElements()) {
            Object ob = keys.nextElement();
            Rectangle rect = (Rectangle)nameHash.get(ob);

            if (x >= rect.x && x <= (rect.x + rect.width) &&
              y >= rect.y && y <= (rect.y + rect.height)) {
              return ob;
            }
  }
  keys = nodeHash.keys();

    while (keys.hasMoreElements()) {
            Object ob = keys.nextElement();
            Rectangle rect = (Rectangle)nodeHash.get(ob);

            if (x >= rect.x && x <= (rect.x + rect.width) &&
              y >= rect.y && y <= (rect.y + rect.height)) {
              return ob;
            }
    }
      return null;

  }

  public void pickNodes(Rectangle pickBox, Selection sel) {
    int width  = getWidth();
    int height = getHeight();

    SequenceNode top = tree.getTopNode();

    float wscale = (float)(width*.8-offx*2)/tree.getMaxHeight()
;
    if (top.count == 0) {
      top.count = ((SequenceNode)top.left()).count + ((SequenceNode)top.right()).count ;
    }
    float chunk = (float)(height-offy*2)/top.count;

    pickNode(pickBox,sel,top,chunk,wscale,width,offx,offy);
  }

  public void pickNode(Rectangle pickBox, Selection sel, SequenceNode node, float chunk, float scale, int width,int offx, int offy) {
    if (node == null) {
      return;
    }

    if (node.left() == null && node.right() == null) {
      float height = node.height;
      float dist   = node.dist;

      int xstart = (int)((height-dist)*scale) + offx;
      int xend   = (int)(height*scale) + offx;

      int ypos = (int)(node.ycount * chunk) + offy;

      if (pickBox.contains(new Point(xend,ypos))) {
        if (node.element() instanceof SequenceI) {
          SequenceI seq = (SequenceI)node.element();
          if (sel.contains(seq)) {
            sel.removeElement(seq);
          } else {
            sel.addElement(seq);
          }
        }
      }
    } else {
      pickNode(pickBox,sel,(SequenceNode)node.left(), chunk,scale,width,offx,offy);
      pickNode(pickBox,sel,(SequenceNode)node.right(),chunk,scale,width,offx,offy);
    }
  }

  public void setColor(SequenceNode node, Color c) {
    if (node == null) {
      return;
    }

    if (node.left() == null && node.right() == null) {
      node.color = c;

      if (node.element() instanceof SequenceI) {
          ((SequenceI)node.element()).setColor(c);
      }
    } else {
      node.color = c;
      setColor((SequenceNode)node.left(),c);
      setColor((SequenceNode)node.right(),c);
    }
  }


  public void paintComponent(Graphics g) {


    font = new Font("Verdana",Font.PLAIN,fontSize);
    g.setFont(font);

    FontMetrics fm = g.getFontMetrics(font);

    if(nameHash.size()==0)
      repaint();


    if( scrollPane.getHeight() > fm.getHeight() * nameHash.size()+offy)
     {
         draw(g,scrollPane.getWidth(),scrollPane.getHeight());
         setPreferredSize(new Dimension(scrollPane.getWidth(), scrollPane.getHeight()));
     }
    else
     {
         setPreferredSize(new Dimension(getWidth(), fm.getHeight() * nameHash.size()));
         draw( g,getWidth(), fm.getHeight() * nameHash.size());
     }


    if (threshold != 0)
    {
        g.setColor(Color.red);
        g.drawLine(threshold,0,threshold,getHeight());
    }

    scrollPane.revalidate();
  }
    public int getFontSize() {
        return fontSize;
    }
    public void setFontSize(int fontSize) {
        this.fontSize = fontSize;
        repaint();
    }
  public void draw(Graphics g, int width, int height) {
      g.setColor(Color.white);
      g.fillRect(0,0,width,height);


      labelLength = g.getFontMetrics(font).stringWidth(longestName)+ 20;//20 allows for scrollbar

      float wscale =(float)(width - labelLength -offx*2)/tree.getMaxHeight();

      SequenceNode top = tree.getTopNode();

      if (top.count == 0) {
          top.count = ((SequenceNode)top.left()).count + ((SequenceNode)top.right()).count ;
      }
      float chunk = (float)(height-offy*2)/top.count ;

      drawNode(g,tree.getTopNode(),chunk,wscale,width,offx,offy);
  }

  public void mouseReleased(MouseEvent e) { }
  public void mouseEntered(MouseEvent e) { }
  public void mouseExited(MouseEvent e) { }
  public void mouseClicked(MouseEvent e) {
  }

  public void mousePressed(MouseEvent e) {
      int x = e.getX();
      int y = e.getY();

      Object ob = findElement(x,y);

      if (ob instanceof SequenceI)
      {
          TreeSelectionChanged((Sequence)ob);
          repaint();
          return;

      } else if (ob instanceof SequenceNode) {
          SequenceNode tmpnode = (SequenceNode)ob;
          tree.swapNodes(tmpnode);
          tree.reCount(tree.getTopNode());
          tree.findHeight(tree.getTopNode());
      } else {
          // Find threshold

          if (tree.getMaxHeight() != 0) {
              float fthreshold = (float)(x - offx)/(float)(getWidth()-labelLength - 2*offx);
              this.threshold = x;
              tree.getGroups().removeAllElements();
              tree.groupNodes(tree.getTopNode(),fthreshold);
              setColor(tree.getTopNode(),Color.black);

              for (int i=0; i < tree.getGroups().size(); i++) {

                  int tmp = i%(7);
                  Color col = new Color((int)(Math.random()*255),
                                        (int)(Math.random()*255),
                                        (int)(Math.random()*255));

                  setColor((SequenceNode)tree.getGroups().elementAt(i),col.brighter());

                  // l is vector of Objects
                  Vector l = tree.findLeaves((SequenceNode)tree.getGroups().elementAt(i),new Vector());

              }
          }
      }

      repaint();

  }

    public void setShowDistances(boolean state) {
        this.showDistances = state;
        repaint();
    }

    public void setShowBootstrap(boolean state) {
      this.showBootstrap = state;
      repaint();
    }

}