sonusai 1.0.9__py3-none-any.whl → 1.0.10__py3-none-any.whl

sonusai/mixture/parse.py

@@ -0,0 +1,145 @@
+"""
+Parse config rules
+
+sai_expand(0, 5)
+sai_expand(.0, 5)
+sai_expand(0., 5)
+sai_expand(0.0, 5)
+sai_expand(0, .5)
+sai_expand(.0, .5)
+sai_expand(0., .5)
+sai_expand(0.0, .5)
+sai_expand(0, 5.)
+sai_expand(.0, 5.)
+sai_expand(0., 5.)
+sai_expand(0.0, 5.)
+sai_expand(0, 5.0)
+sai_expand(.0, 5.0)
+sai_expand(0., 5.0)
+sai_expand(0.0, 5.0)
+
+sai_rand(-1, 1)
+sai_choose(repeat=False)
+sai_choose(repeat=False, tag=tag)
+sai_sequence()
+sai_sequence(tag=tag)
+
+sai_expand(0, sai_rand(-1, 1))
+sai_expand(sai_rand(0, 4), 0)
+sai_expand(sai_rand(0, 4), sai_rand(-1, 1))
+
+sai_choose(num=1, unique=speaker_id, repeat=False)
+sai_choose(num=1, repeat=False)
+sai_choose(num=1, unique=speaker_id, repeat=True)
+sai_choose(num=1, repeat=True)
+sai_sequence(num=0)
+sai_sequence(num=0, unique=speaker_id)
+
+"""
+
+from dataclasses import dataclass
+
+
+@dataclass
+class Match:
+    group: str
+    span: tuple[int, int]
+
+    def start(self) -> int:
+        return self.span[0]
+
+    def end(self) -> int:
+        return self.span[1]
+
+
+def find_sai_expand(text: str) -> list[Match]:
+    import re
+
+    results = []
+    matches = re.finditer(r"sai_expand\(", text)
+    if matches:
+        for match in matches:
+            s = match.start()
+            e = match.end()
+            num_lparen = 1
+            while num_lparen != 0 and e < len(text):
+                if text[e] == "(":
+                    num_lparen += 1
+                elif text[e] == ")":
+                    num_lparen -= 1
+                e += 1
+            if num_lparen != 0:
+                raise ValueError(f"Unbalanced parenthesis in '{text}'")
+
+            results.append(Match(group=text[s:e], span=(s, e)))
+
+    return results
+
+
+def parse_sai_expand(text: str) -> list[str]:
+    """
+    expand_syntax ::= expand_keyword lparen expand_item next_expand_item+ rparen
+    expand_keyword ::= sai_expand
+    lparen ::= (
+    expand_item ::= expand_syntax | rand_syntax | number
+    rand_syntax ::= rand_keyword lparen rand_item comma rand_item rparen
+    rand_item ::= real number
+    comma ::= ,
+    number ::= real number or signed integer
+    next_item ::= comma item
+    rparen ::= )
+    """
+    import pyparsing as pp
+
+    lparen = pp.Literal("(")
+    rparen = pp.Literal(")")
+    comma = pp.Literal(",")
+
+    real_number = pp.pyparsing_common.real
+    signed_integer = pp.pyparsing_common.signed_integer
+    number = real_number | signed_integer
+
+    identifier = pp.Word(pp.alphanums + "_.-")
+
+    rand_literal = pp.Literal("sai_rand")
+    rand_expression = (rand_literal + lparen + number + comma + number + rparen).set_parse_action(
+        lambda tokens: "".join(map(str, tokens))
+    )
+
+    expand_literal = pp.Literal("sai_expand")
+    expand_args = pp.DelimitedList(rand_expression | identifier, min=1)
+    expand_expression = expand_literal + lparen + expand_args("args") + rparen
+
+    try:
+        result = expand_expression.parse_string(text)
+    except pp.ParseException as e:
+        raise ValueError(f"Could not parse '{text}'") from e
+
+    return list(result.args)
+
+
+def sai_expand(text: str) -> list[str]:
+    # initialize with input
+    expanded = [text]
+
+    # look for pattern
+    matches = find_sai_expand(text)
+
+    # if not found, early exit
+    if not matches:
+        return expanded
+
+    # remove entry we are expanding
+    expanded.pop()
+
+    # start with the innermost match
+    match = matches[-1]
+    prelude = text[: match.start()]
+    postlude = text[match.end() :]
+
+    # loop over parsed expand values
+    for value in parse_sai_expand(match.group):
+        # extend result with expand of replacement (for handling multiple expands in a single rule)
+        expanded.extend(sai_expand(prelude + value + postlude))
+
+    return expanded

{sonusai-1.0.9.dist-info → sonusai-1.0.10.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: sonusai
-Version: 1.0.9
+Version: 1.0.10
 Summary: Framework for building deep neural network models for sound, speech, and voice AI
 Home-page: https://aaware.com
 License: GPL-3.0-only

{sonusai-1.0.9.dist-info → sonusai-1.0.10.dist-info}/RECORD

@@ -63,6 +63,7 @@ sonusai/mixture/ir_effects.py,sha256=PqiqD4PS42-7kD6ESnsZi2a3tnKCFa4E0xqUujRBvGg
 sonusai/mixture/log_duration_and_sizes.py,sha256=3ekS27IMKlnxIkQAmprzmBnzHOpRjZh3d7maL2VqWQU,927
 sonusai/mixture/mixdb.py,sha256=5YI0zKisFw_B-jKpB-Y1EYlJ8pHQDvOQLs9LEe0gT1w,84905
 sonusai/mixture/pad_audio.py,sha256=KNxVQAejA0hblLOnMJgLS6lFaeE0n3tWQ5rclaHBnIY,1015
+sonusai/mixture/parse.py,sha256=nqhjuR-J7_3wlGhVitYFvQwLJ1sclU8WZrVF0SyW2Cw,3700
 sonusai/mixture/resample.py,sha256=jXqH6FrZ0mlhQ07XqPx88TT9elu3HHVLw7Q0a7Lh5M4,221
 sonusai/mixture/sox_effects.py,sha256=tndS9qrh3eJOTUPrufyWHCt3UqjbPuh81I4Lo4MNmDg,5328
 sonusai/mixture/sox_help.py,sha256=DOTMBZfYQs3bXrqra25iH92lJFhqcGe9M1am4OrKkBY,29855
@@ -133,7 +134,7 @@ sonusai/utils/tokenized_shell_vars.py,sha256=EDrrAgz5lJ0RBAjLcTJt1MeyjhbNZiqXkym
 sonusai/utils/write_audio.py,sha256=IHzrJoFtFcea_J6wo6QSiojRkgnNOzAEcg-z0rFV7nU,810
 sonusai/utils/yes_or_no.py,sha256=0h1okjXmDNbJp7rZJFR2V-HFU1GJDm3YFTUVmYExkOU,263
 sonusai/vars.py,sha256=m8pdgfR4A6A9TCGf_rok6jPAT5BgrEsYXTSISIh1nrI,1163
-sonusai-1.0.9.dist-info/METADATA,sha256=PnNdFaS3cIoP8dZEDP6bKK-QoR5W_rV-z5OwEhxhlQs,2694
-sonusai-1.0.9.dist-info/WHEEL,sha256=RaoafKOydTQ7I_I3JTrPCg6kUmTgtm4BornzOqyEfJ8,88
-sonusai-1.0.9.dist-info/entry_points.txt,sha256=zMNjEphEPO6B3cD1GNpit7z-yA9tUU5-j3W2v-UWstU,92
-sonusai-1.0.9.dist-info/RECORD,,
+sonusai-1.0.10.dist-info/METADATA,sha256=kliBuHLQIEAUTsv9Hav0VWo1IGQxpTao5bl233yOnaQ,2695
+sonusai-1.0.10.dist-info/WHEEL,sha256=RaoafKOydTQ7I_I3JTrPCg6kUmTgtm4BornzOqyEfJ8,88
+sonusai-1.0.10.dist-info/entry_points.txt,sha256=zMNjEphEPO6B3cD1GNpit7z-yA9tUU5-j3W2v-UWstU,92
+sonusai-1.0.10.dist-info/RECORD,,