jsongrapher 4.2__py3-none-any.whl → 4.4__py3-none-any.whl

JSONGrapher/JSONRecordCreator.py

@@ -610,13 +610,13 @@ class JSONGrapherRecord:
             "comments": comments,  # Top-level comments
             "jsongrapher": "To plot this file, go to www.jsongrapher.com and drag this file into your browser, or use the python version of JSONGrapher. File created with python Version " + JSONGrapher.version.__version__,
             "datatype": datatype,  # Top-level datatype (datatype)
-            "data": data_objects_list if data_objects_list else [],  # Data series list
             "layout": layout if layout else {
                 "title": {"text": graph_title},
                 "xaxis": {"title": {"text": x_axis_label_including_units}},
                 "yaxis": {"title": {"text": y_axis_label_including_units}}
-                   }
-                }
+                   },
+            "data": data_objects_list if data_objects_list else []  # Data series list                
+            }
             )
 
         if plot_style !="":
@@ -1940,7 +1940,8 @@ def apply_plot_style_to_plotly_dict(fig_dict, plot_style=None):
     if str(plot_style["layout_style"]).lower() != 'none': #take no action if received "None" or NoneType
         if plot_style["layout_style"] == '': #in this case, we're going to use the default.
             plot_style["layout_style"] = 'default'
-            print("Warning: No layout_style provided and 'z' field found in first data series. For 'bubble' plots, it is recommended to set layout_style to 'default'. For 'mesh3d' graphs and 'scatter3d' graphs, it is recommended to set layout_style to 'default3d'. Set layout_style to 'none' or another layout_style to avoid this warning.")
+            if "z" in fig_dict["data"][0]:
+                print("Warning: No layout_style provided and 'z' field found in first data series. For 'bubble2d' plots, it is recommended to set layout_style to 'default'. For 'mesh3d' graphs and 'scatter3d' graphs, it is recommended to set layout_style to 'default3d'. Set layout_style to 'none' or another layout_style to avoid this warning.")
         fig_dict = remove_layout_style_from_plotly_dict(fig_dict=fig_dict)
         fig_dict = apply_layout_style_to_plotly_dict(fig_dict=fig_dict, layout_style_to_apply=plot_style["layout_style"])
     #Code logic for trace_styles_collection style.
@@ -2219,16 +2220,19 @@ def apply_trace_style_to_single_data_series(data_series, trace_styles_collection
     if isinstance(trace_style, str): #check if it is a string type.
         if "__" in trace_style:
             trace_style, colorscale = trace_style.split("__")
+        if ("bubble" in trace_style) and ("bubble3d" not in trace_style) and ("bubble2d" not in trace_style):
+            trace_style = trace_style.replace("bubble", "bubble2d")
 
     colorscale_structure = "" #initialize this variable for use later. It tells us which fields to put the colorscale related values in. This should be done before regular trace_style fields are applied.
     #3D and bubble plots will have a colorscale by default.
-    if trace_style == "bubble": #for bubble trace styles, we need to move the z values into the marker size. We also need to do this before the styles_dict collection is accessed, since then the trace_style becomes a dictionary.
-        data_series = prepare_bubble_sizes(data_series)
-        colorscale_structure = "bubble"
-    elif trace_style == "mesh3d": #for bubble trace styles, we need to move the z values into the marker size. We also need to do this before the styles_dict collection is accessed, since then the trace_style becomes a dictionary.
-        colorscale_structure = "mesh3d"
-    elif trace_style == "scatter3d": #for bubble trace styles, we need to move the z values into the marker size. We also need to do this before the styles_dict collection is accessed, since then the trace_style becomes a dictionary.
-        colorscale_structure = "scatter3d"
+    if isinstance(trace_style,str):
+        if "bubble" in trace_style.lower(): #for bubble trace styles (both 2D and 3D), we need to prepare the bubble sizes. We also need to do this before the styles_dict collection is accessed, since then the trace_style becomes a dictionary.
+            data_series = prepare_bubble_sizes(data_series)
+            colorscale_structure = "bubble"
+        elif "mesh3d" in trace_style.lower(): 
+            colorscale_structure = "mesh3d"
+        elif "scatter3d" in trace_style.lower(): 
+            colorscale_structure = "scatter3d"
 
     if trace_style in styles_collection_dict:
         trace_style = styles_collection_dict.get(trace_style)
@@ -2248,7 +2252,7 @@ def apply_trace_style_to_single_data_series(data_series, trace_styles_collection
                 data_series[key] = value  # Direct assignment for non-dictionary values
 
     #Before applying colorscales, we check if we have recieved a colorscale from the user. If so, we'll need to parse the trace_type to assign the colorscale structure.
-    if colorscale != "":
+    if ((colorscale_structure == "") and (colorscale != "")):
         #If it is a scatter plot with markers, then the colorscale_structure will be marker. Need to check for this before the lines alone case.
         if ("markers" in data_series["mode"]) or ("markers+lines" in data_series["mode"]) or ("lines+markers" in data_series["mode"]):
             colorscale_structure = "marker"
@@ -2260,7 +2264,7 @@ def apply_trace_style_to_single_data_series(data_series, trace_styles_collection
     #Block of code to clean color values for 3D plots and 2D plots. It can't be just from the style dictionary because we need to point to data.
     def clean_color_values(list_of_values, variable_string_for_warning):
         if None in list_of_values:
-            print("Warning: A colorscale based on" + variable_string_for_warning + "was requested. None values were found. They are being replaced with 0 values. It is recommended to provide data without None values.")
+            print("Warning: A colorscale based on " + variable_string_for_warning + " was requested. None values were found. They are being replaced with 0 values. It is recommended to provide data without None values.")
             color_values = [0 if value is None else value for value in list_of_values]
         else:
             color_values = list_of_values
@@ -2268,6 +2272,8 @@ def apply_trace_style_to_single_data_series(data_series, trace_styles_collection
 
     if colorscale_structure == "bubble":
         #data_series["marker"]["colorscale"] = "viridis_r" #https://plotly.com/python/builtin-colorscales/
+        if colorscale != "": #this means there is a user specified colorscale.
+            data_series["marker"]["colorscale"] = colorscale
         data_series["marker"]["showscale"] = True
         if "z" in data_series:
             color_values = clean_color_values(list_of_values= data_series["z"], variable_string_for_warning="z")
@@ -2277,6 +2283,8 @@ def apply_trace_style_to_single_data_series(data_series, trace_styles_collection
             data_series["marker"]["color"] = color_values
     elif colorscale_structure == "scatter3d":
         #data_series["marker"]["colorscale"] = "viridis_r" #https://plotly.com/python/builtin-colorscales/
+        if colorscale != "": #this means there is a user specified colorscale.
+            data_series["marker"]["colorscale"] = colorscale
         data_series["marker"]["showscale"] = True
         if "z" in data_series:
             color_values = clean_color_values(list_of_values= data_series["z"], variable_string_for_warning="z")
@@ -2286,6 +2294,8 @@ def apply_trace_style_to_single_data_series(data_series, trace_styles_collection
             data_series["marker"]["color"] = color_values
     elif colorscale_structure == "mesh3d":
         #data_series["colorscale"] = "viridis_r" #https://plotly.com/python/builtin-colorscales/
+        if colorscale != "": #this means there is a user specified colorscale.
+            data_series["colorscale"] = colorscale
         data_series["showscale"] = True
         if "z" in data_series:
             color_values = clean_color_values(list_of_values= data_series["z"], variable_string_for_warning="z")
@@ -2313,10 +2323,18 @@ def prepare_bubble_sizes(data_series):
     #We also will scale them to some maximum bubble size that is specifed.
     if "marker" not in data_series:
         data_series["marker"] = {}   
-    if "z_points" in data_series:
+    if "bubble_sizes" in data_series:
+        if isinstance(data_series["bubble_sizes"], str): #if bubble sizes is a string, it must be a variable name to use for the bubble sizes.
+            bubble_sizes_variable_name = data_series["bubble_sizes"]
+            data_series["marker"]["size"] = data_series[bubble_sizes_variable_name]
+        else:
+            data_series["marker"]["size"] = data_series["bubble_sizes"]
+    elif "z_points" in data_series:
         data_series["marker"]["size"] = data_series["z_points"]
     elif "z" in data_series:
         data_series["marker"]["size"] = data_series["z"]
+    elif "y" in data_series:
+        data_series["marker"]["size"] = data_series["y"]
 
     #now need to normalize to the max value in the list.
     def normalize_to_max(starting_list):
@@ -2338,7 +2356,7 @@ def prepare_bubble_sizes(data_series):
     if "max_bubble_size" in data_series:
         max_bubble_size = data_series["max_bubble_size"]
     else:
-        max_bubble_size = 10       
+        max_bubble_size = 100       
     scaled_sizes = normalized_sizes*max_bubble_size
     data_series["marker"]["size"] = scaled_sizes.tolist() #from numpy array back to list.
     
@@ -2441,7 +2459,11 @@ def extract_trace_style_from_data_series_dict(data_series_dict, new_trace_style_
         additional_attributes_to_extract = []
 
     if new_trace_style_name=='':
-        new_trace_style_name = data_series_dict.get("trace_style", "") #keep blank if not present.
+        #Check if there is a current trace style that is a string, and use that for the name if present.
+        current_trace_style = data_series_dict.get("trace_style", "")
+        if isinstance(current_trace_style, str):
+           new_trace_style_name = current_trace_style
+    #if there is still no new_trace_style_name, we will name it 'custom'
     if new_trace_style_name=='':
         new_trace_style_name = "custom"
 
@@ -3101,11 +3123,10 @@ def convert_to_3d_layout(layout):
     new_layout = copy.deepcopy(layout)
 
     # Add the axis fields inside `scene` first
-    new_layout["scene"] = {
-        "xaxis": layout.get("xaxis", {}),
-        "yaxis": layout.get("yaxis", {}),
-        "zaxis": layout.get("zaxis", {})
-    }
+    scene = new_layout.setdefault("scene", {}) #Create a new dict if not present, otherwise use existing one.
+    scene["xaxis"] = layout.get("xaxis", {})
+    scene["yaxis"] = layout.get("yaxis", {})
+    scene["zaxis"] = layout.get("zaxis", {})
 
     # Remove the original axis fields from the top-level layout
     new_layout.pop("xaxis", None)
@@ -3120,19 +3141,40 @@ def remove_bubble_fields(fig_dict):
     #This code will modify the data_series inside the fig_dict, directly.
     bubble_found = False #initialize with false case.
     for data_series in fig_dict["data"]:
-        if "trace_style" in data_series:
-            if (data_series["trace_style"] == "bubble") or ("max_bubble_size" in data_series):
+        trace_style = data_series.get("trace_style") #trace_style will be None of the key is not present.
+
+        if isinstance(trace_style, str):
+            #If the style is just "bubble" (not explicitly 2D or 3D), default to bubble2d for backward compatibility
+            if ("bubble" in trace_style) and ("bubble3d" not in trace_style) and ("bubble2d" not in trace_style):
+                trace_style = trace_style.replace("bubble", "bubble2d") 
+            if ("bubble" in trace_style.lower()) or ("max_bubble_size" in data_series):
                 bubble_found = True
-            if bubble_found == True:
-                if "z" in data_series:
-                    data_series.pop("z")
-                if "z_points" in data_series:
-                    data_series.pop("z_points")
+            if bubble_found is True:
+                if "bubble2d" in trace_style.lower(): #pop the z variable if it's a bubble2d.
+                    if "z" in data_series:
+                        data_series.pop("z")
+                    if "z_points" in data_series:
+                        data_series.pop("z_points")
                 if "max_bubble_size" in data_series:
                     data_series.pop("max_bubble_size")
-    if bubble_found == True:
-        if "zaxis" in fig_dict["layout"]:
-            fig_dict["layout"].pop("zaxis")
+                if "bubble_sizes" in data_series:
+                    # Now, need to check if the bubble_size is a variable that should be deleted.
+                    # That will be if it is a string, and also not a standard variable. 
+                    if isinstance(data_series["bubble_sizes"], str):
+                        bubble_sizes_variable_name = data_series["bubble_sizes"]
+                        # For bubble2d case, will remove anything that is not x or y.
+                        if "bubble2d" in trace_style.lower():
+                            if bubble_sizes_variable_name not in ("x", "y"):
+                                data_series.pop(bubble_sizes_variable_name, None)
+                        if "bubble3d" in trace_style.lower():
+                            if bubble_sizes_variable_name not in ("x", "y", "z"):
+                                data_series.pop(bubble_sizes_variable_name, None)
+                    # next, remove bubble_sizes since it's not needed anymore and should be removed.
+                    data_series.pop("bubble_sizes")
+                # need to remove "zaxis" if making a bubble2d.
+                if "bubble2d" in trace_style.lower():
+                    if "zaxis" in fig_dict["layout"]:
+                        fig_dict["layout"].pop("zaxis")
     return fig_dict
 
 def update_3d_axes(fig_dict):

JSONGrapher/styles/layout_styles_library.py

@@ -17,6 +17,7 @@ styles_library = {
         },
         "default3d": {
             "layout": {
+                "scene": {"aspectmode":"cube"},
                 "title": {"font": {"size": 32}, "x": 0.5},
                 "xaxis": {"title": {"font": {"size": 12}}, "tickfont": {"size": 12}},
                 "yaxis": {"title": {"font": {"size": 12}}, "tickfont": {"size": 12}},

JSONGrapher/styles/trace_styles_collection_library.py

@@ -42,6 +42,15 @@ styles_library = {
                             "showscale": True
                         }
         },        
+        "bubble2d": {
+            "type": "scatter",
+            "mode": "markers",
+            "marker": {
+                            "color": "auto",
+                            "colorscale": "viridis_r",
+                            "showscale": True
+                        }
+        },     
         "spline": {
             "type": "scatter",
             "mode": "lines",
@@ -76,6 +85,11 @@ styles_library = {
             "colorscale":"rainbow", 
             "showscale":True
         },
+        "bubble3d": {
+            "mode": "markers",
+            "type": "scatter3d",
+            "marker": {"color" : "","colorscale":"rainbow", "showscale":True}
+        },
         "heatmap": {
             "type": "heatmap",
             "colorscale": "Viridis",

JSONGrapher/version.py

@@ -1 +1 @@
-__version__ = '4.2'
+__version__ = '4.4'

{jsongrapher-4.2.dist-info → jsongrapher-4.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: jsongrapher
-Version: 4.2
+Version: 4.4
 Summary: The python version of JSONGrapher with tools for creating JSONGrapher Records.
 Home-page: https://github.com/AdityaSavara/jsongrapher-py
 Author: Aditya Savara

{jsongrapher-4.2.dist-info → jsongrapher-4.4.dist-info}/RECORD

@@ -1,4 +1,4 @@
-JSONGrapher/JSONRecordCreator.py,sha256=Wu5gCafri9g4bAAcyAJwlAoTLkExx7ssxjnQYcuGPUg,215343
+JSONGrapher/JSONRecordCreator.py,sha256=7AlusrLNjMkqOZd3A_Kb_tyhqF-QOvRn7uQQGaXybFY,218318
 JSONGrapher/UnitPytesting.py,sha256=xizJ-2fg9C5oNMFJyfavbBLMusayE9KWQiYIRrQQd4A,4363
 JSONGrapher/UnitpyCustomUnitsTesting.py,sha256=Rwq5p8HXN0FP54lRFgLTV0kgJ9TpcFaD3_C0MEOoEzw,1297
 JSONGrapher/__init__.py,sha256=fA3R6paWq3uskiLe3OglQZY_uJOejZHMFKdyX0dXWjo,310
@@ -6,14 +6,14 @@ JSONGrapher/drag_and_drop_gui.py,sha256=-7QJHLhzadHotWhONH4inerMaZ_xuwoTQSMRF_MP
 JSONGrapher/equation_creator.py,sha256=VFu6dFo-C0wDf1N3e0CyCBI-53cCUrp3oe0umdMLofs,17968
 JSONGrapher/equation_evaluator.py,sha256=Cm7UK5vWa6fT0OylD9f22_eR8zUvxY_Sy_W6n6N9bew,37012
 JSONGrapher/units_list.py,sha256=ROrlAsD66oPozZmOaE65xjNUEg3CxkSmA8iPE2_ihYI,34438
-JSONGrapher/version.py,sha256=bUff73G-qXNfPT8kDpDG9cWR4Adm49E9fp682pqJfbI,19
+JSONGrapher/version.py,sha256=qBnDha-1O1-zjh3MFXQsOlQrB1ykLZIoeKE5jHBMh-k,19
 JSONGrapher/styles/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-JSONGrapher/styles/layout_styles_library.py,sha256=vGb1tE_ETY-Blj2frt8ddHA976ADFSCoabjzKn2ZNYU,3051
-JSONGrapher/styles/trace_styles_collection_library.py,sha256=DUbbO8jTlhn2q-lOY1XYGkHV1RbJX-P5Z1SUCM0qo6M,5952
-jsongrapher-4.2.data/data/LICENSE.txt,sha256=7Guw_pmj_H4sApnM7fld2L4HH6CP-3ZYdFoOXtHFx0Y,1489
-jsongrapher-4.2.data/data/README.md,sha256=u9uRPF0T12Pl6M6qjEleXZoB41ye0VTQPCJpEdldwNU,8157
-jsongrapher-4.2.dist-info/LICENSE.txt,sha256=7Guw_pmj_H4sApnM7fld2L4HH6CP-3ZYdFoOXtHFx0Y,1489
-jsongrapher-4.2.dist-info/METADATA,sha256=SmJRCR_Mh6cYNpdGZnB1AsA8C-k-iBD8Cvd_PcFojCo,9231
-jsongrapher-4.2.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
-jsongrapher-4.2.dist-info/top_level.txt,sha256=5f7Ui2hCKCPTQOjD540WKghKNYOvUDNfdpnDbIlAD3Y,12
-jsongrapher-4.2.dist-info/RECORD,,
+JSONGrapher/styles/layout_styles_library.py,sha256=adGZNVAj_VysYX-9x1NqxDnhIQFtKsMW8swGaY30ocs,3100
+JSONGrapher/styles/trace_styles_collection_library.py,sha256=4WhrmsSotPGRwFplGPoRjTiyFd3PMneg5mUewjZMK88,6437
+jsongrapher-4.4.data/data/LICENSE.txt,sha256=7Guw_pmj_H4sApnM7fld2L4HH6CP-3ZYdFoOXtHFx0Y,1489
+jsongrapher-4.4.data/data/README.md,sha256=u9uRPF0T12Pl6M6qjEleXZoB41ye0VTQPCJpEdldwNU,8157
+jsongrapher-4.4.dist-info/LICENSE.txt,sha256=7Guw_pmj_H4sApnM7fld2L4HH6CP-3ZYdFoOXtHFx0Y,1489
+jsongrapher-4.4.dist-info/METADATA,sha256=XwDZLT51CId5nJG7-Mu-885q8NmVDvt168xza0qEuvc,9231
+jsongrapher-4.4.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
+jsongrapher-4.4.dist-info/top_level.txt,sha256=5f7Ui2hCKCPTQOjD540WKghKNYOvUDNfdpnDbIlAD3Y,12
+jsongrapher-4.4.dist-info/RECORD,,