plotext-plus 1.0.1__py3-none-any.whl

plotext_plus/_global.py

@@ -0,0 +1,370 @@
+# This file contains some plotext functions which are only available to the top main level and not to sub figures (which are written in _figure.py and _monitor.py). These are functions which requires some coding and would be too long to be added directly in _core.py
+
+from plotext_plus._utility import marker_codes, hd_symbols, sin
+from plotext_plus._figure import _figure_class
+from plotext_plus._utility import themes as _themes
+import plotext_plus._utility as ut
+from time import time, sleep
+from math import sqrt, ceil
+import datetime as dt
+
+figure = _figure_class() # the main figure at top level
+
+##############################################
+#######     Simple  Bar Functions    ########
+##############################################
+
+def simple_bar(*args, width = None, marker = None, color = None, title = None):
+    x, y = ut.set_data(*args)
+    marker = ut.correct_marker(marker)
+
+    color_ok = ut.is_color(color) or (isinstance(color, list) and len(color) == len(x))
+    color = [color] if color_ok else None
+
+    simple_stacked_bar(x, [y], width = width, marker = marker, colors = color, title = title)
+
+def simple_stacked_bar(*args, width = None, marker = None, colors = None, title = None, labels = None):
+    x, y, Y, width = ut.bar_data(*args, width = width)
+    marker = ut.correct_marker(marker)
+    
+    bars = len(Y); stacked_bars = len(Y[0])
+
+    colors_ok1 = isinstance(colors, list) and isinstance(colors[0], list) and ut.matrix_size(colors) == [bars, stacked_bars]
+    colors_ok2 = isinstance(colors, list) and len(colors) == stacked_bars
+    colors = ut.transpose(colors) if colors_ok1 else [colors] * bars if colors_ok2 else [ut.color_sequence[:stacked_bars]] * bars
+
+    title = ut.get_title(title, width)
+    bars = [ut.single_bar(x[i], Y[i], y[i], marker, colors[i]) for i in range(bars)]
+    labels = ut.get_simple_labels(marker, labels, colors[0], width)
+    figure.monitor.matrix.canvas = title + '\n'.join(bars) + labels 
+    figure.monitor.fast_plot = True
+
+def simple_multiple_bar(*args, width = None, marker = None, colors = None, title = None, labels = None):
+    x, y, Y, width = ut.bar_data(*args, width = width, mode='multiple')
+    bars = len(Y); multiple_bars = len(Y[0]); lx = len(x[0])
+    marker = ut.correct_marker(marker)
+
+    colors_ok = isinstance(colors, list) and len(colors) == multiple_bars
+    colors = colors if colors_ok else ut.color_sequence[:multiple_bars]
+
+    out = ut.get_title(title, width)
+    for i in range(bars):
+        xn = [x[i] if j == (multiple_bars - 1) // 2 else ut.space * lx for j in range(multiple_bars)]
+        new = [ut.single_bar(xn[j], [Y[i][j]], y[j][i], marker, [colors[j]]) for j in range(multiple_bars)]
+        out += '\n'.join(new)
+        out += '\n\n' if i != bars - 1 else ''
+    labels = ut.get_simple_labels(marker, labels, colors, width)
+    figure.monitor.matrix.canvas = out + labels
+    figure.monitor.fast_plot = True
+
+##############################################
+#############     Play GIF    ################
+##############################################
+
+def play_gif(path):
+    from PIL import Image, ImageSequence
+    path = ut.correct_path(path)
+    if not ut.is_file(path):
+        return
+    im = Image.open(path)
+    index = 1
+    for image in ImageSequence.Iterator(im):
+        load_time = time()
+        figure.clt()
+        image = image.convert('RGB')
+        figure.monitor._draw_image(image, fast = True)
+        figure.show()
+        load_time = time() - load_time
+        frame_time = image.info['duration'] / 10 ** 3
+        if load_time < frame_time:
+            sleep(frame_time - load_time)
+                
+##############################################
+##########     Video Functions    ############
+##############################################
+
+def play_video(path, from_youtube = False):
+    path = ut.correct_path(path)
+    if not ut.is_file(path):
+        return
+    _play_video(path, from_youtube)
+
+def play_youtube(url):
+    import pafy
+    video = pafy.new(url)
+    best = video.getbest()
+    _play_video(best.url, from_youtube = True)
+
+def get_youtube(url, path, log):
+    import pafy
+    video = pafy.new(url)
+    best = video.getbest(preftype = "mp4")
+    path = "youtube-video.mp4" if path is None else path
+    path = ut.correct_path(path)
+    best.download(filepath = path, quiet = not log)
+    print(ut.format_strings('YouTube video downloaded as', path)) if log else None
+
+def _play_video(path, from_youtube = False):
+    import cv2
+    from ffpyplayer.player import MediaPlayer
+    from PIL import Image
+    cap = cv2.VideoCapture(path)
+    player = MediaPlayer(path)#, paused = True, loglevel = 'quiet');
+    fr = 0;
+    while fr == 0:
+        fr = cap.get(cv2.CAP_PROP_FPS)
+    frame_time = 1 / fr 
+    #to_list = lambda frame: [[tuple(int(el) for el in tup) for tup in row] for row in frame]
+    pt = lambda time: '{time:05.1f}  '.format(time=round(10 ** 3 * time, 1))
+    real_time = video_time = 0
+    while True:
+        load_time = time()
+        check_video, frame = cap.read();
+        audio, check_audio = player.get_frame(show = False)
+        load_time = time() - load_time
+        if not check_video:
+            break
+        if load_time >= frame_time:
+            continue
+        real_time += load_time
+        video_time += frame_time
+        show_time = 0
+        shown = False
+        if video_time >= real_time:
+            shown = True
+            show_time = time()
+            frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB) if from_youtube else frame
+            #frame = to_list(frame)
+            image = Image.fromarray(frame)
+            
+            # Synchronized frame rendering - ensure each step completes before next
+            figure.clt()
+            # Small delay to ensure clear operation completes
+            sleep(0.001)  # 1ms delay
+            figure.monitor._draw_image(image, fast = True)
+            # Small delay to ensure image drawing completes before display
+            sleep(0.001)  # 1ms delay
+            figure.show()
+            # Brief delay after display to prevent frame overlap
+            sleep(0.002)  # 2ms delay for terminal rendering
+            
+            show_time = time() - show_time
+        sleep_time = 0
+        if real_time < video_time:
+            sleep_time = time()
+            sleep(video_time - real_time)
+            sleep_time = time() - sleep_time
+        total_time = load_time + show_time + sleep_time
+        real_time += show_time + sleep_time
+        #print('load: ' + pt(load_time), 'show: ' + pt(show_time), 'sleep: ' + pt(sleep_time), 'total: ' + pt(total_time), 'frame: ' + pt(frame_time), 'real: ' + pt(real_time), 'video: ' + pt(video_time), 'r/v:', round(real_time / video_time, 3)) if shown else None
+    player.close_player()
+    cap.release()
+    cv2.destroyAllWindows()
+    figure.clf()
+
+##############################################
+############     Utilities     ###############
+##############################################
+
+test_data_url = "https://raw.githubusercontent.com/piccolomo/plotext/master/data/data.txt"
+test_bar_data_url = "https://raw.githubusercontent.com/piccolomo/plotext/master/data/bar_data.txt"
+test_image_url  = "https://raw.githubusercontent.com/piccolomo/plotext/master/data/cat.jpg"
+test_gif_url    = "https://raw.githubusercontent.com/piccolomo/plotext/master/data/homer.gif"
+test_video_url  = "https://raw.githubusercontent.com/piccolomo/plotext/master/data/moonwalk.mp4"
+test_youtube_url = 'https://www.youtube.com/watch?v=ZNAvVVc4b3E&t=75s'
+
+##############################################
+#########     Matplotlib Backend    ##########
+##############################################
+
+def from_matplotlib(fig, marker = None):
+    fig.canvas.draw()
+    slots = (rows, cols) = fig.axes[0].get_subplotspec().get_gridspec().get_geometry()
+    figure.clf(); #clt()
+    figure.subplots(*slots)
+    round10 = lambda data: [round(el, 10) for el in data]
+    to_rgb = lambda rgb_norm: tuple([round(255 * el) for el in rgb_norm[:3]])
+    figure.axes_color(to_rgb(fig.patch.get_facecolor()))
+    for sub in fig.axes[:]:
+        p = sub.get_subplotspec().get_geometry()[2]
+        row = int((p - 0) / cols + 1)
+        col = p + 1 - (row - 1) * cols
+        monitor = figure.subplot(row, col)
+        monitor.xlabel(sub.get_xlabel())
+        monitor.ylabel(sub.get_ylabel())
+        monitor.title(sub.get_title())
+        monitor.xscale(sub.get_xscale())
+        monitor.yscale(sub.get_yscale())
+        monitor.xticks(round10(sub.get_xticks()))
+        monitor.yticks(round10(sub.get_yticks()))
+        monitor.canvas_color(to_rgb(sub.get_facecolor()))
+        for point in sub.collections:
+            label = point.get_label()
+            label = label if label[0] != '_' else ''
+            #point.set_offset_position('data')
+            x, y = ut.transpose(point.get_offsets())
+            color = [ut.to_rgb(point.to_rgba(el)) for el in point.get_facecolors()[0]]
+            # can't find the right point colors
+            monitor.scatter(x, y, label = label, marker = marker)
+        for line in sub.get_lines():
+            label = line.get_label()
+            label = label if label[0] != '_' else ''
+            x, y = line.get_data()
+            monitor.plot(x, y, marker = marker, color = line.get_c(), label = label)
+        for b in sub.patches:
+            label = b.get_label()
+            label = label if label[0] != '_' else ''
+            color = b.get_facecolor()
+            color = ut.to_rgb(color)
+            box = b.get_bbox()
+            x0, y0, x1, y1 = box.x0, box.y0, box.x1, box.y1
+            x = [x0, x0, x1, x1, x0]
+            y = [y0, y1, y1, y0, y0]
+            fill = b.get_fill()
+            fillx = fill if y0 == 0 else False
+            filly = fill if x0 == 0 else False
+            monitor.plot(x, y, fillx = fillx, filly = filly, marker = marker, color = color, label = label)
+        monitor.xlim(*sub.get_xlim())
+        monitor.ylim(*sub.get_ylim())
+
+##############################################
+#######     Presentation Functions    ########
+##############################################
+        
+def markers():
+    markers = list(hd_symbols.keys())[::-1] + list(marker_codes.keys())
+    l = len(markers)
+    rows = int(sqrt(l))
+    cols = ceil(l / rows)
+    y = ut.sin(1)
+    figure.clf(); figure.theme('pro'); figure.xfrequency(0); figure.yfrequency(0); figure.frame(1)
+    figure.subplots(rows, cols)
+    figure.frame(0)
+    for row in range(1, rows + 1):
+        for col in range(1, cols + 1):
+            i = (row - 1) * cols + col - 1
+            if i < l:
+                subplot = figure.subplot(row, col)
+                figure.frame(1)
+                default = ' [default]' if markers[i] == 'hd' else ''
+                subplot.title(markers[i] + default)
+                subplot.scatter(y, marker = markers[i])
+                subplot.ticks_style('bold')
+                #figure.ticks_color(figure._utility.title_color)
+    figure.show()
+    figure.clf()
+
+def colors():
+    print(ut.colorize("String Color Codes", style = 'bold'))
+    bg = "default"
+    c = ut.no_duplicates([el.replace('+', '') for el in ut.colors if el not in ['default', 'black', 'white']])
+    cp = [ut.colorize(ut.pad_string(el + '+', 10), el + '+', background = bg) for el in c]
+    c = [ut.colorize(ut.pad_string(el, 10), el, background = bg) for el in c]
+    c = [' ' + c[i]  + cp[i] for i in range(len(c))]
+    c = '\n'.join(c)
+    print(' ' + ut.colorize(ut.pad_string('default', 20), background = bg))
+    print(' ' + ut.colorize(ut.pad_string('black', 10), 'black', background = 'gray') + ut.colorize(ut.pad_string('white', 10), 'white', background = bg))
+    print(c)
+    print()
+    #print(colorize("\n\nInteger Color Codes:", style = ''))
+    c = ut.colorize("Integer Color Codes", style = 'bold', show = False) + '\n'
+    for row in range(16):
+        cr = ' '
+        for col in range(16):
+            i = row * 16 + col
+            cr += ut.colorize(ut.pad_string(i, 5), i)
+        c += cr + '\n'
+    print(c)
+    c = '\n'
+    rgb = (100, 200, 85)
+    rgb_string = '(' + ', '.join([str(el) for el in rgb]) + ')'
+    print(ut.colorize("RGB Tuples like:", style = "bold"), ut.colorize(rgb_string, rgb, "bold"))
+
+def styles():
+    from plotext_plus._utility import styles, colorize, title_color
+    c = [colorize(el, style = el) for el in styles]
+    c = '\n'.join(c)
+    print(c)
+    mul = 'bold italic dim'
+    print('\n' + colorize('multiple styles are accepted, ', title_color) + 'eg: ' + colorize(mul, style = mul))
+
+def themes():
+    themes = list(_themes.keys())[::]
+    l = len(themes)
+    rows = int(sqrt(l))
+    cols = ceil(l / rows)
+    y1 = ut.sin(periods = 1)
+    y2 = ut.sin(periods = 1, phase = -1)
+    figure.clf()
+    figure.subplots(rows, cols)
+    for row in range(1, rows + 1):
+        for col in range(1, cols + 1):
+            i = (row - 1) * cols + col - 1
+            if i < l:
+                subplot = figure.subplot(row, col)
+                subplot.theme(themes[i])
+                subplot.title(themes[i])
+                subplot.scatter(y1); subplot.plot(y2)
+    figure.show()
+    figure.clf()
+
+##############################################
+###########     Test Function    #############
+##############################################
+
+def test():
+    import random
+    figure.clf(); figure.clt()
+    figure.date_form("d/m/Y");
+    figure.take_min()
+    figure.plot_size(None, ut.terminal_height())
+    #figure.plot_size(108, 70)
+    
+    figure.plotsize(ut.tw(), ut.th() - 5)
+    figure.subplots(2, 2)
+    
+    subplot = figure.subplot(1, 1)
+    subplot.title("Multiple Axes Plot")
+    subplot.canvas_color(66); subplot.axes_color(4); subplot.ticks_color(216); subplot.ticks_style('bold italic')
+    y = ut.sin(periods = 1); l = len(y)
+    subplot.scatter(y, label = "lower left")
+    x = [figure.today_datetime() + dt.timedelta(days = i) for i in range(l)]; x = figure.datetimes_to_strings(x)
+    subplot.plot(x, x, label = 'upper right - all dates', xside = 2, yside = 2)
+    subplot.vline(l / 2, 'red')
+    subplot.hline(0, 200)
+    subplot.text("origin", l // 2, 0, color = 'red', alignment = 'center')
+    subplot.xlabel('x lower'); subplot.xlabel('x upper', 2)
+    subplot.ylabel('y left', 'left'); subplot.ylabel('y right', 'right')
+    subplot.xfrequency(8); subplot.xfrequency(5, 2);
+    subplot.yfrequency(3); subplot.yfrequency(5, 2);
+    subplot.grid(1,1)
+    
+    subplot = figure.subplot(1, 2)
+    subplot.theme('innocent')
+    xb = ["Sausage", "Pepperoni", "Mushrooms", "Cheese", "Chicken", "Beef"]
+    y1 = [36, 14, 11, 8, 7, 4]
+    y2 = [20, 12, 35, 15, 4, 5]
+    subplot.stacked_bar(xb, [y1, y2], labels = ["men", "women"])
+
+    subplot = figure.subplot(2, 1)
+    subplot.theme('dreamland')
+    ld = 7 * 10 ** 4
+    data = [random.gauss(0, 1) for el in range(10 * ld)]
+    subplot.hist(data, bins = 60, label="mean 0")
+    subplot.frame(1); #subplot.xaxes(1, 0); subplot.yaxes(1, 0)
+    
+    subplot = figure.subplot(2, 2)
+    subplot.canvas_color('gray+'); subplot.axes_color('gray+')
+    ut.download(test_image_url, 'cat.jpg')
+    subplot.image_plot('cat.jpg', grayscale = False)
+    ut.delete_file('cat.jpg')
+    subplot.title('A very Cute Cat')
+    subplot.frame(0)
+
+    #figure.plotsize(0, 0)
+    figure.show()
+    figure._get_time()
+    figure.save_fig('test.txt')
+    figure.save_fig('test.html')
+    #figure.clf()
+

plotext_plus/_matrix.py

@@ -0,0 +1,171 @@
+import plotext_plus._utility as ut
+
+class matrix_class():
+    def __init__(self):
+        self.set_size(0, 0)
+        self.set_axes_color()
+        self.set_canvas_area(0, 0, 0, 0)
+        self.set_canvas_color()
+        self.set_matrices()
+        self.set_canvas()
+        
+    def set_size(self, cols, rows):
+        self.rows = rows; self.Rows = range(self.rows)
+        self.cols = cols; self.Cols = range(self.cols)
+        self.size = [self.rows, self.cols]
+
+    def update_size(self):
+        self.cols, self.rows = ut.matrix_size(self.marker)
+        self.Rows = range(self.rows); self.Cols = range(self.cols)
+        self.size = [self.rows, self.cols]
+
+    def set_axes_color(self, color = ut.no_color):
+        self.axes_color = color
+
+    def set_canvas_area(self, col1, col2, row1, row2):
+        self.Cols_canvas = range(col1, col2)
+        self.Rows_canvas = range(row1, row2)
+
+    def set_canvas_color(self, color = ut.no_color):
+        self.canvas_color = color
+
+    def in_canvas(self, col, row):
+        return col in self.Cols_canvas and row in self.Rows_canvas
+
+    def set_matrices(self):
+        self.marker = [[ut.space] * self.cols for row in self.Rows]
+        self.fullground = [[ut.no_color] * self.cols for row in self.Rows]
+        get_background = lambda col, row: self.canvas_color if self.in_canvas(col, row) else self.axes_color
+        self.background = [[get_background(col, row) for col in self.Cols] for row in self.Rows]
+        self.style = [[ut.no_color] * self.cols for row in self.Rows]
+
+    def legal(self, col, row):
+        return col in self.Cols and row in self.Rows
+
+    def get_marker(self, col, row):
+        return self.marker[row][col] #if self.legal(col, row) else "OUT"
+
+    def get_marker_row(self, row):
+        return self.marker[row] #if self.legal(0, row) else "OUT"
+
+    def get_marker_col(self, col):
+        return [self.marker[r][col] for r in self.Rows]#if self.legal(0, row) else "OUT"
+
+    def set_marker(self, col, row, marker):
+        self.marker[row][col] = marker
+        
+    def set_fullground(self, col, row, fullground = None):
+        self.fullground[row][col] = fullground
+        
+    def set_background(self, col, row, background = None):
+        self.background[row][col] = background
+            
+    def set_style(self, col, row, style = None):
+        self.style[row][col] = style
+
+    def insert_element(self, col, row, marker, fullground = None, style = None, background = None, check_canvas = False):
+        test_canvas = True if check_canvas is False else col in self.Cols_canvas and row in self.Rows_canvas
+        if self.legal(col, row) and test_canvas:
+            self.set_marker(col, row, marker)
+            self.set_fullground(col, row, fullground) if fullground is not None else None
+            self.set_background(col, row, background) if background is not None else None
+            self.set_style(col, row, style) if style is not None else None
+
+    def add_horizontal_string(self, col, row, string, fullground = None, style = None, background = None, alignment = "left", check_space = False, check_canvas = False):
+        l = len(string); L = range(l)
+        col = col if alignment == "left" else col - l // 2 if alignment == "center" else col - l + 1 if alignment == "right" else ut.correct_coord(self.get_marker_row(row), string, col) # if dynamic
+        b, e = max(col - 1, 0), min(col + l + 1, self.cols)
+        test_space = all([self.get_marker(c, row) == ut.space for c in range(b, e)]) and col >= 0 and col + l <= self.cols if check_space else True
+        [self.insert_element(col + i, row, string[i], fullground, style, background, check_canvas) for i in L] if test_space else None
+        return test_space
+
+    def add_vertical_string(self, col, row, string, fullground = None, style = None, background = None,  alignment = "bottom", check_canvas = False):
+        l = len(string); L = range(l)
+        row = row if alignment == "bottom" else row - l // 2 if alignment == "center" else row - l + 1 #if alignment == "top"
+        [self.insert_element(col, row + i, string[i], fullground, style, background, check_canvas) for i in L]
+
+    def add_multiple_horizontal_strings(self, col, row, string, fullground = None, style = None, background = None, alignment = "left", check_space = False, check_canvas = False):
+        strings = ''.join(string).split('\n'); S = len(strings)
+        [self.add_horizontal_string(col, row - s, strings[s], fullground, style, background, alignment, check_space, check_canvas) for s in range(S)]
+
+    def add_multiple_vertical_strings(self, col, row, string, fullground = None, style = None, background = None, alignment = "left", check_canvas = False):
+        strings = ''.join(string).split('\n'); S = len(strings)
+        [self.add_vertical_string(col + s, row, strings[s], fullground, style, background, alignment, check_canvas) for s in range(S)]
+
+    def get_colors(self, col, row):
+        return [self.fullground[row][col], self.style[row][col], self.background[row][col]] #if self.legal(col, row) else ["OUT"] * 3
+
+    def set_canvas(self):
+        canvas = ''
+        for row in self.Rows[::-1]:
+            for col in self.Cols:
+                marker = self.marker[row][col]
+                colors = self.get_colors(col, row)
+                if col == 0 or colors != self.get_colors(col - 1, row):
+                    ansi = ut.colors_to_ansi(*colors)
+                    canvas += ansi
+                canvas += marker
+                if col == self.cols - 1 or colors != self.get_colors(col + 1, row):
+                    #ansi_next = colors_to_ansi(*colors_next)
+                    canvas += ut.ansi_end  #+ ansi_next
+            canvas += '\n'
+
+        self.canvas = canvas
+        #print(repr(canvas))
+        #self.canvas = '\n'.join([''.join(self.marker[row]) for row in self.Rows])
+        return self.canvas
+
+    def get_canvas(self):
+        return self.canvas
+
+    def hstack(self, extra):
+        self.marker = ut.hstack(self.marker, extra.marker)
+        self.fullground = ut.hstack(self.fullground, extra.fullground)
+        self.background = ut.hstack(self.background, extra.background)
+        self.style = ut.hstack(self.style, extra.style)
+        self.update_size()
+        self.canvas = '';
+        
+    def vstack(self, extra):
+        self.marker = ut.vstack(self.marker, extra.marker)
+        self.fullground = ut.vstack(self.fullground, extra.fullground)
+        self.background = ut.vstack(self.background, extra.background)
+        self.style = ut.vstack(self.style, extra.style)
+        self.update_size()
+        self.canvas = ''
+
+    def to_html(self): # turns a matrix of character in html form
+        code = lambda color: "rgb" + str(color).replace(" ", "")
+        #html = "<body>\n<p style=\"font-family:courier; font-size:11pt;\">\n\n"
+        html = "<body> \n <code style = style=\"font-family:courier; \"font-size : 10pt;\"> \n\n"
+        for r in range(self.rows - 1, -1, -1):
+            for c in range(self.cols):
+                marker = self.get_marker(c, r)
+                color, style, background = self.get_colors(c, r)
+                marker = "&nbsp;" if marker == ut.space else marker
+                marker = '<b>' + marker + '</b>' if style == 'bold' else marker
+                marker = '<i>' + marker + '</i>' if style == 'italic' else marker
+                color = 'black' if color == ut.no_color else color
+                background = 'white' if background == ut.no_color else background
+                color = ut.to_rgb(color)
+                background = ut.to_rgb(background)
+                color = code(color)
+                background = code(background)
+                html += "<span style = \"color:" + color + "; background-color: " + background + "\">" + marker + "</span>"
+            html += " <br>\n\n"
+        html += "<code> \n </body>"
+        return html
+
+def join_matrices(matrices): # from a matrix of matrix_class() objects to a single matrix
+    cols, rows = ut.matrix_size(matrices)
+    M = matrix_class()
+    for r in range(rows):
+        m_rows = matrices[r][0].rows
+        m = matrix_class()
+        m.set_size(0, m_rows)
+        m.set_matrices()
+        for c in range(cols):
+            m.hstack(matrices[r][c])
+        M.vstack(m)
+    return M
+