/*
* This program is free software; you can redistribute it and/or modify it under the
* terms of the GNU Lesser General Public License, version 2.1 as published by the Free Software
* Foundation.
*
* You should have received a copy of the GNU Lesser General Public License along with this
* program; if not, you can obtain a copy at http://www.gnu.org/licenses/old-licenses/lgpl-2.1.html
* or from the Free Software Foundation, Inc.,
* 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
*
* This program 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 Lesser General Public License for more details.
*
* Copyright (c) 2006 - 2013 Pentaho Corporation.. All rights reserved.
*/
package org.pentaho.reporting.engine.classic.core.modules.output.fast.csv;
import org.pentaho.reporting.engine.classic.core.InvalidReportStateException;
import org.pentaho.reporting.engine.classic.core.filter.MessageFormatSupport;
import org.pentaho.reporting.engine.classic.core.layout.model.LogicalPageBox;
import org.pentaho.reporting.engine.classic.core.layout.model.RenderBox;
import org.pentaho.reporting.engine.classic.core.layout.output.OutputProcessorMetaData;
import org.pentaho.reporting.engine.classic.core.modules.output.fast.template.FastExportTemplateProducer;
import org.pentaho.reporting.engine.classic.core.modules.output.fast.template.FormattedDataBuilder;
import org.pentaho.reporting.engine.classic.core.modules.output.fast.template.TemplatingOutputProcessor;
import org.pentaho.reporting.engine.classic.core.modules.output.table.base.SheetLayout;
import org.pentaho.reporting.engine.classic.core.modules.output.table.base.TableContentProducer;
import org.pentaho.reporting.engine.classic.core.modules.output.table.base.TableRectangle;
import org.pentaho.reporting.engine.classic.core.modules.output.table.csv.CSVTableModule;
import org.pentaho.reporting.engine.classic.core.util.InstanceID;
import org.pentaho.reporting.libraries.base.util.CSVQuoter;
import org.pentaho.reporting.libraries.fonts.encoding.EncodingRegistry;
import java.io.PrintWriter;
import java.io.StringWriter;
import java.util.HashMap;
import java.util.Locale;
public class CsvTemplateProducer implements FastExportTemplateProducer {
private OutputProcessorMetaData metaData;
private SheetLayout sheetLayout;
private String encoding;
private HashMap<InstanceID, String> idMapping;
private String template;
private CSVQuoter quoter;
public CsvTemplateProducer( final OutputProcessorMetaData metaData, final SheetLayout sheetLayout,
final String encoding ) {
this.metaData = metaData;
this.sheetLayout = sheetLayout;
this.encoding = encoding;
this.idMapping = new HashMap<InstanceID, String>();
final String separator =
metaData.getConfiguration().getConfigProperty( CSVTableModule.SEPARATOR, CSVTableModule.SEPARATOR_DEFAULT );
if ( separator.length() == 0 ) {
throw new IllegalArgumentException( "CSV separate cannot be an empty string." );
}
if ( this.encoding == null ) {
this.encoding =
metaData.getConfiguration().getConfigProperty(
"org.pentaho.reporting.engine.classic.core.modules.output.table.csv.Encoding",
EncodingRegistry.getPlatformDefaultEncoding() );
}
quoter = new CSVQuoter( separator.charAt( 0 ) );
}
public void produceTemplate( final LogicalPageBox pageBox ) {
TableContentProducer contentProducer =
TemplatingOutputProcessor.produceTableLayout( pageBox, sheetLayout, metaData );
final SheetLayout sheetLayout = contentProducer.getSheetLayout();
final int columnCount = contentProducer.getColumnCount();
final int lastColumn = columnCount - 1;
final int startRow = contentProducer.getFinishedRows();
final int finishRow = contentProducer.getFilledRows();
StringWriter swriter = new StringWriter();
PrintWriter writer = new PrintWriter( swriter );
for ( int row = startRow; row < finishRow; row++ ) {
for ( short col = 0; col < columnCount; col++ ) {
final RenderBox content = contentProducer.getContent( row, col );
if ( content == null ) {
writer.print( quoter.getSeparator() );
continue;
}
if ( content.isCommited() == false ) {
throw new InvalidReportStateException( "Uncommited content encountered" );
}
final long contentOffset = contentProducer.getContentOffset( row, col );
final TableRectangle rectangle =
sheetLayout.getTableBounds( content.getX(), content.getY() + contentOffset, content.getWidth(), content
.getHeight(), null );
if ( rectangle.isOrigin( col, row ) == false ) {
// A spanned cell ..
continue;
}
InstanceID instanceId = content.getNodeLayoutProperties().getInstanceId();
String uuid = idMapping.get( instanceId );
if ( uuid == null ) {
uuid = String.valueOf( idMapping.size() );
idMapping.put( instanceId, uuid );
}
final String formattedtext = "$(" + uuid + ")";
writer.write( formattedtext );
if ( col < lastColumn ) {
writer.print( quoter.getSeparator() );
}
content.setFinishedTable( true );
}
writer.println();
}
writer.close();
template = swriter.toString();
}
public CSVQuoter getQuoter() {
return quoter;
}
public String getTemplate() {
return template;
}
public FormattedDataBuilder createDataBuilder() {
MessageFormatSupport messageFormatSupport = new MessageFormatSupport();
messageFormatSupport.setLocale( Locale.ENGLISH );
messageFormatSupport.setFormatString( getTemplate() );
messageFormatSupport.setNullString( "" );
return new CsvFormattedDataBuilder( idMapping, messageFormatSupport, getQuoter(), encoding );
}
}