/* $RCSfile$
 * $Author: hansonr $
 * $Date: 2019-11-17 22:49:25 -0600 (Sun, 17 Nov 2019) $
 * $Revision: 22002 $
 *
 * Copyright (C) 2002-2005  The Jmol Development Team
 *
 * Contact: jmol-developers@lists.sf.net
 *
 *  This library is free software; you can redistribute it and/or
 *  modify it under the terms of the GNU Lesser General Public
 *  License as published by the Free Software Foundation; either
 *  version 2.1 of the License, or (at your option) any later version.
 *
 *  This library is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 *  Lesser General Public License for more details.
 *
 *  You should have received a copy of the GNU Lesser General Public
 *  License along with this library; if not, write to the Free Software
 *  Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
 */
package org.openscience.jmol.app.jmolpanel;

import java.awt.Dimension;
import java.awt.Graphics;
import java.awt.Graphics2D;
import java.awt.Image;
import java.awt.Rectangle;
import java.awt.RenderingHints;
import java.awt.event.ActionEvent;
import java.awt.event.ComponentListener;
import java.awt.print.PageFormat;
import java.awt.print.Printable;
import java.util.Arrays;
import java.util.List;

import javax.swing.AbstractAction;
import javax.swing.AbstractButton;
import javax.swing.Action;
import javax.swing.ButtonGroup;
import javax.swing.JCheckBoxMenuItem;
import javax.swing.JComponent;
import javax.swing.JFrame;
import javax.swing.JPanel;

import org.jmol.awt.JmolFrame;
import org.jmol.i18n.GT;
import org.jmol.viewer.Viewer;

public class DisplayPanel extends JPanel implements JmolFrame,
    ComponentListener, Printable {

  protected Viewer vwr;

  protected String displaySpeed;

  protected Dimension startupDim;
  protected boolean haveDisplay;
  protected JFrame frame;
  protected JmolPanel jmolPanel;

  @Override
  public JFrame getFrame() {
    return frame;
  }

  protected DisplayPanel(JmolPanel jmolPanel) {
    this.jmolPanel = jmolPanel;
    frame = jmolPanel.frame;
    haveDisplay = jmolPanel.jmolApp.haveDisplay;
    startupDim = new Dimension(jmolPanel.startupWidth, jmolPanel.startupHeight);

    setMinimumSize(startupDim);
    setPreferredSize(startupDim);
    setMaximumSize(startupDim);
    setFocusable(true);
    if (System.getProperty("painttime", "false").equals("true"))
      showPaintTime = true;
    displaySpeed = System.getProperty("display.speed");
    if (displaySpeed == null) {
      displaySpeed = "ms";
    }
    setDoubleBuffered(false);
  }

  public void say(String msg) {
    jmolPanel.say(msg);
  }

  protected void setViewer(Viewer vwr) {
    this.vwr = vwr;
    updateSize(false);
  }

  // for now, default to true
  protected boolean showPaintTime = true;

  // current dimensions of the display screen
  protected final Dimension dimSize = new Dimension();
  protected final Rectangle rectClip = new Rectangle();

  public void start() {
    addComponentListener(this);
  }

  AbstractButton buttonRotate;
  AbstractButton buttonModelkit;

  ButtonGroup toolbarButtonGroup = new ButtonGroup();

  boolean isRotateMode() {
    return (buttonRotate != null && buttonRotate.isSelected());
  }

  void setRotateMode() {
    if (buttonRotate != null && !isRotateMode()) {
      buttonRotate.setSelected(true);
      vwr.setSelectionHalosEnabled(false);
    }
  }

  void setModelkitMode() {
    if (buttonModelkit != null)
      buttonModelkit.setSelected(true);
    vwr.setSelectionHalosEnabled(false);
  }

  @Override
  public void componentHidden(java.awt.event.ComponentEvent e) {
    //System.out.println("DisplayPanel.componentHidden");
  }

  @Override
  public void componentMoved(java.awt.event.ComponentEvent e) {
    //System.out.println("DisplayPanel.componentMoved " + e.getComponent().getX() + " " + e.getComponent().getY());
  }

  @Override
  public void componentResized(java.awt.event.ComponentEvent e) {
    //System.out.println("DisplayPanel.componentResized");
    updateSize(true);
  }

  @Override
  public void componentShown(java.awt.event.ComponentEvent e) {
    //System.out.println("DisplayPanel.componentShown");
    updateSize(true);
  }

  protected void updateSize(boolean doAll) {
    if (haveDisplay) {
      getSize(dimSize);
      if (dimSize.width == 0)
        return;
      vwr.setScreenDimension(dimSize.width, dimSize.height);
    } else {
      vwr.setScreenDimension(startupDim.width, startupDim.height);
    }
    if (!doAll)
      return;
    setRotateMode();
    if (haveDisplay)
      jmolPanel.setStatus(StatusBar.STATUS_TEXT, dimSize.width + " x "
          + dimSize.height);
    //vwr.refresh(Viewer.REFRESH_SYNC_MASK, "updateSize");
  }

  @Override
  public void paint(Graphics g) {
    updateSize(false); // important for JavaScript, as the update comes too late. 
    if (dimSize.width == 0)
      return;
    if (showPaintTime)
      startPaintClock();
    vwr.renderScreenImage(g, dimSize.width, dimSize.height);
    if (showPaintTime)
      stopPaintClock();
  }

  @Override
  public int print(Graphics g, PageFormat pf, int pageIndex) {
    Graphics2D g2 = (Graphics2D) g;
    if (pageIndex > 0)
      return Printable.NO_SUCH_PAGE;
    rectClip.x = rectClip.y = 0;
    int screenWidth = rectClip.width = vwr.getScreenWidth();
    int screenHeight = rectClip.height = vwr.getScreenHeight();
    Object image = vwr.getScreenImage();//was null,true
    int pageX = (int) pf.getImageableX();
    int pageY = (int) pf.getImageableY();
    int pageWidth = (int) pf.getImageableWidth();
    int pageHeight = (int) pf.getImageableHeight();
    float scaleWidth = pageWidth / (float) screenWidth;
    float scaleHeight = pageHeight / (float) screenHeight;
    float scale = (scaleWidth < scaleHeight ? scaleWidth : scaleHeight);
    if (scale < 1) {
      int width = (int) (screenWidth * scale);
      int height = (int) (screenHeight * scale);
      g2.setRenderingHint(RenderingHints.KEY_RENDERING,
          RenderingHints.VALUE_RENDER_QUALITY);
      g2.setRenderingHint(RenderingHints.KEY_INTERPOLATION,
          RenderingHints.VALUE_INTERPOLATION_BICUBIC);
      g2.drawImage((Image) image, pageX, pageY, width, height, null);
    } else {
      g2.drawImage((Image) image, pageX, pageY, null);
    }
    vwr.releaseScreenImage();
    return Printable.PAGE_EXISTS;
  }

  // The actions:

  protected HomeAction homeAction = new HomeAction();
  protected DefineCenterAction defineCenterAction = new DefineCenterAction();
  protected Action frontAction = new MoveToAction("front", "moveto 2.0 front");
  protected Action topAction = new MoveToAction("top",
      "moveto 1.0 front;moveto 2.0 top");
  protected Action bottomAction = new MoveToAction("bottom",
      "moveto 1.0 front;moveto 2.0 bottom");
  protected Action rightAction = new MoveToAction("right",
      "moveto 1.0 front;moveto 2.0 right");
  protected Action leftAction = new MoveToAction("left",
      "moveto 1.0 front;moveto 2.0 left");
  protected Action hydrogensAction = new CheckBoxMenuItemAction(
      "hydrogensCheck", "set showHydrogens");
  protected Action measurementsAction = new CheckBoxMenuItemAction(
      "measurementsCheck", "set showMeasurements");
  protected Action perspectiveAction = new CheckBoxMenuItemAction(
      "perspectiveCheck", "set PerspectiveDepth");
  protected Action axesAction = new CheckBoxMenuItemAction("axesCheck",
      "set showAxes");
  protected Action boundboxAction = new CheckBoxMenuItemAction("boundboxCheck",
      "set showBoundBox");
  // next three are not implemented
  protected Action deleteAction = new SetStatusAction("delete",
      GT.$("Delete atoms"));
  protected Action zoomAction = new SetStatusAction("zoom", null);
  protected Action xlateAction = new SetStatusAction("xlate", null);

  //

  // script actions are defined in Properties/Jmol-resource.properties
  // 

  /**
   * Action calling setStatus()
   */
  protected class SetStatusAction extends AbstractAction {
    protected final String statusText;

    public SetStatusAction(String name, String status) {
      super(name);
      this.statusText = status;
      this.setEnabled(true);
    }

    @Override
    public void actionPerformed(ActionEvent e) {
      vwr.setSelectionHalosEnabled(false);
      jmolPanel.setStatus(StatusBar.STATUS_COORD,
          statusText == null ? ((JComponent) e.getSource()).getToolTipText()
              : statusText);
    }
  }

  /**
   * Action calling moveTo()
   */
  protected class MoveToAction extends AbstractAction {
    protected final String action;

    public MoveToAction(String name, String action) {
      super(name);
      this.action = action;
      this.setEnabled(true);
    }

    @Override
    public void actionPerformed(ActionEvent e) {
      if (vwr.getShowBbcage() || vwr.getBooleanProperty("showUnitCell")) {
        vwr.evalStringQuiet(action);
      } else {
        vwr.evalStringQuiet("boundbox on;" + action + ";delay 1;boundbox off");
      }
    }
  }

  class DefineCenterAction extends AbstractAction {

    public DefineCenterAction() {
      super("definecenter");
      this.setEnabled(true);
    }

    @Override
    public void actionPerformed(ActionEvent e) {
      vwr.evalStringQuiet("center (selected)");
      setRotateMode();
    }
  }

  class HomeAction extends AbstractAction {

    public HomeAction() {
      super("home");
      this.setEnabled(true);
    }

    @Override
    public void actionPerformed(ActionEvent e) {
      vwr.homePosition();
      setRotateMode();
    }
  }

  /**
   * Action calling evalStringQuiet(&lt;action&gt; + CheckBoxState)
   */
  protected class CheckBoxMenuItemAction extends AbstractAction {
    protected final String action;

    public CheckBoxMenuItemAction(String name, String action) {
      super(name);
      this.action = action;
      this.setEnabled(true);
    }

    @Override
    public void actionPerformed(ActionEvent e) {
      JCheckBoxMenuItem cbmi = (JCheckBoxMenuItem) e.getSource();
      vwr.evalStringQuiet(action + " " + cbmi.isSelected());
    }
  }

  public void addActions(List<Action> actions) {
    Action[] a = new Action[] { deleteAction, zoomAction, xlateAction,
        frontAction, topAction, bottomAction, rightAction, leftAction,
        defineCenterAction, hydrogensAction, measurementsAction, homeAction,
        perspectiveAction, axesAction, boundboxAction, };
    actions.addAll(Arrays.asList(a));
  }

  // code to record last and average times
  // last and average of all the previous times are shown in the status window

  protected static int timeLast = 0;
  protected static int timeCount;
  protected static int timeTotal;

  protected void resetTimes() {
    timeCount = timeTotal = 0;
    timeLast = -1;
  }

  protected void recordTime(int time) {
    if (timeLast != -1) {
      timeTotal += timeLast;
      ++timeCount;
    }
    timeLast = time;
  }

  protected long timeBegin;
  protected int lastMotionEventNumber;

  protected void startPaintClock() {
    timeBegin = System.currentTimeMillis();
    int motionEventNumber = vwr.getMotionEventNumber();
    if (lastMotionEventNumber != motionEventNumber) {
      lastMotionEventNumber = motionEventNumber;
      resetTimes();
    }
  }

  protected void stopPaintClock() {
    int time = (int) (System.currentTimeMillis() - timeBegin);
    recordTime(time);
    showTimes();
  }

  protected String fmt(int num) {
    if (num < 0)
      return "---";
    if (num < 10)
      return "  " + num;
    if (num < 100)
      return " " + num;
    return "" + num;
  }

  protected void showTimes() {
    // round, don't truncate
    int timeAverage = (timeCount == 0 ? -1 : (timeTotal + timeCount / 2)
        / timeCount);
    jmolPanel.setStatus(StatusBar.STATUS_TIME,
        (displaySpeed.equalsIgnoreCase("fps") ? fmt(1000 / timeLast) + "FPS : "
            + fmt(1000 / timeAverage) + "FPS" : vwr.getP("_memory") + " Mb; "
            + fmt(timeLast) + "/" + timeAverage + " ms"));
  }

  public void setJmolSize(Dimension d) {
    dimSize.width = d.width;
    dimSize.height = d.height;
    setPreferredSize(d);
  }

}