1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
|
pub struct Table {
num_cols: usize,
titles: Vec<String>,
rows: Vec<Vec<String>>
}
impl Table {
pub fn new(titles: Vec<String>) -> Table {
let n = titles.len();
return Table {
num_cols: n,
titles: titles,
rows: Vec::new()
};
}
pub fn get_column_num(&self) -> usize {
return self.num_cols;
}
pub fn get_rows_number(&self) -> usize {
return self.rows.len();
}
pub fn get_mut_row(&mut self, row: usize) -> &mut Vec<String> {
return &mut self.rows[row];
}
pub fn get_row(&self, row: usize) -> &Vec<String> {
return &self.rows[row];
}
pub fn add_row(&mut self, row: Vec<String>) -> Result<&mut Vec<String>, &str> {
if row.len() != self.num_cols {
return Err("Row does not have the proper number of column");
}
self.rows.push(row);
let l = self.rows.len()-1;
return Ok(self.get_mut_row(l));
}
pub fn add_empty_row(&mut self) -> Result<&mut Vec<String>, &str> {
let n = self.num_cols;
return Ok(try!(self.add_row(vec!["".to_string(); n])));
}
pub fn set_element(&mut self, element: String, column: usize, row: usize) -> Result<(), &str> {
if column >= self.num_cols {
return Err("Column index is higher than expected");
}
let rowline: &mut Vec<String>;
if row > self.rows.len() {
rowline = try!(self.add_empty_row());
}
else {
rowline = self.get_mut_row(row);
}
rowline[column] = element;
return Ok(());
}
pub fn remove_row(&mut self, row: usize) {
self.rows.remove(row);
}
fn get_col_width(&self, col_idx: usize) -> Result<usize, &str> {
if col_idx >= self.num_cols {
return Err("Column index is too high");
}
let mut width = self.titles[col_idx].len();
for r in &self.rows {
let l = r[col_idx].len();
if l > width {
width = l;
}
}
return Ok(width);
}
fn print_line_separator(&self, col_width: &[usize]) {
print!("+");
for i in 0..self.num_cols {
for _ in 0..(col_width[i] + 2) {
print!("-");
}
print!("+");
}
println!("");
}
fn print_line(&self, line: &[String], col_width: &[usize]) {
print!("|");
for i in 0..self.num_cols {
print!(" {} ", line[i]);
for _ in 0..(col_width[i] - line[i].len()) {
print!(" ");
}
print!("|");
}
println!("");
}
pub fn print(&self) {
let mut col_width = vec![0usize; self.num_cols];
for i in 0..self.num_cols {
col_width[i] = self.get_col_width(i).unwrap();
}
self.print_line_separator(&col_width);
self.print_line(&self.titles, &col_width);
self.print_line_separator(&col_width);
for r in &self.rows {
self.print_line(r, &col_width);
self.print_line_separator(&col_width);
}
}
}
|
