package org.autoplot.idlsupport; import java.io.File; import java.io.FileOutputStream; import java.io.RandomAccessFile; import java.nio.ByteBuffer; import java.nio.channels.FileChannel; import java.util.Arrays; import java.util.Collections; import java.util.HashSet; import java.util.Set; import java.util.regex.Matcher; import java.util.regex.Pattern; import org.das2.datum.Units; import org.das2.datum.UnitsConverter; import org.das2.datum.UnitsUtil; import org.das2.util.monitor.ProgressMonitor; import org.das2.qds.QDataSet; import org.das2.qds.SemanticOps; import org.autoplot.datasource.AbstractDataSourceFormat; import org.das2.qds.DataSetUtil; import org.das2.qds.ops.Ops; /** * Export to idlsav support. rank 0, rank 1 datasets, rank 2 datasets, rank 3 datasets, and rank 2 bundles are supported. * @author jbf */ public class IdlsavDataSourceFormat extends AbstractDataSourceFormat { /** * Add the data to the container which will be written to an IDLSave file. * @param write the container * @param data the data * @param guessName the name to use, if name is not found within the data. */ private void doOne( WriteIDLSav write, QDataSet data, String guessName ) { String su= getParam( "tunits", "t1970" ); QDataSet wds= Ops.valid(data); Object odd; if ( data.rank()==0 ) { odd= data.value(); } else if ( data.rank()==1 ) { double[] dd= new double[data.length()]; for ( int i=0; i nnames; if ( names==null ) { nnames= Collections.emptySet(); } else { nnames= new HashSet<>(Arrays.asList(names)); } if ( nnames.contains(n) ) { if ( Character.isDigit( n.charAt(n.length()-1) ) ) { Pattern p= Pattern.compile("([a-zA-Z_])(d+)"); Matcher m= p.matcher(n); if ( m.matches() ) { int d= Integer.parseInt( m.group(2) ); return m.group(1)+String.valueOf(d); } } return n+"1"; } else { return n; } } @Override public void formatData( String uri, QDataSet data, ProgressMonitor mon ) throws Exception { setUri(uri); maybeMkdirs(); String append = getParam( "append", "F" ); WriteIDLSav write= new WriteIDLSav(); String guessName= "DATA"; String[] names= new String[0]; if ( append.equals("T") ) { ReadIDLSav reader= new ReadIDLSav(); File f= new File( getResourceURI().getPath() ); if ( f.length()>Integer.MAX_VALUE ) { throw new IllegalArgumentException("Unable to read large IDLSav files"); } ByteBuffer byteBuffer; try (FileChannel fc = new RandomAccessFile( f, "r" ).getChannel()) { byteBuffer = ByteBuffer.allocate((int) f.length()); fc.read(byteBuffer); } names= reader.readVarNames( byteBuffer ); guessName= maybeIncrementName(guessName,names); for ( String n: names ) { QDataSet v= IdlsavDataSource.getArray( reader, byteBuffer, n ); doOne( write, v, n ); } } if ( data.rank()==2 && SemanticOps.isBundle(data) ) { formatRank2Bundle( uri, data, write, names, mon ); } else { QDataSet dep0= (QDataSet) data.property(QDataSet.DEPEND_0); if ( dep0!=null ) { doOne( write,dep0,"dep0" ); } doOne( write,data,guessName ); QDataSet dep1= (QDataSet) data.property(QDataSet.DEPEND_1); if ( dep1!=null ) { doOne( write,dep1,"dep1" ); } } setUri(uri); File f= new File( getResourceURI().toURL().getFile() ); try (FileOutputStream fos = new FileOutputStream(f)) { write.write( fos ); } } @Override public boolean canFormat(QDataSet ds) { return DataSetUtil.isQube(ds) && ( ds.rank()==1 || ds.rank()==2 || ds.rank()==3 ); } @Override public String getDescription() { return "IDL Saveset"; } }