package org.autoplot.datasource; import java.awt.Dimension; import java.awt.Window; import java.awt.event.ActionEvent; import java.awt.event.ActionListener; import java.awt.event.KeyEvent; import java.io.File; import java.util.ArrayList; import java.util.Collections; import java.util.HashMap; import java.util.HashSet; import java.util.Map; import java.util.logging.Logger; import java.util.prefs.Preferences; import javax.swing.AbstractAction; import javax.swing.Action; import javax.swing.Box; import javax.swing.BoxLayout; import javax.swing.ButtonGroup; import javax.swing.JCheckBox; import javax.swing.JFileChooser; import javax.swing.JMenu; import javax.swing.JOptionPane; import javax.swing.JPanel; import javax.swing.SwingUtilities; import javax.swing.filechooser.FileFilter; import javax.swing.filechooser.FileNameExtensionFilter; import org.das2.datum.DatumRange; /** * Additional actions for the DataSetSelector. * @author jbf */ public class DataSetSelectorSupport { private static final Logger logger= Logger.getLogger("apdss.dss"); DataSetSelector ui; private JMenu recentMenu = null; DataSetSelectorSupport(DataSetSelector ui) { this.ui = ui; } private static File userHome() { return new File(System.getProperty("user.home")); } /** * Show a file chooser component, and return the name of a .vap file. * @param parent parent component for focus. If a dataSetSelector is * used then its timerange is used for the initial timerange. * This will now allow a non-local vap to be browsed as well. * @param initialSelection if non-null, then the initial selection. * @return the URI for the vap file, or null if cancel was pressed. */ public static String browseLocalVap( java.awt.Component parent, String initialSelection) { Preferences prefs = AutoplotSettings.getPreferences( AutoplotSettings.class); String currentDirectory = prefs.get( AutoplotSettings.PREF_LAST_OPEN_VAP_FOLDER, prefs.get(AutoplotSettings.PREF_LAST_OPEN_FOLDER, userHome().toString() ) ); String currentFile= prefs.get( AutoplotSettings.PREF_LAST_OPEN_VAP_FILE, "" ); boolean isRemote= initialSelection!=null && ( initialSelection.startsWith("https:/") || initialSelection.startsWith("http:/") || initialSelection.startsWith("ftp:/") || initialSelection.startsWith("sftp:/") ); boolean isLocal= initialSelection==null || initialSelection.isEmpty() || !isRemote; JFileChooser chooser; chooser= new JFileChooser(); try { chooser.setCurrentDirectory( new File( currentDirectory ) ); } catch ( SecurityException ex ) { logger.info("unable to set current directory"); } if ( currentFile.length()>0 ) { try { chooser.setSelectedFile( new File( currentFile ) ); } catch ( SecurityException ex ) { logger.info("unable to set current file"); } } if ( initialSelection!=null && isLocal ) { URISplit split= URISplit.parse(initialSelection); if ( split.file!=null && "vap".equals(split.ext) ) { chooser.setSelectedFile( new File( split.file )); } } JPanel trPanel= new JPanel(); trPanel.setLayout( new BoxLayout(trPanel, BoxLayout.Y_AXIS )); trPanel.setMaximumSize( new Dimension(230,9999) ); trPanel.setPreferredSize( new Dimension(230,200) ); ButtonGroup bg= new ButtonGroup(); JCheckBox b1= new JCheckBox("Use timerange in .vap file"); //b1.setAlignmentX(Component.LEFT_ALIGNMENT); bg.add(b1); final TimeRangeEditor t= new TimeRangeEditor(); if ( parent!=null && parent instanceof DataSetSelector ) { DatumRange tr= ((DataSetSelector)parent).getTimeRange() ; if ( tr!=null ) t.setRange(tr); } t.makeThinner(); final JCheckBox b2= new JCheckBox("Reset the .vap timerange:"); bg.add(b2); //b2.setAlignmentX(Component.LEFT_ALIGNMENT); boolean notimerange= ( initialSelection==null || !initialSelection.contains("?timerange=") ); b1.setSelected( notimerange ); b2.setSelected( !notimerange ); b1.setAlignmentX( 0.f ); b2.setAlignmentX( 0.f ); t.setAlignmentX( 0.f ); trPanel.add(b1); trPanel.add( Box.createVerticalStrut(14) ); trPanel.add(b2); trPanel.add(t); ActionListener enableTR= new ActionListener() { @Override public void actionPerformed(ActionEvent e) { t.setEnabled( b2.isSelected() ); } }; b1.addActionListener(enableTR); b2.addActionListener(enableTR); t.setEnabled(b2.isSelected()); trPanel.add( Box.createVerticalGlue() ); if ( chooser!=null ) { chooser.setAccessory(trPanel); FileFilter ff; ff = new FileNameExtensionFilter( ".vap files", "vap" ); chooser.addChoosableFileFilter(ff); chooser.setFileFilter(ff); Window w= parent==null ? null : SwingUtilities.getWindowAncestor( parent); int result = chooser.showOpenDialog(w); if (result == JFileChooser.APPROVE_OPTION) { prefs.put(AutoplotSettings.PREF_LAST_OPEN_VAP_FOLDER, chooser.getSelectedFile().getParent() ); if ( b2.isSelected() ) { return chooser.getSelectedFile().toURI().toString() + "?timerange="+ t.getRange().toString().replaceAll("\\s+", "+"); } else { return chooser.getSelectedFile().toURI().toString(); } } else { return null; } } else { if ( JOptionPane.showConfirmDialog( parent, trPanel, "New Time Range", JOptionPane.OK_CANCEL_OPTION )==JOptionPane.OK_OPTION ) { URISplit split= URISplit.parse( initialSelection ); Map params= URISplit.parseParams(split.params); if ( b2.isSelected() ) { params.put( "timerange", t.getRange().toString().replaceAll("\\s","+") ); } else { params.remove("timerange"); } if ( params.isEmpty() ) { split.params= null; } else { split.params= URISplit.formatParams(params); } return URISplit.format(split); } else { return null; } } } /** * Show a file chooser component, and return the name of a data or .vap file. * @param parent parent component for focus. * @return the URI for the vap file. */ public static String browseLocal( java.awt.Component parent ) { Preferences prefs = AutoplotSettings.settings().getPreferences(DataSetSelectorSupport.class); String currentDirectory = prefs.get(AutoplotSettings.PREF_LAST_OPEN_FOLDER, userHome().toString()); final HashMap exts = DataSourceRegistry.getInstance().dataSourcesByExt; JFileChooser chooser = new JFileChooser(); try { chooser.setCurrentDirectory( new File(currentDirectory) ); } catch ( SecurityException ex ) { logger.info("unable to set current directory because of security"); } final boolean isAutoplotApp; if ( parent instanceof DataSetSelector ) { isAutoplotApp= ((DataSetSelector)parent).actionTriggers.containsKey("vapfile:(.*)"); //TODO: kludgy } else { isAutoplotApp= true; } FileFilter ff; ff = new FileFilter() { @Override public boolean accept(File f) { if (f.isDirectory()) { return true; } String t = f.getName(); String ext = DataSetURI.getExt(t); if ( ext!=null ) ext= "."+ext; return ( t.endsWith(".zip") || t.endsWith(".ZIP") ) || ( t.endsWith(".tar") || t.endsWith(".tgz") || t.endsWith(".tar.gz") ) || (ext != null && ( exts.containsKey(ext) ) || ( isAutoplotApp && t.endsWith(".vap") )); } @Override public String getDescription() { return "supported formats"; } }; chooser.addChoosableFileFilter(ff); FileFilter select = ff; ArrayList s= new ArrayList( exts.keySet() ); Collections.sort(s); HashSet skip= new HashSet<>(); skip.add(".cdfj"); skip.add(".cdfn"); skip.add(".csv0"); for (String ext1 : s ) { if ( skip.contains(ext1) ) continue; DataSourceFactory factory= DataSourceRegistry.getInstance().getSource(ext1); if ( factory.isFileResource() ) { final String extf = (String) ext1; String desc= DataSourceRegistry.getInstance().describe( factory, extf ); if ( desc.length()==0 || desc.startsWith(".") ) { desc= "*"+ext1; } else { desc= "*"+ext1+" " +desc; } ff = new FileNameExtensionFilter( desc, extf.substring(1) ); chooser.addChoosableFileFilter(ff); } } if ( isAutoplotApp ) { ff = new FileNameExtensionFilter( "*.vap", "vap"); chooser.addChoosableFileFilter(ff); } chooser.setFileFilter(select); int result = chooser.showOpenDialog(parent); if (result == JFileChooser.APPROVE_OPTION) { prefs.put(AutoplotSettings.PREF_LAST_OPEN_FOLDER, chooser.getSelectedFile().getParent() ); return chooser.getSelectedFile().toURI().toString(); } else { return null; } } /** * get "Add Plot from Local File..." action * @return the action */ public Action openLocalAction() { return new AbstractAction("Add Plot from Local File...") { @Override public void actionPerformed(ActionEvent e) { org.das2.util.LoggerManager.logGuiEvent(e); String result= browseLocal(ui); if (result != null ) { ui.setValue(result); ui.maybePlot( KeyEvent.ALT_MASK ); //ui.maybePlot(false); } } }; } /** * get the "Open .vap File..." action * @return the action */ public Action openLocalVapAction() { return new AbstractAction("Open .vap File...") { @Override public void actionPerformed(ActionEvent e) { org.das2.util.LoggerManager.logGuiEvent(e); String result= browseLocalVap(ui, ui.getValue() ); if (result != null ) { ui.setValue(result); ui.maybePlot(false); } } }; } protected void refreshRecentFilesMenu() { if (recentMenu != null) { recentMenu.removeAll(); ArrayList recent = new ArrayList<>(ui.getRecent()); Collections.reverse(recent); for (String s : recent) { final String f = s; Action a = new AbstractAction(String.valueOf(f)) { @Override public void actionPerformed(ActionEvent e) { org.das2.util.LoggerManager.logGuiEvent(e); ui.setValue(f); ui.maybePlot(false); } }; recentMenu.add(a); } } } public static String getPluginsText() { return DataSourceRegistry.getPluginsText(); } JMenu recentMenu() { if (recentMenu == null) { recentMenu = new JMenu("Open Recent"); refreshRecentFilesMenu(); } return recentMenu; } protected void browseSourceTypes() { } }