using CUDA
using .ExpressionProcessing
using .Transpiler

expressions = Vector{Expr}(undef, 2)
variables = Matrix{Float32}(undef, 2,2)
parameters = Vector{Vector{Float32}}(undef, 2)

# Resulting value should be 1.14... for the first expression
expressions[1] = :(1 + 3 * 5 / 7 - sqrt(4))
expressions[2] = :(5 + x1 + 1 * x2 + p1 + p2)
variables[1,1] = 2.0
variables[2,1] = 3.0
variables[1,2] = 0.0
variables[2,2] = 5.0
parameters[1] = Vector{Float32}(undef, 1)
parameters[2] = Vector{Float32}(undef, 2)
parameters[1][1] = 5.0
parameters[2][1] = 5.0
parameters[2][2] = 0.0


@testset "Test TMP transpiler" begin
	postfixExpr = expr_to_postfix(expressions[1])
	postfixExprs = Vector([postfixExpr])
	push!(postfixExprs, expr_to_postfix(expressions[2]))
	push!(postfixExprs, expr_to_postfix(:(5^3 + x1)))

	# generatedCode = Transpiler.transpile(postfixExpr)
	generatedCode = Transpiler.transpile(postfixExprs[3], 2, 3) # TEMP
	# CUDA.@sync interpret(postfixExprs, variables, parameters)

	# This is just here for testing. This will be called inside the execute method in the Transpiler module
	linker = CuLink()
	add_data!(linker, "ExpressionProcessing", generatedCode)

	image = complete(linker)

	mod = CuModule(image)
	func = CuFunction(mod, "ExpressionProcessing")
end

#TODO: test performance of transpiler PTX generation when doing "return String(take!(buffer))" vs "return take!(buffer)"