Skip to content

Commit

Permalink
Reduce number of WASM calls
Browse files Browse the repository at this point in the history
  • Loading branch information
@inthar-raven
inthar-raven committed Aug 19, 2024
1 parent b5fd6f6 commit 0df54e0
Show file tree
Hide file tree
Showing 2 changed files with 89 additions and 60 deletions.
21 changes: 11 additions & 10 deletions index.js
View file
Original file line number Diff line number Diff line change
Expand Up @@ -740,7 +740,7 @@ stack()`
const scaleTable = document.getElementById("table-scales");
const jiTuningTable = document.getElementById("table-ji-tunings");
const edoTuningTable = document.getElementById("table-ed-tunings");
const arr = wasm.sig_to_profiles(
const sigResult = wasm.sig_result(
sig,
document.getElementById("monotone-lm").checked,
document.getElementById("monotone-ms").checked,
Expand All @@ -753,12 +753,12 @@ stack()`
document.querySelector('input[name="mv-constraint"]:checked').value,
document.querySelector('input[name="mos-subst"]:checked').value,
);
const scales = arr.map((j) => j["word"]);
const guideFrames = arr.map((j) => j["structure"]);
const profiles = arr;
const scales = sigResult["profiles"].map((j) => j["word"]);
const guideFrames = sigResult["profiles"].map((j) => j["structure"]);
const profiles = sigResult["profiles"];

const jiTunings = wasm.word_to_ji_tunings(scales[0]);
const edoTunings = wasm.word_to_ed_tunings(scales[0]);
const jiTunings = sigResult["ji_tunings"];
const edoTunings = sigResult["ed_tunings"];
let letters;
if (arity === 3) {
letters = ["L", "m", "s"];
Expand Down Expand Up @@ -877,10 +877,11 @@ stack()`
const query = document.getElementById("input-word").value;
const arity = new Set(Array.from(query)).size;
statusElement.textContent = "Computing...";
const profile = wasm.word_to_profile(query);
const brightestMode = wasm.word_to_brightest(query);
const jiTunings = wasm.word_to_ji_tunings(query);
const edoTunings = wasm.word_to_ed_tunings(query);
const wordResult = wasm.word_result(query);
const profile = wordResult["profile"];
const brightestMode = wordResult["profile"]["word"];
const jiTunings = wordResult["ji_tunings"];
const edoTunings = wordResult["ed_tunings"];
document.getElementById("tables").innerHTML = `
<td>
JI tunings
Expand Down
128 changes: 78 additions & 50 deletions src/lib.rs
View file
Original file line number Diff line number Diff line change
Expand Up @@ -12,7 +12,9 @@ pub mod helpers;
pub mod primes;
pub mod words;

use itertools::Itertools;
use wasm_bindgen::prelude::*;
use words::Letter;

#[wasm_bindgen]
extern "C" {
Expand All @@ -37,15 +39,14 @@ const STEP_LETTERS: [&str; 12] = [
use std::cmp::min;
use std::collections::HashSet;

use itertools::Itertools;
use serde::Serialize;
use serde_wasm_bindgen::to_value;

use guide::guide_structures;
use guide::GuideFrame;
use interval::JiRatio;
use words::{
least_mode, maximum_variety, monotone_lm, monotone_ms, monotone_s0, CountVector, Letter,
least_mode, maximum_variety, monotone_lm, monotone_ms, monotone_s0, CountVector,
};

// for the edo search
Expand All @@ -71,17 +72,37 @@ pub struct GuideResult {
pub complexity: u8,
}

// A representation of a Scale Profile. Doesn't include tunings.
/// A representation of a Scale Profile. Doesn't include tunings.
#[derive(Debug, Serialize)]
pub struct ScaleProfile {
/// brightest word
word: String,
structure: GuideResult,
/// lowest-complexity guide frame structure provided there is one
structure: Option<GuideResult>,
/// whether scale is L=M monotone MOS
lm: bool,
/// whether scale is M=s monotone MOS
ms: bool,
/// whether sclae is s=0 monotone MOS
s0: bool,
/// maximum variety of scale
mv: u8,
}

#[derive(Debug, Serialize)]
pub struct SigResult {
profiles: Vec<ScaleProfile>,
ji_tunings: Vec<Vec<String>>,
ed_tunings: Vec<Vec<String>>,
}

#[derive(Debug, Serialize)]
pub struct WordResult {
profile: ScaleProfile,
ji_tunings: Vec<Vec<String>>,
ed_tunings: Vec<Vec<String>>,
}

fn string_to_numbers(word: &str) -> Vec<usize> {
let mut result = vec![];
let arity = word.chars().collect::<HashSet<_>>().len();
Expand All @@ -93,6 +114,16 @@ fn string_to_numbers(word: &str) -> Vec<usize> {
result
}

fn word_to_sig(input: &[usize]) -> Vec<usize> {
let mut result = vec![0, 0, 0];
for &i in input {
if i < 3 {
result[i] += 1;
}
}
result
}

fn numbers_to_string(word: &[usize]) -> String {
let mut result = "".to_string();
let arity = word.iter().collect::<HashSet<_>>().len();
Expand All @@ -104,11 +135,13 @@ fn numbers_to_string(word: &[usize]) -> String {
result
}

#[wasm_bindgen]
pub fn word_to_profile(query: String) -> Result<JsValue, JsValue> {
let word_in_numbers = string_to_numbers(&query);
let brightest = numbers_to_string(&least_mode(&word_in_numbers));
if let Some(structure) = guide_structures(&string_to_numbers(&query)).first() {
pub fn word_to_profile(query: &[usize]) -> ScaleProfile {
let brightest = numbers_to_string(&least_mode(&query));
let lm = monotone_lm(&query);
let ms = monotone_ms(&query);
let s0 = monotone_s0(&query);
let mv = maximum_variety(&query) as u8;
if let Some(structure) = guide_structures(&query).first() {
let GuideFrame {
gs,
polyoffset,
Expand Down Expand Up @@ -181,23 +214,39 @@ pub fn word_to_profile(query: String) -> Result<JsValue, JsValue> {
complexity: structure.complexity() as u8,
}
};
let lm = monotone_lm(&word_in_numbers);
let ms = monotone_ms(&word_in_numbers);
let s0 = monotone_s0(&word_in_numbers);
let mv = maximum_variety(&word_in_numbers) as u8;
Ok(to_value(&ScaleProfile {
ScaleProfile {
word: brightest,
structure,
structure: Some(structure),
lm,
ms,
s0,
mv,
})?)
}
} else {
Err(JsValue::NULL)
ScaleProfile {
word: brightest,
structure: None,
lm,
ms,
s0,
mv,
}

}
}

#[wasm_bindgen]
pub fn word_result(query: String) -> Result<JsValue, JsValue> {
let word_as_numbers = string_to_numbers(&query);
let step_sig = word_to_sig(&word_as_numbers);

Ok(to_value(&WordResult {
profile: word_to_profile(&word_as_numbers),
ji_tunings: sig_to_ji_tunings(&step_sig),
ed_tunings: sig_to_ed_tunings(&step_sig),
})?)
}

#[wasm_bindgen]
pub fn word_to_brightest(query: String) -> String {
let word_in_numbers = string_to_numbers(&query);
Expand All @@ -211,18 +260,7 @@ pub fn word_to_mv(query: String) -> u8 {
maximum_variety(&word_in_numbers) as u8
}

#[wasm_bindgen]
pub fn word_to_ji_tunings(query: String) -> Vec<JsValue> {
let word_in_numbers = string_to_numbers(&query);
let step_sig = {
let mut r = vec![0; 3];
for letter in word_in_numbers {
if letter < 3 {
r[letter] += 1;
}
}
r
};
pub fn sig_to_ji_tunings(step_sig: &[usize]) -> Vec<Vec<String>> {
let ji_tunings =
crate::ji::solve_step_sig_81_odd_limit(&step_sig, crate::monzo::Monzo::OCTAVE, false);
ji_tunings
Expand All @@ -231,23 +269,11 @@ pub fn word_to_ji_tunings(query: String) -> Vec<JsValue> {
v.into_iter()
.map(|monzo| format!("{}/{}", monzo.numer(), monzo.denom()))
.collect::<Vec<_>>()
.into()
})
.collect::<Vec<_>>()
}

#[wasm_bindgen]
pub fn word_to_ed_tunings(query: String) -> Vec<JsValue> {
let word_in_numbers = string_to_numbers(&query);
let step_sig = {
let mut r = vec![0; 3];
for letter in word_in_numbers {
if letter < 3 {
r[letter] += 1;
}
}
r
};
pub fn sig_to_ed_tunings(step_sig: &[usize]) -> Vec<Vec<String>> {
let ed_tunings = crate::equal::ed_tunings_for_ternary(
&step_sig,
crate::ji_ratio::RawJiRatio::OCTAVE,
Expand All @@ -266,14 +292,13 @@ pub fn word_to_ed_tunings(query: String) -> Vec<JsValue> {
v.iter()
.map(|i| format!("{}\\{}", i, edo))
.collect::<Vec<_>>()
.into()
})
.collect::<Vec<_>>()
}

#[wasm_bindgen]
#[allow(clippy::too_many_arguments)]
pub fn sig_to_profiles(
pub fn sig_result(
query: Vec<u8>,
lm: bool,
ms: bool,
Expand All @@ -285,7 +310,7 @@ pub fn sig_to_profiles(
mv: u8,
mv_constraint: String,
mos_subst: String,
) -> Vec<JsValue> {
) -> Result<JsValue, JsValue> {
let step_sig = query;
let filtering_cond = |scale: &[Letter]| {
(!lm || monotone_lm(scale))
Expand Down Expand Up @@ -341,9 +366,12 @@ pub fn sig_to_profiles(
}
})
.collect::<Vec<_>>();
let mut result = vec![];
for scale in scales {
result.push(word_to_profile(numbers_to_string(&scale)).unwrap_or(JsValue::NULL));
}
result
Ok(to_value(&SigResult {
profiles: scales
.iter()
.map(|scale| word_to_profile(scale))
.collect(),
ji_tunings: sig_to_ji_tunings(&step_sig),
ed_tunings: sig_to_ed_tunings(&step_sig),
})?)
}

0 comments on commit 0df54e0

Please sign in to comment.