Elias Gamma Encoding in Rust

gamma-encoder.rs

// Wikipedia: https://en.wikipedia.org/wiki/Elias_gamma_coding

use std::vec;
use std::env;

fn main() {
    let args: Vec<String> = env::args().collect();
    if args.len() < 2 {
        panic!("Please provide gamma-encoded binary string as input");
    }
    let input = &args[1];
    let chars = input.chars();
    let mut selector = 0;
    let mut nums: Vec<usize> = vec![];
    let mut it = chars.into_iter().peekable();
    while it.peek().is_some() {
        if let Some(mut chr) = it.next() {
            if chr == '0' {
                selector += 1;
            } else {
                let mut num: usize = 0;
                while selector >= 0 {
                    if chr == '1' {
                        num += 1 << selector;
                    }
                    selector -= 1;
                    if selector >= 0 {
                        chr = it.next().expect("Unexpected end of string");
                    }
                }
                selector = 0;
                nums.push(num);
            }
        }
    }
    println!("Input `{}`:", input);
    for num in nums {
        println!("- {}", num);
    }
}