require('Module:No globals')
local getArgs = require('Module:Arguments').getArgs

local p = {}
local pers = {}
local tabella = {}

local function dividi(dati)
	local n = 1
	local resto = 0
	local err = '-'

	while (dati[n]) do n = n+1 end
	n = n-1

	for m=4,n,4 do
		if ((tonumber(dati[m-3])) and (tonumber(dati[m-2]))) then
			if (pers[tonumber(dati[m-3])]) then
				err = 'id ' .. dati[m-3] .. ' utilizzato per elementi diversi'
			else
				pers[tonumber(dati[m-3])] = { padre = tonumber(dati[m-2]), testo = dati[m-1], nota = dati[m], id = -1, x = -1, y = -1, sp = 0, figli = {} }
			end
		else
			err = 'identificativo non numerico in ' .. dati[m-3] .. ', ' .. dati[m-2]
		end
		resto = n-m
	end
	if (err == '-' and resto > 0) then
		err = 'numero di dati ' .. n .. ' non valido (elementi in più: ' .. resto .. ')'
	end
	return err
end

local function organizza(pid, plev)
	local nn = 1
	pers[pid].y = plev -- assegna y ad ogni persona
	if (not tabella[plev]) then tabella[plev] = {} end
	table.insert(tabella[plev], pid) -- inserisce la persona in tabella
	for i, v in pairs(pers[pid].figli) do
		pers[v].id = i -- assegna numero come figlio
		nn = nn + organizza(v, plev+1) -- ricorsione per assegnare y ai figli e contarli
	end
	return nn
end

local function limSx(pid, delta, dt) -- calcola la posizione più a sinistra tra gli elementi del ramo per ogni y
	if (dt[pers[pid].y]) then
		dt[pers[pid].y] = math.min(dt[pers[pid].y], pers[pid].x+delta)
	else
		dt[pers[pid].y] = pers[pid].x + delta
	end
	for _, v in pairs(pers[pid].figli) do
		dt = limSx(v, delta+pers[pid].sp, dt)
	end
	return dt
end

local function limDx(pid, delta, dt) -- calcola la posizione più a destra tra gli elementi del ramo per ogni y
	if (dt[pers[pid].y]) then
		dt[pers[pid].y] = math.max(dt[pers[pid].y], pers[pid].x+delta)
	else
		dt[pers[pid].y] = pers[pid].x + delta
	end
	for _, v in pairs(pers[pid].figli) do
		dt = limDx(v, delta+pers[pid].sp, dt)
	end
	return dt
end

local function riallinea(pid2, n1, n2) -- sposta gli elementi tra i fratelli n1 e n2 per cercare di equispaziarli
	local distanza = n2 - n1
	local vrf = 0
	local pos, inizio, passo
	if (distanza > 1) then
		inizio = pers[pers[pid2].figli[n1]].x
		passo = (pers[pers[pid2].figli[n2]].x - inizio)/distanza
		for cc=1,(distanza-1) do
			pos = inizio + math.floor(cc*passo)
			if (pos - pers[pers[pid2].figli[n1+cc]].x > 0) then
				pers[pers[pid2].figli[n1+cc]].x = pos
				pers[pers[pid2].figli[n1+cc]].sp = pos
			end
		end
		vrf = 1
	end
	return vrf
end

local function verifica(pid) -- verifica che i rami non si sovrappongano e calcola eventuali spostamenti
	local tSx
	local tDx
	local sposta = 0

	local fine = pers[pid].id
	local frt2, n
	
	for frt=1,(fine-1) do -- controlla tutti i rami fratelli da 1 fino al ramo corrente
		frt2 = pers[pers[pid].padre].figli[frt]
		tDx = limDx(frt2, 0, {}) -- elementi più a destra del ramo fratello per ogni y
		tSx = limSx(pid, 0, {}) -- elementi più a sinistra del ramo corrente per ogni y
		n = pers[pid].y
		while ((tSx[n]) and (tDx[n])) do
			if (tSx[n] - tDx[n] + sposta < 2) then -- verifica che ci sia almeno una distanza 2 tra i rami (2 = larghezza casella)
				sposta = 2 + tDx[n] - tSx[n]
			end
			n = n + 1
		end
		if  (sposta > 0) then -- se deve spostare verifica se riallineare elementi intermedi
			pers[pid].x = pers[pid].x + sposta
			pers[pid].sp = pers[pid].sp + sposta
			if (riallinea(pers[pid].padre, frt, fine) == 1) then verifica(pid) end
			sposta = 0
		end
	end
end

local function calcolaX1(pid) -- prima verifica di posizionamento
	for _, v in pairs(pers[pid].figli) do
		calcolaX1(v)
	end
	local tt = #pers[pid].figli
	if (tt == 0) then -- non ha figli
		if (pers[pid].padre == -1) then -- è capostipite
			pers[pid].x = 0
		elseif (pers[pid].id == 1) then -- è primo figlio
			pers[pid].x = 0
		else -- è figlio successivo
			pers[pid].x = pers[pers[pers[pid].padre].figli[pers[pid].id - 1]].x + 2
		end
	elseif (tt == 1) then -- ha un figlio
		if (pers[pid].padre == -1) then -- è capostipite
			pers[pid].x = pers[pers[pid].figli[1]].x
		elseif (pers[pid].id == 1) then -- è primo figlio
			pers[pid].x = pers[pers[pid].figli[1]].x
		else -- è figlio successivo
			pers[pid].x = pers[pers[pers[pid].padre].figli[pers[pid].id - 1]].x + 2
			pers[pid].sp = pers[pid].x - pers[pers[pid].figli[1]].x
			verifica(pid)
		end
	else -- ha più figli
		local media = math.floor((pers[pers[pid].figli[1]].x + pers[pers[pid].figli[tt]].x)/2)
		if (pers[pid].padre == -1) then -- è capostipite
			pers[pid].x = media
		elseif (pers[pid].id == 1) then -- è primo figlio
			pers[pid].x = media
		else -- è figlio successivo
			pers[pid].x = pers[pers[pers[pid].padre].figli[pers[pid].id - 1]].x + 2
			pers[pid].sp = pers[pid].x - media
			verifica(pid)
		end
	end
end

local function calcolaX2(pid)
	local tt = {}
	local sposta = 0

	tt = limSx(pid, 0, tt)
	for _, v in pairs(tt) do
		if (v+sposta<0) then
			sposta = -v
		end
	end
	if (sposta > 0) then
		pers[pid].x = pers[pid].x + sposta
		pers[pid].sp = pers[pid].sp + sposta
	end
end

local function calcolaX3(pid, sposta)
	pers[pid].x = pers[pid].x + sposta
	for _, v in pairs(pers[pid].figli) do
		calcolaX3(v, sposta + pers[pid].sp)
	end
end

local function massimoXY(pid, t) -- calcola numero colonne e righe totali
	if (pers[pid].x > t[1]) then t[1] = pers[pid].x end
	if (pers[pid].y > t[2]) then t[2] = pers[pid].y end
	for _, v in pairs(pers[pid].figli) do
		t = massimoXY(v,t)
	end
	return t
end

local function mostra(pid,allinea,largo) -- crea HTML tabella finale
	local xy = massimoXY(pid, {0, 0})
	local p1, p2, p3 -- posizione all'interno della riga per 1. parte superiore; 2. parte con testo; 3. parte inferiore
	local n1 -- padre dell'ultimo elemento analizzato (per connessione tra fratelli)
	local s1, s2, s3 -- HTML per 1. parte superiore; 2. parte con testo; 3. parte inferiore
	local px, nx -- posizione e padre dell'elemento corrente
	local ss = ''
	if (allinea == 'destra') then
		ss = '<div class="floatright" style="width:'..largo..'px;padding:3px;background:#fff;border:1px solid #c8ccd1"><table cellpadding="1" cellspacing="0" border=0 style="text-align:center;font-size:95%;line-height:110%;margin:0px auto">'
	else
		ss = '<center><table cellpadding="1" cellspacing="0" border=0 style="text-align:center;font-size:95%;line-height:110%;margin:10px auto">'
	end
	local lg = math.floor(100/(xy[1]+2)) -- larghezza percentuale di ogni colonna
	if (lg == 0) then lg = 1 end
	for n=1,xy[2] do -- analisi delle righe della tabella
		p1 = 0; p2 = 0; p3 = 0
		n1 = 0
		if (n==1) then s1 = '<tr>' else s1 = '<tr style="line-height:8px">' end
		s2 = '<tr>'
		if (n<xy[2]) then s3 = '<tr style="line-height:8px">' else s3 = '' end
		for _, v in pairs(tabella[n]) do
			px = pers[v].x
			nx = pers[v].padre

			-- 1. parte superiore con unione dei rami se padre uguale al precedente
			if (n>1) then
				s1 = s1 .. '<td style="border-right:1px solid #000'
				if (n1 == nx) then s1 = s1 ..';border-top:1px solid #000' end
				n1 = nx
				if (px - p1 > 0) then s1 = s1 .. '" colspan="' .. (px+1-p1)  end
				p1 = px + 1
				s1 = s1 ..	'">&nbsp;</td>'
			else
				for m=1,(xy[1]+2) do s1 = s1 .. '<td width='..lg..'%></td>' end
			end

			-- 2. parte con testo
			if (px-p2>0) then
				s2 = s2 .. '<td'
				if (px - p2>1) then s2 = s2 .. ' colspan="'.. (px-p2) ..'"' end
				s2 = s2 .. '>&nbsp;</td>'
			end
			s2 = s2 .. '<td colspan=2>' .. pers[v].testo .. '<br/>'
			if (pers[v].nota == '-') then else s2 = s2 .. '<span style="font-size:90%"><i>' .. pers[v].nota .. '</i></span>' end
			s2 = s2 .. '</td>'
			p2 = px + 2

			-- 3. parte inferiore in caso di figli (non creata per l'ultima riga)
			if (n<xy[2]) then
				if (#pers[v].figli > 0) then
					s3 = s3 .. '<td style="border-right: 1px solid #000'
					if (px - p3 > 0) then s3 = s3 .. '" colspan="' .. (px+1-p3) end
					p3 = px + 1
					s3 = s3 ..	'">&nbsp;</td>'
				end
			end
		end

		-- chiusura delle righe
		if (n>1) then s1 = s1 .. '</tr>' end
		s2 = s2 .. '</tr>'
		if (n<xy[2]) then s3 = s3 .. '</tr>' end

		ss = ss .. s1 .. s2 .. s3
	end

	ss = ss .. '</table>' -- chiusura tabella
	if (allinea == 'destra') then ss = ss .. '</div>' else ss = ss .. '</center>' end

	return ss
end

function p.main(frame)
	local capo = -1
	local n1, n2
	local err = '-'
	local args = getArgs(frame, {
		valueFunc = function (key, value)
			if type(key) == "number" then
				if value == nil then
					return nil
				else
					value = mw.text.trim(value)
				end
			else
				if value == '' then return nil end
            end
			return value
		end
	})
	local lato = args['allinea'] or 'centro'
	local larg = args['larghezza'] or '300'
	err = dividi(args)
	if (err == '-') then
		n1 = 0
		for i, v in pairs(pers) do
			n1 = n1+1
			if (v.padre == -1) then
				if (capo == -1) then
					capo = i
				else
					err = 'più elementi (id: ' .. capo .. ', ' .. i .. ') indicati come capostipite'
				end
			else
				if (v.padre == i) then
					err = 'elemento con id ' .. i .. ' indicato come figlio di se stesso'
				elseif (pers[v.padre]) then
					table.insert(pers[v.padre].figli,i)
				else
					err = 'valore per genitore (' .. v.padre .. ') non valido per l\'elemento con id ' .. i
				end
			end
		end

		if (err == '-') then
			if (capo == -1) then
				err = 'capostipite non definito'
			else
				n2 = organizza(capo, 1)
				if (n1 == n2) then
					calcolaX1(capo)
					calcolaX2(capo)
					calcolaX3(capo, 0)
				else
					err = 'alcuni elementi non sono collegati al capostipite'
				end
			end
		end
	end
	if (err == '-') then
		return mostra(capo, lato, larg)
	else
		return '<p style="color:#b80000"><b>Template:Discendenza - Errore:</b> ' .. err .. '</p>[[Categoria:Errori di compilazione del template Discendenza]]'
	end
end

return p