package net.tz.spss

import net.tz.spss.SpssVariable.DoubleRange
import com.bedatadriven.spss.{SpssVariable => JVar}

import scala.collection.JavaConverters._
import scala.language.postfixOps

case class SpssVariable(name: String, shortName: String, isNumeric: Boolean,
  index: Int, label: String,
  missingRange: Option[DoubleRange] = None,
  missing: List[Double] = Nil,
  missingStrings: List[String] = Nil,
  labels: Map[Double, String] = Map())

object SpssVariable {
  type DoubleRange = (Double, Double)

  def apply(v: JVar): SpssVariable = SpssVariable(
    name = v.getVariableName,
    shortName = v.getShortName,
    isNumeric = v.isNumeric,
    index = v.getIndex,
    label = v.getVariableLabel,
    missingRange = if (v.isHasMissingRange)
      Some((v.getMissingMin.doubleValue(), v.getMissingMax.doubleValue()))
    else None,
    missing = v.getMissing.asScala.map(_.doubleValue()).toList,
    missingStrings = v.getMissingStrings.asScala.toList,
    labels = v.getValueLabels.asScala.map { case (k,x) =>
      (k.doubleValue(), x) } toMap)
}