package jalview.gui;

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

public class SecondaryStructurePanel extends JPanel implements ActionListener
{
  AlignViewport av;
  Vector structures;

  SecondaryStructure amend;

  public SecondaryStructurePanel(AlignViewport av)
  {
    this.av = av;
    structures = new Vector();
    setPreferredSize(new Dimension(500,30));

    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);  }
    });
  }

  int startRes = 0;
  int endRes = 0;
  boolean amendStart = false;
  boolean amendEnd = false;

  public void doMousePressed(MouseEvent evt)
  {
    amend = null;
    amendStart =false;
    amendEnd = false;

    startRes = evt.getX() / av.getCharWidth() + av.getStartRes();
    for(int i=0; i<structures.size(); i++)
    {
      SecondaryStructure ss = (SecondaryStructure) structures.get(i);
      if(startRes>=ss.start && startRes<=ss.end)
      {
        amend = ss;
        if(startRes==ss.start)
          amendStart = true;

        if(startRes==ss.end)
          amendEnd = true;

        if(evt.getClickCount()==2)
        {
          String label = JOptionPane.showInputDialog(this, "Amend label", "Amend label", JOptionPane.QUESTION_MESSAGE );
          if(label==null)
            label = "";
          ss.label = label;
        }
        break;
      }
    }
    repaint();

  }

  public void doMouseReleased(MouseEvent evt)
  {
    endRes = evt.getX() / av.getCharWidth() + av.getStartRes();

    if(amend!=null)
    {
      if(amendStart)
        amend.start = endRes;
      if(amendEnd)
        amend.end = endRes;

      amend=null;
      repaint();
      return;
    }

    endRes = evt.getX() / av.getCharWidth() + av.getStartRes();

    JPopupMenu pop = new JPopupMenu("Structure type");
    JMenuItem item = new JMenuItem("Helix");
    item.addActionListener(this);
    pop.add(item);
    item = new JMenuItem("Sheet");
    item.addActionListener(this);
    pop.add(item);
    pop.show(this, evt.getX(), evt.getY());

  }

  public void actionPerformed(ActionEvent evt)
  {

    int type = 0;
    if(evt.getActionCommand().equals("Helix"))
       type = SecondaryStructure.HELIX;
    else if(evt.getActionCommand().equals("Sheet"))
      type = SecondaryStructure.SHEET;

    String label = JOptionPane.showInputDialog(this, "Enter a label for the structure?", "Enter label", JOptionPane.QUESTION_MESSAGE );
    if(label==null)
      label = "";

    SecondaryStructure ss = new SecondaryStructure(startRes, endRes, type, label);
    structures.add(ss);
    repaint();

  }

  public void doMouseDragged(MouseEvent evt)
  {
    paintNow(getGraphics());
  }

  void paintNow(Graphics g)
  {
    g.setColor(Color.white);
    g.setFont(new Font("Verdana", Font.PLAIN, 10));
    g.fillRect(0,0,getWidth(),getHeight());
    for(int i=0; i<structures.size(); i++)
    {
      SecondaryStructure ss = (SecondaryStructure)structures.get(i);


      if(ss.type==SecondaryStructure.SHEET)
      {
        g.setColor(Color.orange);
        if(ss==amend)
          g.setColor(Color.red);

        g.fillRect( (  ss.start - av.startRes) * av.charWidth,          4,
                    (ss.end - ss.start) * av.charWidth, 7);

        g.fillPolygon( new int[]{ (ss.end - av.startRes )* av.charWidth,
                       ( ss.end - av.startRes )* av.charWidth,
                       ( ss.end - av.startRes )* av.charWidth+7},
                       new int[]{0,14,7}, 3);
      }

      if(ss.type==SecondaryStructure.HELIX)
      {
        g.setColor(Color.magenta);
        if(ss==amend)
          g.setColor(Color.red);
        g.fillRoundRect((ss.start -av.startRes)* av.charWidth, 4,
                   (ss.end - ss.start) * av.charWidth, 7, 8,8);
      }



      g.setColor(Color.black);

      int labelWidth = g.getFontMetrics().stringWidth(ss.label);

      g.drawString( ss.label,  (ss.start +(ss.end-ss.start)/2 -av.startRes)*av.charWidth - labelWidth/2, 24);
    }
  }

  public void paintComponent(Graphics g)
  {
    paintNow(g);
  }
}

class SecondaryStructure
{
  public static int HELIX = 1;
  public static int SHEET = 2;
  public String label="";
  public int start= 20;
  public int end  = 40;
  public int type = HELIX;

  public SecondaryStructure(int s, int e, int t, String l)
  {
    start = s; end = e; type = t; label = l;
  }

}