plot.py

from __future__ import division

import plotly
from plotly.graph_objs import Scatter, Layout, Figure

# bad practice but cba to do proper refactor
from params import *

def calculate_coin_count(blocks):
    altcoin_coin_count = []
    block_rewards = calculate_block_rewards(blocks)
    for i, block in enumerate(blocks):
        if len(altcoin_coin_count) == 0:
            altcoin_coin_count.append(premine)
        else:
            altcoin_coin_count.append((block_rewards[i-1] * halving_interval_blocks) + altcoin_coin_count[i-1])
    return altcoin_coin_count

def calculate_block_rewards(blocks):
    block_rewards = []
    for i, block in enumerate(blocks):
        if len(block_rewards) == 0:
            block_rewards.append(initial_block_reward)
        else:
            block_rewards.append(block_rewards[i-1]/2)
    return block_rewards

def calculate_years(blocks):
    seconds_per_year = 31540000
    years = []
    for i, block in enumerate(blocks):
        if len(years) == 0:
            years.append(init_year)
        else:
            years.append(init_year + ((blocks[i] * block_interval_secs)/seconds_per_year)) 
    return years 

blocks = list(range(0, halving_interval_blocks * halving_count, halving_interval_blocks))
print("blocks", blocks)
print("block reward", calculate_block_rewards(blocks))
print("coin count", calculate_coin_count(blocks))
print("years", calculate_years(blocks))

altcoin_coins = Scatter(
    x=blocks,
    y=calculate_coin_count(blocks),
    name='altcoin_coins'
)
block_reward = Scatter(
    x=calculate_years(blocks),
    y=calculate_block_rewards(blocks),
    name='block_reward',
    xaxis='x2',
    yaxis='y2'
) 

data = [altcoin_coins, block_reward]

layout = Layout(
    title='Altcoin Distribution Schedule',
    xaxis=dict(
        title='Blocks',
        titlefont=dict(
            color='rgb(148,103,189)'
        ),
        tickfont=dict(
            color='rgb(148,103,189)'
        ),
        domain=[0, .45]
    ),
    xaxis2=dict(
        title='Year',
        titlefont=dict(
            color='rgb(148,103,189)'
        ),
        tickfont=dict(
            color='rgb(148,103,189)'
        ),
        domain=[.55,1]
    ),
    yaxis2=dict(
        title='Block Reward',
        titlefont=dict(
            color='rgb(253, 127, 40)'
        ),
        tickfont=dict(
            color='rgb(253, 127, 40)'
        ),
	anchor='x2',
	overlaying='y',
	side='right'
    )
)

fig = Figure(data=data, layout=layout)
plotly.offline.plot(fig, filename='index.html')