darkfi/script/research/lotterysim/draw.py

import matplotlib.pyplot as plt
import numpy as np
import os

LEAD_FILE = 'log'+os.sep+"f_feedback.hist"
F_FILE = 'log'+os.sep+"f_output.hist"

LEAD_PROCESSED_IMG = 'img'+os.sep+"feedback_history_processed.png"
F_PROCESSED_IMG = 'img'+os.sep+"output_history_processed.png"

SEP = ","
NODES = 1000 # number of nodes logged

def draw():
    with open(LEAD_FILE) as f:
        buf = f.read()
        nodes = buf.split(SEP)[:-1]
        node_log = []
        for i in range(0, len(nodes)):
            node_log+=[int(float(nodes[i]))]
        freq_single_lead = sum(np.array(node_log)==1)/float(len(node_log))
        print("single leader frequency: {}".format(freq_single_lead))
        plt.plot(node_log)
        plt.legend(['#leads'])
        plt.savefig(LEAD_PROCESSED_IMG)


    with open(F_FILE) as f:
        buf = f.read()
        nodes = buf.split(SEP)[:-1]
        node_log = []
        for i in range(0, len(nodes)):
            node_log+=[float(nodes[i])]
        plt.plot(node_log)
        plt.legend(['#leads', 'f'])
        plt.savefig(F_PROCESSED_IMG)