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

class FatRedOrangeLion(QCAlgorithm):

    def Initialize(self):
        self.period = 63
        self.SetStartDate(2014, 1, 1)
        self.SetEndDate(2022, 1, 1)
        self.SetCash(100000)
        self.SetWarmup(self.period)
        
        self.stock = self.AddEquity('AAPL', Resolution.Daily).Symbol
        self.volatility = Volatility(self.period)
        
        #self.RegisterIndicator(self.stock, self.volatility, Resolution.Daily)
        #self.RegisterIndicator(self.stock, self.volatility, Resolution.Daily, Field.Volume)
        
        chart = Chart('Results')
        chart.AddSeries(Series('Price', SeriesType.Line, 0))
        chart.AddSeries(Series('Volatility', SeriesType.Line, 1))
        self.AddChart(chart)

    def OnData(self, data):
        if data.ContainsKey(self.stock) and data[self.stock]:
            self.volatility.Update(data[self.stock])
            if not self.volatility.IsReady: return
            self.Plot('Results', 'Price', data[self.stock].Close)
            self.Plot('Results', 'Volatility', self.volatility.Current.Value)
        
class Volatility(PythonIndicator):
    
    def __init__(self, period):
        self.Period = period
        self.Values = RollingWindow[float](period)
        self.Value = 0
        self.Current = IndicatorDataPoint(datetime.min, self.Value)
    
    @property
    def isReady(self):
        return self.Values.Count == self.Period

    def Update(self, input):
        self.Values.Add(input.Volume)
        value = pd.Series(self.Values).iloc[::-1].pct_change().std() * np.sqrt(252)
        self.Current = IndicatorDataPoint(input.Time, value)
        
        return self.IsReady