Merge pull request #344 from plotly/issue236_fixed_stem3

gilbertogalvis · web-flow · commit 3b4fbec421fd · 2021-08-17T09:12:57.000-04:00
Issue236 fixed stem3
diff --git a/plotly/plotlyfig_aux/handlegraphics/updateLineseries.m b/plotly/plotlyfig_aux/handlegraphics/updateLineseries.m
@@ -131,6 +131,9 @@ function updateLineseries(obj,plotIndex)
 
 %-------------------------------------------------------------------------%
 
+%-Fro 3D plots-%
+obj.PlotOptions.is3d = false; % by default
+
 if isfield(plot_data,'ZData')
     
     numbset = unique(plot_data.ZData);
@@ -139,8 +142,11 @@ function updateLineseries(obj,plotIndex)
         %-scatter z-%
         obj.data{plotIndex}.z = plot_data.ZData;
         
-        %overwrite type
+        %-overwrite type-%
         obj.data{plotIndex}.type = 'scatter3d';
+        
+        %-flag to manage 3d plots-%
+        obj.PlotOptions.is3d = true;
     end
 end
 
diff --git a/plotly/plotlyfig_aux/handlegraphics/updateStem.m b/plotly/plotlyfig_aux/handlegraphics/updateStem.m
@@ -5,6 +5,7 @@
 %-update line-%
 updateLineseries(obj,dataIndex);
 stem_temp_data = obj.data{dataIndex};
+isstem3d = obj.PlotOptions.is3d;
 
 %------------------------------------------------------------------------%
 
@@ -17,6 +18,10 @@
 xdata_extended = zeros(1,3*length(stem_temp_data.x)); 
 ydata_extended = zeros(1,3*length(stem_temp_data.y));
 
+if isstem3d
+    zdata_extended = zeros(1,3*length(stem_temp_data.z));
+end
+
 %-format x data-%
 m = 1; 
 for n = 1:length(stem_temp_data.x)
@@ -29,12 +34,28 @@
 %-format y data-%
 m = 1; 
 for n = 1:length(stem_temp_data.y)
-    ydata_extended(m) = 0; 
+    ydata_extended(m) = 0;
+ 
+    if isstem3d
+        ydata_extended(m) = stem_temp_data.y(n); 
+    end
+    
     ydata_extended(m+1) = stem_temp_data.y(n); 
     ydata_extended(m+2) = nan; 
     m = m + 3; 
 end
 
+%-format z data-%
+if isstem3d
+    m = 1; 
+    for n = 1:length(stem_temp_data.z)
+        zdata_extended(m) = 0;
+        zdata_extended(m+1) = stem_temp_data.z(n); 
+        zdata_extended(m+2) = nan; 
+        m = m + 3; 
+    end
+end
+
 %-hide every other marker-%
 color_temp = cell(1,3*length(stem_temp_data.y));
 line_color_temp = cell(1,3*length(stem_temp_data.y));
@@ -57,6 +78,10 @@
 stem_temp_data.x = xdata_extended; 
 stem_temp_data.y = ydata_extended; 
 
+if isstem3d
+    stem_temp_data.z = zdata_extended; 
+end
+
 %------------------------------------------------------------------------%
 
 obj.data{dataIndex} = stem_temp_data;
