feat(watermarker): add support for TrendmarkBuilder to getMinimumInse…

…rtPositions (#75)
FraunhoferISST · Jun 6, 2024 · 751197b · 751197b
1 parent 8159c2d
commit 751197b
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/watermarker/src/commonMain/kotlin/fileWatermarker/TextWatermarker.kt b/watermarker/src/commonMain/kotlin/fileWatermarker/TextWatermarker.kt
@@ -12,6 +12,7 @@ import de.fraunhofer.isst.trend.watermarker.helper.toUnicodeRepresentation
 import de.fraunhofer.isst.trend.watermarker.returnTypes.Event
 import de.fraunhofer.isst.trend.watermarker.returnTypes.Result
 import de.fraunhofer.isst.trend.watermarker.returnTypes.Status
+import de.fraunhofer.isst.trend.watermarker.watermarks.TrendmarkBuilder
 import de.fraunhofer.isst.trend.watermarker.watermarks.Watermark
 import kotlin.js.JsExport
 import kotlin.js.JsName
@@ -331,6 +332,14 @@ class TextWatermarker(
     fun getMinimumInsertPositions(watermark: Watermark): Int =
         getMinimumInsertPositions(watermark.watermarkContent)
 
+    /**
+     * Counts the minimum number of insert positions needed in a text to insert the
+     * [trendmarkBuilder]
+     */
+    @JsName("getMimimumInsertPositionsTrendmarkBuilder")
+    fun getMinimumInsertPositions(trendmarkBuilder: TrendmarkBuilder): Int =
+        getMinimumInsertPositions(trendmarkBuilder.finish())
+
     /** Transforms a [watermark] into a separated watermark */
     private fun getSeparatedWatermark(watermark: List<Byte>): Sequence<Char> {
         val encodedWatermark = transcoding.encode(watermark)