From 8646fcb6c617a10fce5a526b95a7c43210d1355c Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Nicol=C3=B2=20Santilio?= Date: Thu, 31 Dec 2020 13:58:25 +0100 Subject: [PATCH] @structure_data changed --- addons/easy_charts/Utilities/Scripts/chart.gd | 709 ++++++++++++++++++ .../easy_charts/Utilities/Scripts/chart2d.gd | 211 ++++++ 2 files changed, 920 insertions(+) create mode 100644 addons/easy_charts/Utilities/Scripts/chart.gd create mode 100644 addons/easy_charts/Utilities/Scripts/chart2d.gd diff --git a/addons/easy_charts/Utilities/Scripts/chart.gd b/addons/easy_charts/Utilities/Scripts/chart.gd new file mode 100644 index 0000000..435d5c0 --- /dev/null +++ b/addons/easy_charts/Utilities/Scripts/chart.gd @@ -0,0 +1,709 @@ +extends Control +class_name Chart + +# Classes +enum TYPES { Line, Bar, Scatter, Radar, Pie } + +# Signals .................................. +signal chart_plotted(chart) # emit when a chart is plotted (static) or updated (dynamic) +signal point_pressed(point) + +# Onready Vars ............................ +onready var PointData = $PointData/PointData +onready var Points = $Points +onready var Legend = $Legend +onready var ChartName : Label = $ChartName + +# Scenes and Reosurces ...................... +var point_node : PackedScene = preload("../Point/Point.tscn") +var LegendElement : PackedScene = preload("../Legend/FunctionLegend.tscn") + +# Enums ..................................... +enum PointShapes { Dot, Triangle, Square, Cross } + +# Shared Variables ......................... +var SIZE : Vector2 = Vector2() +var OFFSET : Vector2 = Vector2(0,0) +var origin : Vector2 + +var font_size : float = 16 +var const_height : float = font_size/2*font_size/20 +var const_width : float = font_size/2 + +# actual distance between x and y values +var x_pass : float +var y_pass : float + +# vertical distance between y consecutive points used for intervals +var v_dist : float +var h_dist : float + +# define values on x an y axis +var x_chors : Array +var y_chors : Array + +# actual coordinates of points (in pixel) +var x_coordinates : Array +var y_coordinates : Array + +# data contained in file +var data : Array + +# amount of functions to represent +var functions : int = 0 + +# database values +var x_datas : Array +var y_datas : Array + +# labels displayed on chart +var x_label : String + +var x_labels : Array +var y_labels : Array + +var x_margin_min : int = 0 +var y_margin_min : int = 0 + +# actual values of point, from the database +var point_values : Array + +# actual position of points in pixel +var point_positions : Array + +var legend : Array setget set_legend,get_legend + +# ................... Export Shared Variables .................. +export (String) var chart_name : String = "" setget set_chart_name +export (String, FILE, "*.txt, *.csv") var source : String = "" setget set_source +export (String) var delimiter : String = ";" setget set_delimiter + +var origin_at_zero : bool = true setget set_origin_at_zero#, get_origin_at_zero +var are_values_columns : bool = false setget set_are_values_columns#, get_are_values_columns +var show_x_values_as_labels : bool = true setget set_show_x_values_as_labels#, get_show_x_values_as_labels +var labels_index : int = 0 setget set_labels_index#, get_labels_index +var function_names_index : int = 0 setget set_function_names_index#, get_function_names_index + +# for radar +var use_height_as_radius : bool = false setget set_use_height_as_radius +var radius : float = 150.0 setget _set_radius,get_radius + +# for columns +var column_width : float = 10 setget set_column_width +var column_gap : float = 2 setget set_column_gap + +var full_scale : float = 1.0 setget set_full_scale +var x_decim : float = 5.0 setget set_x_decim +var y_decim : float = 1.0 setget set_y_decim + +var points_shape : Array = [PointShapes.Dot] setget set_points_shape +var function_colors = [Color("#1e1e1e")] setget set_function_colors +var outline_color : Color = Color("#1e1e1e") setget set_outline_color +var box_color : Color = Color("#1e1e1e") setget set_box_color +var v_lines_color : Color = Color("#cacaca") setget set_v_lines_color +var h_lines_color : Color = Color("#cacaca") setget set_h_lines_color +var grid_color : Color = Color("#1e1e1e") setget set_grid_color +var font : Font setget set_font +var bold_font : Font setget set_bold_font +var font_color : Color = Color("#1e1e1e") setget set_font_color + +var use_template : bool = true setget set_use_template +var template : int = 0 setget set_template + +# modifiers +var rotation : float = 0 setget set_rotation +var invert_chart : bool = false setget set_invert_chart + +# Only disp a certain range of values: +# (x , 0) -> Only disp first 'x' values +# (0 , y) -> Only disp last 'y' values +# (x , y) -> Only disp values in range [x, y] +# (0 , 0) -> Disp all values (full range) +var only_disp_values : Vector2 = Vector2(0,0) setget set_only_disp_values + +# A vector representing limit values (both on x and y axis) you want to stay over or below +# The treshold value is always relative to your dataset values +# ex. if your dataset is [ [100,100], [300,300] ] a proper treshold would be (200,200) +var treshold : Vector2 setget set_treshold + +# A vector representing @treshold coordinates in its relative chart +# only used to draw treshold values +var treshold_draw : Vector2 + +# !! API v2 +static func instance(chart_type : int): + var chart_t : String = Utilities.get_chart_type(chart_type) + var chart : String = "res://addons/easy_charts/%s/%s.tscn" % [chart_t, chart_t] + return load(chart).instance() + +# .......................... Properties Manager .................................... +func _get(property): + match property: + "Chart_Properties/origin_at_zero": + return origin_at_zero + "Chart_Properties/are_values_columns": + return are_values_columns + "Chart_Properties/show_x_values_as_labels": + return show_x_values_as_labels + "Chart_Properties/labels_index": + return labels_index + "Chart_Properties/function_names_index": + return function_names_index + "Chart_Properties/use_height_as_radius": + return use_height_as_radius + "Chart_Properties/radius": + return radius + "Chart_Properties/column_width": + return column_width + "Chart_Properties/column_gap": + return column_gap + + "Chart_Display/full_scale": + return full_scale + "Chart_Display/x_decim": + return x_decim + "Chart_Display/y_decim": + return y_decim + + "Chart_Style/points_shape": + return points_shape + "Chart_Style/function_colors": + return function_colors + "Chart_Style/template": + return template + "Chart_Style/use_template": + return use_template + "Chart_Style/outline_color": + return outline_color + "Chart_Style/grid_color": + return grid_color + "Chart_Style/box_color": + return box_color + "Chart_Style/v_lines_color": + return v_lines_color + "Chart_Style/h_lines_color": + return h_lines_color + "Chart_Style/font": + return font + "Chart_Style/bold_font": + return bold_font + "Chart_Style/font_color": + return font_color + + "Chart_Modifiers/treshold": + return treshold + "Chart_Modifiers/only_disp_values": + return only_disp_values + "Chart_Modifiers/rotation": + return rotation + "Chart_Modifiers/invert_chart": + return invert_chart + +func _set(property, value): + match property: + "Chart_Properties/origin_at_zero": + origin_at_zero = value + return true + "Chart_Properties/are_values_columns": + are_values_columns = value + return true + "Chart_Properties/show_x_values_as_labels": + show_x_values_as_labels = value + return true + "Chart_Properties/labels_index": + labels_index = value + return true + "Chart_Properties/function_names_index": + function_names_index = value + return true + "Chart_Properties/use_height_as_radius": + use_height_as_radius = value + return true + "Chart_Properties/radius": + radius = value + return true + "Chart_Properties/column_width": + column_width = value + return true + "Chart_Properties/column_gap": + column_gap = value + return true + + "Chart_Display/full_scale": + full_scale = value + return true + "Chart_Display/x_decim": + x_decim = value + return true + "Chart_Display/y_decim": + y_decim = value + return true + + "Chart_Style/points_shape": + points_shape = value + return true + "Chart_Style/function_colors": + function_colors = value + return true + "Chart_Style/use_template": + use_template = value + return true + "Chart_Style/template": + template = value + apply_template(template) + return true + "Chart_Style/outline_color": + outline_color = value + return true + "Chart_Style/grid_color": + grid_color = value + return true + "Chart_Style/box_color": + box_color = value + return true + "Chart_Style/v_lines_color": + v_lines_color = value + return true + "Chart_Style/h_lines_color": + h_lines_color = value + return true + "Chart_Style/font": + font = value + return true + "Chart_Style/bold_font": + bold_font = value + return true + "Chart_Style/font_color": + font_color = value +# apply_template(template) + return true + + "Chart_Modifiers/treshold": + treshold = value + return true + "Chart_Modifiers/only_disp_values": + only_disp_values = value + return true + "Chart_Modifiers/rotation": + rotation = value + return true + "Chart_Modifiers/invert_chart": + invert_chart = value + return true + +# .......................... Shared Functions and virtuals ........................ +func slice_data() -> Array: + var data_to_display : Array + data_to_display.resize(data.size()) + if only_disp_values == Vector2(0,0) : + data_to_display = data.duplicate(true) + elif only_disp_values.x == 0 and only_disp_values.y < data[0].size(): + for row_idx in data.size(): + data_to_display[row_idx] = [data[row_idx][0]] + data[row_idx].slice(data[row_idx].size()-only_disp_values.y, data[row_idx].size()-1) + elif only_disp_values.y == 0 and only_disp_values.x < data[0].size(): + for row_idx in data.size(): + data_to_display[row_idx] = [data[row_idx][0]] + data[row_idx].slice(1, only_disp_values.x) + elif only_disp_values.x != 0 and only_disp_values.y != 0 and only_disp_values.y < data[0].size() and only_disp_values.x < data[0].size(): + for row_idx in data.size(): + data_to_display[row_idx] = [data[row_idx][0]] + data[row_idx].slice(only_disp_values.x, data[row_idx].size()-only_disp_values.y) + else: + data_to_display = data.duplicate(true) + return data_to_display + +func plot(): + load_font() + PointData.hide() + + if source == "" or source == null: + Utilities._print_message("Can't plot a chart without a Source file. Please, choose it in editor, or use the custom function _plot().",1) + return + + + data = read_datas(source) + structure_datas(slice_data()) + build_chart() + count_functions() + calculate_pass() + calculate_colors() + calculate_coordinates() + set_shapes() + create_legend() + emit_signal("chart_plotted",self) + +func plot_from_csv(csv_file : String, _delimiter : String = delimiter): + load_font() + PointData.hide() + + if csv_file == "" or csv_file == null: + Utilities._print_message("Can't plot a chart without a Source file. Please, choose it in editor, or use the custom function _plot().",1) + return + + data = read_datas(csv_file, _delimiter) + structure_datas(slice_data()) + build_chart() + count_functions() + calculate_pass() + calculate_colors() + calculate_coordinates() + set_shapes() + create_legend() + emit_signal("chart_plotted",self) + +func plot_from_array(array : Array) -> void: + clean_variables() + clear_points() + load_font() + PointData.hide() + + if array.empty(): + Utilities._print_message("Can't plot a chart with an empty Array.",1) + return + + data = array.duplicate() + structure_datas(slice_data()) + build_chart() + count_functions() + calculate_pass() + calculate_colors() + calculate_coordinates() + set_shapes() + create_legend() + emit_signal("chart_plotted",self) + + if not is_connected("item_rect_changed",self, "redraw"): connect("item_rect_changed", self, "redraw") + +func plot_from_dataframe(dataframe : DataFrame) -> void: + clean_variables() + clear_points() + load_font() + load_font() + PointData.hide() + + data = dataframe.get_dataframe().duplicate(true) + + if data.empty(): + Utilities._print_message("Can't plot a chart with an empty Array.",1) + return + + structure_datas(slice_data()) + build_chart() + count_functions() + calculate_pass() + calculate_colors() + calculate_coordinates() + set_shapes() + create_legend() + emit_signal("chart_plotted",self) + + if not is_connected("item_rect_changed",self, "redraw"): connect("item_rect_changed", self, "redraw") + +# Append new data (in array format) to the already plotted data. +# All data are stored. +func update_plot_data(array : Array) -> void: + if array.empty(): + Utilities._print_message("Can't plot a chart with an empty Array.",1) + return + + data.append(array) + structure_datas(slice_data()) + redraw() + count_functions() + calculate_colors() + set_shapes() + create_legend() + emit_signal("chart_plotted",self) + + update() + +# Append a new column to data +func append_new_column(dataset : Array, column : Array): + if column.empty(): + Utilities._print_message("Can't update plot with an empty row.",1) + return + for value_idx in column.size(): + dataset[value_idx].append(column[value_idx]) + +func plot_placeholder() -> void: + pass + +func load_font(): + if font != null: + font_size = font.get_height() + var theme : Theme = Theme.new() + theme.set_default_font(font) + set_theme(theme) + + else: + var lbl = Label.new() + font = lbl.get_font("") + lbl.free() + if bold_font != null: + PointData.Data.set("custom_fonts/font",bold_font) + +func calculate_colors(): + if function_colors.size() < functions: + for function in range(functions - function_colors.size() + 1): function_colors.append(Color(randf(),randf(), randf())) + +func set_shapes(): + if points_shape.empty() or points_shape.size() < functions: + for function in functions: + points_shape.append(PointShapes.Dot) + +func read_datas(source : String, _delimiter : String = delimiter): + var file : File = File.new() + file.open(source,File.READ) + var content : Array + while not file.eof_reached(): + var line : PoolStringArray = file.get_csv_line(_delimiter) + content.append(line) + file.close() + for data in content: + if data.size() < 2 or data.empty(): + content.erase(data) + return content + +func count_functions(): + if are_values_columns: functions = data[0].size()-1 + else: functions = y_datas.size() + +func clear_points(): + if $Points.get_children(): + for function in Points.get_children(): + function.queue_free() + for legend in $Legend.get_children(): + legend.queue_free() + +func redraw(): + build_chart() + calculate_pass() + calculate_coordinates() + update() + +func clean_variables(): + x_chors.clear() + y_chors.clear() + x_datas.clear() + y_datas.clear() + x_label = "" + x_labels.clear() + y_labels.clear() + +# .................. VIRTUAL FUNCTIONS ......................... +func structure_datas(database : Array): + pass + +func build_chart(): + pass + +func calculate_pass(): + pass + +func calculate_coordinates(): + pass + +func function_colors(): + pass + +func create_legend(): + for function in functions: + var function_legend : LegendElement + if legend.size() > function: + function_legend = legend[function] + else: + function_legend = LegendElement.instance() + legend.append(function_legend) + var f_name : String = y_labels[function] if not are_values_columns else str(x_datas[function]) + var legend_font : Font + if font != null: + legend_font = font + if bold_font != null: + legend_font = bold_font + function_legend.create_legend(f_name,function_colors[function],bold_font,font_color) + +# ........................... Shared Setters & Getters .............................. +func apply_template(template_name : int): + if Engine.editor_hint: + set_template(template_name) + property_list_changed_notify() + +# !!! API v2 +func set_chart_name(ch_name : String): + chart_name = ch_name + get_node("ChartName").set_text(chart_name) + +# !!! API v2 +func set_source(source_file : String): + source = source_file + +# !!! API v2 +func set_indexes(lb : int = 0, function_names : int = 0): + labels_index = lb + function_names_index = function_names + +# !!! API v2 +func set_radius(use_height : bool = false, f : float = 0): + use_height_as_radius = use_height + radius = f + +# !!! API v2 +func set_chart_colors(f_colors : PoolColorArray, o_color : Color, b_color : Color, g_color : Color, h_lines : Color, v_lines : Color): + function_colors = f_colors + outline_color = o_color + box_color = b_color + grid_color = g_color + h_lines_color = h_lines + v_lines_color = v_lines + +# !!! API v2 +func set_chart_fonts(normal_font : Font, bold_font : Font, f_color : Color = Color.white): + font = normal_font + self.bold_font = bold_font + font_color = f_color + +# !!! API v2 +func set_delimiter(d : String): + delimiter = d + +# ! API +func set_origin_at_zero(b : bool): + origin_at_zero = b + +# ! API +func set_are_values_columns(b : bool): + are_values_columns = b + +# ! API +func set_show_x_values_as_labels(b : bool): + show_x_values_as_labels = b + +func set_labels_index(i : int): + labels_index = i + +func set_function_names_index(i : int): + function_names_index = i + +func set_use_height_as_radius(b : bool): + use_height_as_radius = b + +func _set_radius(r : float): + radius = r + +func get_radius() -> float: + if use_height_as_radius: return get_size().y/2 + else: return radius + +# ! API +func set_column_width(f : float): + column_width = f + +# ! API +func set_column_gap(f : float): + column_gap = f + +# ! API +func set_full_scale(f : float): + full_scale = f + +# ! API +func set_x_decim(f : float): + x_decim = f + +# ! API +func set_y_decim(f : float): + y_decim = f + +# ! API +func set_points_shape(a : Array): + points_shape = a + + +# ! API +func set_function_colors(a : PoolColorArray): + function_colors = a + +# ! API +func set_outline_color(c : Color): + outline_color = c + +# ! API +func set_box_color(c : Color): + box_color = c + +# ! API +func set_grid_color(c : Color): + grid_color = c + +# ! API +func set_v_lines_color(c : Color): + v_lines_color = c + +# ! API +func set_h_lines_color(c : Color): + h_lines_color = c + +# ! API +func set_font(f : Font): + font = f + +# ! API +func set_bold_font(f : Font): + bold_font = f + +# ! API +func set_font_color(c : Color): + font_color = c + +func set_use_template(use : bool): + use_template = use + +# ! API +func set_template(template_name : int): + if not use_template: return + template = template_name + if template_name!=null: + var custom_template = Utilities.templates.get(Utilities.templates.keys()[template_name]) + function_colors = custom_template.function_colors as PoolColorArray + outline_color = Color(custom_template.outline_color) + box_color = Color(custom_template.outline_color) + grid_color = Color(custom_template.v_lines_color) + v_lines_color = Color(custom_template.v_lines_color) + h_lines_color = Color(custom_template.h_lines_color) + box_color = Color(custom_template.outline_color) + font_color = Color(custom_template.font_color) + +# ! API +func set_rotation(f : float): + rotation = f + +# ! API +func set_invert_chart(b : bool): + invert_chart = b + +# ! API +func set_treshold(t : Vector2): + treshold = t + +# ! API +func set_only_disp_values(v : Vector2): + only_disp_values = v + +func set_legend(l : Array): + legend = l + +func get_legend() -> Array: + return legend + +# ............................. Shared Signals .............................. +func point_pressed(point : Point): + emit_signal("point_pressed",point) + +func show_data(point : Point): + PointData.update_datas(point) + PointData.show() + +func hide_data(): + PointData.hide() + +func show_slice_data(slice : Slice): + PointData.update_slice_datas(slice) + PointData.show() diff --git a/addons/easy_charts/Utilities/Scripts/chart2d.gd b/addons/easy_charts/Utilities/Scripts/chart2d.gd new file mode 100644 index 0000000..27f6f65 --- /dev/null +++ b/addons/easy_charts/Utilities/Scripts/chart2d.gd @@ -0,0 +1,211 @@ +tool +extends Node2D +class_name Chart2D + +enum PointShapes { Dot, Triangle, Square, Cross } +enum TemplatesNames { Default, Clean, Gradient, Minimal, Invert } + + +signal chart_plotted(chart) +signal point_pressed(point) + + +const OFFSET: Vector2 = Vector2(0,0) + + +export (Vector2) var SIZE: Vector2 = Vector2() setget _set_size +export (String, FILE, "*.txt, *.csv") var source: String = "" +export (String) var delimiter: String = ";" +export (bool) var origin_at_zero: bool = true + +export (bool) var are_values_columns: bool = false +export (int, 0, 100) var x_values_index: int = 0 +export(bool) var show_x_values_as_labels: bool = true + +export (float,1,20,0.5) var column_width: float = 10 +export (float,0,10,0.5) var column_gap: float = 2 + +export (float, 0.1, 10.0) var x_decim: float = 5.0 +export (float, 0.1, 10.0) var y_decim: float = 5.0 +export (PointShapes) var point_shape: int = 0 +export (PoolColorArray) var function_colors = [Color("#1e1e1e")] +export (Color) var v_lines_color: Color = Color("#cacaca") +export (Color) var h_lines_color: Color = Color("#cacaca") + +export (bool) var boxed: bool = true +export (Color) var box_color: Color = Color("#1e1e1e") +export (Font) var font: Font +export (Font) var bold_font: Font +export (Color) var font_color: Color = Color("#1e1e1e") +export var template : int = 0 setget apply_template +export (float, 0.1, 1) var drawing_duration: float = 0.5 +export (bool) var invert_chart: bool = false + + +var OutlinesTween: Tween +var FunctionsTween: Tween +var PointTween : Tween +var Functions: Node2D +var GridTween: Tween +var PointData: PointData +var Outlines: Line2D +var Grid: Node2D + +var point_node: PackedScene = preload("../Point/Point.tscn") +var FunctionLegend: PackedScene = preload("../Legend/FunctionLegend.tscn") + +var font_size: float = 16 +var const_height: float = font_size / 2 * font_size / 20 +var const_width: float = font_size / 2 + +var origin: Vector2 + +# actual distance between x and y values +var x_pass: float +var y_pass: float + +# vertical distance between y consecutive points used for intervals +var v_dist: float +var h_dist: float + +# quantization, representing the interval in which values will be displayed + +# define values on x an y axis +var x_chors: Array +var y_chors: Array + +# actual coordinates of points (in pixel) +var x_coordinates: Array +var y_coordinates: Array + +# datas contained in file +var datas: Array + +# amount of functions to represent +var functions: int = 0 + +var x_label: String + +# database values +var x_datas: Array +var y_datas: Array + +# labels displayed on chart +var x_labels: Array +var y_labels: Array + +var x_margin_min: int = 0 +var y_margin_min: int = 0 + +# actual values of point, from the database +var point_values: Array + +# actual position of points in pixel +var point_positions: Array + +var legend: Array setget set_legend, get_legend + +var templates: Dictionary = {} + + +# Called when the node enters the scene tree for the first time. +func _ready(): + pass # Replace with function body. + +func build_chart(): + pass + +func calculate_pass(): + pass + +func calculate_coordinates(): + pass + +func _set_size(v : Vector2): + SIZE = v + +func _get_children(): + OutlinesTween = $OutlinesTween + FunctionsTween = $FunctionsTween + Functions = $Functions + GridTween = $GridTween + PointData = $PointData/PointData + Outlines = $Outlines + Grid = $Grid + +func apply_template(template_name: int): + template = template_name + templates = Utilities._load_templates() + if template_name != null: + var custom_template = templates.get(templates.keys()[template_name]) + function_colors = custom_template.function_colors as PoolColorArray + v_lines_color = Color(custom_template.v_lines_color) + h_lines_color = Color(custom_template.h_lines_color) + box_color = Color(custom_template.outline_color) + font_color = Color(custom_template.font_color) + property_list_changed_notify() + + if Engine.editor_hint: + _get_children() + Outlines.set_default_color(box_color) + Grid.get_node("VLine").set_default_color(v_lines_color) + Grid.get_node("HLine").set_default_color(h_lines_color) + +func redraw(): + build_chart() + calculate_pass() + calculate_coordinates() + update() + + +func show_data(point): + PointData.update_datas(point) + PointData.show() + + +func hide_data(): + PointData.hide() + + +func clear_points(): + function_colors.clear() + if Functions.get_children(): + for function in Functions.get_children(): + function.queue_free() + +func count_functions(): + if are_values_columns: + if not invert_chart: + functions = datas[0].size() - 1 + else: + functions = datas.size() - 1 + else: + if invert_chart: + functions = datas[0].size() - 1 + else: + functions = datas.size() - 1 + +func set_legend(l: Array): + legend = l + + +func get_legend(): + return legend + + +func create_legend(): + legend.clear() + for function in functions: + var function_legend = FunctionLegend.instance() + var f_name: String + if invert_chart: + f_name = x_datas[function] as String + else: + f_name = y_labels[function] + var legend_font: Font + if font != null: + legend_font = font + if bold_font != null: + legend_font = bold_font + function_legend.create_legend(f_name, function_colors[function], bold_font, font_color) + legend.append(function_legend)