• Home
• org.scalanlp
• breeze_2.11
• 0.13.1
• source code
• CSVReader.scala

×

Project download

Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $

You can buy this project and download/modify it how often you want.

breeze.io.CSVReader.scala Maven / Gradle / Ivy

package breeze.io

import java.io._
import au.com.bytecode.opencsv.{CSVReader => OpenCSVReader, CSVWriter=>OpenCSVWriter}


/**
 * Just a simple wrapper for OpenCSV's csvreader.
 * @author dlwh
 */
object CSVReader {
  def read(input: Reader,
           separator: Char=',',
           quote: Char='"',
           escape: Char='\\',
           skipLines: Int = 0):IndexedSeq[IndexedSeq[String]] = {
    iterator(input, separator, quote, escape, skipLines).toIndexedSeq
  }

  def parse(input: String, separator: Char=',',
             quote: Char='"',
             escape: Char='\\',
             skipLines: Int = 0):IndexedSeq[IndexedSeq[String]] = {
    read(new StringReader(input), separator, quote, escape, skipLines)

  }

  def iterator(input: Reader,
           separator: Char=',',
           quote: Char='"',
           escape: Char='\\',
           skipLines: Int = 0) = {
    val rdr = new OpenCSVReader(input, separator, quote, escape, skipLines)
    new Iterator[IndexedSeq[String]] {
      var _next = rdr.readNext()
      def hasNext: Boolean = _next ne null

      def next() = {
        if(!hasNext) throw new NoSuchElementException("Next on empty iterator.")
        val x = _next
        _next = rdr.readNext
        x
      }

      def close() = {
        rdr.close()
        _next = null
      }
    }
  }

}

/**
 * Just a simple wrapper for OpenCSV's csvreader.
 * @author dlwh
 */
object CSVWriter {
  def write(output: Writer,
            mat: TraversableOnce[IndexedSeq[String]],
            separator: Char=',',
            quote: Char='"',
            escape: Char='\\') {
    val writer = new OpenCSVWriter(output, separator, quote, escape)
    import scala.collection.JavaConverters._
    mat match {
      case Seq(x @ _*) => writer.writeAll(x.map(_.toArray).asJava)
      case _ =>
        for(l <- mat) {
          writer.writeNext(l.toArray)
        }
    }
    writer.flush()
  }

  def writeFile(file: File,
                mat: IndexedSeq[IndexedSeq[String]],
                separator: Char=',',
                quote: Char='"',
                escape: Char='\\') {
    val out = new FileWriter(file)
    write(out, mat, separator, quote, escape)
    out.close()
  }

  def mkString(mat: IndexedSeq[IndexedSeq[String]],
           separator: Char=',',
           quote: Char='"',
           escape: Char='\\'):String = {
    val out = new StringWriter
     write(out, mat, separator, quote, escape)
    out.toString
  }



}