Added a way to cast to primitive

src/primitives/arc.rs

@@ -171,4 +171,8 @@ impl Parametric for Arc {
     fn set_data(&mut self, data: DVectorView<f64>) {
         self.data = SVector::from_row_slice(data.as_slice());
     }
+
+    fn as_primitive(self) -> super::Primitive {
+        super::Primitive::Arc(self)
+    }
 }

src/primitives/circle.rs

@@ -73,4 +73,8 @@ impl Parametric for Circle {
     fn get_gradient(&self) -> DVector<f64> {
         DVector::from_row_slice(self.gradient.as_slice())
     }
+
+    fn as_primitive(self) -> super::Primitive {
+        super::Primitive::Circle(self)
+    }
 }

src/primitives/line.rs

@@ -73,4 +73,8 @@ impl Parametric for Line {
         // empty vector
         DVector::from_row_slice(&[])
     }
+
+    fn as_primitive(self) -> super::Primitive {
+        super::Primitive::Line(self)
+    }
 }

src/primitives/mod.rs

@@ -14,4 +14,63 @@ pub trait Parametric {
     fn get_data(&self) -> DVector<f64>;
     fn set_data(&mut self, data: DVectorView<f64>);
     fn get_gradient(&self) -> DVector<f64>;
+    fn as_primitive(self) -> Primitive;
+}
+
+pub enum Primitive {
+    Point2(point2::Point2),
+    Line(line::Line),
+    Arc(arc::Arc),
+    Circle(circle::Circle),
+}
+
+impl Parametric for Primitive {
+    fn references(&self) -> Vec<Rc<RefCell<dyn Parametric>>> {
+        match self {
+            Primitive::Point2(p) => p.references(),
+            Primitive::Line(l) => l.references(),
+            Primitive::Arc(a) => a.references(),
+            Primitive::Circle(c) => c.references(),
+        }
+    }
+
+    fn zero_gradient(&mut self) {
+        match self {
+            Primitive::Point2(p) => p.zero_gradient(),
+            Primitive::Line(l) => l.zero_gradient(),
+            Primitive::Arc(a) => a.zero_gradient(),
+            Primitive::Circle(c) => c.zero_gradient(),
+        }
+    }
+
+    fn get_data(&self) -> DVector<f64> {
+        match self {
+            Primitive::Point2(p) => p.get_data(),
+            Primitive::Line(l) => l.get_data(),
+            Primitive::Arc(a) => a.get_data(),
+            Primitive::Circle(c) => c.get_data(),
+        }
+    }
+
+    fn set_data(&mut self, data: DVectorView<f64>) {
+        match self {
+            Primitive::Point2(p) => p.set_data(data),
+            Primitive::Line(l) => l.set_data(data),
+            Primitive::Arc(a) => a.set_data(data),
+            Primitive::Circle(c) => c.set_data(data),
+        }
+    }
+
+    fn get_gradient(&self) -> DVector<f64> {
+        match self {
+            Primitive::Point2(p) => p.get_gradient(),
+            Primitive::Line(l) => l.get_gradient(),
+            Primitive::Arc(a) => a.get_gradient(),
+            Primitive::Circle(c) => c.get_gradient(),
+        }
+    }
+
+    fn as_primitive(self) -> Primitive {
+        self
+    }
 }

src/primitives/point2.rs

@@ -67,4 +67,8 @@ impl Parametric for Point2 {
     fn get_gradient(&self) -> DVector<f64> {
         DVector::from_row_slice(self.gradient.as_slice())
     }
+
+    fn as_primitive(self) -> super::Primitive {
+        super::Primitive::Point2(self)
+    }
 }