package jalview.gui;

import jalview.datamodel.*;
import java.awt.*;
import java.awt.event.*;
import javax.swing.*;

public class ScalePanel extends JPanel
{

  protected int offy = 4;
  public    int width;

  protected AlignViewport av;
  AlignmentPanel ap;

  boolean stretchingGroup = false;

  public ScalePanel(AlignViewport av, AlignmentPanel ap)
  {
    this.av         = av;
    this.ap = ap;

    addMouseListener(new MouseAdapter()
    {
      public void mousePressed(MouseEvent evt)
      {
        doMousePressed(evt);
      }
      public void mouseReleased(MouseEvent evt)
      {
        doMouseReleased(evt);
      }

    });
    addMouseMotionListener(new MouseMotionAdapter()
    {
      public void mouseDragged(MouseEvent evt)
      {
        doMouseDragged(evt);
      }
    });

  }


  public void doMousePressed(MouseEvent evt) {
    int x = evt.getX();
    int res = x/av.getCharWidth() + av.getStartRes();
    SequenceGroup sg = null;

    if (av.getColumnSelection().contains(res))
      av.getColumnSelection().removeElement(res);
    else
    {
      av.getColumnSelection().addElement(res);

      sg = new SequenceGroup();
      for (int i = 0; i < av.alignment.getSequences().size(); i++)
        sg.addSequence(av.alignment.getSequenceAt(i));

      sg.setStartRes(res);
      sg.setEndRes(res);

      ap.annotationPanel.addEditableColumn(res);
    }
    av.setSelectionGroup(sg);
    ap.RefreshPanels();
  }

  public void doMouseReleased(MouseEvent evt)
  {
    if(!stretchingGroup)
      return;

    int x = evt.getX();
    int res = x/av.getCharWidth() + av.getStartRes();

   if (! av.getColumnSelection().contains(res))
     av.getColumnSelection().addElement(res);

   SequenceGroup sg = av.getSelectionGroup();

   if(res>sg.getStartRes())
     sg.setEndRes(res);
   else if(res<sg.getStartRes())
     sg.setStartRes(res);

   stretchingGroup = false;
   ap.RefreshPanels();
  }

  public void doMouseDragged(MouseEvent evt)
  {
    int x = evt.getX();
    int res = x/av.getCharWidth() + av.getStartRes();

    SequenceGroup sg = av.getSelectionGroup();
    if(sg!=null)
    {
      stretchingGroup = true;
      if (res > sg.getStartRes())
        sg.setEndRes(res);
      else if (res < sg.getStartRes())
        sg.setStartRes(res);

       ap.annotationPanel.addEditableColumn(res);
      ap.RefreshPanels();
    }
  }

  public void paintComponent(Graphics g)
  {
    drawScale(g, av.getStartRes(), av.getEndRes(), getWidth());
  }

// scalewidth will normally be screenwidth,
  public void drawScale(Graphics g, int startx, int endx, int scaleWidth)
  {
    Graphics2D gg = (Graphics2D)g;
    gg.setRenderingHint(RenderingHints.KEY_ANTIALIASING,RenderingHints.VALUE_ANTIALIAS_ON);
    int charWidth  = av.getCharWidth();

    gg.setFont(av.getFont());

    //Fill in the background

    gg.setColor(Color.white);
    gg.fillRect(0,0,scaleWidth, getHeight());

    int resWidth = endx - startx + 1;

    // We want roughly one tick every 50 pixels

    double rough = scaleWidth/50.0;

    double roughtick = resWidth/rough;

    int   digits  = (int)(Math.log(roughtick)/Math.log(10));
    int   num     = (int)(roughtick/Math.pow(10,digits));

    if (roughtick>5)
        num = 10;
    else
        num = 5;

  //   else
  //  num = 2;

  //  System.out.println("Dig " + digits + " " + rough + " " + roughtick + " " + getWidth() + " " + num);

    int space = (int)(num * Math.pow(10,digits));

    //Set the text font

    gg.setColor(Color.black);


    int scalestartx = 0;
    if(space>0)
      scalestartx = startx - startx%space + space;


    //Fill the selected columns
    ColumnSelection cs = av.getColumnSelection();
    int width = charWidth;
    if (width == 0)
      width = 1;

    gg.setColor(new Color(220,0,0));
    for (int i=0; i<cs.size(); i++) {
      int sel  = cs.columnAt(i);
      //      System.out.println("Selection = " + sel);
      if ( sel >= startx  && sel <= endx) {

        gg.fillRect((sel-startx)*charWidth,0,width,getHeight());
      }
    }

    // Draw the scale numbers
    gg.setColor(Color.black);


    for (int i=scalestartx;i < endx;i+= space) {

      String string = String.valueOf(i);

      gg.drawString(string,(i-startx-1)*charWidth,  av.charHeight / 2 +offy);
      gg.drawLine( (int)((i-startx-1)*charWidth +charWidth/2), av.charHeight/2 +offy +2, (int)((i-startx-1)*charWidth +charWidth/2),av.charHeight/2 +offy+7);

    }
  }


}