summaryrefslogtreecommitdiff
path: root/src/csvreader.cpp
blob: 08803e7ddadf6781db4d6beaa97ee8c5888bf2a5 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
/*
 * Copyright (C) 2007-2010 Xagasoft, All rights reserved.
 *
 * This file is part of the libbu++ library and is released under the
 * terms of the license contained in the file LICENSE.
 */

#include "bu/csvreader.h"
#include "bu/stream.h"

#include "bu/sio.h"
using namespace Bu;

Bu::CsvReader::CsvReader( Bu::Stream &sIn, Bu::CsvReader::Style eStyle ) :
	sIn( sIn )
{
	switch( eStyle )
	{
		case styleExcel:
			sDecode = Bu::slot( &decodeExcel );
			break;

		case styleC:
			sDecode = Bu::slot( &decodeC );
			break;
	}
}

Bu::CsvReader::CsvReader( Bu::Stream &sIn,
		Bu::CsvReader::DecodeSignal sDecode ) :
	sIn( sIn ),
	sDecode( sDecode )
{
}

Bu::CsvReader::~CsvReader()
{
}

Bu::StrArray Bu::CsvReader::readLine()
{
	Bu::StrArray aVals;

	Bu::FString sLine = sIn.readLine();

	if( !sLine.isSet() )
		return Bu::StrArray();

	Bu::FString::iterator i = sLine.begin();

	aVals.append( sDecode( i ) );

	while( i )
	{
		if( *i == ',' )
		{
			i++;
			if( !i )
				break;
			aVals.append( sDecode( i ) );
		}
		else
		{
			// Blanks and stuff?
			sio << "Out of bound:  '" << *i << "'" << sio.nl;
			i++;
		}
	}

	return aVals;
}

Bu::FString Bu::CsvReader::decodeExcel( Bu::FString::iterator &i )
{
	Bu::FString sRet;

	for(; i && (*i == ' ' || *i == '\t'); i++ ) { }
	
	if( !i )
		return sRet;

	if( *i == '\"' )
	{
		for( i++ ; i; i++ )
		{
			if( *i == '\"' )
			{
				i++;
				if( !i )
				{
					return sRet;
				}
				else if( *i == '\"' )
				{
					sRet += *i;
				}
				else
				{
					return sRet;
				}
			}
			else
			{
				sRet += *i;
			}
		}
	}
	else
	{
		for( ; i; i++ )
		{
			if( *i == ',' )
			{
				return sRet;
			}
			sRet += *i;
		}
	}

	return sRet;
}

Bu::FString Bu::CsvReader::decodeC( Bu::FString::iterator & )
{
	return "";
}