com.rathravane.till.data.csv.csvStream Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of silt Show documentation
A small collection of classes used in various Rathravane systems.
There is a newer version: 2.1.1
package com.rathravane.till.data.csv;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.LinkedList;
import java.util.List;
import java.util.Map;
import java.util.Vector;

import com.rathravane.till.data.stringUtils;
import com.rathravane.till.data.stringUtils.valueInfo;

public class csvStream
{
	public static final String kLineField = "line";
	public static final String kLineField_Default = "line";

	public static final String kHasHeaderRow = "header";
	public static final String kDelimiter = "delimiter";
	public static final String kQuote = "quote";
	public static final String kPassThru = "passthru";

	public csvStream ( boolean header )
	{
		this ( csvEncoder.kDefaultQuoteChar, csvEncoder.kDefaultFieldSeparatorChar, header );
	}

	public csvStream ( char quoteChar, char fieldSepChar, boolean header )
	{
		fDelimiter = fieldSepChar;
		fQuote = quoteChar;
		fColumns = null;
		fHasHeaderRow = header;
		fLineCount = 0;
		fRecords = new ArrayList> ();
	}

	public void reset ()
	{
		fLineCount = 0;
		fRecords.clear ();
	}

	public void read ( InputStream is ) throws IOException
	{
		BufferedReader br = new BufferedReader ( new InputStreamReader ( is ) );
		read ( br );
	}

	public void read ( BufferedReader br ) throws IOException
	{
		String s;
		while ( (s = br.readLine ()) != null )
		{
			s = s.trim ();
			if ( fLineCount == 0 && fHasHeaderRow )
			{
				// deal with header
				parseHeader ( s );
			}
			else
			{
				// deal with regular lines
				final HashMap rec = parseLine ( s );
				fRecords.add ( rec );
			}
			fLineCount++;
		}
	}

	public Vector getColumnNames ()
	{
		final Vector result = new Vector ();
		for ( String c : fColumns )
		{
			result.add ( c );
		}
		return result;
	}

	public int getLinesParsed ()
	{
		return fLineCount;
	}

	public int getRecordCount ()
	{
		return fRecords.size ();
	}

	/**
	 * Get a record by row number. Note that the returned map can include nulls for values.
	 * @param row
	 * @return
	 */
	public Map getRecord ( int row )
	{
		return fRecords.get ( row );
	}

	public String getField ( int row, int field )
	{
		final String name = ( fColumns != null && field> fRecords;

	public static List parseHeaderLine ( String line, char quote, char delim )
	{
		final LinkedList headers = new LinkedList ();

		String remains = line;
		while ( remains.length () > 0 )
		{
			valueInfo result = stringUtils.getLeadingValue ( remains, quote, delim );
			if ( result != null )
			{
				if ( result.fValue == null )
				{
					// odd... an empty value in the column header line
					result = new valueInfo ( "unnamed-" + (headers.size()+1), result.fNextFieldAt );
				}

				headers.add ( result.fValue );
				if ( result.fNextFieldAt > -1 )
				{
					remains = remains.substring ( result.fNextFieldAt );
				}
				else
				{
					remains = "";
				}
			}
		}
		return headers;
	}

	private void parseHeader ( String line )
	{
		LinkedList headers = new LinkedList ();

		String remains = line;
		while ( remains.length () > 0 )
		{
			valueInfo result = stringUtils.getLeadingValue ( remains, fQuote, fDelimiter );
			if ( result != null )
			{
				if ( result.fValue == null )
				{
					// odd... an empty value in the column header line
					result = new valueInfo ( "unnamed-" + (headers.size()+1), result.fNextFieldAt );
				}

				headers.add ( result.fValue );
				if ( result.fNextFieldAt > -1 )
				{
					remains = remains.substring ( result.fNextFieldAt );
				}
				else
				{
					remains = "";
				}
			}
		}
		fColumns = headers.toArray ( new String [headers.size ()] );
	}

	protected HashMap parseLine ( String line )
	{
		final HashMap result = new HashMap ();

		int colNum = 0;
		String remains = new String ( line );
		while ( remains != null )
		{
			final stringUtils.valueInfo vi = stringUtils.getLeadingValue ( remains, fQuote, fDelimiter );
			final String name = ( fColumns != null && colNum -1 )
			{
				remains = remains.substring ( vi.fNextFieldAt );
			}
			else
			{
				remains = null;
			}
		}

		return result;
	}
}