org.pmml4s.xml.XmlAttrs.scala Maven / Gradle / Ivy

Go to download
/*
 * Copyright (c) 2017-2023 AutoDeployAI
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package org.pmml4s.xml

import org.pmml4s.AttributeNotFoundException
import org.pmml4s.util.StringUtils

import javax.xml.stream.events.Attribute

case class XmlAttrs(attrs: Map[String, String]) {

  def has(name: String): Boolean = attrs.contains(name)

  def apply(name: String): String = attrs.getOrElse(name, throw new AttributeNotFoundException(name))

  def apply(n1: String, n2: String): (String, String) = (apply(n1), apply(n2))

  def apply(n1: String, n2: String, n3: String): (String, String, String) = (apply(n1), apply(n2), apply(n3))

  def double(name: String): Double = StringUtils.toDouble(apply(name)).get

  def int(name: String): Int = apply(name).toInt

  def `enum`(name: String, e: Enumeration): e.Value = e.withName(apply(name))

  def getInt(name: String): Option[Int] = get(name).flatMap { x => StringUtils.toInt(x) }

  def getInt(name: String, d: Int): Int = getInt(name).getOrElse(d)

  def getLong(name: String): Option[Long] = get(name).flatMap { x => StringUtils.toLong(x) }

  def getLong(name: String, d: Long): Long = getLong(name).getOrElse(d)

  def getDouble(name: String): Option[Double] = get(name).flatMap { x => StringUtils.toDouble(x) }

  def getDouble(name: String, d: Double): Double = getDouble(name).getOrElse(d)

  def getBoolean(name: String): Option[Boolean] = get(name).flatMap { x => StringUtils.toBool(x) }

  def getBoolean(name: String, d: Boolean): Boolean = getBoolean(name).getOrElse(d)

  def getString(name: String, d: String): String = get(name).getOrElse(d)

  def getEnum(name: String, e: Enumeration): Option[e.Value] = get(name).map(x => e.withName(x))

  def get(name: String): Option[String] = attrs.get(name)

  def -(name: String): XmlAttrs = XmlAttrs(attrs - name)

  def +(name: String, value: String): XmlAttrs = XmlAttrs(attrs + ((name, value)))

  def get(n1: String, n2: String): (Option[String], Option[String]) = (get(n1), get(n2))

  def get(n1: String, n2: String, n3: String): (Option[String], Option[String], Option[String]) = (get(n1), get(n2), get(n3))

  def get(n1: String, n2: String, n3: String, n4: String): (Option[String], Option[String], Option[String], Option[String]) =
    (get(n1), get(n2), get(n3), get(n4))

  def get(n1: String, n2: String, n3: String, n4: String, n5: String): (Option[String], Option[String], Option[String], Option[String], Option[String]) =
    (get(n1), get(n2), get(n3), get(n4), get(n5))

  def get(n1: String, n2: String, n3: String, n4: String, n5: String, n6: String): (Option[String], Option[String], Option[String], Option[String], Option[String], Option[String]) =
    (get(n1), get(n2), get(n3), get(n4), get(n5), get(n6))

  def getDouble(n1: String, n2: String): (Option[Double], Option[Double]) = (getDouble(n1), getDouble(n2))
}

object XmlAttrs {
  def apply(m: java.util.Iterator[Attribute]): XmlAttrs = XmlImplicits.attributes2XmlAttrs(m)

  def apply(name: String, value: String) = new XmlAttrs(Map(name -> value))

  def apply() = new XmlAttrs(Map.empty)
}

object XmlImplicits {
  implicit def attributes2XmlAttrs(attrs: java.util.Iterator[Attribute]): XmlAttrs = XmlAttrs(
    new JIteratorWrapper(attrs).map(x => (x.getName.getLocalPart, x.getValue)).toMap
  )
}

class JIteratorWrapper[A](val underlying: java.util.Iterator[A]) extends scala.collection.Iterator[A] with Serializable {
  def hasNext: Boolean = underlying.hasNext
  def next(): A = underlying.next
}