sanskritnlp.transliteration.transliterator.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of indic-transliteration Show documentation

A collection of scala and java classes for some basic character level processing for the Sanskrit and other Indic (kannada, telugu, etc..) languages, contributed by the open source sanskrit-coders projects and friends. Some notable facilities: * Transliterate text from one script or encoding scheme to another. * Some grammar simulation. Examples: see https://github.com/sanskrit-coders/indic-transliteration Contributions and suggestions are invited at https://github.com/sanskrit-coders/indic-transliteration . (Sister projects there may also be of interest.)

There is a newer version: 1.6

Show newest version

package sanskritnlp.transliteration

import scala.collection.SetLike

/**
  * General transliteration utilities.
  * Created by vvasuki on 2/21/16.
  */
object transliterator {
  val scriptDevanAgarI = "dev"
  val scriptUnknown = null

  def scriptFromString(schemeName: String) : Option[RomanScript] = {
    schemeName match {
      case "hk" => {
        return Some(harvardKyoto)
      }
      case "iast" => {
        return Some(iast)
      }
      case "as" => {
        return Some(as)
      }
      case "slp" => {
        return Some(slp)
      }
      case "optitrans" => {
        return Some(optitrans)
      }
      case _ => {return None}
    }
  }

  // Assumes that words are space separable.
  def transliterateWordsIfIndic(in_str: String, wordSet: Set[String], sourceScheme: String, destScheme: String): String = {
    val words = in_str.split("\\s+")
    words.map(word => {
      if (wordSet.contains(word)) {
        transliterate(word, sourceScheme, destScheme)
      } else if(word.endsWith("s")) {
        if (wordSet.contains(word.dropRight(1))) {
          transliterate(word, sourceScheme, destScheme) + "-s"
        }
      } else {
        word
      }
    }).mkString(" ")

  }

  // Transliterate among roman schemes + devanAgarI via devanAgarI.
  def transliterate(in_str: String, sourceScheme: String, destScheme: String): String = {
    // println("input string: " + in_str)
    var schemeOpt = scriptFromString(sourceScheme)
    var devanAgarIout = in_str
    if (schemeOpt.isDefined) {
      val return_opt = schemeOpt.get.toDevanagari(in_str)
      // println("return_opt: " + return_opt)
      if (return_opt.isEmpty) {
        throw new IllegalArgumentException("Could not transliterate " + in_str)
        return in_str
      } else {
        devanAgarIout = return_opt.get
      }
    }
    schemeOpt = scriptFromString(destScheme)
    if (schemeOpt.isDefined) {
      return schemeOpt.get.fromDevanagari(devanAgarIout)
    } else {
      if (destScheme == scriptDevanAgarI) {
        return devanAgarIout
      } else {
        throw new IllegalArgumentException("Could not transliterate " + in_str)
      }
    }
  }
}