Complete network logic

3 files changed, 110 insertions, 8 deletions

diff --git a/src/main/java/com/detrav/items/behaviours/BehaviourDetravToolElectricProPick.java b/src/main/java/com/detrav/items/behaviours/BehaviourDetravToolElectricProPick.java
index 42faadd8ea..9fa0f06fe5 100644
--- a/src/main/java/com/detrav/items/behaviours/BehaviourDetravToolElectricProPick.java
+++ b/src/main/java/com/detrav/items/behaviours/BehaviourDetravToolElectricProPick.java
@@ -3,18 +3,28 @@ package com.detrav.items.behaviours;
 import com.detrav.items.DetravMetaGeneratedTool01;
 import com.detrav.utils.DetravNetwork;
 import com.detrav.utils.DetravProPickPacket01;
+import gregtech.api.GregTech_API;
 import gregtech.api.items.GT_MetaBase_Item;
+import gregtech.api.util.GT_LanguageManager;
+import gregtech.common.blocks.GT_TileEntity_Ores;
+import net.minecraft.block.Block;
 import net.minecraft.entity.player.EntityPlayer;
 import net.minecraft.entity.player.EntityPlayerMP;
 import net.minecraft.item.ItemStack;
+import net.minecraft.tileentity.TileEntity;
 import net.minecraft.util.ChatComponentText;
 import net.minecraft.world.World;
+import net.minecraft.world.chunk.Chunk;
+import net.minecraft.world.chunk.storage.ChunkLoader;
+import net.minecraftforge.common.ForgeChunkManager;
+
+import java.util.ArrayList;
+import java.util.List;
 
 /**
  * Created by wital_000 on 19.03.2016.
  */
-public class BehaviourDetravToolElectricProPick extends BehaviourDetravToolProPick
-{
+public class BehaviourDetravToolElectricProPick extends BehaviourDetravToolProPick {
 
     public BehaviourDetravToolElectricProPick(int aCosts) {
         super(aCosts);
@@ -24,12 +34,47 @@ public class BehaviourDetravToolElectricProPick extends BehaviourDetravToolProPi
 
         if (!aWorld.isRemote) {
             //aPlayer.openGui();
-
-            aPlayer.addChatMessage(new ChatComponentText("Scanning Begin"));
+            DetravMetaGeneratedTool01 tool = (DetravMetaGeneratedTool01) aItem;
+            //aWorld.getChunkFromBlockCoords()
+            int cX = ((int) aPlayer.posX) >> 4;
+            int cZ = ((int) aPlayer.posZ) >> 4;
+            int size = aItem.getHarvestLevel(aStack, "") + 1;
+            List<Chunk> chunks = new ArrayList<Chunk>();
+            //aPlayer.addChatMessage(new ChatComponentText("Scanning Begin, wait!"));
+            //DetravProPickPacket01 packet = new DetravProPickPacket01();
+            for (int i = -size; i <= size; i++)
+                for (int j = -size; j <= size; j++)
+                    if (i != -size && i != size && j != -size && j != size)
+                        chunks.add(aWorld.getChunkFromChunkCoords(cX + i, cZ + j));
+            size = size - 1;
+            //c.gene
             DetravProPickPacket01 packet = new DetravProPickPacket01();
-            packet.level = ((DetravMetaGeneratedTool01) aItem).getHarvestLevel(aStack,"");
-            DetravNetwork.INSTANCE.sendToPlayer(packet,(EntityPlayerMP)aPlayer);
+            packet.chunkX = cX;
+            packet.chunkZ = cZ;
+            packet.size = size;
+            for (Chunk c : chunks) {
+                for (int x = 0; x < 16; x++)
+                    for (int z = 0; z < 16; z++) {
+                        int ySize = c.getHeightValue(x, z);
+                        for (int y = 1; y < ySize; y++) {
+                            Block b = c.getBlock(x, y, z);
+                            if (b == GregTech_API.sBlockOres1) {
+                                TileEntity entity = c.getTileEntityUnsafe(x, y, z);
+                                if (entity != null && entity instanceof GT_TileEntity_Ores) {
+                                    GT_TileEntity_Ores gt_entity = (GT_TileEntity_Ores) entity;
+                                    String name = GT_LanguageManager.getTranslation(
+                                            b.getUnlocalizedName() + "." + gt_entity.getMetaData() + ".name");
+                                    if (name.startsWith("Small")) continue;
+                                    packet.addBlock(c.xPosition * 16 + x, y, c.zPosition * 16 + z, gt_entity.getMetaData());
+                                }
+                            }
+                        }
+                    }
+            }
+            packet.level = ((DetravMetaGeneratedTool01) aItem).getHarvestLevel(aStack, "");
+            DetravNetwork.INSTANCE.sendToPlayer(packet, (EntityPlayerMP) aPlayer);
+            tool.doDamage(aStack, this.mCosts * chunks.size());
         }
-        return super.onItemRightClick(aItem,aStack,aWorld,aPlayer);
+        return super.onItemRightClick(aItem, aStack, aWorld, aPlayer);
     }
 }
\ No newline at end of file
diff --git a/src/main/java/com/detrav/items/behaviours/BehaviourDetravToolProPick.java b/src/main/java/com/detrav/items/behaviours/BehaviourDetravToolProPick.java
index 5c5b64d53c..2669c1a541 100644
--- a/src/main/java/com/detrav/items/behaviours/BehaviourDetravToolProPick.java
+++ b/src/main/java/com/detrav/items/behaviours/BehaviourDetravToolProPick.java
@@ -32,7 +32,7 @@ public class BehaviourDetravToolProPick extends Behaviour_None {
             "Found a very large sample of ",    //100-**
             "Found "
     };
-    private final int mCosts;
+    protected final int mCosts;
 
     public BehaviourDetravToolProPick(int aCosts)
     {
diff --git a/src/main/java/com/detrav/utils/DetravProPickPacket01.java b/src/main/java/com/detrav/utils/DetravProPickPacket01.java
index e85a570e0b..b06de84037 100644
--- a/src/main/java/com/detrav/utils/DetravProPickPacket01.java
+++ b/src/main/java/com/detrav/utils/DetravProPickPacket01.java
@@ -1,5 +1,6 @@
 package com.detrav.utils;
 
+import com.google.common.base.Strings;
 import com.google.common.io.ByteArrayDataInput;
 import com.google.common.io.ByteArrayDataOutput;
 import com.google.common.io.ByteStreams;
@@ -7,10 +8,17 @@ import net.minecraft.client.Minecraft;
 import net.minecraft.entity.player.EntityPlayer;
 import net.minecraft.util.ChatComponentText;
 
+import java.util.HashMap;
+
 /**
  * Created by wital_000 on 20.03.2016.
  */
 public class DetravProPickPacket01 extends DetravPacket {
+    public int chunkX;
+    public int chunkZ;
+    public int size;
+    HashMap<Byte,Short>[][] map = null;
+
     @Override
     public int getPacketID() {
         return 0;
@@ -22,6 +30,28 @@ public class DetravProPickPacket01 extends DetravPacket {
     public byte[] encode() {
         ByteArrayDataOutput tOut = ByteStreams.newDataOutput(1);
         tOut.writeInt(level);
+        tOut.writeInt(chunkX);
+        tOut.writeInt(chunkZ);
+        tOut.writeInt(size);
+        int aSize = (size*2+1)*16;
+        int checkOut = 0;
+        for(int i =0; i<aSize; i++)
+            for(int j =0; j<aSize; j++)
+            {
+                if(map[i][j]==null)
+                    tOut.writeByte(0);
+                else
+                {
+                    tOut.writeByte(map[i][j].keySet().size());
+                    for(byte key : map[i][j].keySet())
+                    {
+                        tOut.writeByte(key);
+                        tOut.writeShort(map[i][j].get(key));
+                        checkOut++;
+                    }
+                }
+            }
+        tOut.writeInt(checkOut);
         return tOut.toByteArray();
     }
 
@@ -29,6 +59,24 @@ public class DetravProPickPacket01 extends DetravPacket {
     public Object decode(ByteArrayDataInput aData) {
         DetravProPickPacket01 packet = new DetravProPickPacket01();
         packet.level = aData.readInt();
+        packet.chunkX = aData.readInt();
+        packet.chunkZ = aData.readInt();
+        packet.size = aData.readInt();
+        packet.map = new HashMap[(packet.size * 2 + 1) * 16][(packet.size * 2 + 1) * 16];
+        int aSize = (packet.size * 2 + 1) * 16;
+        int checkOut = 0;
+        for (int i = 0; i < aSize; i++)
+            for (int j = 0; j < aSize; j++) {
+                byte kSize = aData.readByte();
+                if(kSize == 0) continue;
+                packet.map[i][j] = new HashMap<Byte, Short>();
+                for (int k = 0; k < kSize; k++) {
+                    packet.map[i][j].put(aData.readByte(),aData.readShort());
+                    checkOut++;
+                }
+            }
+        int checkOut2 = aData.readInt();
+        if(checkOut != checkOut2) return new DetravProPickPacket01();
         return packet;
     }
 
@@ -36,4 +84,13 @@ public class DetravProPickPacket01 extends DetravPacket {
     public void process() {
         Minecraft.getMinecraft().thePlayer.addChatMessage(new ChatComponentText("NetworkTested: " + Integer.toString(level)));
     }
+
+    public void addBlock(int x, int y, int z, short metaData) {
+        if(map == null) map = new HashMap[(size*2+1)*16][(size*2+1)*16];
+        int aX = x - (chunkX-size)*16;
+        int aZ = z - (chunkZ-size)*16;
+        if(map[aX][aZ] == null) map[aX][aZ] = new HashMap<Byte, Short>();
+        map[aX][aZ].put((byte)y,metaData);
+        //String key = String.format(("x_y"))
+    }
 }