cgta.oscala.util.CsvReader.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of oscala-sjs_sjs0.6_2.11 Show documentation

oscala-sjs

The newest version!

package cgta.oscala
package util


//////////////////////////////////////////////////////////////
// Copyright (c) 2014 Ben Jackman, Jeff Gomberg
// All Rights Reserved
// please contact [email protected] or [email protected]
// for licensing inquiries
// Created by bjackman @ 8/11/14 2:45 AM
//////////////////////////////////////////////////////////////

object CsvReader {
  def apply(lines: Iterator[String]): CsvReader = {
    val colByName = lines.next().split(",").zipWithIndex.toMap
    new CsvReader(colByName, lines.map(_.split(",").toArray).toArray)
  }
}

class CsvReader(hdrMap: Map[String, Int], rowArray: Array[Array[String]]) {
  val colNames = hdrMap.iterator.map(_.swap).toList.sortBy(_._1).map(_._2).toArray
  def rows : IndexedSeq[Map[String, String]] = new IndexedSeq[Map[String, String]] {
    override def length: Int = rowArray.length
    override def apply(idx: Int): Map[String, String] = new Map[String, String] {
      override def +[B1 >: String](kv: (String, B1)): Map[String, B1] = sys.error("Don't add to csv rows")
      override def get(key: String): Option[String] = hdrMap.get(key).flatMap(ci => rowArray(idx).getOpt(ci))
      override def iterator: Iterator[(String, String)] = (rowArray(idx) zip colNames).iterator
      override def -(key: String): Map[String, String] = sys.error("Don't remove from csv rows")
    }
  }
}