colorfill.lua

--TODO color fill, works on some areas of screen, not on some other?
--maybe test on blank project ?

--for dbg
maxiterations=500000

--parameterized to fill source layer or other layer
--cpu based impl

--we check floodability to treat later
-- we only fill target pixel is it has the fillColl
--( we only fill areas of same color )
local function floodability(fx,fy,sourcedata,toFillColl)

-- TODO big algorithm mistake
-- to fill coll is the picked color to fill, we fill only if new pixel is equal to picked col


--      print('floodability for '..fx..' '..fy)
      local sr,sg,sb,sa = sourcedata:getPixel(fx,fy)
--      print('tgt color r g b '..sr..' '..sg..' '..sb)
--     print('fill color r g b '..toFillColl.r..' '..toFillColl.g..' '..toFillColl.b)


      if ( sr==toFillColl.r and sg==toFillColl.g and sb==toFillColl.b)  then
	--should be filled
--	print('floodability true')
	return true
      end

    if( sa>0. and sa<1. ) then
      --TODO we fill transparent pixels but we should blend ideally
        return true
    end



      --should be not filled
--	print('floodability FALSE')
      return false
end




floodFill=function(fx,fy,paintingCol,sourceData,targetData)
	iterations=0
	
	--TODO do treated piexels map
	--we have a boolean array of frame size that we navigate
	-- [y*cvsw+x]


	print('broken flood fill '..fx..' '..fy..' with '..paintingCol.r..' '..paintingCol.g..' '..paintingCol.b )

	treatedPixels={}
	local j,i

	for j = 0,conf.cvsh-1
	do
		for i=0,conf.cvsw-1
		do
			treatedPixels[j*conf.cvsw+i]=false
		end
	end
	print('base treatment table is done')


	toTreatQueue=List.new()


	--we fill this color, and this color only
	local fr,fg,fb,fa = sourceData:getPixel(fx,fy)
	floodOrigin={x=fx,y=fy}

	local toFillCol = {}
	toFillCol.r=fr
	toFillCol.g=fg
	toFillCol.b=fb


	List.pushRight(toTreatQueue,floodOrigin)
	treatedPixels[floodOrigin.x+floodOrigin.y*conf.cvsw]=true

	while toTreatQueue.size>0 and iterations<maxiterations
	do
		local point = List.popLeft(toTreatQueue)

		iterations=iterations+1

--		print('about to color '..point.x..' '..point.y..' with r '..paintingCol.r..' ' ..paintingCol.g..' b '..paintingCol.g)
    pr,pg,pb,pa = targetData:getPixel(point.x,point.y)
    
    if pa==1.0 or pa==0.0 then
      --easy
      --test DBG to fill with transp
--      targetData:setPixel(point.x,point.y,paintingCol.r,paintingCol.g,paintingCol.b,1.0)
      targetData:setPixel(point.x,point.y,paintingCol.r,paintingCol.g,paintingCol.b,paintingCol.a)
    else
      -- we blend based on alpha
      targetData:setPixel(point.x,point.y,
          paintingCol.r*(1.-pa)+pa*pr,
          paintingCol.g*(1.-pa)+pa*pg,
          paintingCol.b*(1.-pa)+pa*pb,
          1.0)
      
    end

		-- lets check if we can move in all 4 dirs
		-- left
--		print('checking left pixel x '..(point.x-1) ..'y  '..point.y)
		if (point.x-1)>=0 and treatedPixels[(point.x-1)+point.y*conf.cvsw]==false and floodability((point.x-1),point.y,sourceData,toFillCol)
		then
--			print('left pixel treatable ')
			List.pushRight(toTreatQueue,{x=point.x-1,y=point.y})
			treatedPixels[point.x-1+point.y*conf.cvsw]=true
--		else
--			print('left pixel NOT treatable ')
		
		end


		--right
		if point.x<(conf.cvsw-1) and treatedPixels[(point.x+1)+point.y*conf.cvsw]==false and floodability((point.x+1),point.y,sourceData,toFillCol)
		then
			
			List.pushRight(toTreatQueue,{x=point.x+1,y=point.y})
			treatedPixels[point.x+1+point.y*conf.cvsw]=true
		end

		
--		print('checking top pixel x '..(point.x) ..'y  '..(point.y-1))
		if (point.y-1)>=0 and treatedPixels[(point.x)+(point.y-1)*conf.cvsw]==false and floodability((point.x),(point.y-1),sourceData,toFillCol)
		then
--			print('top pixel treatable ')
			List.pushRight(toTreatQueue,{x=point.x,y=point.y-1})
			treatedPixels[point.x+(point.y-1)*conf.cvsw]=true
			
--		else
--			print('top pixel NOT treatable ')
		
		end

--		print('checking bottom pixel x '..(point.x) ..'y  '..(point.y+1))
		if (point.y+1)<=conf.cvsh and treatedPixels[(point.x)+(point.y+1)*conf.cvsw]==false and floodability((point.x),(point.y+1),sourceData,toFillCol)
		then
--			print('bot pixel treatable ')
			List.pushRight(toTreatQueue,{x=point.x,y=point.y+1})
			treatedPixels[point.x+(point.y+1)*conf.cvsw]=true
		else
--			print('bot pixel NOT treatable ')
		
		end



	end



end