net.sourceforge.plantuml.swing.MainWindow Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of plantuml-lgpl Show documentation
Show all versions of plantuml-lgpl Show documentation
PlantUML is a component that allows to quickly write diagrams from text.
// THIS FILE HAS BEEN GENERATED BY A PREPROCESSOR.
/* +=======================================================================
* |
* | PlantUML : a free UML diagram generator
* |
* +=======================================================================
*
* (C) Copyright 2009-2024, Arnaud Roques
*
* Project Info: https://plantuml.com
*
* If you like this project or if you find it useful, you can support us at:
*
* https://plantuml.com/patreon (only 1$ per month!)
* https://plantuml.com/liberapay (only 1€ per month!)
* https://plantuml.com/paypal
*
*
* PlantUML is free software; you can redistribute it and/or modify it
* under the terms of the GNU Lesser General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* PlantUML distributed in the hope that it will be useful, but
* WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
* or FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public
* License for more details.
*
* You should have received a copy of the GNU Lesser General Public License
* along with this library. If not, see .
*
* PlantUML can occasionally display sponsored or advertising messages. Those
* messages are usually generated on welcome or error images and never on
* functional diagrams.
* See https://plantuml.com/professional if you want to remove them
*
* Images (whatever their format : PNG, SVG, EPS...) generated by running PlantUML
* are owned by the author of their corresponding sources code (that is, their
* textual description in PlantUML language). Those images are not covered by
* this LGPL license.
*
* The generated images can then be used without any reference to the LGPL license.
* It is not even necessary to stipulate that they have been generated with PlantUML,
* although this will be appreciated by the PlantUML team.
*
* There is an exception : if the textual description in PlantUML language is also covered
* by any license, then the generated images are logically covered
* by the very same license.
*
* This is the IGY distribution (Install GraphViz by Yourself).
* You have to install GraphViz and to setup the GRAPHVIZ_DOT environment variable
* (see https://plantuml.com/graphviz-dot )
*
* Icons provided by OpenIconic : https://useiconic.com/open
* Archimate sprites provided by Archi : http://www.archimatetool.com
* Stdlib AWS provided by https://github.com/milo-minderbinder/AWS-PlantUML
* Stdlib Icons provided https://github.com/tupadr3/plantuml-icon-font-sprites
* ASCIIMathML (c) Peter Jipsen http://www.chapman.edu/~jipsen
* ASCIIMathML (c) David Lippman http://www.pierce.ctc.edu/dlippman
* CafeUndZopfli ported by Eugene Klyuchnikov https://github.com/eustas/CafeUndZopfli
* Brotli (c) by the Brotli Authors https://github.com/google/brotli
* Themes (c) by Brett Schwarz https://github.com/bschwarz/puml-themes
* Twemoji (c) by Twitter at https://twemoji.twitter.com/
*
*/
package net.sourceforge.plantuml.swing;
import java.awt.BorderLayout;
import java.awt.Frame;
import java.awt.event.ActionEvent;
import java.awt.event.ActionListener;
import java.awt.event.FocusEvent;
import java.awt.event.FocusListener;
import java.awt.event.KeyAdapter;
import java.awt.event.KeyEvent;
import java.awt.event.MouseAdapter;
import java.awt.event.MouseEvent;
import java.awt.event.MouseListener;
import java.io.File;
import java.io.IOException;
import java.util.ArrayList;
import java.util.Collections;
import java.util.HashSet;
import java.util.Iterator;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.Vector;
import java.util.concurrent.ExecutionException;
import java.util.concurrent.Future;
import java.util.prefs.Preferences;
import java.util.regex.Matcher;
import java.util.regex.Pattern;
import javax.swing.BorderFactory;
import javax.swing.JButton;
import javax.swing.JFileChooser;
import javax.swing.JFrame;
import javax.swing.JLabel;
import javax.swing.JList;
import javax.swing.JMenu;
import javax.swing.JMenuBar;
import javax.swing.JMenuItem;
import javax.swing.JPanel;
import javax.swing.JScrollPane;
import javax.swing.JTextField;
import javax.swing.ListModel;
import javax.swing.SwingUtilities;
import javax.swing.Timer;
import javax.swing.border.CompoundBorder;
import net.sourceforge.plantuml.GeneratedImage;
import net.sourceforge.plantuml.Option;
import net.sourceforge.plantuml.log.Logme;
import net.sourceforge.plantuml.utils.Log;
import net.sourceforge.plantuml.version.PSystemVersion;
public class MainWindow extends JFrame {
final private static Preferences prefs = Preferences.userNodeForPackage(MainWindow.class);
final private static String KEY_DIR = "cur";
final private static String KEY_PATTERN = "pat";
private final JList jList1 = new JList();
private final JScrollPane scrollPane;
private final JButton changeDirButton = new JButton("Change Directory");
private final JTextField extensions = new JTextField();
private final int period = 300;
final private List currentDirectoryListing2 = new ArrayList<>();
final private Set openWindows2 = new HashSet<>();
final private Option option;
private DirWatcher2 dirWatcher;
private String getExtensions() {
return prefs.get(KEY_PATTERN, getDefaultFileExtensions());
}
private String getDefaultFileExtensions() {
return "txt, tex, java, htm, html, c, h, cpp, apt, pu, puml, hpp, hh, md";
}
private void changeExtensions(String ext) {
if (ext.equals(getExtensions())) {
return;
}
final Pattern p = Pattern.compile("\\w+");
final Matcher m = p.matcher(ext);
final StringBuilder sb = new StringBuilder();
while (m.find()) {
final String value = m.group();
if (sb.length() > 0) {
sb.append(", ");
}
sb.append(value);
}
ext = sb.toString();
if (ext.length() == 0) {
ext = getDefaultFileExtensions();
}
extensions.setText(ext);
prefs.put(KEY_PATTERN, ext);
changeDir(dirWatcher.getDir());
}
private String getRegexpPattern(String ext) {
final Pattern p = Pattern.compile("\\w+");
final Matcher m = p.matcher(ext);
final StringBuilder filePattern = new StringBuilder("(?i)^.*\\.(");
while (m.find()) {
final String value = m.group();
if (filePattern.toString().endsWith("(") == false) {
filePattern.append("|");
}
filePattern.append(value);
}
if (filePattern.toString().endsWith("(") == false) {
filePattern.append(")$");
return filePattern.toString();
}
return Option.getPattern();
}
public MainWindow(Option option, File arg) {
super(getDirectory(arg).getAbsolutePath());
System.setProperty("PLANTUML_SECURITY_PROFILE", "UNSECURE");
final File dir = getDirectory(arg);
setIconImage(PSystemVersion.getPlantumlSmallIcon2());
this.option = option;
dirWatcher = new DirWatcher2(dir, option, getRegexpPattern(getExtensions()));
Log.info("Showing MainWindow");
scrollPane = new JScrollPane(jList1);
scrollPane.setBorder(BorderFactory.createEmptyBorder(10, 10, 10, 10));
final JPanel south = new JPanel(new BorderLayout());
final JLabel labelFileExtensions = new JLabel("File extensions: ");
extensions.setText(getExtensions());
labelFileExtensions.setBorder(BorderFactory.createEmptyBorder(6, 6, 6, 6));
CompoundBorder border = BorderFactory.createCompoundBorder(BorderFactory.createEmptyBorder(10, 10, 10, 10),
BorderFactory.createEtchedBorder());
border = BorderFactory.createCompoundBorder(border, BorderFactory.createEmptyBorder(5, 5, 5, 5));
south.setBorder(border);
south.add(labelFileExtensions, BorderLayout.WEST);
south.add(extensions, BorderLayout.CENTER);
south.add(changeDirButton, BorderLayout.SOUTH);
getContentPane().add(south, BorderLayout.SOUTH);
getContentPane().add(scrollPane, BorderLayout.CENTER);
final MouseListener mouseListener = new MouseAdapter() {
@Override
public void mouseClicked(MouseEvent e) {
try {
if (e.getClickCount() == 2) {
final int index = jList1.locationToIndex(e.getPoint());
doubleClick((SimpleLine) jList1.getModel().getElementAt(index), jList1.getModel(), index);
}
} catch (Exception ex) {
}
}
};
jList1.addMouseListener(mouseListener);
changeDirButton.addActionListener(new ActionListener() {
public void actionPerformed(ActionEvent e) {
System.err.println("Opening Directory Window");
displayDialogChangeDir();
}
});
jList1.addKeyListener(new KeyAdapter() {
@Override
public void keyPressed(KeyEvent e) {
if (e.getKeyCode() == KeyEvent.VK_ENTER) {
final int index = jList1.getSelectedIndex();
doubleClick((SimpleLine) jList1.getModel().getElementAt(index), jList1.getModel(), index);
}
}
});
extensions.addActionListener(new ActionListener() {
public void actionPerformed(ActionEvent e) {
changeExtensions(extensions.getText());
}
});
extensions.addFocusListener(new FocusListener() {
public void focusGained(FocusEvent e) {
}
public void focusLost(FocusEvent e) {
changeExtensions(extensions.getText());
}
});
final JMenuBar menuBar = new JMenuBar();
final JMenu mFile = new JMenu("File");
menuBar.add(mFile);
setJMenuBar(menuBar);
final JMenuItem sprite = new JMenuItem("Open Sprite Window");
mFile.add(sprite);
sprite.addActionListener(new ActionListener() {
public void actionPerformed(ActionEvent e) {
new SpriteWindow();
}
});
final JMenuItem about = new JMenuItem("About");
mFile.add(about);
about.addActionListener(new ActionListener() {
public void actionPerformed(ActionEvent e) {
new AboutWindow();
}
});
final JMenuItem exit = new JMenuItem("Exit");
mFile.add(exit);
exit.addActionListener(new ActionListener() {
public void actionPerformed(ActionEvent e) {
System.exit(0);
}
});
setSize(640, 400);
this.setLocationRelativeTo(this.getParent());
setVisible(true);
setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);
startTimer();
}
private static File getDirectory(File arg) {
if (arg != null && arg.exists() && arg.isDirectory()) {
return arg;
}
return new File(prefs.get(KEY_DIR, "."));
}
private void startTimer() {
Log.info("Init done");
final Timer timer = new Timer(period, new ActionListener() {
public void actionPerformed(ActionEvent e) {
tick();
}
});
timer.setInitialDelay(0);
timer.start();
Log.info("Timer started");
}
private void displayDialogChangeDir() {
final JFileChooser chooser = new JFileChooser();
chooser.setDialogType(JFileChooser.CUSTOM_DIALOG);
chooser.setDialogTitle("Directory to watch:");
chooser.setFileSelectionMode(JFileChooser.DIRECTORIES_ONLY);
chooser.setAcceptAllFileFilterUsed(false);
final String currentPath = prefs.get(KEY_DIR, ".");
chooser.setCurrentDirectory(new File(currentPath));
Log.info("Showing OpenDialog");
final int returnVal = chooser.showOpenDialog(this);
Log.info("Closing OpenDialog");
if (returnVal == JFileChooser.APPROVE_OPTION) {
final File dir = chooser.getSelectedFile();
changeDir(dir);
}
}
private void changeDir(File dir) {
prefs.put(KEY_DIR, dir.getAbsolutePath());
dirWatcher.cancel();
dirWatcher = new DirWatcher2(dir, option, getRegexpPattern(getExtensions()));
setTitle(dir.getAbsolutePath());
Log.info("Creating DirWatcher");
currentDirectoryListing2.clear();
jList1.setListData(new Vector<>(currentDirectoryListing2));
jList1.setVisible(true);
}
private void doubleClick(SimpleLine simpleLine, ListModel listModel, int index) {
for (ImageWindow win : openWindows2) {
if (win.getSimpleLine().equals(simpleLine)) {
win.setVisible(true);
win.setExtendedState(Frame.NORMAL);
return;
}
}
if (simpleLine.getGeneratedImage() != null) {
openWindows2.add(new ImageWindow(simpleLine, this, listModel, index));
}
}
private void tick() {
SwingUtilities.invokeLater(new Runnable() {
public void run() {
try {
final boolean changed = refreshDir();
if (changed) {
jList1.setListData(new Vector<>(currentDirectoryListing2));
jList1.setVisible(true);
}
} catch (IOException e) {
Logme.error(e);
} catch (InterruptedException e) {
Logme.error(e);
} catch (ExecutionException e) {
Logme.error(e);
}
}
});
}
private boolean refreshDir() throws IOException, InterruptedException, ExecutionException {
final Map>> createdFiles2 = dirWatcher.buildCreatedFiles();
boolean changed = false;
for (Map.Entry>> ent : createdFiles2.entrySet()) {
final File file = ent.getKey();
removeAllThatUseThisFile(file);
final Future> future = ent.getValue();
final SimpleLine simpleLine = SimpleLine.fromFuture(file, future);
currentDirectoryListing2.add(simpleLine);
changed = true;
}
for (SimpleLine line : new ArrayList<>(currentDirectoryListing2)) {
if (line.pendingAndFinished()) {
currentDirectoryListing2.remove(line);
changed = true;
final Future> future = line.getFuture();
for (GeneratedImage im : future.get()) {
mayRefreshImageWindow(im.getPngFile());
final SimpleLine simpleLine = SimpleLine.fromGeneratedImage(line.getFile(), im);
currentDirectoryListing2.add(simpleLine);
}
}
}
Collections.sort(currentDirectoryListing2);
return changed;
}
private void removeAllThatUseThisFile(File file) {
for (final Iterator it = currentDirectoryListing2.iterator(); it.hasNext();) {
final SimpleLine line = it.next();
if (line.getFile().equals(file)) {
it.remove();
}
}
}
private void mayRefreshImageWindow(File pngFile) {
for (ImageWindow win : openWindows2) {
if (win.getSimpleLine().getGeneratedImage() == null) {
continue;
}
if (pngFile.equals(win.getSimpleLine().getGeneratedImage().getPngFile())) {
win.refreshImage(true);
}
}
}
public void closing(ImageWindow imageWindow) {
final boolean ok = openWindows2.remove(imageWindow);
if (ok == false) {
throw new IllegalStateException();
}
}
public List getCurrentDirectoryListing2() {
return Collections.unmodifiableList(currentDirectoryListing2);
}
}