# sum_plot_counts_for_2stocks_rsm_00d
#https://usethinkscript.com/threads/sum-and-plot-barcount.14386/#post-119602
# Sum and plot barcount
# Shaco 2/7
# Please help me with this code/thought...
# ref1 - in post1 - RSM Indicator
# https://usethinkscript.com/threads/rsm-indicator-for-thinkorswim.5407/#post-50775
# While looking at a SPY chart,
# sum up and plot the UpTrendBarCount from 2 symbols (TSLA and AAPL for example)
# sum up and plot the DownTrendBarCount.
# ref2 - in post3 code- RSI + MACD Confluence
# https://usethinkscript.com/threads/rsi-macd-confluence-indicator-for-thinkorswim.505/page-3
#AddLabel(1, "-");
declare lower;
#def o = open;
#def h = high;
#def l = low;
#def c = close;
#----------------------------
# rsm_counts
script rsm_counts {
#----------------------------
input o = open;
input h = high;
input l = low;
input c = close;
# RSI
#input length = 14;
#input over_Bought = 70;
#input over_Sold = 30;
def len_RSI = 7;
#input averageType = AverageType.WILDERS;
def averageTypeRSI = AverageType.EXPONENTIAL;
def NetChgAvg = MovingAverage(averageTypeRSI, c - c[1], len_RSI);
def TotChgAvg = MovingAverage(averageTypeRSI, AbsValue(c - c[1]), len_RSI);
def ChgRatio = if TotChgAvg != 0 then NetChgAvg / TotChgAvg else 0;
def RSI = 50 * (ChgRatio + 1);
#----------------------------
# Stochastic Slow
def over_boughtSt = 80;
def over_soldSt = 20;
#input KPeriod = 10;
#input DPeriod = 10;
def KPeriod = 14;
def DPeriod = 3;
#input averageType = AverageType.SIMPLE;
def averageTypeStoch = AverageType.WILDERS;
#plot SlowK = reference StochasticFull(over_bought,over_sold,KPeriod,DPeriod,priceH,priceL,priceC,3,averageType).FullK;
#plot SlowD = reference StochasticFull(over_bought,over_sold,KPeriod,DPeriod,priceH,priceL,priceC,3,averageType).FullD;
#def SlowK = reference StochasticFull(over_boughtSt, over_soldSt, KPeriod, DPeriod, hi, lo, cl, 3, averageTypeStoch).FullK;
#def SlowD = reference StochasticFull(over_boughtSt, over_soldSt, KPeriod, DPeriod, hi, lo, cl, 3, averageTypeStoch).FullD;
#/////////////////////////
# stochfull
#input over_bought = 80;
#input over_sold = 20;
#input priceH = high;
#input priceL = low;
#input priceC = close;
def slowing_period = 3;
def averageType = AverageType.SIMPLE;
#input showBreakoutSignals = {default "No", "On FullK", "On FullD", "On FullK & FullD"};
def lowest_k = Lowest(l, KPeriod);
def stc1 = c - lowest_k;
def stc2 = Highest(h, KPeriod) - lowest_k;
def FastK = if stc2 != 0 then stc1 / stc2 * 100 else 0;
def FullK = MovingAverage(averageType, FastK, slowing_period);
def FullD = MovingAverage(averageType, FullK, DPeriod);
#/////////////////////////
#def SlowK = reference StochasticFull(over_boughtSt, over_soldSt, KPeriod, DPeriod, hi, lo, cl, 3).FullK;
#def SlowD = reference StochasticFull(over_boughtSt, over_soldSt, KPeriod, DPeriod, hi, lo, cl, 3, if (averageTypeStoch == 1) then AverageType.SIMPLE else AverageType.EXPONENTIAL).FullD;
def SlowK = fullk;
def SlowD = fulld;
# test code for type numbers
#input Type = AverageType.SIMPLE;
#def u = 1 * Type;
#AddLabel(1, Type + " " + u, Color.YELLOW);
# 0 = simple
# 1 = exp
# 2 = weighted
# 3 = wild
# 4 = hull
#----------------------------
# MACD Calculation
def fastLength = 12;
def slowLength = 26;
def MACDLength = 9;
def averageTypeMACD = AverageType.WEIGHTED;
def Value = MovingAverage(averageTypeMACD, c, fastLength) - MovingAverage(averageTypeMACD, c, slowLength);
def Avg = MovingAverage(averageTypeMACD, Value, MACDLength);
def Diff = Value - Avg;
#----------------------------
# SCAN Variables
def UpTrend = if RSI > 50 and SlowK > 50 and Value > Avg then 1 else 0;
def DownTrend = if RSI < 50 and SlowK < 50 and Value < Avg then 1 else 0;
def bnumUp;
def bnumDown;
def closeUpTrendStart;
def closeDownTrendStart;
def UpTrendBarCount;
def DownTrendBarCount;
if UpTrend and (!UpTrend[1] or DownTrend[1]) {
bnumUp = BarNumber();
bnumDown = 0;
closeUpTrendStart = c;
closeDownTrendStart = 0;
UpTrendBarCount = 1;
DownTrendBarCount = 0;
} else if UpTrend {
bnumUp = bnumUp[1];
bnumDown = 0;
closeUpTrendStart = closeUpTrendStart[1];
closeDownTrendStart = 0;
UpTrendBarCount = UpTrendBarCount[1] + 1;
DownTrendBarCount = 0;
} else if DownTrend and (!DownTrend[1] or UpTrend[1]) {
bnumUp = 0;
bnumDown = BarNumber();
closeDownTrendStart = c;
closeUpTrendStart = 0;
UpTrendBarCount = 0;
DownTrendBarCount = 1;
} else if DownTrend {
bnumDown = bnumDown[1];
closeDownTrendStart = closeDownTrendStart[1];
DownTrendBarCount = DownTrendBarCount[1] + 1;
bnumUp = 0;
closeUpTrendStart = 0;
UpTrendBarCount = 0;
} else {
bnumUp = 0;
bnumDown = 0;
closeUpTrendStart = 0;
closeDownTrendStart = 0;
UpTrendBarCount = 0;
DownTrendBarCount = 0;
}
# The UpTrendJustStarted and DownTrendJustStarted plots can be used to find stocks that have just started a trend in either direction
def UpTrendJustStartedBool = if RSI > 50 and SlowK > 50 and Value > Avg then 1 else 0;
def DownTrendJustStartedBool = if RSI < 50 and SlowK < 50 and Value < Avg then 1 else 0;
def UpTrendJustStarted = if UpTrendJustStartedBool == 1 and UpTrendJustStartedBool[1] == 0 then 1 else 0;
def DownTrendJustStarted = if DownTrendJustStartedBool == 1 and DownTrendJustStartedBool[1] == 0 then 1 else 0;
def UpTrendJustEnded = if UpTrendJustStartedBool[1] == 1 and UpTrendJustStartedBool == 0 then 1 else 0;
def DownTrendJustEnded = if DownTrendJustStartedBool[1] == 1 and DownTrendJustStartedBool == 0 then 1 else 0;
def GetTrend = if UpTrendJustStarted then 3 else if UpTrend then 2 else if UpTrendJustEnded then 1 else if DownTrendJustStarted then -1 else if DownTrend then -2 else if DownTrendJustEnded then -3 else 0;
#def upcnt = UpTrendBarCount;
#def dwncnt = DownTrendBarCount;
# chg to plots , in a script
plot upcnt = UpTrendBarCount;
plot dwncnt = DownTrendBarCount;
#upcnt.setdefaultcolor(color.green);
#dwncnt.setdefaultcolor(color.red);
#addlabel(1, upcnt);
#addlabel(1, dwncnt);
}
#def upbub = (upcnt > upcnt[1] and upcnt > upcnt[-1]);
#def dwnbub = (dwncnt > dwncnt[1] and dwncnt > dwncnt[-1]);
#addchartbubble(upbub, upcnt, upcnt, color.green, yes);
#addchartbubble(dwnbub, dwncnt, dwncnt, color.red, yes);
#===================================
#===================================
# start of main code
def na = double.nan;
def bn = barnumber();
#addlabel(1, "-");
input symbol1 = "SPY";
def o1 = open(symbol1);
def h1 = high(symbol1);
def l1 = low(symbol1);
def c1 = close(symbol1);
#input sym1_base = 0;
input symbol2 = "AAPL";
def o2 = open(symbol2);
def h2 = high(symbol2);
def l2 = low(symbol2);
def c2 = close(symbol2);
#input sym2_base = 25;
input symbol3 = "META";
def o3 = open(symbol3);
def h3 = high(symbol3);
def l3 = low(symbol3);
def c3 = close(symbol3);
#input sym3_base = 50;
input symbol4 = "TSLA";
def o4 = open(symbol4);
def h4 = high(symbol4);
def l4 = low(symbol4);
def c4 = close(symbol4);
#input sym4_base = 75;
input line_spacing = 35;
def sym1_base = 0;
def sym2_base = sym1_base + (1 * line_spacing);
def sym3_base = sym1_base + (2 * line_spacing);
def sym4_base = sym1_base + (3 * line_spacing);
def s1up = rsm_counts(01,h1,l1,c1).upcnt;
def s2up = rsm_counts(02,h2,l2,c2).upcnt;
def s3up = rsm_counts(03,h3,l3,c3).upcnt;
def s4up = rsm_counts(04,h4,l4,c4).upcnt;
def s1dwn = rsm_counts(01,h1,l1,c1).dwncnt;
def s2dwn = rsm_counts(02,h2,l2,c2).dwncnt;
def s3dwn = rsm_counts(03,h3,l3,c3).dwncnt;
def s4dwn = rsm_counts(04,h4,l4,c4).dwncnt;
plot s1u = if isnan(close) then na else sym1_base + s1up;
s1u.setdefaultcolor(color.green);
plot s1d = if isnan(close) then na else sym1_base + s1dwn;
s1d.setdefaultcolor(color.red);
plot s2u = if isnan(close) then na else sym2_base + s2up;
s2u.setdefaultcolor(color.green);
plot s2d = if isnan(close) then na else sym2_base + s2dwn;
s2d.setdefaultcolor(color.red);
plot s3u = if isnan(close) then na else sym3_base + s3up;
s3u.setdefaultcolor(color.green);
plot s3d = if isnan(close) then na else sym3_base + s3dwn;
s3d.setdefaultcolor(color.red);
plot s4u = if isnan(close) then na else sym4_base + s4up;
s4u.setdefaultcolor(color.green);
plot s4d = if isnan(close) then na else sym4_base + s4dwn;
s4d.setdefaultcolor(color.red);
input show_bubbles = yes;
def s1upbub = (s1u > s1u[1] and s1u > s1u[-1]);
def s1dwnbub = (s1d > s1d[1] and s1d > s1d[-1]);
addchartbubble(show_bubbles and s1upbub, s1u, s1u, color.green, yes);
addchartbubble(show_bubbles and s1dwnbub, s1d, s1d, color.red, yes);
def s2upbub = (s2u > s2u[1] and s2u > s2u[-1]);
def s2dwnbub = (s2d > s2d[1] and s2d > s2d[-1]);
addchartbubble(show_bubbles and s2upbub, s2u, s2u, color.green, yes);
addchartbubble(show_bubbles and s2dwnbub, s2d, s2d, color.red, yes);
def s3upbub = (s3u > s3u[1] and s3u > s3u[-1]);
def s3dwnbub = (s3d > s3d[1] and s3d > s3d[-1]);
addchartbubble(show_bubbles and s3upbub, s3u, s3u, color.green, yes);
addchartbubble(show_bubbles and s3dwnbub, s3d, s3d, color.red, yes);
def s4upbub = (s4u > s4u[1] and s4u > s4u[-1]);
def s4dwnbub = (s4d > s4d[1] and s4d > s4d[-1]);
addchartbubble(show_bubbles and s4upbub, s4u, s4u, color.green, yes);
addchartbubble(show_bubbles and s4dwnbub, s4d, s4d, color.red, yes);
input bubblex = 3;
def bubx = (!isnan(close[bubblex]) and isnan(close[bubblex-1]));
addchartbubble(bubx, (sym1_base - (line_spacing/2)), symbol1, color.yellow, yes);
addchartbubble(bubx, (sym2_base - (line_spacing/2)), symbol2, color.yellow, yes);
addchartbubble(bubx, (sym3_base - (line_spacing/2)), symbol3, color.yellow, yes);
addchartbubble(bubx, (sym4_base - (line_spacing/2)), symbol4, color.yellow, yes);
#I want to append the following lines to sum up/plot the UpTrendBarCount and DownTrendBarCount of the 2 symbols:
#plot upSum = if GetTrend == 2 then UpTrendBarCount(symbol="TSLA") + UpTrendBarCount(symbol="AAPL");
#plot downSum = if GetTrend == -2 then DownTrendBarCount(symbol="TSLA") + DownTrendBarCount(symbol="AAPL");
#
