4 changed files with 158 additions and 0 deletions
@ -0,0 +1,59 @@ |
|||||
|
use crate::city::House; |
||||
|
use rusqlite::{Connection, NO_PARAMS, params, Result}; |
||||
|
use std::collections::HashMap; |
||||
|
|
||||
|
pub struct LayoutDB { |
||||
|
connection: Connection, |
||||
|
layouts: Vec<SavedLayout> |
||||
|
} |
||||
|
|
||||
|
pub struct SavedLayout { |
||||
|
id: usize, |
||||
|
houses: Vec<House> |
||||
|
} |
||||
|
|
||||
|
impl LayoutDB { |
||||
|
pub fn from_file(filename: &str) -> Result<Self> { |
||||
|
let connection = Connection::open(filename)?; |
||||
|
let mut layouts: HashMap<u32, Vec<(u32, u32)>> = HashMap::new(); |
||||
|
|
||||
|
// We need to dispose the statement so we can move the connection later
|
||||
|
{ |
||||
|
let mut stmt = connection.prepare("SELECT layout_id, x, y FROM houses")?; |
||||
|
let mut rows = stmt.query(NO_PARAMS)?; |
||||
|
|
||||
|
while let Some(row) = rows.next()? { |
||||
|
let id: u32 = row.get(0)?; |
||||
|
let x: u32 = row.get(1)?; |
||||
|
let y: u32 = row.get(2)?; |
||||
|
layouts.entry(id).or_default().push((x, y)); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
let layouts = layouts.into_iter().map(|(id, xy_pairs)| |
||||
|
SavedLayout |
||||
|
{ |
||||
|
id: id as usize, |
||||
|
houses: xy_pairs.into_iter().map(|(x, y)| House { x: x as usize, y: y as usize }).collect(), |
||||
|
} |
||||
|
).collect(); |
||||
|
|
||||
|
Ok(LayoutDB { connection, layouts }) |
||||
|
} |
||||
|
|
||||
|
pub fn layouts(&self) -> &Vec<SavedLayout> { |
||||
|
&self.layouts |
||||
|
} |
||||
|
|
||||
|
pub fn add_layout(&self, houses: &Vec<House>, fully_optimized: bool) -> Result<()> { |
||||
|
self.connection.execute("INSERT INTO layouts (is_fully_optimized) VALUES (?1)", |
||||
|
params![fully_optimized])?; |
||||
|
let layout_id = self.connection.last_insert_rowid(); |
||||
|
for house in houses { |
||||
|
self.connection.execute("INSERT INTO houses (layout_id, x, y) VALUES (?1, ?2, ?3)", |
||||
|
params![layout_id, house.x as u32, house.y as u32])?; |
||||
|
} |
||||
|
|
||||
|
Ok(()) |
||||
|
} |
||||
|
} |
||||
Loading…
Reference in new issue