commonMain.com.dshatz.fuzzykat.diffutils.ratio.PartialRatio.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of fuzzyKat-jvm Show documentation

Fuzzy search for Kotlin Multiplatform

The newest version!

package com.dshatz.fuzzykat.diffutils.ratio

import com.dshatz.fuzzykat.Ratio
import com.dshatz.fuzzykat.ToStringFunction
import com.dshatz.fuzzykat.diffutils.DiffUtils
import kotlin.math.round


/**
 * Partial ratio of similarity
 */
class PartialRatio : Ratio {

    /**
     * Computes a partial ratio between the strings
     *
     * @param s1 Input string
     * @param s2 Input string
     * @return The partial ratio
     */
    override fun apply(s1: String, s2: String): Int {

        val shorter: String
        val longer: String

        if (s1.length < s2.length) {

            shorter = s1
            longer = s2

        } else {

            shorter = s2
            longer = s1

        }

        val matchingBlocks = DiffUtils.getMatchingBlocks(shorter, longer)

        val scores = ArrayList()

        for (mb in matchingBlocks) {

            val dist = mb.dpos - mb.spos

            val longStart = if (dist > 0) dist else 0
            var longEnd = longStart + shorter.length

            if (longEnd > longer.length) longEnd = longer.length

            val longSubstr = longer.substring(longStart, longEnd)

            val ratio = DiffUtils.getRatio(shorter, longSubstr)

            if (ratio > .995) {
                return 100
            } else {
                scores.add(ratio)
            }

        }

        return round(100 * scores.max()).toInt()

    }

    override fun apply(s1: String, s2: String, sp: ToStringFunction): Int {
        return apply(sp.apply(s1), sp.apply(s2))
    }


}