summaryrefslogtreecommitdiff
path: root/src/cuchaz/enigma/gui/BoxHighlightPainter.java
blob: df63f5a81b1636d41e3c809179f26b276b79b49b (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
/*******************************************************************************
 * Copyright (c) 2014 Jeff Martin.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the GNU Public License v3.0
 * which accompanies this distribution, and is available at
 * http://www.gnu.org/licenses/gpl.html
 * 
 * Contributors:
 *     Jeff Martin - initial API and implementation
 ******************************************************************************/
package cuchaz.enigma.gui;

import java.awt.Color;
import java.awt.Graphics;
import java.awt.Rectangle;
import java.awt.Shape;

import javax.swing.text.BadLocationException;
import javax.swing.text.Highlighter;
import javax.swing.text.JTextComponent;

public abstract class BoxHighlightPainter implements Highlighter.HighlightPainter
{
	private Color m_fillColor;
	private Color m_borderColor;
	
	protected BoxHighlightPainter( Color fillColor, Color borderColor )
	{
		m_fillColor = fillColor;
		m_borderColor = borderColor;
	}
	
	@Override
	public void paint( Graphics g, int start, int end, Shape shape, JTextComponent text )
	{
		Rectangle bounds = getBounds( text, start, end );
		
		// fill the area
		if( m_fillColor != null )
		{
			g.setColor( m_fillColor );
			g.fillRoundRect( bounds.x, bounds.y, bounds.width, bounds.height, 4, 4 );
		}
		
		// draw a box around the area
		g.setColor( m_borderColor );
		g.drawRoundRect( bounds.x, bounds.y, bounds.width, bounds.height, 4, 4 );
	}
	
	protected static Rectangle getBounds( JTextComponent text, int start, int end )
	{
		try
		{
			// determine the bounds of the text
			Rectangle bounds = text.modelToView( start ).union( text.modelToView( end ) );
			
			// adjust the box so it looks nice
			bounds.x -= 2;
			bounds.width += 2;
			bounds.y += 1;
			bounds.height -= 2;
			
			return bounds;
		}
		catch( BadLocationException ex )
		{
			// don't care... just return something
			return new Rectangle( 0, 0, 0, 0 );
		}
	}
}