Added abstract GA code

(cherry picked from commit c534c14)
This commit is contained in:
austinio7116
2018-04-26 22:56:26 +01:00
committed by maustin
parent 7e3b9ccc2d
commit 0d4088ff7a
4 changed files with 336 additions and 1 deletions

View File

@@ -0,0 +1,59 @@
package forge.planarconquestgenerate;
import com.google.common.base.Function;
import com.google.common.base.Predicates;
import com.google.common.collect.Iterables;
import com.google.common.collect.Lists;
import forge.GuiBase;
import forge.GuiDesktop;
import forge.StaticData;
import forge.card.CardRulesPredicates;
import forge.deck.CardRelationMatrixGenerator;
import forge.deck.Deck;
import forge.deck.DeckFormat;
import forge.deck.io.DeckStorage;
import forge.game.GameFormat;
import forge.game.GameRules;
import forge.game.GameType;
import forge.item.PaperCard;
import forge.limited.CardRanker;
import forge.model.FModel;
import forge.properties.ForgeConstants;
import forge.properties.ForgePreferences;
import org.testng.annotations.Test;
import java.io.File;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
@Test
public class PlanarConquestGeneratorGATest {
@Test
public void test(){
GuiBase.setInterface(new GuiDesktop());
FModel.initialize(null, new Function<ForgePreferences, Void>() {
@Override
public Void apply(ForgePreferences preferences) {
preferences.setPref(ForgePreferences.FPref.LOAD_CARD_SCRIPTS_LAZILY, false);
return null;
}
});
PlanarConquestGeneraterGA ga = new PlanarConquestGeneraterGA();
ga.initializeCards(40);
ga.run();
List<Deck> winners = ga.listFinalPopulation();
DeckStorage storage = new DeckStorage(new File(ForgeConstants.DECK_CONSTRUCTED_DIR), ForgeConstants.DECK_BASE_DIR);
for(Deck deck:winners){
storage.save(deck);
}
}
}

View File

@@ -109,6 +109,114 @@ public class PlanarConquestGeneratorTest {
}
}
@Test
public void generateBestDecksGA() {
GuiBase.setInterface(new GuiDesktop());
FModel.initialize(null, new Function<ForgePreferences, Void>() {
@Override
public Void apply(ForgePreferences preferences) {
preferences.setPref(ForgePreferences.FPref.LOAD_CARD_SCRIPTS_LAZILY, false);
return null;
}
});
int cardsToUse=40;
GameFormat format = FModel.getFormats().getStandard();
GameRules rules = new GameRules(GameType.Constructed);
standardMap = CardRelationMatrixGenerator.cardPools.get(format.getName());
List<String> cardNames = new ArrayList<>(standardMap.keySet());
List<PaperCard> cards = new ArrayList<>();
for(String cardName:cardNames){
cards.add(StaticData.instance().getCommonCards().getUniqueByName(cardName));
}
List<PaperCard> rankedList = CardRanker.rankCardsInDeck(cards);
List<String> sets = new ArrayList<>();
sets.add("XLN");
sets.add("RIX");
DeckStorage storage = new DeckStorage(new File(ForgeConstants.DECK_CONSTRUCTED_DIR), ForgeConstants.DECK_BASE_DIR);
GameFormat planarConquestFormat = new GameFormat("conquest",sets,null);
DeckFormat deckFormat = DeckFormat.PlanarConquest;
Iterable<PaperCard> filtered= Iterables.filter(rankedList, Predicates.and(
Predicates.compose(CardRulesPredicates.IS_KEPT_IN_AI_DECKS, PaperCard.FN_GET_RULES),
Predicates.compose(CardRulesPredicates.Presets.IS_NON_LAND, PaperCard.FN_GET_RULES),
planarConquestFormat.getFilterPrinted()));
List<PaperCard> filteredList = Lists.newArrayList(filtered);
DeckGroup deckGroup = new DeckGroup("SimulatedTournament");
List<TournamentPlayer> players = new ArrayList<>();
int numPlayers=0;
for(PaperCard card: filteredList.subList(0,cardsToUse)){
System.out.println(card.getName());
for( int i=0; i<4;i++){
Deck genDeck = DeckgenUtil.buildPlanarConquestDeck(card, planarConquestFormat, deckFormat);
Deck d = new Deck(genDeck,genDeck.getName()+"_"+i+"_"+0);
deckGroup.addAiDeck(d);
players.add(new TournamentPlayer(GamePlayerUtil.createAiPlayer(d.getName(), 0), numPlayers));
numPlayers++;
}
}
TournamentSwiss tourney = null;
for (int m=1;m<10;++m) {
//get best decks in population
tourney = new TournamentSwiss(players, 2);
tourney = runTournament(tourney, rules, numPlayers, deckGroup);
players = new ArrayList<>();
DeckGroup newDeckGroup = new DeckGroup("SimulatedTournament"+m);
numPlayers=0;
int winnerCount = new Float(tourney.getAllPlayers().size()* .5f).intValue();
for (int k = 0; k < winnerCount; k++) {
String deckName = tourney.getAllPlayers().get(k).getPlayer().getName();
for (Deck winningDeck : deckGroup.getAiDecks()) {
if (winningDeck.getName().equals(deckName)) {
newDeckGroup.addAiDeck(winningDeck);
players.add(new TournamentPlayer(GamePlayerUtil.createAiPlayer(winningDeck.getName(), 0), numPlayers));
numPlayers++;
break;
}
}
}
deckGroup = newDeckGroup;
//add random decks
for(PaperCard card: filteredList.subList(0,cardsToUse)){
System.out.println(card.getName());
for( int i=0; i<2;i++){
Deck genDeck = DeckgenUtil.buildPlanarConquestDeck(card, planarConquestFormat, deckFormat);
Deck d = new Deck(genDeck,genDeck.getName()+"_"+i+"_"+m);
deckGroup.addAiDeck(d);
players.add(new TournamentPlayer(GamePlayerUtil.createAiPlayer(d.getName(), 0), numPlayers));
numPlayers++;
}
}
}
tourney = new TournamentSwiss(players, 2);
tourney = runTournament(tourney, rules, numPlayers, deckGroup);
int winnerCount = new Float(tourney.getAllPlayers().size()* 0.25f).intValue();
for (int k = 0; k < winnerCount; k++) {
String deckName = tourney.getAllPlayers().get(k).getPlayer().getName();
Deck winningDeck;
for(Deck deck:deckGroup.getAiDecks()){
if(deck.getName().equals(deckName)){
winningDeck=deck;
storage.save(winningDeck);
System.out.println(winningDeck.getName());
System.out.println(winningDeck.getAllCardsInASinglePool().toString());
break;
}
}
}
}
@Test
public void generatePlanarConquestCommanderDecks() {
@@ -155,7 +263,7 @@ public class PlanarConquestGeneratorTest {
DeckGroup deckGroup = new DeckGroup("SimulatedTournament");
List<TournamentPlayer> players = new ArrayList<>();
int numPlayers=0;
for( int i=0; i<2;i++){
for( int i=0; i<16;i++){
Deck genDeck = DeckgenUtil.buildPlanarConquestCommanderDeck(card, planarConquestFormat, deckFormat);
Deck d = new Deck(genDeck,genDeck.getName()+"_"+i);
deckGroup.addAiDeck(d);