diff options
| -rwxr-xr-x | gen-prime.sh | 2 | ||||
| -rw-r--r-- | model.py | 11 |
2 files changed, 12 insertions, 1 deletions
diff --git a/gen-prime.sh b/gen-prime.sh index d9fa887..3a38d46 100755 --- a/gen-prime.sh +++ b/gen-prime.sh @@ -54,7 +54,7 @@ function gen_prime_set () { # gen_prime $1 6 44100 'noise' # gen_prime $1 6 44100 'sin' 440 # gen_prime $1 6 44100 'noise' 0 0 True - gen_prime $1 6 22050 'sin' 440 0 True + gen_prime $1 6 22050 'spike' 440 0 True ./latest.pl -n 'fixed_gradient' -l $exp_name } @@ -350,11 +350,22 @@ class PrimedGenerator(Runner): x[:, i] = int((math.sin(i/44100 * primer_freq) + 1) / 2 * (q_width) + q_min) return x + def _saw(x): + primer_width = int(n_samples / 2) + for i in range(n_samples): + # x[:, i] = #int((math.sin(i/44100 * primer_freq) + 1) / 2 * (q_width) + q_min) + if (i % primer_width) == 0: + x[:, i] = q_levels + else: + x[:, i] = int(q_levels / 2) + return x + sequence_lookup = { 'zero': lambda x: x.fill_(utils.q_zero(self.model.q_levels)), 'noise': _noise, 'sin': _sin, 'saw': _saw, + 'spike': _spike, } sequences = sequence_lookup.get(primer, 'zero')(sequences) |