Full rewrite of the pie algorithm

This commit is contained in:
pancake 2024-03-25 17:25:17 +01:00
parent 1c5f42999d
commit 283544dbfd
2 changed files with 51 additions and 126 deletions

View File

@ -283,7 +283,7 @@ R_API void r_print_init_rowoffsets(RPrint *p);
R_API ut32 r_print_rowoff(RPrint *p, int i);
R_API void r_print_set_rowoff(RPrint *p, int i, ut32 offset, bool overwrite);
R_API int r_print_row_at_off(RPrint *p, ut32 offset);
R_API int r_print_pie(RPrint *p, int nvalues, int *values, const char **text, int size);
R_API void r_print_pie(RPrint *p, int nvalues, int *values, const char **text, int size);
R_API void r_print_graphline(RPrint *print, const ut8 *buf, size_t len);
// WIP

View File

@ -4,39 +4,20 @@
#include <r_util/r_print.h>
#include <math.h>
#define PI 3.1415
#define O out[x + (y * size)]
#define USE_SINCOS 0
// TODO: [ ] add colors
// TODO: [ ] better rounding ascii art
// TODO: [ ] add support for xy_factor
#if 1
#define A(c) out[(x+half) + ((y+half) * (half*2))] = c
#else
printf ("%c%c", c, c);
#endif
static void drawSectorLine(char *out, int size, int percent, int ch) {
int i, x, y;
double foo = 0.02;
double A = (percent * PI) / 50;
for (i = (size - 1) / 2; i < (size*2); i++) {
x = y = (size - 1) / 2;
x += (int)(cos (A) * foo);
y += (int)(sin (A) * foo);
foo += 0.3; // 1.3;
O = ch; // '.';
}
}
static void fillSectorLine(char *out, int size, int percent, int ch) {
int i;
for (i = 0; i < size; i++) {
drawSectorLine (out, size, percent +i, ch);
}
}
static void drawPieChart(char *out, int size, int numSlices, int sliceSizes[], const char **text) {
static void drawPieChart(char *out, int diameter, int numSlices, int sliceSizes[], const char **text) {
int totalDegrees = 0;
char sliceChars[] = "0123456789"; // {'*', '+', '=', '-', '#', '%', '&', '$'}; // Characters for different slices
char sliceChars[] = "0123456789";
int sliceCharCount = sizeof (sliceChars) / sizeof (sliceChars[0]);
int RADIUS = size / 2;
int radius = diameter / 2;
int x, y, i;
// Calculate the start and end angle for each slice
int sliceAngles[11]; // max slices is 10
sliceAngles[0] = 0; // Starting angle for the first slice
@ -45,20 +26,19 @@ static void drawPieChart(char *out, int size, int numSlices, int sliceSizes[], c
sliceAngles[i+1] = totalDegrees;
// printf ("%d\n", totalDegrees);
}
// if (numSlices < 11 && totalDegrees < 360) {
// Add the remaining slice if the sum of parts don't cover the whole pie
if (totalDegrees < 360) {
numSlices++;
sliceAngles[numSlices] = 360; // - totalDegrees;
sliceAngles[numSlices] = 360;
}
int half = RADIUS;
#define A out[(x+half) + ((y+half) * (half*2))]
int half = radius;
// Draw the pie chart
for (y = -RADIUS; y <= RADIUS; y++) {
for (x = -RADIUS; x <= RADIUS; x++) {
for (y = -radius; y <= radius; y++) {
for (x = -radius; x <= radius; x++) {
double distance = sqrt (x * x + y * y);
// Check if the current point is within the circle
if (distance < RADIUS) {
if (distance < radius) {
double angle = atan2(y, x) * 180 / PI;
if (angle < 0) {
angle += 360;
@ -72,14 +52,14 @@ static void drawPieChart(char *out, int size, int numSlices, int sliceSizes[], c
}
if (sliceIndex != -1) {
const char ch = sliceChars[sliceIndex % sliceCharCount];
A = ch;
A (ch);
// printf ("%c%c", ch, ch);
} else {
A = ' ';
A (' ');
// printf (" "); // Should not happen, but just in case
}
} else {
A = ' ';
A (' ');
// printf (" ");
}
}
@ -87,13 +67,17 @@ static void drawPieChart(char *out, int size, int numSlices, int sliceSizes[], c
}
}
R_API int r_print_pie(R_NULLABLE RPrint *p, int nvalues, int *values, const char **text, int size) {
R_API void r_print_pie(R_NULLABLE RPrint *p, int nvalues, int *values, const char **text, int size) {
if (size < 1) {
R_LOG_WARN ("No one cant eat such smol pie");
return;
}
if (nvalues > 9) {
R_LOG_WARN ("Cant render more than 10 portions in a pie chart");
nvalues = 9;
}
ut8 *nv = calloc (nvalues, sizeof (ut8));
char *out = calloc (size, size);
char *out = calloc (size, size * size);
drawPieChart (out, size, nvalues, values, text);
@ -101,11 +85,11 @@ R_API int r_print_pie(R_NULLABLE RPrint *p, int nvalues, int *values, const char
const char *fg_colors[10] = {
Color_GREEN,
Color_RED,
Color_YELLOW,
Color_BLUE,
Color_WHITE,
Color_MAGENTA,
Color_CYAN,
Color_YELLOW,
Color_WHITE,
Color_GREEN,
Color_RED,
Color_YELLOW,
@ -113,11 +97,11 @@ R_API int r_print_pie(R_NULLABLE RPrint *p, int nvalues, int *values, const char
const char *bg_colors[10] = {
Color_BGGREEN,
Color_BGRED,
Color_BGYELLOW,
Color_BGBLUE,
Color_BGWHITE,
Color_BGMAGENTA,
Color_BGCYAN,
Color_BGYELLOW,
Color_BGWHITE,
Color_BGGREEN,
Color_BGRED,
Color_BGYELLOW,
@ -133,67 +117,15 @@ R_API int r_print_pie(R_NULLABLE RPrint *p, int nvalues, int *values, const char
}
}
int legend_idx = 0;
if (nv && out) {
#if 0
ut64 total = 0;
for (i = 0; i < nvalues; i++) {
total += values[i];
}
total /= 100;
if (total < 1) {
total = 1;
}
for (i = 0; i < nvalues; i++) {
nv[i] = values[i] / total;
}
for (x = 0; x < size; x++) {
for (y = 0; y < size; y++) {
O = ' ';
}
}
#if USE_SINCOS
float a = 0.0;
int s = size / 2;
while (a < 2 * PI) {
x = s * cos (a) + (size / 2);
y = s * sin (a) + (size / 2);
O = '.';
a += 0.1;
}
#else
int radius = (size - 1) / 2;
// draw portions
for (x = 0; x <= 2 * radius; x++) {
for (y = 0; y <= 2 * radius; y++) {
double distance = sqrt ((double)(x - radius) * (x - radius) + (y - radius) * (y - radius));
O = (distance > radius - 0.5 && distance < radius + 0.5) ? '*' : ' ';
}
}
#endif
int amount = 0;
for (i = 0; i < nvalues; i++) {
fillSectorLine (out, size, nv[i] + amount, '0' + i);
amount += nv[i];
}
#if 0
for (i = 1; i < nvalues; i++) {
drawSectorLine (out, size, nv[i] + amount, ',');
amount += nv[i];
}
#endif
#endif
/// render
if (p && p->cb_printf) {
if (nv && out && p && p->cb_printf) {
for (y = 0; y < size; y++) {
for (x = 0; x < size; x++) {
if (use_colors && isdigit (O)) {
const int index = O - '0';
const char ch = out[x + (y * size)];
if (use_colors && isdigit (ch)) {
const int index = ch - '0';
p->cb_printf ("%s··"Color_RESET, fg_colors[index]);
} else {
p->cb_printf ("%c%c", O, O);
p->cb_printf ("%c%c", ch, ch);
}
}
if (y > 0 && legend_idx < nvalues) {
@ -214,15 +146,8 @@ R_API int r_print_pie(R_NULLABLE RPrint *p, int nvalues, int *values, const char
}
p->cb_printf ("\n");
}
}
p->cb_printf ("\n");
}
free (out);
free (nv);
return 0;
}
#if 0
main() {
ut64 v[] = { 10, 20, 30, 40 };
r_print_pie (NULL, &v, 4, 20);
}
#endif