Dessia-tech · Tanguylo · Jun 3, 2022 · Jun 7, 2022 · Jun 7, 2022 · Jun 8, 2022
diff --git a/plot_data/core.py b/plot_data/core.py
@@ -716,6 +716,30 @@ def __init__(self, x_variable: str, y_variable: str,
         PlotDataObject.__init__(self, type_='scatterplot', name=name)
 
 
+class PieChart(PlotDataObject):
+    """
+    A class for drawing pie plots.
+
+    :param data_samples: [{'name_param_1': any,..., 'name_param_p': any},...,
+                          {'name_param_1': any,..., 'name_param_p': any}]
+    :type data_samples: List[dict]
+    :param slicing_variable: variable that you want to use to fill pie parts
+    :type slicing_variable: str
+    """
+
+    def __init__(self, slicing_variable: str, 
+                 data_samples: List[Any] = None,
+                 name: str = ''):
+
+        self.slicing_variable = slicing_variable
+        if not data_samples:
+            self.data_samples = []
+        else:
+            self.data_samples = data_samples
+
+        PlotDataObject.__init__(self, type_='piechart', name=name)
+
+
 class ScatterMatrix(PlotDataObject):
     def __init__(self, elements: List[Any] = None, axes: List[str] = None,
                  point_style: PointStyle = None, surface_style: SurfaceStyle = None,
@@ -1151,6 +1175,8 @@ def plot_canvas(plot_data_object: Subclass[PlotDataObject],
         template = templates.histogram_template
     elif plot_type == "scattermatrix":
         template = templates.scatter_matrix_template
+    elif plot_type == "piechart":
+        template = templates.piechart_template
     else:
         raise NotImplementedError('Type {} not implemented'.format(plot_type))
 

diff --git a/plot_data/templates.py b/plot_data/templates.py
@@ -225,3 +225,35 @@
   </body>
 </html>
 ''')
+
+
+piechart_template = Template('''
+<!DOCTYPE html>
+<html lang="en">
+  <head>
+    <script src=$core_path></script>
+  </head>
+    <div id="app">
+        <canvas id="$canvas_id" width="2000" height="490"
+                    style="border: 1px solid black;">
+        </canvas>
+
+        <!-- Sets the basepath for the library if not in same directory -->
+
+        <script>
+            var width = 0.95*window.innerWidth;
+            var height = Math.max(0.95*window.innerHeight, 350);
+
+            var data = $data;
+            var number_plot_data = data.length;
+
+            var plot_data = new PlotData.PlotPieChart(
+                data, width, height, true, 0, 0, $canvas_id.id
+            );
+            plot_data.define_canvas($canvas_id.id);
+            plot_data.draw_initial();
+            plot_data.mouse_interaction(false);
+        </script>
+    </div>
+</html>
+''')
diff --git a/script/multiple_plots.py b/script/multiple_plots.py
@@ -16,15 +16,15 @@
 a list of vectors (dictionaries) that are displayed
 through different representations such as parallel plots and scatter plots
 """
-elements = []
+data_samples = []
 
 nb_elements = 50
 available_colors = [colors.VIOLET, colors.BLUE, colors.GREEN, colors.RED, colors.YELLOW, colors.CYAN, colors.ROSE]
 directions = ['north', 'south', 'west', 'east']
 for i in range(nb_elements):
     random_color = available_colors[random.randint(0, len(available_colors) - 1)]
     random_direction = directions[random.randint(0, len(directions) - 1)]
-    elements.append({'x': random.uniform(0, 200),
+    data_samples.append({'x': random.uniform(0, 200),
                      'y': random.uniform(0, 100),
                      'color': random_color,
                      'direction': random_direction})
@@ -35,12 +35,14 @@
 
 """Scatterplots"""
 scatterplot1 = plot_data.Scatter(x_variable='x', y_variable='y')
+piechart1 = plot_data.PieChart(data_samples=data_samples,
+                               slicing_variable='mass')
 
 scatterplot2 = plot_data.Scatter(x_variable='y', y_variable='color',
                                  point_style=plot_data.PointStyle(shape='square'))  # optional argument that changes
                                                                                     # points' appearance
 
-scatterplot3 = plot_data.Scatter(x_variable='x', y_variable='direction')
+# scatterplot3 = plot_data.Scatter(x_variable='x', y_variable='direction')
 
 """PrimitiveGroupContainers"""
 contour = plot_data.Contour2D(plot_data_primitives=[plot_data.LineSegment2D([1, 1], [1, 2]),
@@ -66,13 +68,13 @@
 histogram = plot_data.Histogram(x_variable='x')
 
 """Creating the multiplot"""
-plots = [parallelplot1, parallelplot2, scatterplot1,
-         scatterplot2, scatterplot3, graph2d, primitive_group_container,
+plots = [piechart1, parallelplot1, parallelplot2, scatterplot1,
+         scatterplot2, parallelplot1, graph2d, primitive_group_container,
          histogram]
 
 # plots = [scatterplot1, scatterplot2]
 
-multiplot = plot_data.MultiplePlots(plots=plots, elements=elements,
+multiplot = plot_data.MultiplePlots(plots=plots, elements=data_samples,
                                     initial_view_on=True)
 
 # Display

diff --git a/script/pie_chart.py b/script/pie_chart.py
@@ -0,0 +1,35 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+"""
+Created on Fri Jun  3 16:18:31 2022
+
+@author: tanguy
+"""
+
+import plot_data
+import plot_data.colors as colors
+import random
+
+
+data_samples = []
+SHAPES = ['round', 'square', 'triangle', 'ellipse']
+COLORS = [colors.RED, colors.BLUE, colors.GREEN, colors.YELLOW, colors.ORANGE, colors.VIOLET]
+for i in range(50):
+    random_shape = SHAPES[random.randint(0, len(SHAPES) - 1)]
+    random_color = COLORS[random.randint(0, len(SHAPES) - 1)]
+    data_samples.append({'mass': random.uniform(0, 50),
+                     'length': random.uniform(0, 100),
+                     'shape': random_shape,
+                     'color': random_color
+                     })
+
+
+piechart1 = plot_data.PieChart(data_samples=data_samples,
+                               slicing_variable='mass')
+
+piechart2 = plot_data.PieChart(data_samples=data_samples,
+                               slicing_variable='length')
+
+
+plot_data.plot_canvas(plot_data_object=piechart1, debug_mode=True)
+plot_data.plot_canvas(plot_data_object=piechart2, debug_mode=True)
diff --git a/src/multiplots.ts b/src/multiplots.ts
@@ -1,7 +1,7 @@
 import {PlotData, Interactions} from './plot-data';
 import {Point2D} from './primitives';
 import { Attribute, PointFamily, check_package_version, Window, TypeOf, equals, Sort, download } from './utils';
-import { PlotContour, PlotScatter, ParallelPlot, PrimitiveGroupContainer, Histogram } from './subplots';
+import { PlotContour, PlotScatter, ParallelPlot, PrimitiveGroupContainer, Histogram, PlotPieChart } from './subplots';
 import { List, Shape, MyObject } from './toolbox';
 import { string_to_hex, string_to_rgb, rgb_to_string } from './color_conversion';
 
@@ -83,7 +83,7 @@ export class MultiplePlots {
         } else if (object_type_ === 'parallelplot') {
           this.dataObjects[i]['elements'] = elements;
           newObject = new ParallelPlot(this.dataObjects[i], this.sizes[i]['width'], this.sizes[i]['height'], buttons_ON, this.initial_coords[i][0], this.initial_coords[i][1], canvas_id, true);
-        } else if (object_type_ === 'primitivegroup') {
+         } else if (object_type_ === 'primitivegroup') {
           newObject = new PlotContour(this.dataObjects[i], this.sizes[i]['width'], this.sizes[i]['height'], buttons_ON, this.initial_coords[i][0], this.initial_coords[i][1], canvas_id, true);
         } else if (object_type_ === 'primitivegroupcontainer') {
           newObject = new PrimitiveGroupContainer(this.dataObjects[i], this.sizes[i]['width'], this.sizes[i]['height'], buttons_ON, this.initial_coords[i][0], this.initial_coords[i][1], canvas_id, true);
@@ -101,6 +101,7 @@ export class MultiplePlots {
         }
         this.initializeObjectContext(newObject);
         this.objectList.push(newObject);
+        console.log(newObject)
       }
       if (elements) {this.initialize_point_families();}
 
@@ -109,6 +110,7 @@ export class MultiplePlots {
         this.display_order.push(i);
         this.to_display_plots.push(i);
       }
+
       this.mouse_interaction();
 
       if (buttons_ON) {