src/bin/encode_vector.rs

   1 use stud_rust_base::{io::*, cli::CliErr};
   2 use std::{env, error::Error};
   3
   4 fn main() -> Result<(), Box<dyn Error>> {
   5     match &env::args().skip(1).collect::<Vec<String>>()[..] {
   6         [data_type, output] => {
   7             match data_type.as_ref() {
   8                 "i8" => { parse_input::<i8>()?.write_to(output)?; Ok(()) },
   9                 "u8" => { parse_input::<u8>()?.write_to(output)?; Ok(()) },
  10                 "i16" => { parse_input::<i16>()?.write_to(output)?; Ok(()) },
  11                 "u16" => { parse_input::<u16>()?.write_to(output)?; Ok(()) },
  12                 "i32" => { parse_input::<i32>()?.write_to(output)?; Ok(()) },
  13                 "u32" => { parse_input::<u32>()?.write_to(output)?; Ok(()) },
  14                 "i64" => { parse_input::<i64>()?.write_to(output)?; Ok(()) },
  15                 "u64" => { parse_input::<u64>()?.write_to(output)?; Ok(()) },
  16                 "f32" => { parse_input::<f32>()?.write_to(output)?; Ok(()) },
  17                 "f64" => { parse_input::<f64>()?.write_to(output)?; Ok(()) },
  18                 "int8" => { parse_input::<i8>()?.write_to(output)?; Ok(()) },
  19                 "uint8" => { parse_input::<u8>()?.write_to(output)?; Ok(()) },
  20                 "int16" => { parse_input::<i16>()?.write_to(output)?; Ok(()) },
  21                 "uint16" => { parse_input::<u16>()?.write_to(output)?; Ok(()) },
  22                 "int32" => { parse_input::<i32>()?.write_to(output)?; Ok(()) },
  23                 "uint32" => { parse_input::<u32>()?.write_to(output)?; Ok(()) },
  24                 "int64" => { parse_input::<i64>()?.write_to(output)?; Ok(()) },
  25                 "uint64" => { parse_input::<u64>()?.write_to(output)?; Ok(()) },
  26                 "float32" => { parse_input::<f32>()?.write_to(output)?; Ok(()) },
  27                 "float64" => { parse_input::<f64>()?.write_to(output)?; Ok(()) },
  28                 _ => {
  29                     print_usage();
  30                     Err(Box::new(CliErr("Invalid data type")))
  31                 }
  32             }
  33         },
  34         _ => {
  35             print_usage();
  36             Err(Box::new(CliErr("Invalid arguments")))
  37         },
  38     }
  39 }
  40
  41 fn print_usage() {
  42     eprintln!("Usage: encode_vector data_type output_vector_file
  43
  44 Reads textual data from the standard input and writes it in a binary format to output_vector_file. The input data should be one data element per line. The data is only written once an end of file is encountered on the input. data_type can be one of
  45 * i8
  46 * u8
  47 * i16
  48 * u16
  49 * i32
  50 * u32
  51 * i64
  52 * u64
  53 * f32
  54 * f64
  55
  56 ");
  57 }
  58
  59 use std::str::FromStr;
  60
  61 fn parse_input<T>() -> Result<Vec<T>, Box<dyn Error>> where
  62     T: FromStr,
  63     <T as FromStr>::Err: Error + 'static
  64 {
  65     use std::io::{BufRead, stdin};
  66
  67     let mut values = Vec::new();
  68
  69     let stdin = stdin();
  70     for line in stdin.lock().lines() {
  71         values.push(line.unwrap().parse::<T>()?)
  72     }
  73
  74     Ok(values)
  75 }