eu.agrosense.client.io.csv.CsvDataObject Maven / Gradle / Ivy
/**
* Copyright (C) 2008-2013 LimeTri. All rights reserved.
*
* AgroSense is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* There are special exceptions to the terms and conditions of the GPLv3 as it is applied to
* this software, see the FLOSS License Exception
* .
*
* AgroSense is 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 General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with AgroSense. If not, see if (value.trim().length() != 1) {
// so can't store tab or space separator as characters...
String str = ch == null ? null : ch.toString();
getPrimaryFile().setAttribute(ATT_SEPARATOR_CHAR, str);
firePropertyChange(PROP_SEPARATOR_CHAR, old, ch);
} catch (IOException ex) {
LOGGER.log(Level.WARNING, "could not save separator char property for csv data object {0}", getName());
}
}
public Character getQuoteChar() {
return (Character) getPrimaryFile().getAttribute(ATT_QUOTE_CHAR);
}
void setQuoteChar(Character ch) {
try {
Character old = getQuoteChar();
getPrimaryFile().setAttribute(ATT_QUOTE_CHAR, ch);
firePropertyChange(PROP_QUOTE_CHAR, old, ch);
} catch (IOException ex) {
LOGGER.log(Level.WARNING, "could not save quote char property for csv data object {0}", getName());
}
}
public Character getEscapeChar() {
return (Character) getPrimaryFile().getAttribute(ATT_ESCAPE_CHAR);
}
void setEscapeChar(Character ch) {
try {
Character old = getEscapeChar();
getPrimaryFile().setAttribute(ATT_ESCAPE_CHAR, ch);
firePropertyChange(PROP_ESCAPE_CHAR, old, ch);
} catch (IOException ex) {
LOGGER.log(Level.WARNING, "could not save escape char property for csv data object {0}", getName());
}
}
public Integer getSkipLines() {
return (Integer) getPrimaryFile().getAttribute(ATT_SKIP_LINES);
}
void setSkipLines(Integer i) {
try {
Integer old = getSkipLines();
getPrimaryFile().setAttribute(ATT_SKIP_LINES, i);
firePropertyChange(PROP_SKIP_LINES, old, i);
} catch (IOException ex) {
LOGGER.log(Level.WARNING, "could not save skip lines property for csv data object {0}", getName());
}
}
@MultiViewElement.Registration(
displayName = "#LBL_Csv_EDITOR",
iconBase = ICON_BASE,
mimeType = MIME_TYPE,
persistenceType = TopComponent.PERSISTENCE_ONLY_OPENED,
preferredID = "Csv",
position = 2000)
@Messages("LBL_Csv_EDITOR=Source")
public static MultiViewEditorElement createEditor(Lookup lkp) {
final MultiViewEditorElement mvee = new MultiViewEditorElement(lkp);
// make source view read only:
SwingUtilities.invokeLater(new Runnable() {
@Override public void run() {
JEditorPane pane = mvee.getEditorPane();
pane.setEditable(false);
// FIXME: skipping BOM workaround for now, as it has some side effects:
// It opens the source tab at the bottom of the document, and marks it as savable.
// The extra character in a read only view is less harmful.
// pane.setText(stripBOM(pane.getText()));
}
});
return mvee;
}
private String getHeaderLine() {
// can't use first value from getPrimaryFile().asLines().iterator():
// FileObjectLineIterator keeps a Reader open unless iterated all the way...
String s = null;
try (InputStream is = getPrimaryFile().getInputStream();
Reader r = new InputStreamReader(is, charset);
BufferedReader br = new BufferedReader(r)) {
s = stripBOM(br.readLine());
} catch (IOException ex) {
LOGGER.warning(ex.toString());
// Exceptions.printStackTrace(ex);
}
return s;
}
private String getContents() {
try {
return stripBOM(getPrimaryFile().asText(charset.name()));
} catch (IOException ex) {
Exceptions.printStackTrace(ex);
return null;
}
}
// Remove byte order mark when present
// https://netbeans.org/bugzilla/show_bug.cgi?id=161015
private static String stripBOM(String s) {
if (s == null) return null;
return s.startsWith("\uFEFF") ? s.substring(1) : s;
}
private class CsvMappableSource implements MappableSource {
@Override
public String[] getColumnNames() {
String header = getHeaderLine();
if (header != null) {
CSVReader csvReader = new CSVReader(new StringReader(header), getSeparatorChar());
try {
return csvReader.readNext();
} catch (IOException ex) {
LOGGER.log(Level.WARNING, "could not read CSV column names: {0}", ex.toString());
}
}
return new String[0];
}
@Override
public String getName() {
return getPrimaryFile().getNameExt();
}
@Override
public List getAttributeDescriptors() {
List descriptors = new ArrayList<>();
for (String name : getColumnNames()) {
descriptors.add(new SourceAttributeDescriptor(name, String.class));
}
return descriptors;
}
@Override
public String[][] getContents() {
try {
String contents = CsvDataObject.this.getContents();
int skipLines = getSkipLines();
CSVReader csvReader = new CSVReader(new StringReader(contents), getSeparatorChar(), DEFAULT_QUOTE_CHAR, skipLines);
List rows = csvReader.readAll();
return rows.toArray(new String[0][0]);
} catch (IOException ex) {
LOGGER.warning(ex.toString());
}
return new String[0][0];
}
@Override
public SourceRecordIterator getRecordIterator() {
return new CsvRecordGenerator();
}
@Override
public SourceRecord getGlobals() {
return SourceRecord.empty();
}
}
private class CsvRecordGenerator extends SourceRecordGenerator {
private final List descriptors = mappableSource.getAttributeDescriptors();
private final CSVReader csvReader;
public CsvRecordGenerator() {
String contents = CsvDataObject.this.getContents();
int skipLines = getSkipLines();
csvReader = new CSVReader(new StringReader(contents), getSeparatorChar(), DEFAULT_QUOTE_CHAR, skipLines);
}
@Override
protected SourceRecord getNextRecord() {
if (csvReader == null) { return null; }
try {
String[] next = csvReader.readNext();
return next == null ? null : SourceRecord.createFromColumnValues(descriptors, next);
} catch (IOException ex) {
LOGGER.warning(ex.toString());
return null;
}
}
@Override
public void close() throws IOException {
if (csvReader != null) {
csvReader.close();
}
}
}
}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy