hardcode HeatMap zmin/zmax/zmid type to f64

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

Author: andrei-ng

examples/custom_controls/src/main.rs

@@ -44,7 +44,7 @@ fn bar_plot_with_dropdown_for_different_data() {
 /// Display a heat map, with buttons to allow for toggling of different
 /// colorscales.
 fn heat_map_with_modifiable_colorscale() {
-    type HeatMapType = HeatMap<f64, f64, f64>;
+    type HeatMapType = HeatMap<f64, f64, Vec<f64>>;
     let gauss = |v: i32| (-v as f64 * v as f64 / 200.0).exp();
     let z = (-30..30)
         .map(|x| (-30..30).map(|y| gauss(x) * gauss(y)).collect_vec())

examples/scientific_charts/src/main.rs

@@ -102,7 +102,7 @@ fn customizing_spacing_between_x_and_y_ticks() {
 // Heatmaps
 fn basic_heat_map() {
     let z = vec![vec![1, 20, 30], vec![20, 1, 60], vec![30, 60, 1]];
-    let trace = HeatMap::new_z(z);
+    let trace = HeatMap::new_z(z).zmin(1.0).zmax(60.0);
     let mut plot = Plot::new();
     plot.add_trace(trace);
 

plotly/src/traces/heat_map.rs

@@ -106,21 +106,21 @@ where
     y_axis: Option<String>,
     #[serde(rename = "ycalendar")]
     y_calendar: Option<Calendar>,
-    z: Option<Vec<Vec<Z>>>,
+    z: Option<Vec<Z>>,
     zauto: Option<bool>,
     #[serde(rename = "zhoverformat")]
     zhover_format: Option<String>,
-    zmax: Option<Z>,
-    zmid: Option<Z>,
-    zmin: Option<Z>,
+    zmax: Option<f64>,
+    zmid: Option<f64>,
+    zmin: Option<f64>,
     zsmooth: Option<Smoothing>,
 }
 
 impl<Z> HeatMap<f64, f64, Z>
 where
     Z: Serialize + Clone,
 {
-    pub fn new_z(z: Vec<Vec<Z>>) -> Box<Self> {
+    pub fn new_z(z: Vec<Z>) -> Box<Self> {
         Box::new(Self {
             z: Some(z),
             ..Default::default()
@@ -134,7 +134,7 @@ where
     Y: Serialize + Clone,
     Z: Serialize + Clone,
 {
-    pub fn new(x: Vec<X>, y: Vec<Y>, z: Vec<Vec<Z>>) -> Box<Self> {
+    pub fn new(x: Vec<X>, y: Vec<Y>, z: Vec<Z>) -> Box<Self> {
         Box::new(Self {
             x: Some(x),
             y: Some(y),