Extract category definitions into individual functions

1 files changed, 14 insertions, 0 deletions

diff --git a/library/src/visualize/mod.rs b/library/src/visualize/mod.rs
index 4cf5d04f..3adf4e77 100644
--- a/library/src/visualize/mod.rs
+++ b/library/src/visualize/mod.rs
@@ -11,3 +11,17 @@ pub use self::line::*;
 pub use self::path::*;
 pub use self::polygon::*;
 pub use self::shape::*;
+
+use crate::prelude::*;
+
+/// Hook up all visualize definitions.
+pub(super) fn define(global: &mut Scope) {
+    global.define("image", ImageElem::func());
+    global.define("line", LineElem::func());
+    global.define("rect", RectElem::func());
+    global.define("square", SquareElem::func());
+    global.define("ellipse", EllipseElem::func());
+    global.define("circle", CircleElem::func());
+    global.define("polygon", PolygonElem::func());
+    global.define("path", PathElem::func());
+}