com.mayabot.mynlp.fasttext.IOUtils.kt Maven / Gradle / Ivy

Go to download

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

Java(kotlin) implementation of the Fasebook 's FastText

The newest version!

package com.mayabot.mynlp.fasttext

import com.mayabot.blas.ByteUtils
import java.io.IOException
import java.nio.ByteBuffer
import java.nio.channels.FileChannel


fun FileChannel.writeBoolean(value: Boolean) {
    this.write(ByteBuffer.allocate(2).put(if(value) 1.toByte() else 0.toByte()).apply { flip() })
}
fun FileChannel.writeInt(value: Int) {
    this.write(ByteBuffer.allocate(4).putInt(value).apply { flip() })
}
fun FileChannel.writeLong(value: Long) {
    this.write(ByteBuffer.allocate(8).putLong(value).apply { flip() })
}

fun FileChannel.writeShortArray(value: ShortArray) {
    val b = ByteBuffer.allocate(value.size*2)
    for (i in 0 until value.size) {
        b.put(ByteUtils.short2Byte(value[i]))
    }
    b.flip()
    while (b.hasRemaining()) {
        write(b)
    }
}

fun FileChannel.writeDouble(value: Double) {
    this.write(ByteBuffer.allocate(8).putDouble(value).apply { flip() })
}


const val byteZero = 0.toByte()



@Throws(IOException::class)
fun ByteBuffer.writeUTF(string: String){
    string.toByteArray().forEach {
        this.put(it)
    }

    this.put(com.mayabot.mynlp.fasttext.byteZero)
}


fun ByteBuffer.writeFloatArray(source:FloatArray){
    for (i in 0 until source.size) {
        this.putFloat(source[i])
    }
}