ntcharts 0.1.0-aarch64-linux-musl

data/ext/ntcharts/wavelinechart.c

@@ -0,0 +1,248 @@
+#include "extension.h"
+
+static void wavelinechart_free(void *pointer) {
+  ntcharts_wavelinechart_t *chart = (ntcharts_wavelinechart_t *)pointer;
+  if (chart->handle != 0) {
+    ntcharts_wavelinechart_free(chart->handle);
+  }
+  xfree(chart);
+}
+
+static size_t wavelinechart_memsize(const void *pointer) {
+  return sizeof(ntcharts_wavelinechart_t);
+}
+
+const rb_data_type_t wavelinechart_type = {
+  .wrap_struct_name = "Ntcharts::Wavelinechart",
+  .function = {
+    .dmark = NULL,
+    .dfree = wavelinechart_free,
+    .dsize = wavelinechart_memsize,
+  },
+  .flags = RUBY_TYPED_FREE_IMMEDIATELY
+};
+
+static VALUE wavelinechart_alloc(VALUE klass) {
+  ntcharts_wavelinechart_t *chart = ALLOC(ntcharts_wavelinechart_t);
+  chart->handle = 0;
+  return TypedData_Wrap_Struct(klass, &wavelinechart_type, chart);
+}
+
+static VALUE wavelinechart_initialize(VALUE self, VALUE width, VALUE height) {
+  GET_WAVELINECHART(self, chart);
+  chart->handle = ntcharts_wavelinechart_new(NUM2INT(width), NUM2INT(height));
+  return self;
+}
+
+static VALUE wavelinechart_width(VALUE self) {
+  GET_WAVELINECHART(self, chart);
+  return INT2NUM(ntcharts_wavelinechart_width(chart->handle));
+}
+
+static VALUE wavelinechart_height(VALUE self) {
+  GET_WAVELINECHART(self, chart);
+  return INT2NUM(ntcharts_wavelinechart_height(chart->handle));
+}
+
+static VALUE wavelinechart_graph_width(VALUE self) {
+  GET_WAVELINECHART(self, chart);
+  return INT2NUM(ntcharts_wavelinechart_graph_width(chart->handle));
+}
+
+static VALUE wavelinechart_graph_height(VALUE self) {
+  GET_WAVELINECHART(self, chart);
+  return INT2NUM(ntcharts_wavelinechart_graph_height(chart->handle));
+}
+
+static VALUE wavelinechart_min_x(VALUE self) {
+  GET_WAVELINECHART(self, chart);
+  return DBL2NUM(ntcharts_wavelinechart_min_x(chart->handle));
+}
+
+static VALUE wavelinechart_max_x(VALUE self) {
+  GET_WAVELINECHART(self, chart);
+  return DBL2NUM(ntcharts_wavelinechart_max_x(chart->handle));
+}
+
+static VALUE wavelinechart_min_y(VALUE self) {
+  GET_WAVELINECHART(self, chart);
+  return DBL2NUM(ntcharts_wavelinechart_min_y(chart->handle));
+}
+
+static VALUE wavelinechart_max_y(VALUE self) {
+  GET_WAVELINECHART(self, chart);
+  return DBL2NUM(ntcharts_wavelinechart_max_y(chart->handle));
+}
+
+static VALUE wavelinechart_set_y_range(VALUE self, VALUE min_val, VALUE max_val) {
+  GET_WAVELINECHART(self, chart);
+  ntcharts_wavelinechart_set_y_range(chart->handle, NUM2DBL(min_val), NUM2DBL(max_val));
+  return self;
+}
+
+static VALUE wavelinechart_set_view_x_range(VALUE self, VALUE min_val, VALUE max_val) {
+  GET_WAVELINECHART(self, chart);
+  ntcharts_wavelinechart_set_view_x_range(chart->handle, NUM2DBL(min_val), NUM2DBL(max_val));
+  return self;
+}
+
+static VALUE wavelinechart_set_view_y_range(VALUE self, VALUE min_val, VALUE max_val) {
+  GET_WAVELINECHART(self, chart);
+  ntcharts_wavelinechart_set_view_y_range(chart->handle, NUM2DBL(min_val), NUM2DBL(max_val));
+  return self;
+}
+
+static VALUE wavelinechart_resize(VALUE self, VALUE width, VALUE height) {
+  GET_WAVELINECHART(self, chart);
+  ntcharts_wavelinechart_resize(chart->handle, NUM2INT(width), NUM2INT(height));
+  return self;
+}
+
+static VALUE wavelinechart_clear(VALUE self) {
+  GET_WAVELINECHART(self, chart);
+  ntcharts_wavelinechart_clear(chart->handle);
+  return self;
+}
+
+static VALUE wavelinechart_clear_all_data(VALUE self) {
+  GET_WAVELINECHART(self, chart);
+  ntcharts_wavelinechart_clear_all_data(chart->handle);
+  return self;
+}
+
+static VALUE wavelinechart_clear_data_set(VALUE self, VALUE name) {
+  GET_WAVELINECHART(self, chart);
+  ntcharts_wavelinechart_clear_data_set(chart->handle, StringValueCStr(name));
+  return self;
+}
+
+static VALUE wavelinechart_plot(VALUE self, VALUE x, VALUE y) {
+  GET_WAVELINECHART(self, chart);
+  ntcharts_wavelinechart_plot(chart->handle, NUM2DBL(x), NUM2DBL(y));
+  return self;
+}
+
+static VALUE wavelinechart_plot_data_set(VALUE self, VALUE name, VALUE x, VALUE y) {
+  GET_WAVELINECHART(self, chart);
+  ntcharts_wavelinechart_plot_data_set(chart->handle, StringValueCStr(name), NUM2DBL(x), NUM2DBL(y));
+  return self;
+}
+
+static VALUE wavelinechart_set_styles(VALUE self, VALUE line_style, VALUE style) {
+  GET_WAVELINECHART(self, chart);
+  ntcharts_style_t *style_ptr;
+  TypedData_Get_Struct(style, ntcharts_style_t, &style_type, style_ptr);
+  ntcharts_wavelinechart_set_styles(chart->handle, NUM2INT(line_style), style_ptr->handle);
+  return self;
+}
+
+static VALUE wavelinechart_set_data_set_styles(VALUE self, VALUE name, VALUE line_style, VALUE style) {
+  GET_WAVELINECHART(self, chart);
+  ntcharts_style_t *style_ptr;
+  TypedData_Get_Struct(style, ntcharts_style_t, &style_type, style_ptr);
+  ntcharts_wavelinechart_set_data_set_styles(chart->handle, StringValueCStr(name), NUM2INT(line_style), style_ptr->handle);
+  return self;
+}
+
+static VALUE wavelinechart_set_style(VALUE self, VALUE style) {
+  GET_WAVELINECHART(self, chart);
+  ntcharts_style_t *style_ptr;
+  TypedData_Get_Struct(style, ntcharts_style_t, &style_type, style_ptr);
+  ntcharts_wavelinechart_set_style(chart->handle, style_ptr->handle);
+  return self;
+}
+
+static VALUE wavelinechart_set_axis_style(VALUE self, VALUE style) {
+  GET_WAVELINECHART(self, chart);
+  ntcharts_style_t *style_ptr;
+  TypedData_Get_Struct(style, ntcharts_style_t, &style_type, style_ptr);
+  ntcharts_wavelinechart_set_axis_style(chart->handle, style_ptr->handle);
+  return self;
+}
+
+static VALUE wavelinechart_set_label_style(VALUE self, VALUE style) {
+  GET_WAVELINECHART(self, chart);
+  ntcharts_style_t *style_ptr;
+  TypedData_Get_Struct(style, ntcharts_style_t, &style_type, style_ptr);
+  ntcharts_wavelinechart_set_label_style(chart->handle, style_ptr->handle);
+  return self;
+}
+
+static VALUE wavelinechart_draw(VALUE self) {
+  GET_WAVELINECHART(self, chart);
+  ntcharts_wavelinechart_draw(chart->handle);
+  return self;
+}
+
+static VALUE wavelinechart_draw_all(VALUE self) {
+  GET_WAVELINECHART(self, chart);
+  ntcharts_wavelinechart_draw_all(chart->handle);
+  return self;
+}
+
+static VALUE wavelinechart_view(VALUE self) {
+  GET_WAVELINECHART(self, chart);
+  char *result = ntcharts_wavelinechart_view(chart->handle);
+  VALUE rb_result = rb_utf8_str_new_cstr(result);
+  ntcharts_free(result);
+  return rb_result;
+}
+
+static VALUE wavelinechart_render(VALUE self) {
+  wavelinechart_draw(self);
+  return wavelinechart_view(self);
+}
+
+static VALUE wavelinechart_render_all(VALUE self) {
+  wavelinechart_draw_all(self);
+  return wavelinechart_view(self);
+}
+
+static VALUE wavelinechart_to_s(VALUE self) {
+  return wavelinechart_view(self);
+}
+
+void Init_ntcharts_wavelinechart(void) {
+  cWavelinechart = rb_define_class_under(mNtcharts, "Wavelinechart", rb_cObject);
+
+  rb_define_alloc_func(cWavelinechart, wavelinechart_alloc);
+  rb_define_method(cWavelinechart, "initialize", wavelinechart_initialize, 2);
+
+  rb_define_method(cWavelinechart, "width", wavelinechart_width, 0);
+  rb_define_method(cWavelinechart, "height", wavelinechart_height, 0);
+  rb_define_method(cWavelinechart, "graph_width", wavelinechart_graph_width, 0);
+  rb_define_method(cWavelinechart, "graph_height", wavelinechart_graph_height, 0);
+
+  rb_define_method(cWavelinechart, "min_x", wavelinechart_min_x, 0);
+  rb_define_method(cWavelinechart, "max_x", wavelinechart_max_x, 0);
+  rb_define_method(cWavelinechart, "min_y", wavelinechart_min_y, 0);
+  rb_define_method(cWavelinechart, "max_y", wavelinechart_max_y, 0);
+
+  rb_define_method(cWavelinechart, "set_y_range", wavelinechart_set_y_range, 2);
+  rb_define_method(cWavelinechart, "set_view_x_range", wavelinechart_set_view_x_range, 2);
+  rb_define_method(cWavelinechart, "set_view_y_range", wavelinechart_set_view_y_range, 2);
+
+  rb_define_method(cWavelinechart, "resize", wavelinechart_resize, 2);
+  rb_define_method(cWavelinechart, "clear", wavelinechart_clear, 0);
+  rb_define_method(cWavelinechart, "clear_all_data", wavelinechart_clear_all_data, 0);
+  rb_define_method(cWavelinechart, "clear_data_set", wavelinechart_clear_data_set, 1);
+
+  rb_define_method(cWavelinechart, "plot", wavelinechart_plot, 2);
+  rb_define_method(cWavelinechart, "plot_data_set", wavelinechart_plot_data_set, 3);
+
+  rb_define_method(cWavelinechart, "set_styles", wavelinechart_set_styles, 2);
+  rb_define_method(cWavelinechart, "set_data_set_styles", wavelinechart_set_data_set_styles, 3);
+  rb_define_method(cWavelinechart, "_set_style", wavelinechart_set_style, 1);
+  rb_define_method(cWavelinechart, "_set_axis_style", wavelinechart_set_axis_style, 1);
+  rb_define_method(cWavelinechart, "_set_label_style", wavelinechart_set_label_style, 1);
+
+  rb_define_method(cWavelinechart, "draw", wavelinechart_draw, 0);
+  rb_define_method(cWavelinechart, "draw_all", wavelinechart_draw_all, 0);
+  rb_define_method(cWavelinechart, "view", wavelinechart_view, 0);
+  rb_define_method(cWavelinechart, "render", wavelinechart_render, 0);
+  rb_define_method(cWavelinechart, "render_all", wavelinechart_render_all, 0);
+  rb_define_method(cWavelinechart, "to_s", wavelinechart_to_s, 0);
+
+  rb_define_const(cWavelinechart, "LINE_STYLE_THIN", INT2NUM(0));
+  rb_define_const(cWavelinechart, "LINE_STYLE_ARC", INT2NUM(1));
+}

data/go/barchart.go

@@ -0,0 +1,356 @@
+package main
+
+import "C"
+
+import (
+	"encoding/json"
+	"github.com/NimbleMarkets/ntcharts/barchart"
+	"github.com/charmbracelet/lipgloss"
+)
+
+type BarValueJSON struct {
+	Name        string `json:"name"`
+	Value       float64 `json:"value"`
+	StyleHandle uint64 `json:"style_handle,omitempty"`
+}
+
+type BarDataJSON struct {
+	Label  string         `json:"label"`
+	Values []BarValueJSON `json:"values"`
+}
+
+func parseBarData(jsonStr string) barchart.BarData {
+	var data BarDataJSON
+
+	if err := json.Unmarshal([]byte(jsonStr), &data); err != nil {
+		return barchart.BarData{}
+	}
+
+	barData := barchart.BarData{Label: data.Label}
+
+	for _, v := range data.Values {
+		style := lipgloss.NewStyle()
+
+		if v.StyleHandle != 0 {
+			if s, ok := getStyle(v.StyleHandle); ok {
+				style = s
+			}
+		}
+
+		barValue := barchart.BarValue{
+			Name:  v.Name,
+			Value: v.Value,
+			Style: style,
+		}
+
+		barData.Values = append(barData.Values, barValue)
+	}
+
+	return barData
+}
+
+//export ntcharts_barchart_new
+func ntcharts_barchart_new(width C.int, height C.int) C.ulonglong {
+	b := barchart.New(int(width), int(height))
+	return C.ulonglong(allocBarchart(&b))
+}
+
+//export ntcharts_barchart_free
+func ntcharts_barchart_free(id C.ulonglong) {
+	freeBarchart(uint64(id))
+}
+
+//export ntcharts_barchart_width
+func ntcharts_barchart_width(id C.ulonglong) C.int {
+	b := getBarchart(uint64(id))
+
+	if b == nil {
+		return 0
+	}
+
+	return C.int(b.Width())
+}
+
+//export ntcharts_barchart_height
+func ntcharts_barchart_height(id C.ulonglong) C.int {
+	b := getBarchart(uint64(id))
+
+	if b == nil {
+		return 0
+	}
+
+	return C.int(b.Height())
+}
+
+//export ntcharts_barchart_max_value
+func ntcharts_barchart_max_value(id C.ulonglong) C.double {
+	b := getBarchart(uint64(id))
+
+	if b == nil {
+		return 0
+	}
+
+	return C.double(b.MaxValue())
+}
+
+//export ntcharts_barchart_scale
+func ntcharts_barchart_scale(id C.ulonglong) C.double {
+	b := getBarchart(uint64(id))
+
+	if b == nil {
+		return 0
+	}
+
+	return C.double(b.Scale())
+}
+
+//export ntcharts_barchart_bar_width
+func ntcharts_barchart_bar_width(id C.ulonglong) C.int {
+	b := getBarchart(uint64(id))
+
+	if b == nil {
+		return 0
+	}
+
+	return C.int(b.BarWidth())
+}
+
+//export ntcharts_barchart_bar_gap
+func ntcharts_barchart_bar_gap(id C.ulonglong) C.int {
+	b := getBarchart(uint64(id))
+
+	if b == nil {
+		return 0
+	}
+
+	return C.int(b.BarGap())
+}
+
+//export ntcharts_barchart_show_axis
+func ntcharts_barchart_show_axis(id C.ulonglong) C.int {
+	b := getBarchart(uint64(id))
+
+	if b == nil {
+		return 0
+	}
+
+	if b.ShowAxis() {
+		return 1
+	}
+
+	return 0
+}
+
+//export ntcharts_barchart_horizontal
+func ntcharts_barchart_horizontal(id C.ulonglong) C.int {
+	b := getBarchart(uint64(id))
+
+	if b == nil {
+		return 0
+	}
+
+	if b.Horizontal() {
+		return 1
+	}
+
+	return 0
+}
+
+//export ntcharts_barchart_set_max
+func ntcharts_barchart_set_max(id C.ulonglong, maxVal C.double) {
+	b := getBarchart(uint64(id))
+
+	if b == nil {
+		return
+	}
+
+	b.SetMax(float64(maxVal))
+}
+
+//export ntcharts_barchart_set_bar_width
+func ntcharts_barchart_set_bar_width(id C.ulonglong, width C.int) {
+	b := getBarchart(uint64(id))
+
+	if b == nil {
+		return
+	}
+
+	b.AutoBarWidth = false
+	b.SetBarWidth(int(width))
+}
+
+//export ntcharts_barchart_set_bar_gap
+func ntcharts_barchart_set_bar_gap(id C.ulonglong, gap C.int) {
+	b := getBarchart(uint64(id))
+
+	if b == nil {
+		return
+	}
+
+	b.SetBarGap(int(gap))
+}
+
+//export ntcharts_barchart_set_horizontal
+func ntcharts_barchart_set_horizontal(id C.ulonglong, horizontal C.int) {
+	b := getBarchart(uint64(id))
+
+	if b == nil {
+		return
+	}
+
+	b.SetHorizontal(horizontal != 0)
+}
+
+//export ntcharts_barchart_set_show_axis
+func ntcharts_barchart_set_show_axis(id C.ulonglong, showAxis C.int) {
+	b := getBarchart(uint64(id))
+
+	if b == nil {
+		return
+	}
+
+	b.SetShowAxis(showAxis != 0)
+}
+
+//export ntcharts_barchart_set_auto_max_value
+func ntcharts_barchart_set_auto_max_value(id C.ulonglong, autoMax C.int) {
+	b := getBarchart(uint64(id))
+
+	if b == nil {
+		return
+	}
+
+	b.AutoMaxValue = autoMax != 0
+}
+
+//export ntcharts_barchart_set_auto_bar_width
+func ntcharts_barchart_set_auto_bar_width(id C.ulonglong, autoWidth C.int) {
+	b := getBarchart(uint64(id))
+
+	if b == nil {
+		return
+	}
+
+	b.AutoBarWidth = autoWidth != 0
+}
+
+//export ntcharts_barchart_set_axis_style
+func ntcharts_barchart_set_axis_style(id C.ulonglong, styleID C.ulonglong) {
+	b := getBarchart(uint64(id))
+
+	if b == nil {
+		return
+	}
+
+	if style, ok := getStyle(uint64(styleID)); ok {
+		b.AxisStyle = style
+	}
+}
+
+//export ntcharts_barchart_set_label_style
+func ntcharts_barchart_set_label_style(id C.ulonglong, styleID C.ulonglong) {
+	b := getBarchart(uint64(id))
+
+	if b == nil {
+		return
+	}
+
+	if style, ok := getStyle(uint64(styleID)); ok {
+		b.LabelStyle = style
+	}
+}
+
+//export ntcharts_barchart_resize
+func ntcharts_barchart_resize(id C.ulonglong, width C.int, height C.int) {
+	b := getBarchart(uint64(id))
+
+	if b == nil {
+		return
+	}
+
+	b.Resize(int(width), int(height))
+}
+
+//export ntcharts_barchart_clear
+func ntcharts_barchart_clear(id C.ulonglong) {
+	b := getBarchart(uint64(id))
+
+	if b == nil {
+		return
+	}
+
+	b.Clear()
+}
+
+//export ntcharts_barchart_push
+func ntcharts_barchart_push(id C.ulonglong, barDataJSON *C.char) {
+	b := getBarchart(uint64(id))
+
+	if b == nil {
+		return
+	}
+
+	barData := parseBarData(C.GoString(barDataJSON))
+	b.Push(barData)
+}
+
+//export ntcharts_barchart_push_all
+func ntcharts_barchart_push_all(id C.ulonglong, barDataArrayJSON *C.char) {
+	b := getBarchart(uint64(id))
+
+	if b == nil {
+		return
+	}
+
+	var dataArray []BarDataJSON
+
+	if err := json.Unmarshal([]byte(C.GoString(barDataArrayJSON)), &dataArray); err != nil {
+		return
+	}
+
+	for _, data := range dataArray {
+		barData := barchart.BarData{Label: data.Label}
+
+		for _, v := range data.Values {
+			style := lipgloss.NewStyle()
+
+			if v.StyleHandle != 0 {
+				if s, ok := getStyle(v.StyleHandle); ok {
+					style = s
+				}
+			}
+
+			barValue := barchart.BarValue{
+				Name:  v.Name,
+				Value: v.Value,
+				Style: style,
+			}
+
+			barData.Values = append(barData.Values, barValue)
+		}
+
+		b.Push(barData)
+	}
+}
+
+//export ntcharts_barchart_draw
+func ntcharts_barchart_draw(id C.ulonglong) {
+	b := getBarchart(uint64(id))
+
+	if b == nil {
+		return
+	}
+
+	b.Draw()
+}
+
+//export ntcharts_barchart_view
+func ntcharts_barchart_view(id C.ulonglong) *C.char {
+	b := getBarchart(uint64(id))
+
+	if b == nil {
+		return C.CString("")
+	}
+
+	return C.CString(b.View())
+}