source: josm/trunk/src/org/openstreetmap/josm/actions/SessionSaveAsAction.java@ 8781

Last change on this file since 8781 was 8510, checked in by Don-vip, 9 years ago

checkstyle: enable relevant whitespace checks and fix them
  • Property svn:eol-style set to native
File size: 9.0 KB
Line 
1// License: GPL. For details, see LICENSE file.
2package org.openstreetmap.josm.actions;
3
4import static org.openstreetmap.josm.gui.help.HelpUtil.ht;
5import static org.openstreetmap.josm.tools.I18n.tr;
6
7import java.awt.Component;
8import java.awt.Dimension;
9import java.awt.GridBagLayout;
10import java.awt.event.ActionEvent;
11import java.io.File;
12import java.io.IOException;
13import java.util.ArrayList;
14import java.util.Arrays;
15import java.util.Collection;
16import java.util.HashMap;
17import java.util.HashSet;
18import java.util.List;
19import java.util.Map;
20import java.util.Set;
21
22import javax.swing.BorderFactory;
23import javax.swing.JCheckBox;
24import javax.swing.JFileChooser;
25import javax.swing.JLabel;
26import javax.swing.JOptionPane;
27import javax.swing.JPanel;
28import javax.swing.JScrollPane;
29import javax.swing.JTabbedPane;
30import javax.swing.SwingConstants;
31import javax.swing.border.EtchedBorder;
32import javax.swing.filechooser.FileFilter;
33
34import org.openstreetmap.josm.Main;
35import org.openstreetmap.josm.gui.ExtendedDialog;
36import org.openstreetmap.josm.gui.HelpAwareOptionPane;
37import org.openstreetmap.josm.gui.layer.Layer;
38import org.openstreetmap.josm.gui.widgets.AbstractFileChooser;
39import org.openstreetmap.josm.io.session.SessionLayerExporter;
40import org.openstreetmap.josm.io.session.SessionWriter;
41import org.openstreetmap.josm.tools.GBC;
42import org.openstreetmap.josm.tools.MultiMap;
43import org.openstreetmap.josm.tools.Utils;
44import org.openstreetmap.josm.tools.WindowGeometry;
45
46/**
47 * Saves a JOSM session
48 * @since 4685
49 */
50public class SessionSaveAsAction extends DiskAccessAction {
51
52 private transient List<Layer> layers;
53 private transient Map<Layer, SessionLayerExporter> exporters;
54 private transient MultiMap<Layer, Layer> dependencies;
55
56 /**
57 * Constructs a new {@code SessionSaveAsAction}.
58 */
59 public SessionSaveAsAction() {
60 super(tr("Save Session As..."), "save_as", tr("Save the current session to a new file."), null, true, "save_as-session", true);
61 putValue("help", ht("/Action/SessionSaveAs"));
62 }
63
64 @Override
65 public void actionPerformed(ActionEvent e) {
66 if (!isEnabled()) {
67 return;
68 }
69
70 SessionSaveAsDialog dlg = new SessionSaveAsDialog();
71 dlg.showDialog();
72 if (dlg.getValue() != 1) return;
73
74 boolean zipRequired = false;
75 for (Layer l : layers) {
76 SessionLayerExporter ex = exporters.get(l);
77 if (ex != null && ex.requiresZip()) {
78 zipRequired = true;
79 break;
80 }
81 }
82
83 FileFilter joz = new ExtensionFileFilter("joz", "joz", tr("Session file (archive) (*.joz)"));
84 FileFilter jos = new ExtensionFileFilter("jos", "jos", tr("Session file (*.jos)"));
85
86 AbstractFileChooser fc;
87
88 if (zipRequired) {
89 fc = createAndOpenFileChooser(false, false, tr("Save session"), joz, JFileChooser.FILES_ONLY, "lastDirectory");
90 } else {
91 fc = createAndOpenFileChooser(false, false, tr("Save session"), Arrays.asList(new FileFilter[]{jos, joz}), jos,
92 JFileChooser.FILES_ONLY, "lastDirectory");
93 }
94
95 if (fc == null)
96 return;
97
98 File file = fc.getSelectedFile();
99 String fn = file.getName();
100
101 boolean zip;
102 FileFilter ff = fc.getFileFilter();
103 if (zipRequired) {
104 zip = true;
105 } else if (joz.equals(ff)) {
106 zip = true;
107 } else if (jos.equals(ff)) {
108 zip = false;
109 } else {
110 if (Utils.hasExtension(fn, "joz")) {
111 zip = true;
112 } else {
113 zip = false;
114 }
115 }
116 if (fn.indexOf('.') == -1) {
117 file = new File(file.getPath() + (zip ? ".joz" : ".jos"));
118 if (!SaveActionBase.confirmOverwrite(file))
119 return;
120 }
121
122 List<Layer> layersOut = new ArrayList<>();
123 for (Layer layer : layers) {
124 if (exporters.get(layer) == null || !exporters.get(layer).shallExport()) continue;
125 // TODO: resolve dependencies for layers excluded by the user
126 layersOut.add(layer);
127 }
128
129 int active = -1;
130 Layer activeLayer = Main.map.mapView.getActiveLayer();
131 if (activeLayer != null) {
132 active = layersOut.indexOf(activeLayer);
133 }
134
135 SessionWriter sw = new SessionWriter(layersOut, active, exporters, dependencies, zip);
136 try {
137 sw.write(file);
138 } catch (IOException ex) {
139 Main.error(ex);
140 HelpAwareOptionPane.showMessageDialogInEDT(
141 Main.parent,
142 tr("<html>Could not save session file ''{0}''.<br>Error is:<br>{1}</html>", file.getName(), ex.getMessage()),
143 tr("IO Error"),
144 JOptionPane.ERROR_MESSAGE,
145 null
146 );
147 }
148 }
149
150 /**
151 * The "Save Session" dialog
152 */
153 public class SessionSaveAsDialog extends ExtendedDialog {
154
155 /**
156 * Constructs a new {@code SessionSaveAsDialog}.
157 */
158 public SessionSaveAsDialog() {
159 super(Main.parent, tr("Save Session"), new String[] {tr("Save As"), tr("Cancel")});
160 initialize();
161 setButtonIcons(new String[] {"save_as", "cancel"});
162 setDefaultButton(1);
163 setRememberWindowGeometry(getClass().getName() + ".geometry",
164 WindowGeometry.centerInWindow(Main.parent, new Dimension(350, 450)));
165 setContent(build(), false);
166 }
167
168 /**
169 * Initializes action.
170 */
171 public final void initialize() {
172 layers = new ArrayList<>(Main.map.mapView.getAllLayersAsList());
173 exporters = new HashMap<>();
174 dependencies = new MultiMap<>();
175
176 Set<Layer> noExporter = new HashSet<>();
177
178 for (Layer layer : layers) {
179 SessionLayerExporter exporter = SessionWriter.getSessionLayerExporter(layer);
180 if (exporter != null) {
181 exporters.put(layer, exporter);
182 Collection<Layer> deps = exporter.getDependencies();
183 if (deps != null) {
184 dependencies.putAll(layer, deps);
185 } else {
186 dependencies.putVoid(layer);
187 }
188 } else {
189 noExporter.add(layer);
190 exporters.put(layer, null);
191 }
192 }
193
194 int numNoExporter = 0;
195 WHILE: while (numNoExporter != noExporter.size()) {
196 numNoExporter = noExporter.size();
197 for (Layer layer : layers) {
198 if (noExporter.contains(layer)) continue;
199 for (Layer depLayer : dependencies.get(layer)) {
200 if (noExporter.contains(depLayer)) {
201 noExporter.add(layer);
202 exporters.put(layer, null);
203 break WHILE;
204 }
205 }
206 }
207 }
208 }
209
210 protected final Component build() {
211 JPanel p = new JPanel(new GridBagLayout());
212 JPanel ip = new JPanel(new GridBagLayout());
213 for (Layer layer : layers) {
214 JPanel wrapper = new JPanel(new GridBagLayout());
215 wrapper.setBorder(BorderFactory.createEtchedBorder(EtchedBorder.RAISED));
216 Component exportPanel;
217 SessionLayerExporter exporter = exporters.get(layer);
218 if (exporter == null) {
219 if (!exporters.containsKey(layer)) throw new AssertionError();
220 exportPanel = getDisabledExportPanel(layer);
221 } else {
222 exportPanel = exporter.getExportPanel();
223 }
224 wrapper.add(exportPanel, GBC.std().fill(GBC.HORIZONTAL));
225 ip.add(wrapper, GBC.eol().fill(GBC.HORIZONTAL).insets(2, 2, 4, 2));
226 }
227 ip.add(GBC.glue(0, 1), GBC.eol().fill(GBC.VERTICAL));
228 JScrollPane sp = new JScrollPane(ip);
229 sp.setBorder(BorderFactory.createEmptyBorder());
230 p.add(sp, GBC.eol().fill());
231 final JTabbedPane tabs = new JTabbedPane();
232 tabs.addTab(tr("Layers"), p);
233 return tabs;
234 }
235
236 protected final Component getDisabledExportPanel(Layer layer) {
237 JPanel p = new JPanel(new GridBagLayout());
238 JCheckBox include = new JCheckBox();
239 include.setEnabled(false);
240 JLabel lbl = new JLabel(layer.getName(), layer.getIcon(), SwingConstants.LEFT);
241 lbl.setToolTipText(tr("No exporter for this layer"));
242 lbl.setLabelFor(include);
243 lbl.setEnabled(false);
244 p.add(include, GBC.std());
245 p.add(lbl, GBC.std());
246 p.add(GBC.glue(1, 0), GBC.std().fill(GBC.HORIZONTAL));
247 return p;
248 }
249 }
250
251 @Override
252 protected void updateEnabledState() {
253 setEnabled(Main.isDisplayingMapView());
254 }
255}
Note: See TracBrowser for help on using the repository browser.