package appeng.me; import java.util.Collection; import java.util.HashMap; import java.util.Iterator; import java.util.Set; import appeng.api.AEApi; import appeng.api.networking.IGrid; import appeng.api.networking.IGridCache; import appeng.api.networking.IGridHost; import appeng.api.networking.IGridNode; import appeng.api.networking.IMachineSet; import appeng.api.networking.events.MENetworkEvent; import appeng.api.networking.events.MENetworkPostCacheConstruction; import appeng.api.util.IReadOnlyCollection; import appeng.core.WorldSettings; import appeng.hooks.TickHandler; import appeng.util.ReadOnlyCollection; public class Grid implements IGrid { GridStorage myStorage; NetworkEventBus bus = new NetworkEventBus(); HashMap, Set> Machines = new HashMap, Set>(); HashMap, GridCacheWrapper> caches = new HashMap, GridCacheWrapper>(); GridNode pivot; public Grid(GridNode center) { this.pivot = center; HashMap, IGridCache> myCaches = AEApi.instance().registries().gridCache().createCacheInstance( this ); for (Class c : myCaches.keySet()) { bus.readClass( c, myCaches.get( c ).getClass() ); caches.put( c, new GridCacheWrapper( myCaches.get( c ) ) ); } TickHandler.instance.addNetwork( this ); center.setGrid( this ); postEvent( new MENetworkPostCacheConstruction() ); } public Set> getMachineClasses() { return Machines.keySet(); } @Override public IGridNode getPivot() { return pivot; } public int size() { int out = 0; for (Collection x : Machines.values()) out += x.size(); return out; } public void remove(GridNode gridNode) { for (IGridCache c : caches.values()) c.removeNode( gridNode, gridNode.getMachine() ); Collection nodes = Machines.get( gridNode.getMachineClass() ); if ( nodes != null ) nodes.remove( gridNode ); gridNode.setGridStorage( null ); if ( pivot == gridNode ) { Iterator n = getNodes().iterator(); if ( n.hasNext() ) pivot = (GridNode) n.next(); else { pivot = null; TickHandler.instance.removeNetwork( this ); myStorage.remove(); } } } public void add(GridNode gridNode) { Class mClass = gridNode.getMachineClass(); Set nodes = Machines.get( mClass ); if ( nodes == null ) { Machines.put( mClass, nodes = new MachineSet( mClass ) ); bus.readClass( mClass, mClass ); } // handle loading grid storages. if ( gridNode.getGridStorage() != null ) { GridStorage gs = gridNode.getGridStorage(); if ( gs.getGrid() == null ) { myStorage = gs; myStorage.setGrid( this ); for (IGridCache gc : caches.values()) gc.onJoin( myStorage ); } else if ( gs.getGrid() != this ) { if ( myStorage == null ) { myStorage = WorldSettings.getInstance().getNewGridStorage(); myStorage.setGrid( this ); } GridStorage tmp = new GridStorage(); if ( !gs.hasDivided( myStorage ) ) { gs.addDivided( myStorage ); for (IGridCache gc : ((Grid) gs.getGrid()).caches.values()) gc.onSplit( tmp ); for (IGridCache gc : caches.values()) gc.onJoin( tmp ); } } } else if ( myStorage == null ) { myStorage = WorldSettings.getInstance().getNewGridStorage(); myStorage.setGrid( this ); } // update grid node... gridNode.setGridStorage( myStorage ); // track node. nodes.add( gridNode ); for (IGridCache c : caches.values()) c.addNode( gridNode, gridNode.getMachine() ); gridNode.gridProxy.gridChanged(); // postEventTo( gridNode, networkChanged ); } @Override public IReadOnlyCollection getNodes() { return new NodeIteratable( Machines ); } @Override public IReadOnlyCollection> getMachinesClasses() { return new ReadOnlyCollection>( Machines.keySet() ); } @Override public IMachineSet getMachines(Class c) { MachineSet s = (MachineSet) Machines.get( c ); if ( s == null ) return new MachineSet( c ); return s; } @Override public C getCache(Class iface) { return (C) caches.get( iface ).myCache; } @Override public MENetworkEvent postEventTo(IGridNode node, MENetworkEvent ev) { return bus.postEventTo( this, (GridNode) node, ev ); } @Override public MENetworkEvent postEvent(MENetworkEvent ev) { return bus.postEvent( this, ev ); } public void requestSave() { myStorage.markDirty(); WorldSettings.getInstance().save(); } public void update() { // are ther any nodes left? if ( pivot != null ) { for (IGridCache gc : caches.values()) { gc.onUpdateTick(); } } } public Iterable getCacheWrappers() { return caches.values(); } @Override public boolean isEmpty() { return pivot == null; } public void saveState() { for (IGridCache c : caches.values()) { c.populateGridStorage( myStorage ); } } }