Add build script for cbor file generation for color names dataset

blenderskool · Sep 3, 2020 · 3c3e398 · 3c3e398
1 parent a0a6fc9
commit 3c3e398
Show file tree

Hide file tree

Showing 5 changed files with 43 additions and 3 deletions.
diff --git a/.gitignore b/.gitignore
@@ -1,3 +1,4 @@
 /target
 **/*.rs.bk
-pkg
+pkg
+*.cbor
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/Cargo.toml b/Cargo.toml
@@ -16,6 +16,12 @@ termion = "1.5.5"
 prettytable-rs = "0.8.0"
 pigmnts = { path = "lib" }
 reqwest = { version = "0.10", features = ["blocking"] }
+serde_cbor = "0.11.1"
+lazy_static = "^1.4.0"
+
+[build-dependencies]
+serde_json = "^1.0.52"
+serde_cbor = "^0.11.1"
 
 [[bin]]
 name = "pigmnts"

diff --git a/build.rs b/build.rs
@@ -0,0 +1,13 @@
+extern crate serde_cbor;
+extern crate serde_json;
+
+use std::fs::File;
+
+fn main() {
+  // This dataset has been taken from https://github.com/meodai/color-names
+  let color_names_file = File::open("./data/colornames.bestof.min.json").unwrap();
+  let color_names: serde_json::Value = serde_json::from_reader(color_names_file).unwrap();
+
+  let dest_file = File::create("./data/colornames.cbor").unwrap();
+  serde_cbor::to_writer(dest_file, &color_names).unwrap();
+}
diff --git a/data/colornames.bestof.min.json b/data/colornames.bestof.min.json