From 619752373579850591b07c14b29942110e7f4937 Mon Sep 17 00:00:00 2001 From: Benjamin Calderon Date: Sun, 18 Aug 2024 22:07:17 -0400 Subject: [PATCH] add join delimiter --- src/core/operations/Ngram.mjs | 16 ++++++++++++---- 1 file changed, 12 insertions(+), 4 deletions(-) diff --git a/src/core/operations/Ngram.mjs b/src/core/operations/Ngram.mjs index 6471e4d8..269c788d 100644 --- a/src/core/operations/Ngram.mjs +++ b/src/core/operations/Ngram.mjs @@ -5,6 +5,7 @@ */ import Operation from "../Operation.mjs"; +import {JOIN_DELIM_OPTIONS} from "../lib/Delim.mjs"; /** * ngram operation @@ -29,6 +30,11 @@ class Ngram extends Operation { type: "number", value: 3 }, + { + "name": "Join delimiter", + "type": "editableOptionShort", + "value": JOIN_DELIM_OPTIONS + } ]; } @@ -38,13 +44,15 @@ class Ngram extends Operation { * @returns {string} */ run(input, args) { - const n = args[0]; + const nGramSize = args[0], + joinDelim = args[1]; + const ngrams = []; - for (let i = 0; i <= input.length - n; i++) { - ngrams.push(input.slice(i, i + n)); + for (let i = 0; i <= input.length - nGramSize; i++) { + ngrams.push(input.slice(i, i + nGramSize)); } - return ngrams.join("\n"); + return ngrams.join(joinDelim); } }