kr.bydelta.koala.kryo.SentenceSerializer.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of koalanlp-kryo_2.11 Show documentation

koalaNLP-kryo

The newest version!

package kr.bydelta.koala.kryo

import com.esotericsoftware.kryo.io.{Input, Output}
import com.esotericsoftware.kryo.{Kryo, Serializer}
import kr.bydelta.koala.data.{Relationship, Sentence, Word}

/**
  * KryoSerializer object for Sentence class
  */
object SentenceSerializer extends Serializer[Sentence] {
  override def write(kryo: Kryo, output: Output, value: Sentence): Unit = {
    output.writeInt(value.size)
    value.words.foreach(kryo.writeObject(output, _))
    output.writeInt(value.root.dependents.size)
    value.root.dependents.foreach(kryo.writeObject(output, _))
  }

  override def read(kryo: Kryo, input: Input, `type`: Class[Sentence]): Sentence = {
    val szWords = input.readInt
    val words = (0 until szWords).map {
      _ => kryo.readObject(input, classOf[Word])
    }

    val sent = Sentence(words)
    val szTops = input.readInt
    sent.root.deps = (0 until szTops).map {
      _ => kryo.readObject(input, classOf[Relationship])
    }.toSet

    sent
  }
}