-
-
Notifications
You must be signed in to change notification settings - Fork 62
/
VillagerEntry.java
86 lines (71 loc) · 2.54 KB
/
VillagerEntry.java
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
package jeresources.entry;
import jeresources.collection.TradeList;
import mezz.jei.api.recipe.IFocus;
import net.minecraft.entity.passive.EntityVillager;
import net.minecraft.item.ItemStack;
import java.util.ArrayList;
import java.util.LinkedList;
import java.util.List;
import java.util.stream.Collectors;
public class VillagerEntry {
private final List<TradeList> tradeList;
private final int profession, career;
private final String name;
public VillagerEntry(String name, int profession, int career, List<List<EntityVillager.ITradeList>> tradesLists) {
this.name = name;
this.profession = profession;
this.career = career;
this.tradeList = new LinkedList<>();
addITradeLists(tradesLists);
}
public void addITradeLists(List<List<EntityVillager.ITradeList>> tradesLists) {
int i = 0;
for (List<EntityVillager.ITradeList> levelList : tradesLists) {
TradeList trades = this.tradeList.size() > i ? this.tradeList.get(i) : new TradeList(this);
levelList.forEach(trades::addITradeList);
this.tradeList.add(trades);
i++;
}
}
public TradeList getVillagerTrades(int level) {
return tradeList.get(level);
}
public List<ItemStack> getInputs() {
List<ItemStack> list = new LinkedList<>();
for (List<TradeList.Trade> trades : this.tradeList) {
for (TradeList.Trade trade : trades) {
list.add(trade.getMinBuyStack1());
list.add(trade.getMinBuyStack2());
}
}
return list;
}
public List<ItemStack> getOutputs() {
List<ItemStack> list = new LinkedList<>();
for (List<TradeList.Trade> trades : this.tradeList)
list.addAll(trades.stream().map(TradeList.Trade::getMinSellStack).collect(Collectors.toList()));
return list;
}
public int getMaxLevel() {
return tradeList.size();
}
public int getCareer() {
return career;
}
public String getName() {
return this.name;
}
public String getDisplayName() {
return "entity.Villager." + this.name;
}
public int getProfession() {
return this.profession;
}
public List<Integer> getPossibleLevels(IFocus<ItemStack> focus) {
List<Integer> levels = new ArrayList<>();
for (int i = 0; i < tradeList.size(); i++)
if (tradeList.get(i) != null && tradeList.get(i).getFocusedList(focus).size() > 0)
levels.add(i + 1);
return levels;
}
}