ch.ninecode.cim.CIMRegistrator.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of CIMReader Show documentation

Expose CIM data files as Spark RDD

There is a newer version: 2.12-3.0.1-5.1.1

package ch.ninecode.cim

import com.esotericsoftware.kryo.Kryo
import org.apache.spark.serializer.KryoRegistrator

import ch.ninecode.model.BasicElement
import ch.ninecode.model.BasicElementSerializer
import ch.ninecode.model.Unknown
import ch.ninecode.model.UnknownSerializer

class CIMRegistrator extends KryoRegistrator
{
    override def registerClasses (kryo: Kryo)
    {
        kryo.register (classOf [BasicElement], BasicElementSerializer)
        kryo.register (classOf [Unknown], UnknownSerializer)
        val infos = new CHIM ("").classes
        infos.foreach (info => kryo.register (info.subsetter.runtime_class, info.serializer))
    }
}