refactor

Signed-off-by: Andrei Gherghescu <[email protected]>

Author: andrei-ng

plotly/src/layout/layout.rs

@@ -1,4 +1,4 @@
-use plotly_derive::generate_layout_structs;
+use plotly_derive::layout_structs;
 use plotly_derive::FieldSetter;
 use serde::Serialize;
 use std::borrow::Cow;
@@ -37,7 +37,7 @@ impl Into<Cow<'static, Template>> for &'static Template {
     }
 }
 
-#[generate_layout_structs]
+#[layout_structs]
 pub struct LayoutFields {
     title: Option<Title>,
     #[serde(rename = "showlegend")]

plotly/src/layout/mod.rs

@@ -36,9 +36,11 @@ pub use self::layout::{Layout, LayoutTemplate, Template};
 pub use self::legend::{Legend, TraceOrder};
 pub use self::mapbox::{Mapbox, MapboxStyle};
 pub use self::modes::{
-    BarMode, BarNorm, BoxMode, ClickMode, UniformTextMode, ViolinMode, WaterfallMode,
+    AspectMode, BarMode, BarNorm, BoxMode, ClickMode, UniformTextMode, ViolinMode, WaterfallMode,
+};
+pub use self::scene::{
+    Camera, CameraCenter, DragMode, DragMode3D, HoverMode, LayoutScene, Projection, ProjectionType,
 };
-pub use self::scene::{DragMode, DragMode3D, HoverMode, LayoutScene, Projection, ProjectionType};
 pub use self::shape::{
     ActiveShape, DrawDirection, FillRule, NewShape, Shape, ShapeLayer, ShapeLine, ShapeSizeMode,
     ShapeType,

plotly/src/layout/modes.rs

@@ -1,5 +1,27 @@
 use serde::{Serialize, Serializer};
 
+#[derive(Serialize, Debug, Clone)]
+/// If "cube", this scene's axes are drawn as a cube, regardless of the axes'
+/// ranges. If "data", this scene's axes are drawn in proportion with the axes'
+/// ranges. If "manual", this scene's axes are drawn in proportion with the
+/// input of "aspectratio" (the default behavior if "aspectratio" is provided).
+/// If "auto", this scene's axes are drawn using the results of "data" except
+/// when one axis is more than four times the size of the two others, where in
+/// that case the results of "cube" are used.
+/// Default: "auto"
+#[derive(Default)]
+pub enum AspectMode {
+    #[serde(rename = "auto")]
+    #[default]
+    Auto,
+    #[serde(rename = "cube")]
+    Cube,
+    #[serde(rename = "data")]
+    Data,
+    #[serde(rename = "manual")]
+    Manual,
+}
+
 #[derive(Serialize, Debug, Clone)]
 #[serde(rename_all = "lowercase")]
 pub enum BoxMode {
@@ -123,4 +145,19 @@ mod tests {
         assert_eq!(to_value(WaterfallMode::Group).unwrap(), json!("group"));
         assert_eq!(to_value(WaterfallMode::Overlay).unwrap(), json!("overlay"));
     }
+
+    #[test]
+    fn serialize_aspect_mode() {
+        let aspect_mode = AspectMode::default();
+
+        assert_eq!(to_value(aspect_mode).unwrap(), json!("auto"));
+
+        let aspect_mode = AspectMode::Data;
+
+        assert_eq!(to_value(aspect_mode).unwrap(), json!("data"));
+
+        let aspect_mode = AspectMode::Cube;
+
+        assert_eq!(to_value(aspect_mode).unwrap(), json!("cube"));
+    }
 }

plotly/src/layout/scene.rs

@@ -2,29 +2,7 @@ use plotly_derive::FieldSetter;
 use serde::Serialize;
 
 use crate::color::Color;
-use crate::layout::{Annotation, Axis};
-
-#[derive(Serialize, Debug, Clone)]
-/// If "cube", this scene's axes are drawn as a cube, regardless of the axes'
-/// ranges. If "data", this scene's axes are drawn in proportion with the axes'
-/// ranges. If "manual", this scene's axes are drawn in proportion with the
-/// input of "aspectratio" (the default behavior if "aspectratio" is provided).
-/// If "auto", this scene's axes are drawn using the results of "data" except
-/// when one axis is more than four times the size of the two others, where in
-/// that case the results of "cube" are used.
-/// Default: "auto"
-#[derive(Default)]
-pub enum AspectMode {
-    #[serde(rename = "auto")]
-    #[default]
-    Auto,
-    #[serde(rename = "cube")]
-    Cube,
-    #[serde(rename = "data")]
-    Data,
-    #[serde(rename = "manual")]
-    Manual,
-}
+use crate::layout::{Annotation, AspectMode, Axis};
 
 #[serde_with::skip_serializing_none]
 #[derive(Serialize, Debug, Clone, FieldSetter)]
@@ -494,21 +472,6 @@ mod tests {
         assert_eq!(to_value(camera_center).unwrap(), expected);
     }
 
-    #[test]
-    fn serialize_aspect_mode() {
-        let aspect_mode = AspectMode::default();
-
-        assert_eq!(to_value(aspect_mode).unwrap(), json!("auto"));
-
-        let aspect_mode = AspectMode::Data;
-
-        assert_eq!(to_value(aspect_mode).unwrap(), json!("data"));
-
-        let aspect_mode = AspectMode::Cube;
-
-        assert_eq!(to_value(aspect_mode).unwrap(), json!("cube"));
-    }
-
     #[test]
     fn serialize_up() {
         let up = Up::new();

plotly_derive/src/lib.rs

@@ -14,7 +14,7 @@ pub fn field_setter(input: proc_macro::TokenStream) -> proc_macro::TokenStream {
 }
 
 #[proc_macro_attribute]
-pub fn generate_layout_structs(_attr: TokenStream, item: TokenStream) -> TokenStream {
+pub fn layout_structs(_attr: TokenStream, item: TokenStream) -> TokenStream {
     let input = parse_macro_input!(item as ItemStruct);
     let vis = &input.vis;
     let fields = &input.fields;