summaryrefslogtreecommitdiff
path: root/src/main/java/cuchaz/enigma/gui/ProgressDialog.java
blob: 70bca15bdaa9462c82baaa83820df7abcefd0140 (plain) (blame)
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
/*******************************************************************************
 * Copyright (c) 2015 Jeff Martin.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the GNU Lesser General Public
 * License v3.0 which accompanies this distribution, and is available at
 * http://www.gnu.org/licenses/lgpl.html
 * <p>
 * Contributors:
 * Jeff Martin - initial API and implementation
 ******************************************************************************/
package cuchaz.enigma.gui;

import java.awt.BorderLayout;
import java.awt.Container;
import java.awt.Dimension;
import java.awt.FlowLayout;

import javax.swing.*;

import cuchaz.enigma.Constants;
import cuchaz.enigma.Deobfuscator.ProgressListener;

public class ProgressDialog implements ProgressListener, AutoCloseable {

    private JFrame frame;
    private JLabel labelTitle;
    private JLabel labelText;
    private JProgressBar progress;

    public ProgressDialog(JFrame parent) {

        // init frame
        this.frame = new JFrame(Constants.NAME + " - Operation in progress");
        final Container pane = this.frame.getContentPane();
        FlowLayout layout = new FlowLayout();
        layout.setAlignment(FlowLayout.LEFT);
        pane.setLayout(layout);

        this.labelTitle = new JLabel();
        pane.add(this.labelTitle);

        // set up the progress bar
        JPanel panel = new JPanel();
        pane.add(panel);
        panel.setLayout(new BorderLayout());
        this.labelText = GuiTricks.unboldLabel(new JLabel());
        this.progress = new JProgressBar();
        this.labelText.setBorder(BorderFactory.createEmptyBorder(0, 0, 10, 0));
        panel.add(this.labelText, BorderLayout.NORTH);
        panel.add(this.progress, BorderLayout.CENTER);
        panel.setPreferredSize(new Dimension(360, 50));

        // show the frame
        pane.doLayout();
        this.frame.setSize(400, 120);
        this.frame.setResizable(false);
        this.frame.setLocationRelativeTo(parent);
        this.frame.setVisible(true);
        this.frame.setDefaultCloseOperation(WindowConstants.DO_NOTHING_ON_CLOSE);
    }

    public void close() {
        this.frame.dispose();
    }

    @Override
    public void init(int totalWork, String title) {
        this.labelTitle.setText(title);
        this.progress.setMinimum(0);
        this.progress.setMaximum(totalWork);
        this.progress.setValue(0);
    }

    @Override
    public void onProgress(int numDone, String message) {
        this.labelText.setText(message);
        this.progress.setValue(numDone);

        // update the frame
        this.frame.validate();
        this.frame.repaint();
    }

    public interface ProgressRunnable {
        void run(ProgressListener listener) throws Exception;
    }

    public static void runInThread(final JFrame parent, final ProgressRunnable runnable) {
        new Thread() {
            @Override
            public void run() {
                try (ProgressDialog progress = new ProgressDialog(parent)) {
                    runnable.run(progress);
                } catch (Exception ex) {
                    throw new Error(ex);
                }
            }
        }.start();
    }
}