149 lines
3.7 KiB
Java
149 lines
3.7 KiB
Java
package ee.futur.easygiantsfoundry;
|
|
|
|
import ee.futur.easygiantsfoundry.enums.MetalBarSource;
|
|
import ee.futur.easygiantsfoundry.enums.MetalBarType;
|
|
import lombok.Getter;
|
|
import lombok.Value;
|
|
import net.runelite.api.Item;
|
|
import net.runelite.api.ItemContainer;
|
|
import net.runelite.api.gameval.InventoryID;
|
|
import net.runelite.api.gameval.ItemID;
|
|
|
|
import javax.inject.Inject;
|
|
import javax.inject.Singleton;
|
|
import java.util.HashMap;
|
|
import java.util.Map;
|
|
|
|
@Singleton
|
|
public class MetalBarCounter
|
|
{
|
|
private final Map<Integer, Map<MetalBarType, CountsBySource>> index = new HashMap<>(); // container id -> bar counts
|
|
|
|
@Inject
|
|
private EasyGiantsFoundryConfig config;
|
|
|
|
@Getter
|
|
private boolean seenBank = false;
|
|
|
|
public int get(MetalBarType type)
|
|
{
|
|
int count = 0;
|
|
for (Map<MetalBarType, CountsBySource> counts : index.values())
|
|
{
|
|
count += counts.get(type).sum(config);
|
|
}
|
|
|
|
return count;
|
|
}
|
|
|
|
public void clear()
|
|
{
|
|
index.clear();
|
|
seenBank = false;
|
|
}
|
|
|
|
public void put(ItemContainer container)
|
|
{
|
|
int tinOre = 0;
|
|
int copperOre = 0;
|
|
int ironOre = 0;
|
|
|
|
if (container.getId() == InventoryID.BANK)
|
|
{
|
|
seenBank = true;
|
|
}
|
|
|
|
Map<MetalBarType, CountsBySource> counts = newCounts();
|
|
for (Item item : container.getItems())
|
|
{
|
|
MetalBarValues.Record record = MetalBarValues.get(item.getId());
|
|
if (record == null)
|
|
{
|
|
continue;
|
|
}
|
|
|
|
// ore special cases:
|
|
// * add bronze bars equal to min(tin, copper)
|
|
// * there is an edge case here where it won't sum quite right multiple item containers but I don't really
|
|
// care to code for that right now
|
|
// * iron ore counts for both iron and steel
|
|
switch (item.getId())
|
|
{
|
|
case ItemID.TIN_ORE:
|
|
case ItemID.Cert.TIN_ORE:
|
|
tinOre += item.getQuantity();
|
|
break;
|
|
case ItemID.COPPER_ORE:
|
|
case ItemID.Cert.COPPER_ORE:
|
|
copperOre += item.getQuantity();
|
|
break;
|
|
case ItemID.IRON_ORE:
|
|
case ItemID.Cert.IRON_ORE:
|
|
ironOre += item.getQuantity();
|
|
break;
|
|
default:
|
|
counts.compute(record.getType(), (k, v) ->
|
|
v.add(record.getSource(), record.getValue() * item.getQuantity()));
|
|
}
|
|
}
|
|
|
|
int finalTinOre = tinOre;
|
|
int finalCopperOre = copperOre;
|
|
int finalIronOre = ironOre;
|
|
counts.compute(MetalBarType.BRONZE, (k, v) ->
|
|
v.add(MetalBarSource.ORE, Math.min(finalTinOre, finalCopperOre)));
|
|
counts.compute(MetalBarType.IRON, (k, v) ->
|
|
v.add(MetalBarSource.ORE, finalIronOre));
|
|
counts.compute(MetalBarType.STEEL, (k, v) ->
|
|
v.add(MetalBarSource.ORE, finalIronOre));
|
|
|
|
index.put(container.getId(), counts);
|
|
}
|
|
|
|
private static Map<MetalBarType, CountsBySource> newCounts()
|
|
{
|
|
Map<MetalBarType, CountsBySource> counts = new HashMap<>();
|
|
counts.put(MetalBarType.BRONZE, CountsBySource.empty());
|
|
counts.put(MetalBarType.IRON, CountsBySource.empty());
|
|
counts.put(MetalBarType.STEEL, CountsBySource.empty());
|
|
counts.put(MetalBarType.MITHRIL, CountsBySource.empty());
|
|
counts.put(MetalBarType.ADAMANT, CountsBySource.empty());
|
|
counts.put(MetalBarType.RUNITE, CountsBySource.empty());
|
|
return counts;
|
|
}
|
|
|
|
@Value
|
|
private static class CountsBySource
|
|
{
|
|
int ores, bars, equipment;
|
|
|
|
public static CountsBySource empty()
|
|
{
|
|
return new CountsBySource(0, 0, 0);
|
|
}
|
|
|
|
public CountsBySource add(MetalBarSource source, int count)
|
|
{
|
|
switch (source)
|
|
{
|
|
case ORE:
|
|
return new CountsBySource(ores + count, bars, equipment);
|
|
case BAR:
|
|
return new CountsBySource(ores, bars + count, equipment);
|
|
case EQUIPMENT:
|
|
return new CountsBySource(ores, bars, equipment + count);
|
|
default:
|
|
return this;
|
|
}
|
|
}
|
|
|
|
public int sum(EasyGiantsFoundryConfig config)
|
|
{
|
|
int sum = config.countOre() ? ores : 0;
|
|
sum += config.countBars() ? bars : 0;
|
|
sum += config.countEquipment() ? equipment : 0;
|
|
return sum;
|
|
}
|
|
}
|
|
}
|