Add EMA & SMA

Skinok · Skinok · commit 8b1fc84e65d9 · 2021-12-03T22:46:09.000+01:00
diff --git a/finplotWindow.py b/finplotWindow.py
@@ -6,6 +6,8 @@
 from indicators import rsi
 from indicators import stochastic
 from indicators import stochasticRsi
+from indicators import sma
+from indicators import ema
 
 sys.path.append('../finplot')
 import finplot as fplt
@@ -66,6 +68,16 @@ def drawCandles(self):
         #self.createControlPanel(self.ax0.ax_widget)
         pass
 
+    def drawSma(self, period):
+        self.sma_indicator = sma.Sma(self.data, period)
+        self.sma_indicator.draw(self.ax0)
+        pass
+
+    def drawEma(self, period):
+        self.ema_indicator = ema.Ema(self.data, period)
+        self.ema_indicator.draw(self.ax0)
+        pass
+
     def drawRsi(self, period):
         self.rsi_indicator = rsi.Rsi(self.data, period)
         self.rsi_indicator.draw(self.ax_rsi)
diff --git a/indicators/ema.py b/indicators/ema.py
@@ -0,0 +1,20 @@
+#!/usr/bin/env python3
+import sys
+
+sys.path.append('../finplot')
+import finplot as fplt
+
+from common import calc_rsi
+
+class Ema():
+
+    def __init__(self, dataFrames, ema_periods=9):
+        self.ema_df =  dataFrames["Close"].ewm(span=ema_periods, adjust=False).mean()
+        pass
+
+    def draw(self, ax, ema_color = "yellow"):
+        self.ema_plot = fplt.plot(self.ema_df, ax = ax, color=ema_color, width=1 )
+        pass
+
+    def clear(self):
+        pass
diff --git a/indicators/sma.py b/indicators/sma.py
@@ -0,0 +1,20 @@
+#!/usr/bin/env python3
+import sys
+
+sys.path.append('../finplot')
+import finplot as fplt
+
+from common import calc_rsi
+
+class Sma():
+
+    def __init__(self, dataFrames, sma_periods=14):
+        self.sma_df =  dataFrames["Close"].rolling(window=sma_periods).mean()
+        pass
+
+    def draw(self, ax, sma_color = "green"):
+        self.sma_plot = fplt.plot(self.sma_df, ax = ax, color=sma_color, width=1 )
+        pass
+
+    def clear(self):
+        pass
diff --git a/userInterface.py b/userInterface.py
@@ -497,6 +497,26 @@ def createControlPanel(self):
         spacer = QtWidgets.QSpacerItem(50,20,QtWidgets.QSizePolicy.Minimum)
         layout.addSpacerItem(spacer)
 
+        # SMA
+        self.SmaPB = QtWidgets.QPushButton(self.panel)
+        self.SmaPB.setText("SMA")
+        self.SmaPB.setCheckable(True)
+        self.SmaPB.setMaximumWidth(100)
+        self.SmaPB.toggled.connect(self.addSma)
+        layout.addWidget(self.SmaPB)
+
+        # EMA
+        self.EmaPB = QtWidgets.QPushButton(self.panel)
+        self.EmaPB.setText("EMA")
+        self.EmaPB.setCheckable(True)
+        self.EmaPB.setMaximumWidth(100)
+        self.EmaPB.toggled.connect(self.addEma)
+        layout.addWidget(self.EmaPB)
+
+        # Spacer
+        spacer = QtWidgets.QSpacerItem(50,20,QtWidgets.QSizePolicy.Minimum)
+        layout.addSpacerItem(spacer)
+
         # RSI
         self.RsiPB = QtWidgets.QPushButton(self.panel)
         self.RsiPB.setText("RSI")
@@ -568,6 +588,38 @@ def resetChart(self):
         self.fpltWindow.updateChart()
         pass
 
+    # On chart indicators
+    def addSma(self):
+
+        # Show indicator parameter dialog
+        paramDialog = indicatorParametersUI.IndicatorParametersUI(self.dock_chart)
+        paramDialog.setWindowFlags(QtCore.Qt.CustomizeWindowHint)
+        paramDialog.setTitle("SMA Indicator parameters")
+        paramDialog.addParameter("SMA Period", 14)
+        paramDialog.adjustSize()
+
+        if (paramDialog.exec() == QtWidgets.QDialog.Accepted ):
+            period = paramDialog.getValue("SMA Period")
+            self.fpltWindow.drawSma( period )
+
+        pass
+
+    # On chart indicators
+    def addEma(self):
+
+        # Show indicator parameter dialog
+        paramDialog = indicatorParametersUI.IndicatorParametersUI(self.dock_chart)
+        paramDialog.setWindowFlags(QtCore.Qt.CustomizeWindowHint)
+        paramDialog.setTitle("EMA Indicator parameters")
+        paramDialog.addParameter("EMA Period", 9)
+        paramDialog.adjustSize()
+
+        if (paramDialog.exec() == QtWidgets.QDialog.Accepted ):
+            period = paramDialog.getValue("EMA Period")
+            self.fpltWindow.drawEma( period )
+
+        pass
+
     # indicators in external windows
     def toogleRsi(self):
 
