366 lines
8.8 KiB
Java
366 lines
8.8 KiB
Java
/*
|
|
* This file is part of Applied Energistics 2.
|
|
* Copyright (c) 2013 - 2015, AlgorithmX2, All rights reserved.
|
|
*
|
|
* Applied Energistics 2 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 3 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* Applied Energistics 2 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 Applied Energistics 2. If not, see <http://www.gnu.org/licenses/lgpl>.
|
|
*/
|
|
|
|
package appeng.services;
|
|
|
|
|
|
import java.io.File;
|
|
import java.util.HashMap;
|
|
import java.util.Map;
|
|
import java.util.Optional;
|
|
import java.util.concurrent.ExecutorService;
|
|
import java.util.concurrent.Executors;
|
|
import java.util.concurrent.Future;
|
|
import java.util.concurrent.ThreadFactory;
|
|
import java.util.concurrent.TimeUnit;
|
|
import javax.annotation.Nonnull;
|
|
|
|
import com.google.common.base.Preconditions;
|
|
|
|
import net.minecraft.block.Block;
|
|
import net.minecraft.world.World;
|
|
import net.minecraft.world.chunk.Chunk;
|
|
import net.minecraftforge.event.world.WorldEvent;
|
|
import net.minecraftforge.fml.common.eventhandler.SubscribeEvent;
|
|
|
|
import appeng.api.AEApi;
|
|
import appeng.api.util.DimensionalCoord;
|
|
import appeng.services.compass.CompassReader;
|
|
import appeng.services.compass.ICompassCallback;
|
|
import appeng.util.Platform;
|
|
|
|
|
|
public final class CompassService
|
|
{
|
|
private static final int CHUNK_SIZE = 16;
|
|
|
|
private final Map<World, CompassReader> worldSet = new HashMap<World, CompassReader>( 10 );
|
|
private final ExecutorService executor;
|
|
|
|
/**
|
|
* AE2 Folder for each world
|
|
*/
|
|
private final File worldCompassFolder;
|
|
|
|
private int jobSize;
|
|
|
|
public CompassService( @Nonnull final File worldCompassFolder, @Nonnull final ThreadFactory factory )
|
|
{
|
|
Preconditions.checkNotNull( worldCompassFolder );
|
|
|
|
this.worldCompassFolder = worldCompassFolder;
|
|
this.executor = Executors.newSingleThreadExecutor( factory );
|
|
this.jobSize = 0;
|
|
}
|
|
|
|
public Future<?> getCompassDirection( final DimensionalCoord coord, final int maxRange, final ICompassCallback cc )
|
|
{
|
|
this.jobSize++;
|
|
return this.executor.submit( new CMDirectionRequest( coord, maxRange, cc ) );
|
|
}
|
|
|
|
/**
|
|
* Ensure the a compass service is removed once a world gets unloaded by forge.
|
|
*
|
|
* @param event the event containing the unloaded world.
|
|
*/
|
|
@SubscribeEvent
|
|
public void unloadWorld( final WorldEvent.Unload event )
|
|
{
|
|
if( Platform.isServer() && this.worldSet.containsKey( event.getWorld() ) )
|
|
{
|
|
final CompassReader compassReader = this.worldSet.remove( event.getWorld() );
|
|
|
|
compassReader.close();
|
|
}
|
|
}
|
|
|
|
private int jobSize()
|
|
{
|
|
return this.jobSize;
|
|
}
|
|
|
|
private void cleanUp()
|
|
{
|
|
for( final CompassReader cr : this.worldSet.values() )
|
|
{
|
|
cr.close();
|
|
}
|
|
}
|
|
|
|
public void updateArea( final World w, final int chunkX, final int chunkZ )
|
|
{
|
|
final int x = chunkX << 4;
|
|
final int z = chunkZ << 4;
|
|
|
|
this.updateArea( w, x, CHUNK_SIZE, z );
|
|
this.updateArea( w, x, CHUNK_SIZE + 32, z );
|
|
this.updateArea( w, x, CHUNK_SIZE + 64, z );
|
|
this.updateArea( w, x, CHUNK_SIZE + 96, z );
|
|
|
|
this.updateArea( w, x, CHUNK_SIZE + 128, z );
|
|
this.updateArea( w, x, CHUNK_SIZE + 160, z );
|
|
this.updateArea( w, x, CHUNK_SIZE + 192, z );
|
|
this.updateArea( w, x, CHUNK_SIZE + 224, z );
|
|
}
|
|
|
|
public Future<?> updateArea( final World w, final int x, final int y, final int z )
|
|
{
|
|
this.jobSize++;
|
|
|
|
final int cx = x >> 4;
|
|
final int cdy = y >> 5;
|
|
final int cz = z >> 4;
|
|
|
|
final int low_y = cdy << 5;
|
|
final int hi_y = low_y + 32;
|
|
|
|
// lower level...
|
|
final Chunk c = w.getChunkFromChunkCoords( cx, cz );
|
|
|
|
Optional<Block> maybeBlock = AEApi.instance().definitions().blocks().skyStoneBlock().maybeBlock();
|
|
if( maybeBlock.isPresent() )
|
|
{
|
|
Block skyStoneBlock = maybeBlock.get();
|
|
for( int i = 0; i < CHUNK_SIZE; i++ )
|
|
{
|
|
for( int j = 0; j < CHUNK_SIZE; j++ )
|
|
{
|
|
for( int k = low_y; k < hi_y; k++ )
|
|
{
|
|
final Block blk = c.getBlockState( i, k, j ).getBlock();
|
|
if( blk == skyStoneBlock )
|
|
{
|
|
return this.executor.submit( new CMUpdatePost( w, cx, cz, cdy, true ) );
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
return this.executor.submit( new CMUpdatePost( w, cx, cz, cdy, false ) );
|
|
}
|
|
|
|
public void kill()
|
|
{
|
|
this.executor.shutdown();
|
|
|
|
try
|
|
{
|
|
this.executor.awaitTermination( 6, TimeUnit.MINUTES );
|
|
this.jobSize = 0;
|
|
|
|
for( final CompassReader cr : this.worldSet.values() )
|
|
{
|
|
cr.close();
|
|
}
|
|
|
|
this.worldSet.clear();
|
|
}
|
|
catch( final InterruptedException e )
|
|
{
|
|
// wrap this up..
|
|
}
|
|
}
|
|
|
|
private CompassReader getReader( final World w )
|
|
{
|
|
CompassReader cr = this.worldSet.get( w );
|
|
|
|
if( cr == null )
|
|
{
|
|
cr = new CompassReader( w.provider.getDimension(), this.worldCompassFolder );
|
|
this.worldSet.put( w, cr );
|
|
}
|
|
|
|
return cr;
|
|
}
|
|
|
|
private int dist( final int ax, final int az, final int bx, final int bz )
|
|
{
|
|
final int up = ( bz - az ) * CHUNK_SIZE;
|
|
final int side = ( bx - ax ) * CHUNK_SIZE;
|
|
|
|
return up * up + side * side;
|
|
}
|
|
|
|
private double rad( final int ax, final int az, final int bx, final int bz )
|
|
{
|
|
final int up = bz - az;
|
|
final int side = bx - ax;
|
|
|
|
return Math.atan2( -up, side ) - Math.PI / 2.0;
|
|
}
|
|
|
|
private class CMUpdatePost implements Runnable
|
|
{
|
|
|
|
public final World world;
|
|
|
|
public final int chunkX;
|
|
public final int chunkZ;
|
|
public final int doubleChunkY; // 32 blocks instead of 16.
|
|
public final boolean value;
|
|
|
|
public CMUpdatePost( final World w, final int cx, final int cz, final int dcy, final boolean val )
|
|
{
|
|
this.world = w;
|
|
this.chunkX = cx;
|
|
this.doubleChunkY = dcy;
|
|
this.chunkZ = cz;
|
|
this.value = val;
|
|
}
|
|
|
|
@Override
|
|
public void run()
|
|
{
|
|
CompassService.this.jobSize--;
|
|
|
|
final CompassReader cr = CompassService.this.getReader( this.world );
|
|
cr.setHasBeacon( this.chunkX, this.chunkZ, this.doubleChunkY, this.value );
|
|
|
|
if( CompassService.this.jobSize() < 2 )
|
|
{
|
|
CompassService.this.cleanUp();
|
|
}
|
|
}
|
|
}
|
|
|
|
private class CMDirectionRequest implements Runnable
|
|
{
|
|
|
|
public final int maxRange;
|
|
public final DimensionalCoord coord;
|
|
public final ICompassCallback callback;
|
|
|
|
public CMDirectionRequest( final DimensionalCoord coord, final int getMaxRange, final ICompassCallback cc )
|
|
{
|
|
this.coord = coord;
|
|
this.maxRange = getMaxRange;
|
|
this.callback = cc;
|
|
}
|
|
|
|
@Override
|
|
public void run()
|
|
{
|
|
CompassService.this.jobSize--;
|
|
|
|
final int cx = this.coord.x >> 4;
|
|
final int cz = this.coord.z >> 4;
|
|
|
|
final CompassReader cr = CompassService.this.getReader( this.coord.getWorld() );
|
|
|
|
// Am I standing on it?
|
|
if( cr.hasBeacon( cx, cz ) )
|
|
{
|
|
this.callback.calculatedDirection( true, true, -999, 0 );
|
|
|
|
if( CompassService.this.jobSize() < 2 )
|
|
{
|
|
CompassService.this.cleanUp();
|
|
}
|
|
|
|
return;
|
|
}
|
|
|
|
// spiral outward...
|
|
for( int offset = 1; offset < this.maxRange; offset++ )
|
|
{
|
|
final int minX = cx - offset;
|
|
final int minZ = cz - offset;
|
|
final int maxX = cx + offset;
|
|
final int maxZ = cz + offset;
|
|
|
|
int closest = Integer.MAX_VALUE;
|
|
int chosen_x = cx;
|
|
int chosen_z = cz;
|
|
|
|
for( int z = minZ; z <= maxZ; z++ )
|
|
{
|
|
if( cr.hasBeacon( minX, z ) )
|
|
{
|
|
final int closeness = CompassService.this.dist( cx, cz, minX, z );
|
|
if( closeness < closest )
|
|
{
|
|
closest = closeness;
|
|
chosen_x = minX;
|
|
chosen_z = z;
|
|
}
|
|
}
|
|
|
|
if( cr.hasBeacon( maxX, z ) )
|
|
{
|
|
final int closeness = CompassService.this.dist( cx, cz, maxX, z );
|
|
if( closeness < closest )
|
|
{
|
|
closest = closeness;
|
|
chosen_x = maxX;
|
|
chosen_z = z;
|
|
}
|
|
}
|
|
}
|
|
|
|
for( int x = minX + 1; x < maxX; x++ )
|
|
{
|
|
if( cr.hasBeacon( x, minZ ) )
|
|
{
|
|
final int closeness = CompassService.this.dist( cx, cz, x, minZ );
|
|
if( closeness < closest )
|
|
{
|
|
closest = closeness;
|
|
chosen_x = x;
|
|
chosen_z = minZ;
|
|
}
|
|
}
|
|
|
|
if( cr.hasBeacon( x, maxZ ) )
|
|
{
|
|
final int closeness = CompassService.this.dist( cx, cz, x, maxZ );
|
|
if( closeness < closest )
|
|
{
|
|
closest = closeness;
|
|
chosen_x = x;
|
|
chosen_z = maxZ;
|
|
}
|
|
}
|
|
}
|
|
|
|
if( closest < Integer.MAX_VALUE )
|
|
{
|
|
this.callback.calculatedDirection( true, false, CompassService.this.rad( cx, cz, chosen_x, chosen_z ), CompassService.this.dist( cx, cz, chosen_x, chosen_z ) );
|
|
|
|
if( CompassService.this.jobSize() < 2 )
|
|
{
|
|
CompassService.this.cleanUp();
|
|
}
|
|
|
|
return;
|
|
}
|
|
}
|
|
|
|
// didn't find shit...
|
|
this.callback.calculatedDirection( false, true, -999, 999 );
|
|
|
|
if( CompassService.this.jobSize() < 2 )
|
|
{
|
|
CompassService.this.cleanUp();
|
|
}
|
|
}
|
|
}
|
|
}
|