dmolony-DiskBrowser/src/com/bytezone/diskbrowser/gui/DiskLayoutImage.java
2018-06-30 19:40:38 +10:00

309 lines
8.2 KiB
Java

package com.bytezone.diskbrowser.gui;
import java.awt.*;
import java.awt.event.InputEvent;
import java.awt.event.KeyAdapter;
import java.awt.event.KeyEvent;
import java.awt.event.MouseAdapter;
import java.awt.event.MouseEvent;
import java.awt.geom.AffineTransform;
import java.util.List;
import javax.swing.JPanel;
import javax.swing.Scrollable;
import javax.swing.SwingConstants;
import com.bytezone.diskbrowser.disk.Disk;
import com.bytezone.diskbrowser.disk.DiskAddress;
import com.bytezone.diskbrowser.disk.FormattedDisk;
import com.bytezone.diskbrowser.disk.SectorType;
import com.bytezone.diskbrowser.gui.DiskLayoutPanel.LayoutDetails;
import com.bytezone.diskbrowser.gui.RedoHandler.RedoEvent;
import com.bytezone.diskbrowser.gui.RedoHandler.RedoListener;
class DiskLayoutImage extends JPanel implements Scrollable, RedoListener
{
private static final Cursor crosshairCursor = new Cursor (Cursor.CROSSHAIR_CURSOR);
private static final Color[] lightColors =
{ Color.WHITE, Color.YELLOW, Color.PINK, Color.CYAN, Color.ORANGE, Color.GREEN };
private FormattedDisk disk;
private LayoutDetails layoutDetails;
private boolean showFreeSectors;
private final DiskLayoutSelection selectionHandler = new DiskLayoutSelection ();
private boolean redo;
private boolean isRetina;
// set defaults (used until a real disk is set)
private int bw = 30;
private int bh = 15;
private int gw = 8;
private int gh = 35;
public DiskLayoutImage ()
{
setPreferredSize (new Dimension (240 + 1, 525 + 1));
addMouseListener (new MyMouseListener ());
setBackground (new Color (0xE0, 0xE0, 0xE0));
setOpaque (true);
addKeyListener (new MyKeyListener ());
}
public void setDisk (FormattedDisk disk, LayoutDetails details)
{
this.disk = disk;
layoutDetails = details;
bw = layoutDetails.block.width;
bh = layoutDetails.block.height;
gw = layoutDetails.grid.width;
gh = layoutDetails.grid.height;
setPreferredSize (new Dimension (gw * bw + 1, gh * bh + 1));
selectionHandler.setSelection (null);
Graphics2D g = (Graphics2D) this.getGraphics ();
if (g != null) // panel might not be showing
isRetina = g.getFontRenderContext ().getTransform ()
.equals (AffineTransform.getScaleInstance (2.0, 2.0));
repaint ();
}
public FormattedDisk getDisk ()
{
return disk;
}
public void setShowFreeSectors (boolean showFree)
{
showFreeSectors = showFree;
repaint ();
}
void setSelection (List<DiskAddress> sectors)
{
selectionHandler.setSelection (sectors);
if (sectors != null && sectors.size () > 0)
{
DiskAddress da = sectors.size () == 1 ? sectors.get (0) : sectors.get (1);
// if (da != null)
scrollRectToVisible (layoutDetails.getLocation (da));
}
repaint ();
}
@Override
protected void paintComponent (Graphics g)
{
super.paintComponent (g);
if (disk == null)
return;
Rectangle clipRect = g.getClipBounds ();
Point p1 = new Point (clipRect.x / bw * bw, clipRect.y / bh * bh);
Point p2 = new Point ((clipRect.x + clipRect.width - 1) / bw * bw,
(clipRect.y + clipRect.height - 1) / bh * bh);
int maxBlock = gw * gh;
Disk d = disk.getDisk ();
// this stops an index error when using alt-5 to switch to 512-byte blocks
// if (maxBlock > d.getTotalBlocks ())
// maxBlock = d.getTotalBlocks ();
// the index error is caused by not recalculating the grid layout
Graphics2D g2d = (Graphics2D) g;
for (int y = p1.y; y <= p2.y; y += bh)
for (int x = p1.x; x <= p2.x; x += bw)
{
int blockNo = y / bh * gw + x / bw;
if (blockNo < maxBlock)
{
DiskAddress da = d.getDiskAddress (blockNo);
boolean free = showFreeSectors && disk.isSectorFree (da);
boolean selected = selectionHandler.isSelected (da);
drawBlock (g2d, blockNo, x, y, free, selected);
}
}
}
private void drawBlock (Graphics2D g, int blockNo, int x, int y, boolean flagFree,
boolean selected)
{
SectorType type = disk.getSectorType (blockNo);
if (type == null)
{
System.out.println ("Sector type is null " + blockNo);
return;
}
int offset = (bw - 4) / 2 + 1;
Rectangle rect = new Rectangle (x, y, bw, bh);
int width = rect.width - (isRetina ? 2 : 3) + 1;
int height = rect.height - (isRetina ? 2 : 3) + 1;
int offset2 = isRetina ? 1 : 2;
// draw frame
// if (false)
// {
// g.setColor (Color.GRAY);
// // g.drawRect (rect.x, rect.y, rect.width, rect.height);
// g.draw (rect);
// }
// draw blocks
g.setColor (type.colour);
g.fillRect (rect.x + offset2, rect.y + offset2, width, height);
// draw an indicator in free blocks
if (flagFree)
{
g.setColor (getContrastColor (type));
g.drawOval (rect.x + offset - 2, rect.y + 4, 7, 7);
}
// draw an indicator in selected blocks
if (selected)
{
g.setColor (getContrastColor (type));
g.fillOval (rect.x + offset, rect.y + 6, 3, 3);
}
}
private Color getContrastColor (SectorType type)
{
for (Color color : lightColors)
if (type.colour == color)
return Color.BLACK;
return Color.WHITE;
}
@Override
public Dimension getPreferredScrollableViewportSize ()
{
return new Dimension (240 + 1, 525 + 1); // floppy disk size
}
@Override
public int getScrollableUnitIncrement (Rectangle visibleRect, int orientation,
int direction)
{
return orientation == SwingConstants.HORIZONTAL ? bw : bh;
}
@Override
public int getScrollableBlockIncrement (Rectangle visibleRect, int orientation,
int direction)
{
return orientation == SwingConstants.HORIZONTAL ? bw * 4 : bh * 10;
}
@Override
public boolean getScrollableTracksViewportHeight ()
{
return false;
}
@Override
public boolean getScrollableTracksViewportWidth ()
{
return false;
}
@Override
public void redo (RedoEvent redoEvent)
{
redo = true;
SectorSelectedEvent event = (SectorSelectedEvent) redoEvent.value;
setSelection (event.getSectors ());
fireSectorSelectionEvent (event);
redo = false;
requestFocusInWindow ();
}
private void fireSectorSelectionEvent ()
{
SectorSelectedEvent event =
new SectorSelectedEvent (this, selectionHandler.getHighlights (), disk);
fireSectorSelectionEvent (event);
}
private void fireSectorSelectionEvent (SectorSelectedEvent event)
{
event.redo = redo;
SectorSelectionListener[] listeners =
(listenerList.getListeners (SectorSelectionListener.class));
for (SectorSelectionListener listener : listeners)
listener.sectorSelected (event);
}
public void addSectorSelectionListener (SectorSelectionListener listener)
{
listenerList.add (SectorSelectionListener.class, listener);
}
public void removeSectorSelectionListener (SectorSelectionListener listener)
{
listenerList.remove (SectorSelectionListener.class, listener);
}
class MyKeyListener extends KeyAdapter
{
@Override
public void keyPressed (KeyEvent e)
{
switch (e.getKeyCode ())
{
case KeyEvent.VK_LEFT:
case KeyEvent.VK_RIGHT:
case KeyEvent.VK_UP:
case KeyEvent.VK_DOWN:
selectionHandler.cursorMove (disk, e);
fireSectorSelectionEvent ();
repaint ();
}
}
}
class MyMouseListener extends MouseAdapter
{
private Cursor currentCursor;
@Override
public void mouseClicked (MouseEvent e)
{
int x = e.getX () / bw;
int y = e.getY () / bh;
int blockNo = y * gw + x;
DiskAddress da = disk.getDisk ().getDiskAddress (blockNo);
boolean extend = ((e.getModifiersEx () & InputEvent.SHIFT_DOWN_MASK) > 0);
boolean append = ((e.getModifiersEx () & InputEvent.CTRL_DOWN_MASK) > 0);
selectionHandler.doClick (disk.getDisk (), da, extend, append);
fireSectorSelectionEvent ();
repaint ();
requestFocusInWindow ();
}
@Override
public void mouseEntered (MouseEvent e)
{
currentCursor = getCursor ();
setCursor (crosshairCursor);
}
@Override
public void mouseExited (MouseEvent e)
{
setCursor (currentCursor);
}
}
}