made into library, better morse formatting

src/lib.rs

@@ -0,0 +1,102 @@
+use hound::{self, WavWriter};
+use std::collections::HashMap;
+use std::f32::consts::PI;
+use std::{fs, i16, io};
+
+const SAMPLE_RATE: f32 = 44100.0;
+const MORSE_FREQ: f32 = 600.0;
+const UNIT_TIME: f32 = 100.0 / 1000.0; // IN MS
+
+const SPEC: hound::WavSpec = hound::WavSpec {
+    channels: 1,
+    sample_rate: 44100,
+    bits_per_sample: 16,
+    sample_format: hound::SampleFormat::Int,
+};
+
+fn write_freq(writer: &mut WavWriter<io::BufWriter<fs::File>>, freq: f32, duration: f32) {
+    let sample_count = (duration * SAMPLE_RATE).ceil() as i32;
+    for t in (0..sample_count).map(|x| x as f32 / SAMPLE_RATE) {
+        let sample = (t * freq * 2.0 * PI).sin();
+        let amplitude = i16::MAX as f32;
+        writer.write_sample((sample * amplitude) as i16).unwrap();
+    }
+}
+
+pub fn to_morse(text: &str) -> String {
+    let morse_map: HashMap<char, &str> = HashMap::from([
+        ('a', ".-"),
+        ('b', "-..."),
+        ('c', "-.-."),
+        ('d', "-.."),
+        ('e', "."),
+        ('f', "..-."),
+        ('g', "--."),
+        ('h', "...."),
+        ('i', ".."),
+        ('j', ".---"),
+        ('k', "-.-"),
+        ('l', ".-.."),
+        ('m', "--"),
+        ('n', "-."),
+        ('o', "---"),
+        ('p', ".--."),
+        ('q', "--.-"),
+        ('r', ".-."),
+        ('s', "..."),
+        ('t', "-"),
+        ('u', "..-"),
+        ('v', "...-"),
+        ('w', ".--"),
+        ('x', "-..-"),
+        ('y', "-.--"),
+        ('z', "--.."),
+        ('1', ".----"),
+        ('2', "..---"),
+        ('3', "...--"),
+        ('4', "....-"),
+        ('5', "....."),
+        ('6', "-...."),
+        ('7', "--..."),
+        ('8', "---.."),
+        ('9', "----."),
+        ('0', "-----"),
+    ]);
+
+    let code = text.to_lowercase();
+    let mut ret = String::new();
+
+    for c in code.chars() {
+        if c == ' ' {
+            ret += " / ";
+        } else if let Some(morse) = morse_map.get(&c) {
+            ret += morse;
+        }
+        ret += " ";
+    }
+
+    ret
+}
+
+pub fn write_morse(filename: &str, morse: &str) {
+    let mut writer = hound::WavWriter::create(filename, SPEC).unwrap();
+    for c in morse.chars() {
+        // We give a margin of 0.5 * UNIT_TIME between dots and dashes
+        // give 1.5 * UNIT_TIME between characters
+        // and give 2 * UNIT_TIME between words
+        if c == '.' {
+            write_freq(&mut writer, MORSE_FREQ, UNIT_TIME);
+            write_freq(&mut writer, 0.0, 0.5 * UNIT_TIME);
+        } else if c == '-' {
+            write_freq(&mut writer, MORSE_FREQ, 3.0 * UNIT_TIME);
+            write_freq(&mut writer, 0.0, 0.5 * UNIT_TIME);
+        } else if c == ' ' {
+            write_freq(&mut writer, 0.0, 1.5 * UNIT_TIME);
+        } else if c == '/' {
+            write_freq(&mut writer, 0.0, 2.0 * UNIT_TIME);
+        } else {
+            panic!("The morse code given is not valid!");
+        }
+    }
+    write_freq(&mut writer, 0.0, UNIT_TIME); // End padding
+}

src/main.rs

@@ -1,59 +1,7 @@
-use hound::{self, WavWriter};
+use morse_translator::{to_morse, write_morse};
-use std::f32::consts::PI;
-use std::{fs, i16, io};
-
-const SAMPLE_RATE: f32 = 44100.0;
-const MORSE_FREQ: f32 = 600.0;
-const UNIT_TIME: f32 = 100.0 / 1000.0; // IN MS
-const CHAR_TIME: f32 = 50.0 / 1000.0;
-
-const SPEC: hound::WavSpec = hound::WavSpec {
-    channels: 1,
-    sample_rate: 44100,
-    bits_per_sample: 16,
-    sample_format: hound::SampleFormat::Int,
-};
-
-fn write_freq(writer: &mut WavWriter<io::BufWriter<fs::File>>, freq: f32, duration: f32) {
-    let sample_count = (duration * SAMPLE_RATE).ceil() as i32;
-    for t in (0..sample_count).map(|x| x as f32 / SAMPLE_RATE) {
-        let sample = (t * freq * 2.0 * PI).sin();
-        let amplitude = i16::MAX as f32;
-        writer.write_sample((sample * amplitude) as i16).unwrap();
-    }
-}
-
-fn write_morse(filename: &str, morse: &str) {
-    let mut writer = hound::WavWriter::create(filename, SPEC).unwrap();
-
-    //for t in (0..sample_count).map(|x| x as f32 / SAMPLE_RATE) {
-    //    let sample = (t * MORSE_FREQ * 2.0 * PI).sin();
-    //    let amplitude = i16::MAX as f32;
-    //    writer.write_sample((sample * amplitude) as i16).unwrap();
-    //}
-
-    for c in morse.chars() {
-        // We give a margin of CHAR_TIME between dots and dashes
-        // and we give 1.5 * UNIT_TIME between characters
-        if c == '.' {
-            write_freq(&mut writer, MORSE_FREQ, UNIT_TIME);
-            write_freq(&mut writer, 0.0, CHAR_TIME);
-        } else if c == '-' {
-            write_freq(&mut writer, MORSE_FREQ, 3.0 * UNIT_TIME);
-            write_freq(&mut writer, 0.0, CHAR_TIME);
-        } else if c == ' ' {
-            write_freq(&mut writer, 0.0, 1.5 * UNIT_TIME);
-        } else {
-            panic!("The morse code given is not valid!");
-        }
-    }
-    write_freq(&mut writer, 0.0, UNIT_TIME); // End padding
-}
 
 fn main() {
-    // -.. .- -. .. ... ...
+    let text = to_morse("danish ghaisan");
 
-    write_morse("morse_danish.wav", "-.. .- -. .. ... ....");
+    write_morse("danish.wav", &text);
-    //write_freq(&mut writer, 350.0, 1.0);
-    //write_freq(&mut writer, 650.0, 1.0);
 }