-- Id: 12347

-- More information about this indicator can be found at:
-- http://fxcodebase.com/code/viewtopic.php?f=17&t=61065

--+------------------------------------------------------------------+
--|                               Copyright © 2018, Gehtsoft USA LLC |
--|                                            http://fxcodebase.com |
--+------------------------------------------------------------------+
--|                                      Developed by : Mario Jemic  |
--|                                          mario.jemic@gmail.com   |
--+------------------------------------------------------------------+
--|                                 Support our efforts by donating  |
--|                                    Paypal: https://goo.gl/9Rj74e |
--|                    BitCoin : 15VCJTLaz12Amr7adHSBtL9v8XomURo9RF  |
--|                BitCoin Cash: 1BEtS465S3Su438Kc58h2sqvVvHK9Mijtg  |
--|           Ethereum : 0x8C110cD61538fb6d7A2B47858F0c0AaBd663068D  |
--|                   LiteCoin : LLU8PSY2vsq7B9kRELLZQcKf5nJQrdeqwD  |
--+------------------------------------------------------------------+

-- Indicator profile initialization routine
-- Defines indicator profile properties and indicator parameters
-- TODO: Add minimal and maximal value of numeric parameters and default color of the streams
function Init()
	indicator:name("Pitchfork Tool")
	indicator:description("")
	indicator:requiredSource(core.Bar)
	indicator:type(core.Indicator)

	indicator.parameters:addGroup("Calculation")
	indicator.parameters:addString("Method", "Method", "Method", "Andrews Pitchfork")
	indicator.parameters:addStringAlternative("Method", "Andrews Pitchfork", "Andrews Pitchfork", "Andrews Pitchfork")
	indicator.parameters:addStringAlternative("Method", "Schiff Pitchfork ", "Schiff Pitchfork ", "Schiff Pitchfork")
	indicator.parameters:addStringAlternative(
		"Method",
		"Modified Schiff Pitchfork",
		"Modified Schiff Pitchfork",
		"Modified Schiff Pitchfork"
	)

	indicator.parameters:addBoolean("Extend", "Extend Lines", "", true)

	indicator.parameters:addGroup("Style")
	indicator.parameters:addInteger("Size", "Label Size", "", 10)
	indicator.parameters:addInteger("Width", "Line Width", "", 3)
	indicator.parameters:addColor("Color", "Line Color", "", core.rgb(0, 0, 0))
end

-- Indicator instance initialization routine
-- Processes indicator parameters and creates output streams
-- TODO: Refine the first period calculation for each of the output streams.
-- TODO: Calculate all constants, create instances all subsequent indicators and load all required libraries
-- Parameters block
local first
local source = nil
local Color
local db
local name
local Size
local init = false
local Cell
local Method
local Width
local Extend
function Prepare(nameOnly)
	Color = instance.parameters.Color
	Size = instance.parameters.Size
	Method = instance.parameters.Method
	Width = instance.parameters.Width
	Extend = instance.parameters.Extend
	source = instance.source
	first = source:first()

	name = profile:id() .. "(" .. source:name() .. ", " .. Method .. ")"
	instance:name(name)

	if (nameOnly) then
		return
	end

	core.host:execute("addCommand", 1, "Set A Point", "")
	core.host:execute("addCommand", 2, "Set B Point", "")
	core.host:execute("addCommand", 3, "Set C Point", "")
	core.host:execute("addCommand", 4, "Reset")

	require("storagedb")
	db = storagedb.get_db(profile:id() .. "(" .. source:name())

	instance:ownerDrawn(true)
end

-- Indicator calculation routine
-- TODO: Add your code for calculation output values
function Update(period)
end

function Draw(stage, context)
	if stage == 2 then
		return
	end

	local AL = db:get("AL", 0)
	local AD = db:get("AD", 0)

	local BL = db:get("BL", 0)
	local BD = db:get("BD", 0)

	local CL = db:get("CL", 0)
	local CD = db:get("CD", 0)

	if not init then
		Cell = context:pointsToPixels(Size)
		context:createFont(1, "Wingdings", Cell, Cell, 0)
		context:createPen(2, context.SOLID, context:pointsToPixels(Width), Color)
		init = true
	end

	if AL == 0 or BL == 0 or CL == 0 then
		return
	end

	local p1 = core.findDate(source, AD, false)
	local p2 = core.findDate(source, BD, false)
	local p3 = core.findDate(source, CD, false)

	if p1 == -1 or p2 == -1 or p3 == -1 then
		return
	end

	local x1, y1 = Add(context, AL, p1)
	local x2, y2 = Add(context, BL, p2)
	local x3, y3 = Add(context, CL, p3)

	if x1 == nil or x2 == nil or x3 == nil then
		return
	end

	local X2 = x2
	local Y2 = y2
	local X3 = x3
	Y3 = y3

	if Method == "Andrews Pitchfork" then
		X1 = x1
		Y1 = y1
	elseif Method == "Schiff Pitchfork" then
		X1 = x1
		if y1 > y2 then
			Y1 = y1 - (y1 - y2) / 2
		else
			Y1 = y1 + (y2 - y1) / 2
		end
	elseif Method == "Modified Schiff Pitchfork" then
		X1 = math.min(x1, x2) + (math.max(x1, x2) - math.min(x1, x2)) / 2

		if y1 > y2 then
			Y1 = y1 - (y1 - y2) / 2
		else
			Y1 = y1 + (y2 - y1) / 2
		end
	end

	local mX = math.min(X2, X3) + (math.max(X2, X3) - math.min(X2, X3)) / 2
	local mY = math.min(Y2, Y3) + (math.max(Y2, Y3) - math.min(Y2, Y3)) / 2

	---Srednja Start
	AddNew(context, mY, mX)

	AddNew(context, Y1, X1)

	context:drawLine(2, X1, Y1, mX, mY)
	context:drawLine(2, X2, Y2, X3, Y3)

	--a, c=  math2d.lineEquation (X1, Y1, mX, mY);

	--To calculate y coordinate of a point on the line use
	--y = a * x + c

	--To calculate x coordinate of a point on the line use
	--x = (y - b) / a

	xDelta = mX - X1
	xA = X2 + xDelta
	xB = X3 + xDelta

	yDelta = Y1 - mY
	yA = Y2 - yDelta
	yB = Y3 - yDelta

	AddNew(context, yA, xA)
	AddNew(context, yB, xB)

	--Srednja Mid
	local mXm = math.min(xA, xB) + (math.max(xA, xB) - math.min(xA, xB)) / 2
	local mYm = math.min(yA, yB) + (math.max(yA, yB) - math.min(yA, yB)) / 2

	AddNew(context, mYm, mXm)

	if Extend then
		ex1, ey1, ex2, ey2 =
			math2d.lineRectangleIntersection(context:left(), context:top(), context:right(), context:bottom(), xA, yA, X2, Y2)
		ex3, ey3, ex4, ey4 =
			math2d.lineRectangleIntersection(context:left(), context:top(), context:right(), context:bottom(), xB, yB, X3, Y3)
		ex5, ey5, ex6, ey6 =
			math2d.lineRectangleIntersection(context:left(), context:top(), context:right(), context:bottom(), mX, mY, mXm, mYm)

		if
			ex1 == nil or ex2 == nil or ey1 == nil or ey2 == nil or ex3 == nil or ex4 == nil or ey3 == nil or ey4 == nil or
				ex5 == nil or
				ex6 == nil or
				ey5 == nil or
				ey6 == nil
		 then
			return
		end

		if X1 > mX then
			x1X = math.min(ex1, ex2)
			x2X = math.min(ex3, ex4)
		else
			x1X = math.max(ex1, ex2)
			x2X = math.max(ex3, ex4)
		end

		if Y1 > mY then
			x1Y = math.min(ey1, ey2)
			x2Y = math.min(ey3, ey4)
		else
			x1Y = math.max(ey1, ey2)
			x2Y = math.max(ey3, ey4)
		end

		------------------------------

		if X1 < mX then
			eX = math.max(ex5, ex6)
		else
			eX = math.min(ex5, ex6)
		end

		if Y1 < mY then
			eY = math.max(ey5, ey6)
		else
			eY = math.min(ey5, ey6)
		end

		----------------------------

		context:drawLine(2, x1X, x1Y, X2, Y2)
		context:drawLine(2, x2X, x2Y, X3, Y3)
		context:drawLine(2, mX, mY, eX, eY)
	else
		context:drawLine(2, xA, yA, X2, Y2)
		context:drawLine(2, xB, yB, X3, Y3)
		context:drawLine(2, mX, mY, mXm, mYm)
	end
end

function Parallel(x, y, x1, y1)
	local a = x
	local b = y1 - a * x1

	return a, b
end

function Add(context, L, D)
	if S == -1 or L == 0 then
		return
	end

	x, x1 = context:positionOfBar(D)
	visible, y1 = context:pointOfPrice(L)
	x2 = x1 + Cell / 2
	y2 = y1 + Cell / 2
	context:drawText(1, "\164", Color, -1, x1 - Cell / 2, y1 - Cell / 2, x2, y2, 0)

	return x1, y1
end

function AddNew(context, y2, x2)
	context:drawText(1, "\164", Color, -1, x2 - Cell / 2, y2 - Cell / 2, x2 + Cell / 2, y2 + Cell / 2, 0)
end

local pattern = "([^;]*);([^;]*)"
function AsyncOperationFinished(cookie, success, message)
	local date
	local level

	level, date = string.match(message, pattern, pos)

	if cookie == 1 then
		db:put("AL", tostring(level))
		db:put("AD", tostring(date))
	elseif cookie == 2 then
		db:put("BL", tostring(level))
		db:put("BD", tostring(date))
	elseif cookie == 3 then
		db:put("CL", tostring(level))
		db:put("CD", tostring(date))
	elseif cookie == 4 then
		db:put("AL", tostring(0))
		db:put("AD", tostring(0))
		db:put("BL", tostring(0))
		db:put("BD", tostring(0))
		db:put("CL", tostring(0))
		db:put("CD", tostring(0))
	end
end