From d95a3e470ce3b4d72ae58326fd5225d2029c6e51 Mon Sep 17 00:00:00 2001 From: LevelX2 Date: Tue, 12 Sep 2017 17:25:18 +0200 Subject: [PATCH] Updated some logic. --- .../card/dl/sources/TokensMtgImageSource.java | 19 ++- .../plugins/card/images/DownloadPictures.java | 123 ++++++++++++++---- 2 files changed, 117 insertions(+), 25 deletions(-) diff --git a/Mage.Client/src/main/java/org/mage/plugins/card/dl/sources/TokensMtgImageSource.java b/Mage.Client/src/main/java/org/mage/plugins/card/dl/sources/TokensMtgImageSource.java index 95f4294a05..f04dad377e 100644 --- a/Mage.Client/src/main/java/org/mage/plugins/card/dl/sources/TokensMtgImageSource.java +++ b/Mage.Client/src/main/java/org/mage/plugins/card/dl/sources/TokensMtgImageSource.java @@ -34,11 +34,14 @@ import java.io.InputStreamReader; import java.net.URL; import java.util.ArrayList; import java.util.HashMap; +import java.util.LinkedHashSet; import java.util.List; import java.util.Map; +import java.util.Set; import java.util.logging.Level; import org.apache.log4j.Logger; import org.mage.plugins.card.images.CardDownloadData; +import org.mage.plugins.card.images.DownloadPictures; /** * @@ -51,6 +54,7 @@ public enum TokensMtgImageSource implements CardImageSource { // [[EXP/Name, TokenData> private HashMap> tokensData; + private static final Set supportedSets = new LinkedHashSet(); private final Object tokensDataSync = new Object(); @@ -175,6 +179,7 @@ public enum TokensMtgImageSource implements CardImageSource { private HashMap> getTokensData() throws IOException { synchronized (tokensDataSync) { if (tokensData == null) { + DownloadPictures.getInstance().updateAndViewMessage("Creating token data..."); tokensData = new HashMap<>(); // get tokens data from resource file @@ -186,6 +191,7 @@ public enum TokensMtgImageSource implements CardImageSource { if (list == null) { list = new ArrayList<>(); tokensData.put(key, list); + supportedSets.add(tokenData.getExpansionSetCode()); logger.info("Added key: " + key); } list.add(tokenData); @@ -219,8 +225,10 @@ public enum TokensMtgImageSource implements CardImageSource { } } } - } catch (Exception exception) { - logger.warn("Failed to get tokens description from tokens.mtg.onl", exception); + DownloadPictures.getInstance().updateAndViewMessage(""); + } catch (Exception ex) { + logger.warn("Failed to get tokens description from tokens.mtg.onl", ex); + DownloadPictures.getInstance().updateAndViewMessage(ex.getMessage()); } } } @@ -319,6 +327,13 @@ public enum TokensMtgImageSource implements CardImageSource { public void doPause(String httpImageUrl) { } + @Override + public ArrayList getSupportedSets() { + ArrayList supportedSetsCopy = new ArrayList<>(); + supportedSetsCopy.addAll(supportedSets); + return supportedSetsCopy; + } + @Override public boolean isImageProvided(String setCode, String cardName) { try { diff --git a/Mage.Client/src/main/java/org/mage/plugins/card/images/DownloadPictures.java b/Mage.Client/src/main/java/org/mage/plugins/card/images/DownloadPictures.java index 2e81ebd36c..92e0d20c1a 100644 --- a/Mage.Client/src/main/java/org/mage/plugins/card/images/DownloadPictures.java +++ b/Mage.Client/src/main/java/org/mage/plugins/card/images/DownloadPictures.java @@ -26,6 +26,7 @@ import mage.cards.Sets; import mage.cards.repository.CardCriteria; import mage.cards.repository.CardInfo; import mage.cards.repository.CardRepository; +import mage.client.MageFrame; import mage.client.constants.Constants; import mage.client.dialog.PreferencesDialog; import mage.client.util.sets.ConstructedFormats; @@ -44,12 +45,15 @@ import org.mage.plugins.card.utils.CardImageUtils; public class DownloadPictures extends DefaultBoundedRangeModel implements Runnable { + private static DownloadPictures instance; + private static final Logger logger = Logger.getLogger(DownloadPictures.class); public static final String ALL_IMAGES = "- All images from that source"; public static final String ALL_STANDARD_IMAGES = "- All images from standard from that source"; public static final String ALL_TOKENS = "- Only all token images from that source"; + private JDialog dialog; private final JProgressBar bar; private final JOptionPane dlg; private boolean cancel; @@ -57,7 +61,7 @@ public class DownloadPictures extends DefaultBoundedRangeModel implements Runnab private final JButton startDownloadButton; private int cardIndex; private List allCardsMissingImage; - List cardsToDownload = new ArrayList<>(); + private List cardsToDownload; private int missingCards = 0; private int missingTokens = 0; @@ -65,6 +69,7 @@ public class DownloadPictures extends DefaultBoundedRangeModel implements Runnab List selectedSetCodes = new ArrayList<>(); private final JComboBox jComboBoxServer; + private final JLabel jLabelMessage; private final JLabel jLabelAllMissing; private final JLabel jLabelServer; @@ -107,6 +112,10 @@ public class DownloadPictures extends DefaultBoundedRangeModel implements Runnab } + public static DownloadPictures getInstance() { + return instance; + } + public static void main(String[] args) { startDownload(); } @@ -118,18 +127,15 @@ public class DownloadPictures extends DefaultBoundedRangeModel implements Runnab * JOptionPane.showMessageDialog(null, * "All card pictures have been downloaded."); return; } */ - DownloadPictures download = new DownloadPictures(); - JDialog dlg = download.getDlg(null); - dlg.setVisible(true); - dlg.dispose(); - download.cancel = true; + instance = new DownloadPictures(MageFrame.getInstance()); + Thread t1 = new Thread(new LoadMissingCardData(instance)); + t1.start(); + instance.getDlg().setVisible(true); + instance.getDlg().dispose(); + instance.cancel = true; } - public JDialog getDlg(JFrame frame) { - String title = "Downloading images"; - - final JDialog dialog = this.dlg.createDialog(frame, title); - closeButton.addActionListener(e -> dialog.setVisible(false)); + public JDialog getDlg() { return dialog; } @@ -137,12 +143,23 @@ public class DownloadPictures extends DefaultBoundedRangeModel implements Runnab this.cancel = cancel; } - public DownloadPictures() { + static int WIDTH = 400; + + public DownloadPictures(JFrame frame) { + + cardsToDownload = new ArrayList<>(); + JPanel p0 = new JPanel(); p0.setLayout(new BoxLayout(p0, BoxLayout.Y_AXIS)); p0.add(Box.createVerticalStrut(5)); + jLabelMessage = new JLabel(); + jLabelMessage.setAlignmentX(Component.CENTER_ALIGNMENT); + jLabelMessage.setText("Initializing image download..."); + p0.add(jLabelMessage); + p0.add(Box.createVerticalStrut(5)); + jLabelAllMissing = new JLabel(); jLabelAllMissing.setAlignmentX(Component.LEFT_ALIGNMENT); @@ -153,6 +170,7 @@ public class DownloadPictures extends DefaultBoundedRangeModel implements Runnab jLabelServer = new JLabel(); jLabelServer.setText("Please select image source:"); jLabelServer.setAlignmentX(Component.LEFT_ALIGNMENT); + jLabelServer.setVisible(false); p0.add(jLabelServer); p0.add(Box.createVerticalStrut(5)); @@ -160,12 +178,18 @@ public class DownloadPictures extends DefaultBoundedRangeModel implements Runnab jComboBoxServer = new JComboBox(); jComboBoxServer.setModel(new DefaultComboBoxModel(DownloadSources.values())); jComboBoxServer.setAlignmentX(Component.LEFT_ALIGNMENT); + jComboBoxServer.setAlignmentY(Component.LEFT_ALIGNMENT); jComboBoxServer.addItemListener((ItemEvent event) -> { if (event.getStateChange() == ItemEvent.SELECTED) { comboBoxServerItemSelected(event); } }); + Dimension d = jComboBoxServer.getPreferredSize(); + d.width = WIDTH; + jComboBoxServer.setPreferredSize(d); p0.add(jComboBoxServer); + jComboBoxServer.setVisible(false); + // set the first source as default cardImageSource = WizardCardsImageSource.instance; @@ -175,10 +199,11 @@ public class DownloadPictures extends DefaultBoundedRangeModel implements Runnab jLabelSet = new JLabel(); jLabelSet.setText("Please select sets to download the images for:"); jLabelSet.setAlignmentX(Component.LEFT_ALIGNMENT); + jLabelSet.setVisible(false); p0.add(jLabelSet); jComboBoxSet = new JComboBox(); - jComboBoxSet.setModel(new DefaultComboBoxModel<>(getSetsForCurrentImageSource())); +// jComboBoxSet.setModel(new DefaultComboBoxModel<>(getSetsForCurrentImageSource())); jComboBoxSet.setAlignmentX(Component.LEFT_ALIGNMENT); jComboBoxSet.addItemListener((ItemEvent event) -> { if (event.getStateChange() == ItemEvent.SELECTED) { @@ -187,6 +212,7 @@ public class DownloadPictures extends DefaultBoundedRangeModel implements Runnab }); p0.add(jComboBoxSet); + jComboBoxSet.setVisible(false); p0.add(Box.createVerticalStrut(5)); @@ -203,28 +229,58 @@ public class DownloadPictures extends DefaultBoundedRangeModel implements Runnab p0.add(bar); bar.setStringPainted(true); - Dimension d = bar.getPreferredSize(); - d.width = 400; + d = bar.getPreferredSize(); + d.width = WIDTH; bar.setPreferredSize(d); + bar.setVisible(false); // JOptionPane Object[] options = {startDownloadButton, closeButton = new JButton("Cancel")}; - dlg = new JOptionPane(p0, JOptionPane.PLAIN_MESSAGE, JOptionPane.DEFAULT_OPTION, null, options, options[1]); + startDownloadButton.setVisible(false); + closeButton.addActionListener(e -> dialog.setVisible(false)); + closeButton.setVisible(false); - setAllMissingCards(); + dlg = new JOptionPane(p0, JOptionPane.PLAIN_MESSAGE, JOptionPane.DEFAULT_OPTION, null, options, options[1]); + dialog = this.dlg.createDialog(frame, "Downloading images"); } public void setAllMissingCards() { + updateAndViewMessage("Get all available cards from the repository..."); List cards = CardRepository.instance.findCards(new CardCriteria()); + updateAndViewMessage("Check which images are missing ..."); this.allCardsMissingImage = getNeededCards(cards); + updateAndViewMessage("Check which images the current source is providing ..."); + jComboBoxSet.setModel(new DefaultComboBoxModel<>(getSetsForCurrentImageSource())); + updateCardsToDownload(jComboBoxSet.getSelectedItem().toString()); + + jComboBoxServer.setVisible(true); + jLabelServer.setVisible(true); + jComboBoxSet.setVisible(true); + jLabelSet.setVisible(true); + bar.setVisible(true); + startDownloadButton.setVisible(true); + closeButton.setVisible(true); + + updateAndViewMessage(""); } private void comboBoxServerItemSelected(ItemEvent evt) { - cardImageSource = ((DownloadSources) evt.getItem()).getSource(); - // update the available sets / token comboBox - jComboBoxSet.setModel(new DefaultComboBoxModel<>(getSetsForCurrentImageSource())); - updateCardsToDownload(jComboBoxSet.getSelectedItem().toString()); + if (jComboBoxServer.isEnabled()) { + cardImageSource = ((DownloadSources) evt.getItem()).getSource(); + // update the available sets / token comboBox + jComboBoxSet.setModel(new DefaultComboBoxModel<>(getSetsForCurrentImageSource())); + updateCardsToDownload(jComboBoxSet.getSelectedItem().toString()); + } + } + + public void updateAndViewMessage(String text) { + jLabelMessage.setText(text); + if (dialog != null) { + dialog.pack(); + dialog.validate(); + dialog.repaint(); + } } private Object[] getSetsForCurrentImageSource() { @@ -290,8 +346,8 @@ public class DownloadPictures extends DefaultBoundedRangeModel implements Runnab int numberTokenImagesAvailable = 0; int numberCardImagesAvailable = 0; for (CardDownloadData data : allCardsMissingImage) { - if (data.isToken() && cardImageSource.isTokenSource()) { - if (cardImageSource.isImageProvided(data.getSet(), data.getName())) { + if (data.isToken()) { + if (cardImageSource.isTokenSource() && cardImageSource.isImageProvided(data.getSet(), data.getName())) { numberTokenImagesAvailable++; cardsToDownload.add(data); } @@ -809,4 +865,25 @@ public class DownloadPictures extends DefaultBoundedRangeModel implements Runnab } private static final long serialVersionUID = 1L; + +} + +class LoadMissingCardData implements Runnable { + + private static DownloadPictures downloadPictures; + + public LoadMissingCardData(DownloadPictures downloadPictures) { + LoadMissingCardData.downloadPictures = downloadPictures; + } + + @Override + public void run() { + downloadPictures.setAllMissingCards(); + } + + public static void main() { + + (new Thread(new LoadMissingCardData(downloadPictures))).start(); + } + }