"""
This file demonstrates a bokeh applet, which can either be viewed
directly on a bokeh-server, or embedded into a flask application.
See the README.md file in this directory for instructions on running.
"""

import logging

logging.basicConfig(level=logging.DEBUG)

from os import listdir
from os.path import dirname, join, splitext

import numpy as np
import pandas as pd
import datetime as datetime

from bokeh.models import ColumnDataSource, Plot


from bokeh.plotting import figure, curdoc
from bokeh.properties import String, Instance
from bokeh.server.app import bokeh_app
from bokeh.server.utils.plugins import object_page
from bokeh.models.widgets import (HBox, VBox, VBoxForm, PreText,
                                  Select, AppHBox, AppVBox, AppVBoxForm)

#==================================================================
from bokeh.models import Callback,Slider
from bokeh.models.glyphs import Circle
from bokeh.models import (
    GMapPlot, Range1d, ColumnDataSource, LinearAxis,
    PanTool, WheelZoomTool, BoxSelectTool,HoverTool,ResetTool,
    BoxSelectionOverlay, GMapOptions,ResizeTool,
    BoxZoomTool,NumeralTickFormatter, PrintfTickFormatter)
#==================================================================


from bokeh.simpleapp import simpleapp

slider_1 = Slider(name='ticker_slider_day',start=1, end=18, value=1, step=1, title="\n Day")#,callback=None) #, callback=callback
select_job_site = Select(name='ticker_job_site', value='Site1', options=['Site1', 'Site2', 'Site3']) #
select1 = Select(name='ticker1', value='AAPL', options=['AAPL', 'GOOG', 'INTC', 'BRCM', 'YHOO'])
select2 = Select(name='ticker2', value='GOOG', options=['AAPL', 'GOOG', 'INTC', 'BRCM', 'YHOO'])

#@simpleapp(select1, select2)
#def stock(ticker1, ticker2):

@simpleapp(select_job_site,select1, select2,slider_1)
def stock(ticker_job_site,ticker1, ticker2,ticker_slider_day):
    pretext = PreText(text=" ", width=500)
    df = get_data(ticker1, ticker2,ticker_slider_day)


    pretext = PreText(text="", width=500)
    df = get_data(ticker1, ticker2,ticker_slider_day)
    source = ColumnDataSource(data=df)
    source.tags = ['main_source']
    p = figure(
        title="%s vs %s" % (ticker1, ticker2),
        plot_width=400, plot_height=400,
        tools="pan,wheel_zoom,box_select,reset",
        title_text_font_size="10pt",
    )
    p.circle(ticker1 + "_returns", ticker2 + "_returns",
             size=2,
             nonselection_alpha=0.02,
             source=source
    )




    # #==========Dummy JobSite Longitude and Latitude================#
    # if ticker_job_site == "Site1":
    #     Job_Site_1_lat=33.052343
    #     Job_Site_1_lng=-94.642177
    #     Job_Site_loc=[Job_Site_1_lat,Job_Site_1_lng]
    # if ticker_job_site == "Site2":
    #     Job_Site_1_lat=30.2861
    #     Job_Site_1_lng=-97.7394
    #     Job_Site_loc=[Job_Site_1_lat,Job_Site_1_lng]
    # if ticker_job_site == "Site3":
    #     Job_Site_1_lat=40.74118
    #     Job_Site_1_lng=-89.484877
    #     Job_Site_loc=[Job_Site_1_lat,Job_Site_1_lng]
    #
    #
    # map_options = GMapOptions(lat=Job_Site_loc[0], lng=Job_Site_loc[1], map_type="roadmap", zoom=14) #,, zoom=14
    #
    # x_range=None
    # y_range=None
    # x_range=Range1d()
    # y_range=Range1d()
    #
    #
    #
    # #=========== Google Map and AT lat and longitude Data =============#
    # lat_list=[33.052343]
    # long_list=[-94.642177]
    #
    # #========Dummy Loading and Dumping Zones=====================#
    # source_map = ColumnDataSource(
    #     data=dict(
    #         lat=[33.055, 33.0552, 33.056], #lat_list,  #
    #         lon=[-94.660, -94.640, -94.632], #longt_list,  #
    #         #bodyup=BodyUp_list,
    #
    #         fill=['orange', 'blue', 'green']
    #     )
    # )
    #
    # #=============================================================
    # plot=None
    # plot = GMapPlot(
    #     x_range=x_range, y_range=y_range,
    #     map_options=map_options,
    #     title = ticker_job_site,  #"HOUSTON SS99",
    #     plot_width=1200,
    #     plot_height=400,
    #
    # )
    # plot.map_options.map_type="hybrid"
    # #time.sleep(3)
    # circle = Circle(x="lon", y="lat", size=15, line_color="green",fill_color="fill") #fill_color="fill",
    # plot.add_glyph(source_map, circle)

    plot = create_map(ticker_job_site)

    # pan = PanTool()
    # wheel_zoom = WheelZoomTool()
    # box_zoom = BoxZoomTool()
    # #hover = HoverTool()
    # resize = ResizeTool()
    # reset = ResetTool()
    # plot.add_tools(pan,wheel_zoom, box_zoom,resize,reset)

    #=========== Google Map and AT lat and longitude Data =============#



















    row0 = HBox(children=[plot])

    stats = df.describe()
    pretext.text = str(stats)
    row1 = HBox(children=[p, pretext])
    hist1 = hist_plot(df, ticker1)
    hist2 = hist_plot(df, ticker2)
    row2 = HBox(children=[hist1, hist2])
    line1 = line_plot(ticker1, source)
    line2 = line_plot(ticker2, source, line1.x_range)




    output =  VBox(children=[row0,row1, row2, line1, line2])


    #output =  VBox(children=[row1, row2, line1, line2])
    # return output


    return dict(
        mainplot=plot,
        scatterplot=p,
            statstext=pretext,
            hist1=hist1,
            hist2=hist2,
            line1=line1,
            line2=line2)

stock.route("/bokeh/site1/")


@stock.layout
def stock_layout(app):
    widgets = AppVBoxForm(app=app, children=['ticker_slider_day', 'ticker1', 'ticker2'])
    row1 = AppHBox(app=app, children=['scatterplot', 'statstext'])
    row2 = AppHBox(app=app, children=['hist1', 'hist2'])
    all_plots = AppVBox(app=app, children=['mainplot', row1, row2, 'line1', 'line2'])
    app = AppHBox(app=app, children=[widgets, all_plots])
    return app


def create_map(ticker_job_site):
    if ticker_job_site == "Site1":
        Job_Site_1_lat=33.052343
        Job_Site_1_lng=-94.642177
        Job_Site_loc=[Job_Site_1_lat,Job_Site_1_lng]
    if ticker_job_site == "Site2":
        Job_Site_1_lat=30.2861
        Job_Site_1_lng=-97.7394
        Job_Site_loc=[Job_Site_1_lat,Job_Site_1_lng]
    if ticker_job_site == "Site3":
        Job_Site_1_lat=40.74118
        Job_Site_1_lng=-89.484877
        Job_Site_loc=[Job_Site_1_lat,Job_Site_1_lng]


    map_options = GMapOptions(lat=Job_Site_loc[0], lng=Job_Site_loc[1], map_type="roadmap", zoom=14) #,, zoom=14

    x_range=None
    y_range=None
    x_range=Range1d()
    y_range=Range1d()



    #=========== Google Map and AT lat and longitude Data =============#
    lat_list=[33.052343]
    long_list=[-94.642177]

    #========Dummy Loading and Dumping Zones=====================#
    source_map = ColumnDataSource(
        data=dict(
            lat=[33.055, 33.0552, 33.056], #lat_list,  #
            lon=[-94.660, -94.640, -94.632], #longt_list,  #
            #bodyup=BodyUp_list,

            fill=['orange', 'blue', 'green']
        )
    )

    #=============================================================
    plot=None
    plot = GMapPlot(
        x_range=x_range, y_range=y_range,
        map_options=map_options,
        title = ticker_job_site,  #"HOUSTON SS99",
        plot_width=1200,
        plot_height=400,

    )
    plot.map_options.map_type="hybrid"
    #time.sleep(3)
    circle = Circle(x="lon", y="lat", size=15, line_color="green",fill_color="fill") #fill_color="fill",
    plot.add_glyph(source_map, circle)

    pan = PanTool()
    wheel_zoom = WheelZoomTool()
    box_zoom = BoxZoomTool()
    #hover = HoverTool()
    resize = ResizeTool()
    reset = ResetTool()
    plot.add_tools(pan,wheel_zoom, box_zoom,resize,reset)

    return plot

@stock.update(['ticker_slider_day'])
def stock_update_input(ticker_job_site,ticker1, ticker2,ticker_slider_day):
    plot = create_map(ticker_job_site)

    return {
        'mainplot': plot
    }



def hist_plot(df, ticker, selected_df=None):
    if selected_df is None:
        selected_df = df
    global_hist, global_bins = np.histogram(df[ticker + "_returns"], bins=50)
    hist, bins = np.histogram(selected_df[ticker + "_returns"], bins=50)
    width = 0.7 * (bins[1] - bins[0])
    center = (bins[:-1] + bins[1:]) / 2
    start = global_bins.min()
    end = global_bins.max()
    top = hist.max()
    p = figure(
        title="%s hist" % ticker,
        plot_width=500, plot_height=200,
        tools="",
        title_text_font_size="10pt",
        x_range=[start, end],
        y_range=[0, top],
    )
    p.rect(center, hist / 2.0, width, hist)
    return p

def line_plot(ticker, source, x_range=None):
    p = figure(
        title=ticker,
        x_range=x_range,
        x_axis_type='datetime',
        plot_width=1000, plot_height=200,
        title_text_font_size="10pt",
        tools="pan,wheel_zoom,box_select,reset"
    )
    p.circle(
        'date', ticker,
        size=2,
        source=source,
        nonselection_alpha=0.02
    )
    return p

# build up list of stock data in the daily folder
data_dir = join(dirname(__file__), "daily")
try:
    tickers = listdir(data_dir)
except OSError as e:
    print('Stock data not available, see README for download instructions.')
    raise e
tickers = [splitext(x)[0].split("table_")[-1] for x in tickers]

# cache stock data as dict of pandas DataFrames
pd_cache = {}


def get_ticker_data(ticker,ticker_slider_day):
    fname = join(data_dir, "table_%s.csv" % ticker.lower())
    data = pd.read_csv(
        fname,
        names=['date', 'foo', 'o', 'h', 'l', 'c', 'v'],
        header=False,
        parse_dates=['date']
    )
    data = data.set_index('date')
    data = pd.DataFrame({ticker: data.c, ticker + "_returns": data.c.diff()})
    return data



def get_data(ticker1, ticker2,ticker_slider_day):
    if pd_cache.get((ticker1, ticker2)) is not None:
        return pd_cache.get((ticker1, ticker2))

    # only append columns if it is the same ticker
    if ticker1 != ticker2:
        print (" @get data: if not")
        data1 = get_ticker_data(ticker1,ticker_slider_day)
        data2 = get_ticker_data(ticker2,ticker_slider_day)
        data = pd.concat([data1, data2], axis=1)
    else:

        print (" @get data: else")
        data = get_ticker_data(ticker1,ticker_slider_day)

    data = data.dropna()
    pd_cache[(ticker1, ticker2)] = data
    return data































##
##
##
##
##
##
##
##@simpleapp(select1, select2)
##def stock2(ticker1, ticker2):
##    pretext = PreText(text="", width=500)
##    df = get_data(ticker1, ticker2)
##    source = ColumnDataSource(data=df)
##    source.tags = ['main_source']
##    p = figure(
##        title="%s vs %s" % (ticker1, ticker2),
##        plot_width=400, plot_height=400,
##        tools="pan,wheel_zoom,box_select,reset",
##        title_text_font_size="10pt",
##    )
##    p.circle(ticker1 + "_returns", ticker2 + "_returns",
##             size=2,
##             nonselection_alpha=0.02,
##             source=source
##    )
##
##
##
##
##
##
##
##
##
##
##
##
##
##
##
##
##
##
##
##
##
##
##    stats = df.describe()
##    pretext.text = str(stats)
##
##    row0 = HBox(children=[plot]) #, pretext
##
##    row1 = HBox(children=[p,pretext]) #, pretext
##
##
##
##    hist1 = hist_plot(df, ticker1)
##    hist2 = hist_plot(df, ticker2)
##    line1 = line_plot(ticker1, source)
##    line2 = line_plot(ticker2, source, line1.x_range)
##    return dict(scatterplot=p,
##                statstext=pretext,
##                hist1=hist1,
##                hist2=hist2,
##                line1=line1,
##                line2=line2)
##
##@stock2.layout
##def stock2_layout(app):
##    widgets = AppVBoxForm(app=app, children=['ticker1', 'ticker2'])
##    row1 = AppHBox(app=app, children=['scatterplot', 'statstext'])
##    row2 = AppHBox(app=app, children=['hist1', 'hist2'])
##    all_plots = AppVBox(app=app, children=[row1, row2, 'line1', 'line2'])
##    app = AppHBox(app=app, children=[widgets, all_plots])
##    return app
##
##@stock2.update(['ticker1', 'ticker2'])
##def stock2_update_input(ticker1, ticker2, app):
##    return stock2(ticker1, ticker2)
##
##@stock2.update([({'tags' : 'main_source'}, ['selected'])])
##def stock2_update_selection(ticker1, ticker2, app):
##    source = app.select_one({'tags' : 'main_source'})
##    df = get_data(ticker1, ticker2)
##    if source.selected:
##        selected_df = df.iloc[source.selected['1d']['indices'], :]
##    else:
##        selected_df = df
##
##    stats_text = app.objects['statstext']
##    stats_text.text = str(selected_df.describe())
##    return {
##        'hist1': hist_plot(df, ticker1, selected_df=selected_df),
##        'hist2': hist_plot(df, ticker2, selected_df=selected_df),
##        'statstext': stats_text,
##    }
##
##stock2.route("/bokeh/jobsite_ss99_2/")
##
##def hist_plot(df, ticker, selected_df=None):
##    if selected_df is None:
##        selected_df = df
##    global_hist, global_bins = np.histogram(df[ticker + "_returns"], bins=50)
##    hist, bins = np.histogram(selected_df[ticker + "_returns"], bins=50)
##    width = 0.7 * (bins[1] - bins[0])
##    center = (bins[:-1] + bins[1:]) / 2
##    start = global_bins.min()
##    end = global_bins.max()
##    top = hist.max()
##    p = figure(
##        title="%s hist" % ticker,
##        plot_width=500, plot_height=200,
##        tools="",
##        title_text_font_size="10pt",
##        x_range=[start, end],
##        y_range=[0, top],
##    )
##    p.rect(center, hist / 2.0, width, hist)
##    return p
##
##def line_plot(ticker, source, x_range=None):
##    p = figure(
##        title=ticker,
##        x_range=x_range,
##        x_axis_type='datetime',
##        plot_width=1000, plot_height=200,
##        title_text_font_size="10pt",
##        tools="pan,wheel_zoom,box_select,reset"
##    )
##    p.circle(
##        'date', ticker,
##        size=2,
##        source=source,
##        nonselection_alpha=0.02
##    )
##    return p
##
### build up list of stock data in the daily folder
##data_dir = join(dirname(__file__), "daily")
##try:
##    tickers = listdir(data_dir)
##except OSError as e:
##    print('Stock data not available, see README for download instructions.')
##    raise e
##tickers = [splitext(x)[0].split("table_")[-1] for x in tickers]
##
### cache stock data as dict of pandas DataFrames
##pd_cache = {}
##
##
##def get_ticker_data(ticker,ticker_slider_day):
##    fname = join(data_dir, "table_%s.csv" % ticker.lower())
##    data = pd.read_csv(
##        fname,
##        names=['date', 'foo', 'o', 'h', 'l', 'c', 'v'],
##        header=False,
##        parse_dates=['date']
##    )
##    data = data.set_index('date')
##    data = pd.DataFrame({ticker: data.c, ticker + "_returns": data.c.diff()})
##    return data
##
##
##
##def get_data(ticker1, ticker2,ticker_slider_day):
##    if pd_cache.get((ticker1, ticker2)) is not None:
##        return pd_cache.get((ticker1, ticker2))
##
##    # only append columns if it is the same ticker
##    if ticker1 != ticker2:
##        print (" @get data: if not")
##        data1 = get_ticker_data(ticker1,ticker_slider_day)
##        data2 = get_ticker_data(ticker2,ticker_slider_day)
##        data = pd.concat([data1, data2], axis=1)
##    else:
##
##        print (" @get data: else")
##        data = get_ticker_data(ticker1,ticker_slider_day)
##
##    data = data.dropna()
##    pd_cache[(ticker1, ticker2)] = data
##    return data
##
##
####def get_data(ticker1, ticker2):
####    if pd_cache.get((ticker1, ticker2)) is not None:
####        return pd_cache.get((ticker1, ticker2))
####
####    # only append columns if it is the same ticker
####    if ticker1 != ticker2:
####        data1 = get_ticker_data(ticker1)
####        data2 = get_ticker_data(ticker2)
####        data = pd.concat([data1, data2], axis=1)
####    else:
####        data = get_ticker_data(ticker1)
####
####    data = data.dropna()
####    pd_cache[(ticker1, ticker2)] = data
####    return data
##
##
##
### class StockApp(VBox):
###     extra_generated_classes = [["StockApp", "StockApp", "VBox"]]
###     jsmodel = "VBox"
##
###     # text statistics
###     pretext = Instance(PreText)
##
###     # plots
###     plot = Instance(Plot)
###     line_plot1 = Instance(Plot)
###     line_plot2 = Instance(Plot)
###     hist1 = Instance(Plot)
###     hist2 = Instance(Plot)
##
###     # data source
###     source = Instance(ColumnDataSource)
##
###     # layout boxes
###     mainrow = Instance(HBox)
###     histrow = Instance(HBox)
###     statsbox = Instance(VBox)
##
###     # inputs
###     ticker1 = String(default="AAPL")
###     ticker2 = String(default="GOOG")
###     ticker1_select = Instance(Select)
###     ticker2_select = Instance(Select)
###     input_box = Instance(VBoxForm)
##
###     def __init__(self, *args, **kwargs):
###         super(StockApp, self).__init__(*args, **kwargs)
###         self._dfs = {}
##
###     @classmethod
###     def create(cls):
###         """
###         This function is called once, and is responsible for
###         creating all objects (plots, datasources, etc)
###         """
###         # create layout widgets
###         obj = cls()
##
###         # create input widgets
###         obj.make_inputs()
##
###         # outputs
###         obj.pretext = PreText(text="", width=500)
###         obj.make_source()
###         obj.make_plots()
###         obj.make_stats()
##
###         # layout
###         obj.set_children()
###         return obj
##
###     def make_inputs(self):
##
###         self.ticker1_select = Select(
###             name='ticker1',
###             value='AAPL',
###             options=['AAPL', 'GOOG', 'INTC', 'BRCM', 'YHOO']
###         )
###         self.ticker2_select = Select(
###             name='ticker2',
###             value='GOOG',
###             options=['AAPL', 'GOOG', 'INTC', 'BRCM', 'YHOO']
###         )
##
###     @property
###     def selected_df(self):
###         pandas_df = self.df
###         selected = self.source.selected
###         if selected:
###             pandas_df = pandas_df.iloc[selected, :]
###         return pandas_df
##
###     def make_source(self):
###         self.source = ColumnDataSource(data=self.df)
##
###     def line_plot(self, ticker, x_range=None):
###         p = figure(
###             title=ticker,
###             x_range=x_range,
###             x_axis_type='datetime',
###             plot_width=1000, plot_height=200,
###             title_text_font_size="10pt",
###             tools="pan,wheel_zoom,box_select,reset"
###         )
###         p.circle(
###             'date', ticker,
###             size=2,
###             source=self.source,
###             nonselection_alpha=0.02
###         )
###         return p
##
###     def hist_plot(self, ticker):
###         global_hist, global_bins = np.histogram(self.df[ticker + "_returns"], bins=50)
###         hist, bins = np.histogram(self.selected_df[ticker + "_returns"], bins=50)
###         width = 0.7 * (bins[1] - bins[0])
###         center = (bins[:-1] + bins[1:]) / 2
###         start = global_bins.min()
###         end = global_bins.max()
###         top = hist.max()
##
###         p = figure(
###             title="%s hist" % ticker,
###             plot_width=500, plot_height=200,
###             tools="",
###             title_text_font_size="10pt",
###             x_range=[start, end],
###             y_range=[0, top],
###         )
###         p.rect(center, hist / 2.0, width, hist)
###         return p
##
###     def make_plots(self):
###         ticker1 = self.ticker1
###         ticker2 = self.ticker2
###         p = figure(
###             title="%s vs %s" % (ticker1, ticker2),
###             plot_width=400, plot_height=400,
###             tools="pan,wheel_zoom,box_select,reset",
###             title_text_font_size="10pt",
###         )
###         p.circle(ticker1 + "_returns", ticker2 + "_returns",
###                  size=2,
###                  nonselection_alpha=0.02,
###                  source=self.source
###         )
###         self.plot = p
##
###         self.line_plot1 = self.line_plot(ticker1)
###         self.line_plot2 = self.line_plot(ticker2, self.line_plot1.x_range)
###         self.hist_plots()
##
###     def hist_plots(self):
###         ticker1 = self.ticker1
###         ticker2 = self.ticker2
###         self.hist1 = self.hist_plot(ticker1)
###         self.hist2 = self.hist_plot(ticker2)
##
###     def set_children(self):
###         self.children = [self.mainrow, self.histrow, self.line_plot1, self.line_plot2]
###         self.mainrow.children = [self.input_box, self.plot, self.statsbox]
###         self.input_box.children = [self.ticker1_select, self.ticker2_select]
###         self.histrow.children = [self.hist1, self.hist2]
###         self.statsbox.children = [self.pretext]
##
###     def input_change(self, obj, attrname, old, new):
###         if obj == self.ticker2_select:
###             self.ticker2 = new
###         if obj == self.ticker1_select:
###             self.ticker1 = new
##
###         self.make_source()
###         self.make_plots()
###         self.set_children()
###         curdoc().add(self)
##
###     def setup_events(self):
###         super(StockApp, self).setup_events()
###         if self.source:
###             self.source.on_change('selected', self, 'selection_change')
###         if self.ticker1_select:
###             self.ticker1_select.on_change('value', self, 'input_change')
###         if self.ticker2_select:
###             self.ticker2_select.on_change('value', self, 'input_change')
##
###     def make_stats(self):
###         stats = self.selected_df.describe()
###         self.pretext.text = str(stats)
##
###     def selection_change(self, obj, attrname, old, new):
###         self.make_stats()
###         self.hist_plots()
###         self.set_children()
###         curdoc().add(self)
##
###     @property
###     def df(self):
###         return get_data(self.ticker1, self.ticker2)
##
##
### # The following code adds a "/bokeh/stocks/" url to the bokeh-server. This URL
### # will render this StockApp. If you don't want serve this applet from a Bokeh
### # server (for instance if you are embedding in a separate Flask application),
### # then just remove this block of code.
### @bokeh_app.route("/bokeh/stocks/")
### @object_page("stocks")
### def make_object():
###     app = StockApp.create()
###     return app