aboutsummaryrefslogtreecommitdiff
path: root/src/main/java/common/tileentities/TE_ItemProxyEndpoint.java
blob: 88a8ef8bdadafcb6b3f9248e1e6c8563e62d0051 (plain)
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
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
package common.tileentities;

import java.util.HashSet;
import net.minecraft.entity.player.EntityPlayer;
import net.minecraft.inventory.IInventory;
import net.minecraft.inventory.ISidedInventory;
import net.minecraft.item.ItemStack;
import net.minecraft.tileentity.TileEntity;
import net.minecraftforge.common.util.ForgeDirection;

public class TE_ItemProxyEndpoint extends TileEntity implements ISidedInventory {

    private byte channel = -1;
    private IInventory proxyInventory = null;
    private int tickCounter = 0;

    public TE_ItemProxyEndpoint() {
        channel = 0;
    }

    public void setChannel(byte channel) {
        this.channel = channel;
    }

    public int getChannel() {
        return channel;
    }

    @Override
    public void updateEntity() {
        if (tickCounter == 20) {
            if (channel != -1) {
                proxyInventory = searchSource();
            }
            tickCounter = 0;
        }
        tickCounter++;
    }

    private TE_ItemProxySource searchSource() {

        final HashSet<TE_ItemProxySource> sources = new HashSet<>();
        final HashSet<String> visited = new HashSet<>();

        for (ForgeDirection next : ForgeDirection.VALID_DIRECTIONS) {
            final TileEntity te = super.getWorldObj()
                    .getTileEntity(
                            super.xCoord + next.offsetX, super.yCoord + next.offsetY, super.zCoord + next.offsetZ);
            if (te instanceof TE_ItemProxyCable) {
                final TE_ItemProxyCable cable = (TE_ItemProxyCable) te;
                if (cable.isConnected(next.getOpposite())) {
                    searchSourceRecursive(sources, visited, next.getOpposite(), cable);
                }
            }
        }

        if (sources.isEmpty()) {
            return null;
        } else {
            return sources.iterator().next();
        }
    }

    private void searchSourceRecursive(
            HashSet<TE_ItemProxySource> sources,
            HashSet<String> visited,
            ForgeDirection from,
            TE_ItemProxyCable nextTarget) {

        if (!visited.contains(nextTarget.getIdentifier())) {
            visited.add(nextTarget.getIdentifier());

            for (ForgeDirection next : ForgeDirection.VALID_DIRECTIONS) {
                if (next != from) {
                    final TileEntity te = super.getWorldObj()
                            .getTileEntity(
                                    nextTarget.xCoord + next.offsetX,
                                    nextTarget.yCoord + next.offsetY,
                                    nextTarget.zCoord + next.offsetZ);
                    if (te instanceof TE_ItemProxyCable) {
                        final TE_ItemProxyCable cable = (TE_ItemProxyCable) te;
                        if (cable.isConnected(next.getOpposite())) {
                            searchSourceRecursive(sources, visited, next.getOpposite(), cable);
                        }
                    } else if (te instanceof TE_ItemProxySource) {
                        sources.add((TE_ItemProxySource) te);
                    }
                }
            }
        }
    }

    @Override
    public int getSizeInventory() {
        return 1;
    }

    @Override
    public ItemStack getStackInSlot(int slot) {
        if (proxyInventory != null && slot == 0) {
            return proxyInventory.getStackInSlot(channel);
        } else {
            return null;
        }
    }

    @Override
    public ItemStack decrStackSize(int slot, int amount) {
        if (proxyInventory != null && slot == 0) {
            return proxyInventory.decrStackSize(channel, amount);
        } else {
            return null;
        }
    }

    @Override
    public ItemStack getStackInSlotOnClosing(int slot) {
        return (proxyInventory != null) ? proxyInventory.getStackInSlotOnClosing(channel) : null;
    }

    @Override
    public void setInventorySlotContents(int slot, ItemStack itemStack) {
        if (proxyInventory != null && slot == 0) {
            proxyInventory.setInventorySlotContents(channel, itemStack);
        }
    }

    @Override
    public String getInventoryName() {
        return (proxyInventory != null) ? "Connected: " + proxyInventory.getInventoryName() : "Untethered Proxy";
    }

    @Override
    public boolean hasCustomInventoryName() {
        return true;
    }

    @Override
    public int getInventoryStackLimit() {
        return (proxyInventory != null) ? proxyInventory.getInventoryStackLimit() : 0;
    }

    @Override
    public boolean isUseableByPlayer(EntityPlayer player) {
        return true;
    }

    @Override
    public void openInventory() {}

    @Override
    public void closeInventory() {}

    @Override
    public boolean isItemValidForSlot(int slot, ItemStack itemStack) {
        if (proxyInventory != null && slot == 0) {
            return proxyInventory.isItemValidForSlot(channel, itemStack);
        } else {
            return false;
        }
    }

    @Override
    public int[] getAccessibleSlotsFromSide(int side) {
        return new int[] {0};
    }

    @Override
    public boolean canInsertItem(int slot, ItemStack itemStack, int side) {
        return isItemValidForSlot(slot, itemStack);
    }

    @Override
    public boolean canExtractItem(int slot, ItemStack itemStack, int side) {
        return slot == 0;
    }
}