mirror of
https://github.com/SquidDev-CC/CC-Tweaked
synced 2024-06-25 22:53:22 +00:00
413 lines
13 KiB
Java
413 lines
13 KiB
Java
/**
|
|
* This file is part of ComputerCraft - http://www.computercraft.info
|
|
* Copyright Daniel Ratcliffe, 2011-2016. Do not distribute without permission.
|
|
* Send enquiries to dratcliffe@gmail.com
|
|
*/
|
|
|
|
package dan200.computercraft.core.apis;
|
|
|
|
import dan200.computercraft.api.lua.ILuaContext;
|
|
import dan200.computercraft.api.lua.LuaException;
|
|
import dan200.computercraft.core.computer.IComputerEnvironment;
|
|
import dan200.computercraft.core.terminal.Terminal;
|
|
import dan200.computercraft.shared.util.Palette;
|
|
import org.apache.commons.lang3.ArrayUtils;
|
|
|
|
import java.util.HashMap;
|
|
import java.util.Map;
|
|
import java.util.Set;
|
|
|
|
public class TermAPI implements ILuaAPI
|
|
{
|
|
private final Terminal m_terminal;
|
|
private final IComputerEnvironment m_environment;
|
|
|
|
public TermAPI( IAPIEnvironment _environment )
|
|
{
|
|
m_terminal = _environment.getTerminal();
|
|
m_environment = _environment.getComputerEnvironment();
|
|
}
|
|
|
|
@Override
|
|
public String[] getNames()
|
|
{
|
|
return new String[] {
|
|
"term"
|
|
};
|
|
}
|
|
|
|
@Override
|
|
public void startup( )
|
|
{
|
|
}
|
|
|
|
@Override
|
|
public void advance( double _dt )
|
|
{
|
|
}
|
|
|
|
@Override
|
|
public void shutdown( )
|
|
{
|
|
}
|
|
|
|
@Override
|
|
public String[] getMethodNames()
|
|
{
|
|
return new String[] {
|
|
"write",
|
|
"scroll",
|
|
"setCursorPos",
|
|
"setCursorBlink",
|
|
"getCursorPos",
|
|
"getSize",
|
|
"clear",
|
|
"clearLine",
|
|
"setTextColour",
|
|
"setTextColor",
|
|
"setBackgroundColour",
|
|
"setBackgroundColor",
|
|
"isColour",
|
|
"isColor",
|
|
"getTextColour",
|
|
"getTextColor",
|
|
"getBackgroundColour",
|
|
"getBackgroundColor",
|
|
"blit",
|
|
"setPaletteColour",
|
|
"setPaletteColor",
|
|
"getPaletteColour",
|
|
"getPaletteColor"
|
|
};
|
|
}
|
|
|
|
public static int parseColour( Object[] args, boolean _enableColours ) throws LuaException
|
|
{
|
|
if( args.length < 1 || args[0] == null || !(args[0] instanceof Double) )
|
|
{
|
|
throw new LuaException( "Expected number" );
|
|
}
|
|
int colour = (int)((Double)args[0]).doubleValue();
|
|
if( colour <= 0 )
|
|
{
|
|
throw new LuaException( "Colour out of range" );
|
|
}
|
|
colour = getHighestBit( colour ) - 1;
|
|
if( colour < 0 || colour > 15 )
|
|
{
|
|
throw new LuaException( "Colour out of range" );
|
|
}
|
|
if( !_enableColours && (colour != 0 && colour != 15 && colour != 7 && colour != 8) )
|
|
{
|
|
throw new LuaException( "Colour not supported" );
|
|
}
|
|
return colour;
|
|
}
|
|
|
|
public static Object[] encodeColour( int colour ) throws LuaException
|
|
{
|
|
return new Object[] {
|
|
1 << colour
|
|
};
|
|
}
|
|
|
|
public static void setColour( Terminal terminal, int colour, float r, float g, float b )
|
|
{
|
|
if( terminal.getPalette() != null )
|
|
{
|
|
terminal.getPalette().setColour( colour, r, g, b );
|
|
terminal.setChanged();
|
|
}
|
|
}
|
|
|
|
public static void setColour( Terminal terminal, HashMap<Object, Object> colours) throws LuaException
|
|
{
|
|
final double lg2 = Math.log( 2 );
|
|
|
|
for(Map.Entry<Object, Object> e : colours.entrySet())
|
|
{
|
|
if(e.getKey() instanceof Double)
|
|
{
|
|
int index = 15 - (int)( Math.log( (Double)e.getKey() ) / lg2 );
|
|
|
|
try
|
|
{
|
|
if (e.getValue() instanceof HashMap)
|
|
{
|
|
@SuppressWarnings({ "unchecked" }) // There isn't really a nice way around this :(
|
|
HashMap<Object, Object> colour = (HashMap<Object, Object>) e.getValue();
|
|
|
|
setColour(
|
|
terminal,
|
|
index,
|
|
( (Double)colour.get( 1.0 ) ).floatValue(),
|
|
( (Double)colour.get( 2.0 ) ).floatValue(),
|
|
( (Double)colour.get( 3.0 ) ).floatValue()
|
|
);
|
|
}
|
|
else if (e.getValue() instanceof Double)
|
|
{
|
|
float[] rgb = Palette.decodeRGB8( ((Double)e.getValue()).intValue() );
|
|
|
|
setColour(
|
|
terminal,
|
|
index,
|
|
rgb[0], rgb[1], rgb[2]
|
|
);
|
|
}
|
|
}
|
|
catch(ClassCastException cce)
|
|
{
|
|
throw new LuaException( "Malformed colour table " + cce.getMessage() );
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public Object[] callMethod( ILuaContext context, int method, Object[] args ) throws LuaException
|
|
{
|
|
switch( method )
|
|
{
|
|
case 0:
|
|
{
|
|
// write
|
|
String text;
|
|
if( args.length > 0 && args[0] != null ) {
|
|
text = args[0].toString();
|
|
} else {
|
|
text = "";
|
|
}
|
|
|
|
synchronized( m_terminal )
|
|
{
|
|
m_terminal.write( text );
|
|
m_terminal.setCursorPos( m_terminal.getCursorX() + text.length(), m_terminal.getCursorY() );
|
|
}
|
|
return null;
|
|
}
|
|
case 1:
|
|
{
|
|
// scroll
|
|
if( args.length != 1 || args[0] == null || !(args[0] instanceof Double) )
|
|
{
|
|
throw new LuaException( "Expected number" );
|
|
}
|
|
|
|
int y = (int)((Double)args[0]).doubleValue();
|
|
synchronized( m_terminal )
|
|
{
|
|
m_terminal.scroll(y);
|
|
}
|
|
return null;
|
|
}
|
|
case 2:
|
|
{
|
|
// setCursorPos
|
|
if( args.length != 2 || args[0] == null || !(args[0] instanceof Double) || args[1] == null || !(args[1] instanceof Double) )
|
|
{
|
|
throw new LuaException( "Expected number, number" );
|
|
}
|
|
int x = (int)((Double)args[0]).doubleValue() - 1;
|
|
int y = (int)((Double)args[1]).doubleValue() - 1;
|
|
synchronized( m_terminal )
|
|
{
|
|
m_terminal.setCursorPos( x, y );
|
|
}
|
|
return null;
|
|
}
|
|
case 3:
|
|
{
|
|
// setCursorBlink
|
|
if( args.length != 1 || args[0] == null || !(args[0] instanceof Boolean) )
|
|
{
|
|
throw new LuaException( "Expected boolean" );
|
|
}
|
|
boolean b = (Boolean) args[ 0 ];
|
|
synchronized( m_terminal )
|
|
{
|
|
m_terminal.setCursorBlink( b );
|
|
}
|
|
return null;
|
|
}
|
|
case 4:
|
|
{
|
|
// getCursorPos
|
|
int x, y;
|
|
synchronized( m_terminal )
|
|
{
|
|
x = m_terminal.getCursorX();
|
|
y = m_terminal.getCursorY();
|
|
}
|
|
return new Object[] { x + 1, y + 1 };
|
|
}
|
|
case 5:
|
|
{
|
|
// getSize
|
|
int width, height;
|
|
synchronized( m_terminal )
|
|
{
|
|
width = m_terminal.getWidth();
|
|
height = m_terminal.getHeight();
|
|
}
|
|
return new Object[] { width, height };
|
|
}
|
|
case 6:
|
|
{
|
|
// clear
|
|
synchronized( m_terminal )
|
|
{
|
|
m_terminal.clear();
|
|
}
|
|
return null;
|
|
}
|
|
case 7:
|
|
{
|
|
// clearLine
|
|
synchronized( m_terminal )
|
|
{
|
|
m_terminal.clearLine();
|
|
}
|
|
return null;
|
|
}
|
|
case 8:
|
|
case 9:
|
|
{
|
|
// setTextColour/setTextColor
|
|
int colour = parseColour( args, m_environment.isColour() );
|
|
synchronized( m_terminal )
|
|
{
|
|
m_terminal.setTextColour( colour );
|
|
}
|
|
return null;
|
|
}
|
|
case 10:
|
|
case 11:
|
|
{
|
|
// setBackgroundColour/setBackgroundColor
|
|
int colour = parseColour( args, m_environment.isColour() );
|
|
synchronized( m_terminal )
|
|
{
|
|
m_terminal.setBackgroundColour( colour );
|
|
}
|
|
return null;
|
|
}
|
|
case 12:
|
|
case 13:
|
|
{
|
|
// isColour/isColor
|
|
return new Object[] { m_environment.isColour() };
|
|
}
|
|
case 14:
|
|
case 15:
|
|
{
|
|
// getTextColour/getTextColor
|
|
return encodeColour( m_terminal.getTextColour() );
|
|
}
|
|
case 16:
|
|
case 17:
|
|
{
|
|
// getBackgroundColour/getBackgroundColor
|
|
return encodeColour( m_terminal.getBackgroundColour() );
|
|
}
|
|
case 18:
|
|
{
|
|
// blit
|
|
if( args.length < 3 || !(args[0] instanceof String) || !(args[1] instanceof String) || !(args[2] instanceof String) )
|
|
{
|
|
throw new LuaException( "Expected string, string, string" );
|
|
}
|
|
|
|
String text = (String)args[0];
|
|
String textColour = (String)args[1];
|
|
String backgroundColour = (String)args[2];
|
|
if( textColour.length() != text.length() || backgroundColour.length() != text.length() )
|
|
{
|
|
throw new LuaException( "Arguments must be the same length" );
|
|
}
|
|
|
|
synchronized( m_terminal )
|
|
{
|
|
m_terminal.blit( text, textColour, backgroundColour );
|
|
m_terminal.setCursorPos( m_terminal.getCursorX() + text.length(), m_terminal.getCursorY() );
|
|
}
|
|
return null;
|
|
}
|
|
case 19:
|
|
case 20:
|
|
{
|
|
// setColour/setColor
|
|
if( !m_environment.isColour() )
|
|
{
|
|
// Make sure you can't circumvent greyscale terminals with this function.
|
|
throw new LuaException( "Colour not supported" );
|
|
}
|
|
|
|
if(args.length >= 1 && args[0] instanceof HashMap)
|
|
{
|
|
@SuppressWarnings( { "unchecked" } ) // There isn't really a nice way around this :(
|
|
HashMap<Object, Object> colourTbl = (HashMap<Object, Object>)args[0];
|
|
setColour( m_terminal, colourTbl );
|
|
return null;
|
|
}
|
|
|
|
if(args.length == 2 && args[0] instanceof Double && args[1] instanceof Double)
|
|
{
|
|
int colour = 15 - parseColour( args, m_environment.isColour() );
|
|
int hex = ((Double)args[1]).intValue();
|
|
float[] rgb = Palette.decodeRGB8( hex );
|
|
setColour( m_terminal, colour, rgb[0], rgb[1], rgb[2] );
|
|
return null;
|
|
}
|
|
|
|
if(args.length >= 4 && args[0] instanceof Double && args[1] instanceof Double && args[2] instanceof Double && args[3] instanceof Double)
|
|
{
|
|
int colour = 15 - parseColour( args, m_environment.isColour() );
|
|
float r = ((Double)args[1]).floatValue();
|
|
float g = ((Double)args[2]).floatValue();
|
|
float b = ((Double)args[3]).floatValue();
|
|
setColour( m_terminal, colour, r, g, b );
|
|
return null;
|
|
}
|
|
|
|
throw new LuaException( "Expected table or number, number or number, number, number, number" );
|
|
}
|
|
case 21:
|
|
case 22:
|
|
{
|
|
// getColour/getColor
|
|
if (args.length < 1 || !(args[0] instanceof Double))
|
|
{
|
|
throw new LuaException( "Expected number" );
|
|
}
|
|
|
|
int colour = 15 - parseColour( args, m_environment.isColour() );
|
|
|
|
synchronized( m_terminal )
|
|
{
|
|
if ( m_terminal.getPalette() != null )
|
|
{
|
|
return ArrayUtils.toObject( m_terminal.getPalette().getColour( colour ) );
|
|
}
|
|
}
|
|
return null;
|
|
}
|
|
default:
|
|
{
|
|
return null;
|
|
}
|
|
}
|
|
}
|
|
|
|
private static int getHighestBit( int group )
|
|
{
|
|
int bit = 0;
|
|
while( group > 0 )
|
|
{
|
|
group >>= 1;
|
|
bit++;
|
|
}
|
|
return bit;
|
|
}
|
|
}
|