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

Last change on this file since 6252 was 6245, checked in by Don-vip, 11 years ago

see #9032, see #4029 - Allow to load session from standard I/O mechanisms (local and remote files) + javadoc
  • Property svn:eol-style set to native
File size: 8.6 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.io.session.SessionLayerExporter;
39import org.openstreetmap.josm.io.session.SessionWriter;
40import org.openstreetmap.josm.tools.GBC;
41import org.openstreetmap.josm.tools.MultiMap;
42import org.openstreetmap.josm.tools.WindowGeometry;
43
44/**
45 * Saves a JOSM session
46 * @since 4685
47 */
48public class SessionSaveAsAction extends DiskAccessAction {
49
50 private List<Layer> layers;
51 private Map<Layer, SessionLayerExporter> exporters;
52 private MultiMap<Layer, Layer> dependencies;
53
54 private boolean zipRequired;
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 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 JFileChooser 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, JFileChooser.FILES_ONLY, "lastDirectory");
92 }
93
94 if (fc == null)
95 return;
96
97 File file = fc.getSelectedFile();
98 String fn = file.getName();
99
100 boolean zip;
101 FileFilter ff = fc.getFileFilter();
102 if (zipRequired) {
103 zip = true;
104 } else if (ff == joz) {
105 zip = true;
106 } else if (ff == jos) {
107 zip = false;
108 } else {
109 if (fn.toLowerCase().endsWith(".joz")) {
110 zip = true;
111 } else {
112 zip = false;
113 }
114 }
115 if (fn.indexOf('.') == -1) {
116 file = new File(file.getPath() + (zip ? ".joz" : ".jos"));
117 if (!SaveActionBase.confirmOverwrite(file))
118 return;
119 }
120
121 List<Layer> layersOut = new ArrayList<Layer>();
122 for (Layer layer : layers) {
123 if (exporters.get(layer) == null || !exporters.get(layer).shallExport()) continue;
124 // TODO: resolve dependencies for layers excluded by the user
125 layersOut.add(layer);
126 }
127
128 SessionWriter sw = new SessionWriter(layersOut, exporters, dependencies, zip);
129 try {
130 sw.write(file);
131 } catch (IOException ex) {
132 ex.printStackTrace();
133 HelpAwareOptionPane.showMessageDialogInEDT(
134 Main.parent,
135 tr("<html>Could not save session file ''{0}''.<br>Error is:<br>{1}</html>", file.getName(), ex.getMessage()),
136 tr("IO Error"),
137 JOptionPane.ERROR_MESSAGE,
138 null
139 );
140 }
141 }
142
143 /**
144 * The "Save Session" dialog
145 */
146 public class SessionSaveAsDialog extends ExtendedDialog {
147
148 /**
149 * Constructs a new {@code SessionSaveAsDialog}.
150 */
151 public SessionSaveAsDialog() {
152 super(Main.parent, tr("Save Session"), new String[] {tr("Save As"), tr("Cancel")});
153 initialize();
154 setButtonIcons(new String[] {"save_as", "cancel"});
155 setDefaultButton(1);
156 setRememberWindowGeometry(getClass().getName() + ".geometry",
157 WindowGeometry.centerInWindow(Main.parent, new Dimension(350, 450)));
158 setContent(build(), false);
159 }
160
161 public void initialize() {
162 layers = new ArrayList<Layer>(Main.map.mapView.getAllLayersAsList());
163 exporters = new HashMap<Layer, SessionLayerExporter>();
164 dependencies = new MultiMap<Layer, Layer>();
165
166 Set<Layer> noExporter = new HashSet<Layer>();
167
168 for (Layer layer : layers) {
169 SessionLayerExporter exporter = SessionWriter.getSessionLayerExporter(layer);
170 if (exporter != null) {
171 exporters.put(layer, exporter);
172 Collection<Layer> deps = exporter.getDependencies();
173 if (deps != null) {
174 dependencies.putAll(layer, deps);
175 } else {
176 dependencies.putVoid(layer);
177 }
178 } else {
179 noExporter.add(layer);
180 exporters.put(layer, null);
181 }
182 }
183
184 int numNoExporter = 0;
185 WHILE:while (numNoExporter != noExporter.size()) {
186 numNoExporter = noExporter.size();
187 for (Layer layer : layers) {
188 if (noExporter.contains(layer)) continue;
189 for (Layer depLayer : dependencies.get(layer)) {
190 if (noExporter.contains(depLayer)) {
191 noExporter.add(layer);
192 exporters.put(layer, null);
193 break WHILE;
194 }
195 }
196 }
197 }
198 }
199
200 public Component build() {
201 JPanel p = new JPanel(new GridBagLayout());
202 JPanel ip = new JPanel(new GridBagLayout());
203 for (Layer layer : layers) {
204 JPanel wrapper = new JPanel(new GridBagLayout());
205 wrapper.setBorder(BorderFactory.createEtchedBorder(EtchedBorder.RAISED));
206 Component exportPanel;
207 SessionLayerExporter exporter = exporters.get(layer);
208 if (exporter == null) {
209 if (!exporters.containsKey(layer)) throw new AssertionError();
210 exportPanel = getDisabledExportPanel(layer);
211 } else {
212 exportPanel = exporter.getExportPanel();
213 }
214 wrapper.add(exportPanel, GBC.std().fill(GBC.HORIZONTAL));
215 ip.add(wrapper, GBC.eol().fill(GBC.HORIZONTAL).insets(2,2,4,2));
216 }
217 ip.add(GBC.glue(0,1), GBC.eol().fill(GBC.VERTICAL));
218 JScrollPane sp = new JScrollPane(ip);
219 sp.setBorder(BorderFactory.createEmptyBorder());
220 p.add(sp, GBC.eol().fill());
221 final JTabbedPane tabs = new JTabbedPane();
222 tabs.addTab(tr("Layers"), p);
223 return tabs;
224 }
225
226 protected Component getDisabledExportPanel(Layer layer) {
227 JPanel p = new JPanel(new GridBagLayout());
228 JCheckBox include = new JCheckBox();
229 include.setEnabled(false);
230 JLabel lbl = new JLabel(layer.getName(), layer.getIcon(), SwingConstants.LEFT);
231 lbl.setToolTipText(tr("No exporter for this layer"));
232 lbl.setEnabled(false);
233 p.add(include, GBC.std());
234 p.add(lbl, GBC.std());
235 p.add(GBC.glue(1,0), GBC.std().fill(GBC.HORIZONTAL));
236 return p;
237 }
238 }
239
240 @Override
241 protected void updateEnabledState() {
242 setEnabled(Main.isDisplayingMapView());
243 }
244}
Note: See TracBrowser for help on using the repository browser.