CSV.html 19 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273
  1. <!DOCTYPE HTML PUBLIC "-//W3C//DTD HTML 4.01 Transitional//EN" "http://www.w3.org/TR/html4/loose.dtd">
  2. <html lang="en">
  3. <head>
  4. <title>Source code</title>
  5. <link rel="stylesheet" type="text/css" href="../../../../../stylesheet.css" title="Style">
  6. </head>
  7. <body>
  8. <div class="sourceContainer">
  9. <pre><span class="sourceLineNo">001</span>/* ===========================================================<a name="line.1"></a>
  10. <span class="sourceLineNo">002</span> * JFreeChart : a free chart library for the Java(tm) platform<a name="line.2"></a>
  11. <span class="sourceLineNo">003</span> * ===========================================================<a name="line.3"></a>
  12. <span class="sourceLineNo">004</span> *<a name="line.4"></a>
  13. <span class="sourceLineNo">005</span> * (C) Copyright 2000-2013, by Object Refinery Limited and Contributors.<a name="line.5"></a>
  14. <span class="sourceLineNo">006</span> *<a name="line.6"></a>
  15. <span class="sourceLineNo">007</span> * Project Info: http://www.jfree.org/jfreechart/index.html<a name="line.7"></a>
  16. <span class="sourceLineNo">008</span> *<a name="line.8"></a>
  17. <span class="sourceLineNo">009</span> * This library is free software; you can redistribute it and/or modify it<a name="line.9"></a>
  18. <span class="sourceLineNo">010</span> * under the terms of the GNU Lesser General Public License as published by<a name="line.10"></a>
  19. <span class="sourceLineNo">011</span> * the Free Software Foundation; either version 2.1 of the License, or<a name="line.11"></a>
  20. <span class="sourceLineNo">012</span> * (at your option) any later version.<a name="line.12"></a>
  21. <span class="sourceLineNo">013</span> *<a name="line.13"></a>
  22. <span class="sourceLineNo">014</span> * This library is distributed in the hope that it will be useful, but<a name="line.14"></a>
  23. <span class="sourceLineNo">015</span> * WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY<a name="line.15"></a>
  24. <span class="sourceLineNo">016</span> * or FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public<a name="line.16"></a>
  25. <span class="sourceLineNo">017</span> * License for more details.<a name="line.17"></a>
  26. <span class="sourceLineNo">018</span> *<a name="line.18"></a>
  27. <span class="sourceLineNo">019</span> * You should have received a copy of the GNU Lesser General Public<a name="line.19"></a>
  28. <span class="sourceLineNo">020</span> * License along with this library; if not, write to the Free Software<a name="line.20"></a>
  29. <span class="sourceLineNo">021</span> * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301,<a name="line.21"></a>
  30. <span class="sourceLineNo">022</span> * USA.<a name="line.22"></a>
  31. <span class="sourceLineNo">023</span> *<a name="line.23"></a>
  32. <span class="sourceLineNo">024</span> * [Oracle and Java are registered trademarks of Oracle and/or its affiliates. <a name="line.24"></a>
  33. <span class="sourceLineNo">025</span> * Other names may be trademarks of their respective owners.]<a name="line.25"></a>
  34. <span class="sourceLineNo">026</span> *<a name="line.26"></a>
  35. <span class="sourceLineNo">027</span> * --------<a name="line.27"></a>
  36. <span class="sourceLineNo">028</span> * CSV.java<a name="line.28"></a>
  37. <span class="sourceLineNo">029</span> * --------<a name="line.29"></a>
  38. <span class="sourceLineNo">030</span> * (C) Copyright 2003-2008, by Object Refinery Limited.<a name="line.30"></a>
  39. <span class="sourceLineNo">031</span> *<a name="line.31"></a>
  40. <span class="sourceLineNo">032</span> * Original Author: David Gilbert (for Object Refinery Limited);<a name="line.32"></a>
  41. <span class="sourceLineNo">033</span> * Contributor(s): -;<a name="line.33"></a>
  42. <span class="sourceLineNo">034</span> *<a name="line.34"></a>
  43. <span class="sourceLineNo">035</span> * Changes<a name="line.35"></a>
  44. <span class="sourceLineNo">036</span> * -------<a name="line.36"></a>
  45. <span class="sourceLineNo">037</span> * 24-Nov-2003 : Version 1 (DG);<a name="line.37"></a>
  46. <span class="sourceLineNo">038</span> *<a name="line.38"></a>
  47. <span class="sourceLineNo">039</span> */<a name="line.39"></a>
  48. <span class="sourceLineNo">040</span><a name="line.40"></a>
  49. <span class="sourceLineNo">041</span>package org.jfree.data.io;<a name="line.41"></a>
  50. <span class="sourceLineNo">042</span><a name="line.42"></a>
  51. <span class="sourceLineNo">043</span>import java.io.BufferedReader;<a name="line.43"></a>
  52. <span class="sourceLineNo">044</span>import java.io.IOException;<a name="line.44"></a>
  53. <span class="sourceLineNo">045</span>import java.io.Reader;<a name="line.45"></a>
  54. <span class="sourceLineNo">046</span>import java.util.List;<a name="line.46"></a>
  55. <span class="sourceLineNo">047</span><a name="line.47"></a>
  56. <span class="sourceLineNo">048</span>import org.jfree.data.category.CategoryDataset;<a name="line.48"></a>
  57. <span class="sourceLineNo">049</span>import org.jfree.data.category.DefaultCategoryDataset;<a name="line.49"></a>
  58. <span class="sourceLineNo">050</span><a name="line.50"></a>
  59. <span class="sourceLineNo">051</span>/**<a name="line.51"></a>
  60. <span class="sourceLineNo">052</span> * A utility class for reading {@link CategoryDataset} data from a CSV file.<a name="line.52"></a>
  61. <span class="sourceLineNo">053</span> * This initial version is very basic, and won't handle errors in the data<a name="line.53"></a>
  62. <span class="sourceLineNo">054</span> * file very gracefully.<a name="line.54"></a>
  63. <span class="sourceLineNo">055</span> */<a name="line.55"></a>
  64. <span class="sourceLineNo">056</span>public class CSV {<a name="line.56"></a>
  65. <span class="sourceLineNo">057</span><a name="line.57"></a>
  66. <span class="sourceLineNo">058</span> /** The field delimiter. */<a name="line.58"></a>
  67. <span class="sourceLineNo">059</span> private char fieldDelimiter;<a name="line.59"></a>
  68. <span class="sourceLineNo">060</span><a name="line.60"></a>
  69. <span class="sourceLineNo">061</span> /** The text delimiter. */<a name="line.61"></a>
  70. <span class="sourceLineNo">062</span> private char textDelimiter;<a name="line.62"></a>
  71. <span class="sourceLineNo">063</span><a name="line.63"></a>
  72. <span class="sourceLineNo">064</span> /**<a name="line.64"></a>
  73. <span class="sourceLineNo">065</span> * Creates a new CSV reader where the field delimiter is a comma, and the<a name="line.65"></a>
  74. <span class="sourceLineNo">066</span> * text delimiter is a double-quote.<a name="line.66"></a>
  75. <span class="sourceLineNo">067</span> */<a name="line.67"></a>
  76. <span class="sourceLineNo">068</span> public CSV() {<a name="line.68"></a>
  77. <span class="sourceLineNo">069</span> this(',', '"');<a name="line.69"></a>
  78. <span class="sourceLineNo">070</span> }<a name="line.70"></a>
  79. <span class="sourceLineNo">071</span><a name="line.71"></a>
  80. <span class="sourceLineNo">072</span> /**<a name="line.72"></a>
  81. <span class="sourceLineNo">073</span> * Creates a new reader with the specified field and text delimiters.<a name="line.73"></a>
  82. <span class="sourceLineNo">074</span> *<a name="line.74"></a>
  83. <span class="sourceLineNo">075</span> * @param fieldDelimiter the field delimiter (usually a comma, semi-colon,<a name="line.75"></a>
  84. <span class="sourceLineNo">076</span> * colon, tab or space).<a name="line.76"></a>
  85. <span class="sourceLineNo">077</span> * @param textDelimiter the text delimiter (usually a single or double<a name="line.77"></a>
  86. <span class="sourceLineNo">078</span> * quote).<a name="line.78"></a>
  87. <span class="sourceLineNo">079</span> */<a name="line.79"></a>
  88. <span class="sourceLineNo">080</span> public CSV(char fieldDelimiter, char textDelimiter) {<a name="line.80"></a>
  89. <span class="sourceLineNo">081</span> this.fieldDelimiter = fieldDelimiter;<a name="line.81"></a>
  90. <span class="sourceLineNo">082</span> this.textDelimiter = textDelimiter;<a name="line.82"></a>
  91. <span class="sourceLineNo">083</span> }<a name="line.83"></a>
  92. <span class="sourceLineNo">084</span><a name="line.84"></a>
  93. <span class="sourceLineNo">085</span> /**<a name="line.85"></a>
  94. <span class="sourceLineNo">086</span> * Reads a {@link CategoryDataset} from a CSV file or input source.<a name="line.86"></a>
  95. <span class="sourceLineNo">087</span> *<a name="line.87"></a>
  96. <span class="sourceLineNo">088</span> * @param in the input source.<a name="line.88"></a>
  97. <span class="sourceLineNo">089</span> *<a name="line.89"></a>
  98. <span class="sourceLineNo">090</span> * @return A category dataset.<a name="line.90"></a>
  99. <span class="sourceLineNo">091</span> *<a name="line.91"></a>
  100. <span class="sourceLineNo">092</span> * @throws IOException if there is an I/O problem.<a name="line.92"></a>
  101. <span class="sourceLineNo">093</span> */<a name="line.93"></a>
  102. <span class="sourceLineNo">094</span> public CategoryDataset readCategoryDataset(Reader in) throws IOException {<a name="line.94"></a>
  103. <span class="sourceLineNo">095</span><a name="line.95"></a>
  104. <span class="sourceLineNo">096</span> DefaultCategoryDataset dataset = new DefaultCategoryDataset();<a name="line.96"></a>
  105. <span class="sourceLineNo">097</span> BufferedReader reader = new BufferedReader(in);<a name="line.97"></a>
  106. <span class="sourceLineNo">098</span> List columnKeys = null;<a name="line.98"></a>
  107. <span class="sourceLineNo">099</span> int lineIndex = 0;<a name="line.99"></a>
  108. <span class="sourceLineNo">100</span> String line = reader.readLine();<a name="line.100"></a>
  109. <span class="sourceLineNo">101</span> while (line != null) {<a name="line.101"></a>
  110. <span class="sourceLineNo">102</span> if (lineIndex == 0) { // first line contains column keys<a name="line.102"></a>
  111. <span class="sourceLineNo">103</span> columnKeys = extractColumnKeys(line);<a name="line.103"></a>
  112. <span class="sourceLineNo">104</span> }<a name="line.104"></a>
  113. <span class="sourceLineNo">105</span> else { // remaining lines contain a row key and data values<a name="line.105"></a>
  114. <span class="sourceLineNo">106</span> extractRowKeyAndData(line, dataset, columnKeys);<a name="line.106"></a>
  115. <span class="sourceLineNo">107</span> }<a name="line.107"></a>
  116. <span class="sourceLineNo">108</span> line = reader.readLine();<a name="line.108"></a>
  117. <span class="sourceLineNo">109</span> lineIndex++;<a name="line.109"></a>
  118. <span class="sourceLineNo">110</span> }<a name="line.110"></a>
  119. <span class="sourceLineNo">111</span> return dataset;<a name="line.111"></a>
  120. <span class="sourceLineNo">112</span><a name="line.112"></a>
  121. <span class="sourceLineNo">113</span> }<a name="line.113"></a>
  122. <span class="sourceLineNo">114</span><a name="line.114"></a>
  123. <span class="sourceLineNo">115</span> /**<a name="line.115"></a>
  124. <span class="sourceLineNo">116</span> * Extracts the column keys from a string.<a name="line.116"></a>
  125. <span class="sourceLineNo">117</span> *<a name="line.117"></a>
  126. <span class="sourceLineNo">118</span> * @param line a line from the input file.<a name="line.118"></a>
  127. <span class="sourceLineNo">119</span> *<a name="line.119"></a>
  128. <span class="sourceLineNo">120</span> * @return A list of column keys.<a name="line.120"></a>
  129. <span class="sourceLineNo">121</span> */<a name="line.121"></a>
  130. <span class="sourceLineNo">122</span> private List extractColumnKeys(String line) {<a name="line.122"></a>
  131. <span class="sourceLineNo">123</span> List keys = new java.util.ArrayList();<a name="line.123"></a>
  132. <span class="sourceLineNo">124</span> int fieldIndex = 0;<a name="line.124"></a>
  133. <span class="sourceLineNo">125</span> int start = 0;<a name="line.125"></a>
  134. <span class="sourceLineNo">126</span> for (int i = 0; i &lt; line.length(); i++) {<a name="line.126"></a>
  135. <span class="sourceLineNo">127</span> if (line.charAt(i) == this.fieldDelimiter) {<a name="line.127"></a>
  136. <span class="sourceLineNo">128</span> if (fieldIndex &gt; 0) { // first field is ignored, since<a name="line.128"></a>
  137. <span class="sourceLineNo">129</span> // column 0 is for row keys<a name="line.129"></a>
  138. <span class="sourceLineNo">130</span> String key = line.substring(start, i);<a name="line.130"></a>
  139. <span class="sourceLineNo">131</span> keys.add(removeStringDelimiters(key));<a name="line.131"></a>
  140. <span class="sourceLineNo">132</span> }<a name="line.132"></a>
  141. <span class="sourceLineNo">133</span> start = i + 1;<a name="line.133"></a>
  142. <span class="sourceLineNo">134</span> fieldIndex++;<a name="line.134"></a>
  143. <span class="sourceLineNo">135</span> }<a name="line.135"></a>
  144. <span class="sourceLineNo">136</span> }<a name="line.136"></a>
  145. <span class="sourceLineNo">137</span> String key = line.substring(start, line.length());<a name="line.137"></a>
  146. <span class="sourceLineNo">138</span> keys.add(removeStringDelimiters(key));<a name="line.138"></a>
  147. <span class="sourceLineNo">139</span> return keys;<a name="line.139"></a>
  148. <span class="sourceLineNo">140</span> }<a name="line.140"></a>
  149. <span class="sourceLineNo">141</span><a name="line.141"></a>
  150. <span class="sourceLineNo">142</span> /**<a name="line.142"></a>
  151. <span class="sourceLineNo">143</span> * Extracts the row key and data for a single line from the input source.<a name="line.143"></a>
  152. <span class="sourceLineNo">144</span> *<a name="line.144"></a>
  153. <span class="sourceLineNo">145</span> * @param line the line from the input source.<a name="line.145"></a>
  154. <span class="sourceLineNo">146</span> * @param dataset the dataset to be populated.<a name="line.146"></a>
  155. <span class="sourceLineNo">147</span> * @param columnKeys the column keys.<a name="line.147"></a>
  156. <span class="sourceLineNo">148</span> */<a name="line.148"></a>
  157. <span class="sourceLineNo">149</span> private void extractRowKeyAndData(String line,<a name="line.149"></a>
  158. <span class="sourceLineNo">150</span> DefaultCategoryDataset dataset,<a name="line.150"></a>
  159. <span class="sourceLineNo">151</span> List columnKeys) {<a name="line.151"></a>
  160. <span class="sourceLineNo">152</span> Comparable rowKey = null;<a name="line.152"></a>
  161. <span class="sourceLineNo">153</span> int fieldIndex = 0;<a name="line.153"></a>
  162. <span class="sourceLineNo">154</span> int start = 0;<a name="line.154"></a>
  163. <span class="sourceLineNo">155</span> for (int i = 0; i &lt; line.length(); i++) {<a name="line.155"></a>
  164. <span class="sourceLineNo">156</span> if (line.charAt(i) == this.fieldDelimiter) {<a name="line.156"></a>
  165. <span class="sourceLineNo">157</span> if (fieldIndex == 0) { // first field contains the row key<a name="line.157"></a>
  166. <span class="sourceLineNo">158</span> String key = line.substring(start, i);<a name="line.158"></a>
  167. <span class="sourceLineNo">159</span> rowKey = removeStringDelimiters(key);<a name="line.159"></a>
  168. <span class="sourceLineNo">160</span> }<a name="line.160"></a>
  169. <span class="sourceLineNo">161</span> else { // remaining fields contain values<a name="line.161"></a>
  170. <span class="sourceLineNo">162</span> Double value = Double.valueOf(<a name="line.162"></a>
  171. <span class="sourceLineNo">163</span> removeStringDelimiters(line.substring(start, i))<a name="line.163"></a>
  172. <span class="sourceLineNo">164</span> );<a name="line.164"></a>
  173. <span class="sourceLineNo">165</span> dataset.addValue(<a name="line.165"></a>
  174. <span class="sourceLineNo">166</span> value, rowKey,<a name="line.166"></a>
  175. <span class="sourceLineNo">167</span> (Comparable) columnKeys.get(fieldIndex - 1)<a name="line.167"></a>
  176. <span class="sourceLineNo">168</span> );<a name="line.168"></a>
  177. <span class="sourceLineNo">169</span> }<a name="line.169"></a>
  178. <span class="sourceLineNo">170</span> start = i + 1;<a name="line.170"></a>
  179. <span class="sourceLineNo">171</span> fieldIndex++;<a name="line.171"></a>
  180. <span class="sourceLineNo">172</span> }<a name="line.172"></a>
  181. <span class="sourceLineNo">173</span> }<a name="line.173"></a>
  182. <span class="sourceLineNo">174</span> Double value = Double.valueOf(<a name="line.174"></a>
  183. <span class="sourceLineNo">175</span> removeStringDelimiters(line.substring(start, line.length()))<a name="line.175"></a>
  184. <span class="sourceLineNo">176</span> );<a name="line.176"></a>
  185. <span class="sourceLineNo">177</span> dataset.addValue(<a name="line.177"></a>
  186. <span class="sourceLineNo">178</span> value, rowKey, (Comparable) columnKeys.get(fieldIndex - 1)<a name="line.178"></a>
  187. <span class="sourceLineNo">179</span> );<a name="line.179"></a>
  188. <span class="sourceLineNo">180</span> }<a name="line.180"></a>
  189. <span class="sourceLineNo">181</span><a name="line.181"></a>
  190. <span class="sourceLineNo">182</span> /**<a name="line.182"></a>
  191. <span class="sourceLineNo">183</span> * Removes the string delimiters from a key (as well as any white space<a name="line.183"></a>
  192. <span class="sourceLineNo">184</span> * outside the delimiters).<a name="line.184"></a>
  193. <span class="sourceLineNo">185</span> *<a name="line.185"></a>
  194. <span class="sourceLineNo">186</span> * @param key the key (including delimiters).<a name="line.186"></a>
  195. <span class="sourceLineNo">187</span> *<a name="line.187"></a>
  196. <span class="sourceLineNo">188</span> * @return The key without delimiters.<a name="line.188"></a>
  197. <span class="sourceLineNo">189</span> */<a name="line.189"></a>
  198. <span class="sourceLineNo">190</span> private String removeStringDelimiters(String key) {<a name="line.190"></a>
  199. <span class="sourceLineNo">191</span> String k = key.trim();<a name="line.191"></a>
  200. <span class="sourceLineNo">192</span> if (k.charAt(0) == this.textDelimiter) {<a name="line.192"></a>
  201. <span class="sourceLineNo">193</span> k = k.substring(1);<a name="line.193"></a>
  202. <span class="sourceLineNo">194</span> }<a name="line.194"></a>
  203. <span class="sourceLineNo">195</span> if (k.charAt(k.length() - 1) == this.textDelimiter) {<a name="line.195"></a>
  204. <span class="sourceLineNo">196</span> k = k.substring(0, k.length() - 1);<a name="line.196"></a>
  205. <span class="sourceLineNo">197</span> }<a name="line.197"></a>
  206. <span class="sourceLineNo">198</span> return k;<a name="line.198"></a>
  207. <span class="sourceLineNo">199</span> }<a name="line.199"></a>
  208. <span class="sourceLineNo">200</span><a name="line.200"></a>
  209. <span class="sourceLineNo">201</span>}<a name="line.201"></a>
  210. </pre>
  211. </div>
  212. </body>
  213. </html>