#!/home/hoogca/virtualenv/bri/3.7/bin/python
###!/home/debian/miniconda3/bin/python3
#print("Incl wyckoff")

import sys, os, math

global n, nRecords, Records

sideBorder=20
PriceBarpx=100
candleWidth=10
canvasW=(sideBorder+PriceBarpx)+(nRecords*candleWidth)
canvasH=850

VolumeBarHeight=100

topBorder=50
HL=HighestHigh-LowestLow
PriceRange=HL+(2*(HL*0.05))
PriceWindowPx=canvasH-VolumeBarHeight#-(2*topBorder)

basePrice=(2*topBorder)+PriceWindowPx
baseVolume=basePrice+VolumeBarHeight 

#print(HighestHigh)

svg=f"""<svg width="{canvasW}" height="{canvasH}">
    <rect x="0" y="0" width="{canvasW}" height="{canvasH}" style="fill:rgb(0,0,0);" />
    <!-- Volume --><line x1="0" y1="{canvasH-VolumeBarHeight}" x2="{canvasW}" y2="{canvasH-VolumeBarHeight}" stroke="white" />
    <!-- Price  --><line x1="{canvasW-PriceBarpx}" y1="0" x2="{canvasW-PriceBarpx}" y2="{canvasH}" stroke="white" />
"""

barWidth=candleWidth-2
wickWith=4
if candleWidth>14: wickWith=6
if candleWidth<16: wickWith=2
slope=0.5
direction25="up"
direction5="up"
closeMAdirection="up"
prevX5=0

global NewHigh10,NewLow10
NewHigh10=[0 for i in range(nRecords)]
NewLow10=[0 for i in range(nRecords)]
def checkHL(n,m): 
    nNewHigh=n
    nNewLow=n
    for z in reversed(range(n-m,n)):
        if z>m:
            if Records['high'][z]>Records['high'][nNewHigh] and Records['high'][z+1]<=Records['high'][z]: nNewHigh=z
            if Records['low'][z] <Records['low'][nNewLow] and Records['low'][z+1]>=Records['low'][z]: nNewLow=z
            
    if NewHigh10[nNewHigh]>0 and Records['high'][NewHigh10[nNewHigh]]>Records['high'][nNewHigh]: nNewHigh=NewHigh10[nNewHigh]
    if NewLow10[nNewLow]>0 and Records['low'][NewLow10[nNewLow]]<Records['low'][nNewLow]: nNewLow=NewLow10[nNewLow]
    #if nNewHigh==n: nNewHigh=0
    #if nNewLow==n: nNewLow=0
    return nNewHigh,nNewLow

directionH="up"
directionL="up"
HighPeaks=[]
LowPeaks=[]
nPrevHigh=0
nPrevLow=0
peaks=""
debug=False
for n in range(1,nRecords):

    NewHigh10[n],NewLow10[n]=checkHL(n,5)
    #print(f"[{n},{NewHigh10[n]}]")
    
    X=(sideBorder)+(n*candleWidth)  
    
    nhX=(sideBorder)+(NewHigh10[n]*candleWidth)
    hY=round((PriceWindowPx)-( ((Records['high'][n]-LowestLow)*PriceWindowPx) / PriceRange ))
    nhY=round((PriceWindowPx)-( ((Records['high'][NewHigh10[n]]-LowestLow)*PriceWindowPx) / PriceRange ))
    #svg+=f"""   <!-- NewHigh10 --><line x1="{nhX-(barWidth/2)}" y1="{nhY}" x2="{X-(barWidth/2)}" y2="{hY}" stroke="rgb(255,255,0)" />\n"""
    if debug: svg+=f"""   <!-- NewHigh10   --><rect x="{nhX-(barWidth/2)}" y="{nhY-5}" width="5" height="5" style="fill:rgb(255,255,0)"  fill-opacity="0.25" />\n"""
            
    nlX=(sideBorder)+(NewLow10[n]*candleWidth)
    lY=round((PriceWindowPx)-( ((Records['low'][n]-LowestLow)*PriceWindowPx) / PriceRange ))
    nlY=round((PriceWindowPx)-( ((Records['low'][NewLow10[n]]-LowestLow)*PriceWindowPx) / PriceRange ))
    #svg+=f"""   <!-- NewHigh10 --><line x1="{nlX-(barWidth/2)}" y1="{nlY}" x2="{X-(barWidth/2)}" y2="{lY}" stroke="rgb(0,255,255)" />\n"""
    if debug: svg+=f"""   <!-- NewHigh10   --><rect x="{nlX-(barWidth/2)}" y="{nlY}" width="5" height="5" style="fill:rgb(0,255,255)"  fill-opacity="0.25" />\n"""
    
    
 
    cY=round((PriceWindowPx)-( ((Records['close'][n]-LowestLow)*PriceWindowPx) / PriceRange ))
    #oY=cY+( ((openX-close)*PriceWindowPx) / PriceRange )
    #oY=(PriceWindowPx)-( ((openX-LowestLow)*PriceWindowPx) / PriceRange )
    hY=round((PriceWindowPx)-( ((Records['high'][n]-LowestLow)*PriceWindowPx) / PriceRange ))
    lY=round((PriceWindowPx)-( ((Records['low'][n]-LowestLow)*PriceWindowPx) / PriceRange ))

    svg+=f"""   <!-- High-Close  --><rect x="{X-(barWidth/2)}" y="{hY}" width="{wickWith}" height="{abs(cY-hY)}" style="fill:rgb(100,100,100)" />\n"""#100,0,0
    svg+=f"""   <!-- Close-Low   --><rect x="{X-(barWidth/2)}" y="{cY}" width="{wickWith}" height="{abs(lY-cY)}" style="fill:rgb(100,100,100)" />\n"""#0,100,0
    svg+=f"""   <!-- CLOSE --><line x1="{X-(barWidth/2)}" y1="{cY}" x2="{X+(barWidth)}" y2="{cY}" stroke="rgb(100,100,100)"  stroke-width="2" />\n"""
    
    if n>2:
        
        zY=round((PriceWindowPx)-( ((Records['lowMA'][n]-LowestLow)*PriceWindowPx) / PriceRange ))
        pzY=round((PriceWindowPx)-( ((Records['lowMA'][n-1]-LowestLow)*PriceWindowPx) / PriceRange ))
        if Records['pcChange_lowMA'][n]>slope: clr="0,200,0"
        elif Records['pcChange_lowMA'][n]<-slope: clr="200,0,0"
        else: clr="0,0,200" 
        #svg+=f"""   <!-- lowMA --><line x1="{X-candleWidth-(barWidth/2)}" y1="{pzY}" x2="{X-(barWidth/2)}" y2="{zY}" stroke="rgb({clr})" />\n"""
        
        
        #zY=round((PriceWindowPx)-( ((Records['pZZMA5'][n]-LowestLow)*PriceWindowPx) / PriceRange ))
        #pzY=round((PriceWindowPx)-( ((Records['pZZMA5'][n-1]-LowestLow)*PriceWindowPx) / PriceRange ))
        #if Records['pcChange_pZZMA5'][n]>slope: clr="0,200,0"
        #elif Records['pcChange_pZZMA5'][n]<-slope: clr="200,0,0"
        #else: clr="0,0,200"
        #svg+=f"""   <!-- pZZMA5 --><line x1="{X-candleWidth-(barWidth/2)}" y1="{pzY}" x2="{X-(barWidth/2)}" y2="{zY}" stroke="rgb({clr})" />\n"""
        
        zY=round((PriceWindowPx)-( ((Records['pZZMA5'][n]-LowestLow)*PriceWindowPx) / PriceRange ))
        pzY=round((PriceWindowPx)-( ((Records['pZZMA5'][n-1]-LowestLow)*PriceWindowPx) / PriceRange ))
        sumChange=Records['pcChange_pZZMA5'][n]#+Records['pcChange_lZZMA5'][n]+Records['pcChange_hZZMA5'][n]
        if sumChange>0.5: clr="0,200,0"
        elif sumChange<-0.5: clr="200,0,0" 
        else: clr="0,0,200"
        #svg+=f"""   <!-- pZZMA5 --><line x1="{X-candleWidth-(barWidth/2)}" y1="{pzY}" x2="{X-(barWidth/2)}" y2="{zY}" stroke="rgb({clr})" />\n"""
        
        
        zY=round((PriceWindowPx)-( ((Records['closeMA'][n]-LowestLow)*PriceWindowPx) / PriceRange ))
        pzY=round((PriceWindowPx)-( ((Records['closeMA'][n-1]-LowestLow)*PriceWindowPx) / PriceRange ))
        if Records['pcChange_closeMA'][n]>0.40:
            if closeMAdirection=="dn":
                nlX=(sideBorder)+(NewLow10[n]*candleWidth)
                lY=round((PriceWindowPx)-( ((Records['low'][n]-LowestLow)*PriceWindowPx) / PriceRange ))
                nlY=round((PriceWindowPx)-( ((Records['low'][NewLow10[n]]-LowestLow)*PriceWindowPx) / PriceRange ))
                if debug: svg+=f"""   <!-- NewHigh10 --><line x1="{nlX-(barWidth/2)}" y1="{nlY}" x2="{X-(barWidth/2)}" y2="{lY}" stroke="rgb(0,255,255)" />\n"""
                #svg+=f"""   <!-- NewHigh10   --><rect x="{nlX-(barWidth/2)}" y="{nlY}" width="5" height="5" style="fill:rgb(0,255,255)"  fill-opacity="0.25" />\n"""
                if not NewLow10[n] in LowPeaks:
                    LowPeaks.append(NewLow10[n])
                    nPrevLow=NewLow10[n]
                    peaks+=f"L{NewLow10[n]},"
                else: 
                    LowPeaks.append(n)
                    nPrevLow=n
                    peaks+=f"L{n},"
                    
            closeMAdirection="up"
            clr="0,200,0" 
        elif Records['pcChange_closeMA'][n]<-0.40:
            if closeMAdirection=="up":
                nhX=(sideBorder)+(NewHigh10[n]*candleWidth)
                hY=round((PriceWindowPx)-( ((Records['high'][n]-LowestLow)*PriceWindowPx) / PriceRange ))
                nhY=round((PriceWindowPx)-( ((Records['high'][NewHigh10[n]]-LowestLow)*PriceWindowPx) / PriceRange ))
                if debug: svg+=f"""   <!-- NewHigh10 --><line x1="{nhX-(barWidth/2)}" y1="{nhY}" x2="{X-(barWidth/2)}" y2="{hY}" stroke="rgb(255,255,0)" />\n"""
                #svg+=f"""   <!-- NewHigh10   --><rect x="{nhX-(barWidth/2)}" y="{nhY-5}" width="5" height="5" style="fill:rgb(255,255,0)"  fill-opacity="0.25" />\n"""
                if not NewHigh10[n] in HighPeaks:
                    HighPeaks.append(NewHigh10[n])
                    nPrevHigh=NewHigh10[n]
                    peaks+=f"H{NewHigh10[n]},"
                else: 
                    HighPeaks.append(n)
                    nPrevHigh=n
                    peaks+=f"H{n},"
            
            closeMAdirection="dn" 
            clr="200,0,0" 
        else: 
            if closeMAdirection=="up": clr="0,200,0"
            else:  clr="200,0,0"
            #clr="0,0,200"
        if debug: svg+=f"""   <!-- pZZMA5 --><line x1="{X-(barWidth/2)}" y1="{pzY}" x2="{X-(barWidth/2)+candleWidth}" y2="{zY}" stroke="rgb({clr})" />\n"""
        
        
        zY15=round((PriceWindowPx)-( ((Records['pZZMA15'][n]-LowestLow)*PriceWindowPx) / PriceRange ))
        pzY15=round((PriceWindowPx)-( ((Records['pZZMA15'][n-1]-LowestLow)*PriceWindowPx) / PriceRange ))
        if Records['pcChange_pZZMA15'][n]>0.1:
            direction25="up"
            clr15="0,200,0"
            zY=round((PriceWindowPx)-( ((Records['hZZMA5'][n]-LowestLow)*PriceWindowPx) / PriceRange ))
            #pzY=round((PriceWindowPx)-( ((Records['hZZMA5'][n-1]-LowestLow)*PriceWindowPx) / PriceRange ))
            if Records['pcChange_hZZMA5'][n]>slope: 
                clr="0,200,0"
                direction5="up"
            elif Records['pcChange_hZZMA5'][n]<-slope: 
                clr="200,0,0"
                direction5="dn"
            else: 
                if direction5=="up": clr="0,200,0"
                else:  clr="200,0,0"
            #if prevX5>0: svg+=f"""   <!-- hZZMA5 --><line x1="{X-candleWidth-(barWidth/2)}" y1="{prevX5}" x2="{X-(barWidth/2)}" y2="{zY}" stroke="rgb({clr})" />\n"""
            prevX5=zY
        elif Records['pcChange_pZZMA15'][n]<-0.1:
            direction25="dn"
            clr15="200,0,0"
            zY=round((PriceWindowPx)-( ((Records['lZZMA5'][n]-LowestLow)*PriceWindowPx) / PriceRange ))
            #pzY=round((PriceWindowPx)-( ((Records['lZZMA5'][n-1]-LowestLow)*PriceWindowPx) / PriceRange ))
            if Records['pcChange_lZZMA5'][n]>slope: 
                clr="0,200,0"
                direction5="up"
            elif Records['pcChange_lZZMA5'][n]<-slope: 
                clr="200,0,0"
                direction5="dn"
            else: 
                if direction5=="up": clr="0,200,0"
                else:  clr="200,0,0"
            #if prevX5>0: svg+=f"""   <!-- lZZMA5 --><line x1="{X-candleWidth-(barWidth/2)}" y1="{prevX5}" x2="{X-(barWidth/2)}" y2="{zY}" stroke="rgb({clr})" />\n"""
            prevX5=zY
            
        else: 
            clr15="0,0,200"
            if direction25=="up":
                clr15="0,200,0"
                zY=round((PriceWindowPx)-( ((Records['hZZMA5'][n]-LowestLow)*PriceWindowPx) / PriceRange ))
                #pzY=round((PriceWindowPx)-( ((Records['hZZMA5'][n-1]-LowestLow)*PriceWindowPx) / PriceRange ))
                if Records['pcChange_hZZMA5'][n]>slope: 
                    clr="0,200,0"
                    direction5="up"
                elif Records['pcChange_hZZMA5'][n]<-slope: 
                    clr="200,0,0"
                    direction5="dn"
                else: 
                    if direction5=="up": clr="0,200,0"
                    else:  clr="200,0,0"
                #if prevX5>0: svg+=f"""   <!-- hZZMA5 --><line x1="{X-candleWidth-(barWidth/2)}" y1="{prevX5}" x2="{X-(barWidth/2)}" y2="{zY}" stroke="rgb({clr})" />\n"""
                prevX5=zY
            else:
                clr15="200,0,0"
                zY=round((PriceWindowPx)-( ((Records['lZZMA5'][n]-LowestLow)*PriceWindowPx) / PriceRange ))
                #pzY=round((PriceWindowPx)-( ((Records['lZZMA5'][n-1]-LowestLow)*PriceWindowPx) / PriceRange ))
                if Records['pcChange_lZZMA5'][n]>slope: clr="0,200,0"
                elif Records['pcChange_lZZMA5'][n]<-slope: clr="200,0,0"
                else:  
                    if direction5=="up": clr="0,200,0"
                    else:  clr="200,0,0"
                #if prevX5>0: svg+=f"""   <!-- lZZMA5 --><line x1="{X-candleWidth-(barWidth/2)}" y1="{prevX5}" x2="{X-(barWidth/2)}" y2="{zY}" stroke="rgb({clr})" />\n"""
                prevX5=zY
           
        if direction25=="up":
            zY=round((PriceWindowPx)-( ((Records['high'][n]-Records['pZZMA15'][n])*PriceWindowPx) / PriceRange ))
            #if prevX5>0: svg+=f"""   <!-- xxx --><line x1="{X-(barWidth/2)}" y1="{PriceWindowPx}" x2="{X-(barWidth/2)}" y2="{zY}" stroke="rgb(0,200,0)" />\n"""
        else:
            zY=round((PriceWindowPx)-( ((Records['low'][n]-Records['pZZMA15'][n])*PriceWindowPx) / PriceRange )) 
            #if prevX5>0: svg+=f"""   <!-- xxx --><line x1="{X-(barWidth/2)}" y1="{PriceWindowPx}" x2="{X-(barWidth/2)}" y2="{zY}" stroke="rgb(200,0,0)" />\n"""
        # svg+=f"""   <!-- pZZMA15 --><line x1="{X-candleWidth-(barWidth/2)}" y1="{pzY15}" x2="{X-(barWidth/2)}" y2="{zY15}" stroke="rgb({clr15})" />\n"""

#hprint(len(HighPeaks))  
#hprint(len(LowPeaks))
peaks=peaks.rstrip(",")
#hprint(peaks)

peakArr=peaks.split(",")
for npeak in range(0,len(peakArr)):
    if npeak>0:
        
        FROM=peakArr[npeak-1]
        if FROM.find("H")==0:
            nFROM=int(FROM.replace("H",""))
            FROMprice=Records['high'][nFROM]
        else:
            nFROM=int(FROM.replace("L",""))
            FROMprice=Records['low'][nFROM]
        TO=peakArr[npeak]
        if TO.find("H")==0:
            nTO=int(TO.replace("H",""))
            TOprice=Records['high'][nTO]
        else:
            nTO=int(TO.replace("L",""))
            TOprice=Records['low'][nTO]
            
        #hprint(f"{nFROM}|{FROMprice} .. {nTO}|{TOprice}")
        XF=(sideBorder)+(nFROM*candleWidth)  
        XT=(sideBorder)+(nTO*candleWidth)
        YF=round((PriceWindowPx)-( ((FROMprice-LowestLow)*PriceWindowPx) / PriceRange ))
        YT=round((PriceWindowPx)-( ((TOprice-LowestLow)*PriceWindowPx) / PriceRange ))
        svg+=f"""   <!-- PeakLines --><line x1="{XF-(barWidth/2)}" y1="{YF}" x2="{XT-(barWidth/2)}" y2="{YT}" stroke="rgb(255,255,255)" />\n"""
        #svg+=f"""   <!-- NewHigh10   --><rect x="{nhX-(barWidth/2)}" y="{nhY-5}" width="5" height="5" style="fill:rgb(255,255,0)"  fill-opacity="0.25" />\n"""
    
Overalldirection="up"
for n in (range(1,nRecords)):#reversed
    X=(sideBorder)+(n*candleWidth) 
    
    if n>2 and n<nRecords-1:
        
        if Records['avgPcChange'][n]>slope: 
            if Records['avgPcChange'][n+1]>slope: # Filter out single-bar trand changes
                Overalldirection="up"
        elif Records['avgPcChange'][n]<-slope: 
            if Records['avgPcChange'][n+1]<-slope: # Filter out single-bar trand changes
                Overalldirection="dn"
            
        if Overalldirection=="up": trendclr="0,200,0"
        else:  trendclr="200,0,0"

        zhY=round((PriceWindowPx)-( ((Records['hZZMA5'][n]-LowestLow)*PriceWindowPx) / PriceRange ))
        pzhY=round((PriceWindowPx)-( ((Records['hZZMA5'][n-1]-LowestLow)*PriceWindowPx) / PriceRange ))
        if Records['pcChange_hZZMA5'][n]>slope: 
            clr="0,200,0"
            directionH="up"
        elif Records['pcChange_hZZMA5'][n]<-slope: 
            clr="200,0,0"
            directionH="dn"
        else: 
            if directionH=="up": clr="0,200,0"
            else:  clr="200,0,0"
        svg+=f"""   <!-- hZZMA5 --><line x1="{X-candleWidth-(barWidth/2)}" y1="{pzhY}" x2="{X-(barWidth/2)}" y2="{zhY}" stroke="rgb({trendclr})" />\n"""
        
        zlY=round((PriceWindowPx)-( ((Records['lZZMA5'][n]-LowestLow)*PriceWindowPx) / PriceRange ))
        pzlY=round((PriceWindowPx)-( ((Records['lZZMA5'][n-1]-LowestLow)*PriceWindowPx) / PriceRange ))
        if Records['pcChange_lZZMA5'][n]>slope: 
            clr="0,200,0"
            directionL="up"
        elif Records['pcChange_lZZMA5'][n]<-slope: 
            clr="200,0,0"
            directionL="dn"
        else: 
            if directionH=="up": clr="0,200,0"
            else:  clr="200,0,0"
        svg+=f"""   <!-- lZZMA5 --><line x1="{X-candleWidth-(barWidth/2)}" y1="{pzlY}" x2="{X-(barWidth/2)}" y2="{zlY}" stroke="rgb({trendclr})" />\n"""
         

        
svg+=f"""</svg>"""

print(svg)