Moho Forum

Code: Select all

function HS_GradientPlay:Run(moho, view)

	HS_Test_harness:tracePrint (thisUUT, self:Name(), self:Version())

	moho.document:PrepUndo(moho.layer)
	moho.document:SetDirty()
	
	local layer = moho.layer
	local doc = moho.document

	local i, j, k, m, n, p, s, w
	local Pos = LM.Vector2:new_local()

	local mesh = moho:Mesh()

	local shape


	local orange = LM.ColorOps:RgbColor(255, 150, 0, 255)
	local indigo = LM.ColorOps:RgbColor(75, 0, 130, 255)
	local violet = LM.ColorOps:RgbColor(238, 130, 238, 255)



	local colours = {
		LM.ColorOps.Clear,
		LM.ColorOps.Red,
		orange,
		LM.ColorOps.Yellow,
		LM.ColorOps.Green,
		LM.ColorOps.Blue,
		indigo
	}

	local interval = 15 -- frames

--[[
--diagnostic
	HS_Test_harness:diagnosePrint (thisUUT, "# colours", #colours)
	local rgbCol = LM.rgb_color:new_local()
	for i = 1, #colours do
		rgbCol.r = colours[i].r
		rgbCol.g = colours[i].g
		rgbCol.b = colours[i].b
		rgbCol.a = colours[i].a
		HS_Test_harness:diagnosePrint (thisUUT, "colour", i, rgbCol)
	end
]]

	local cSS
	local selShapes = {}

	local selShapeEffectsChan, selSubChans
	local ctChannels
	local chInfo = MOHO.MohoLayerChannel:new_local()


-- find selected shapes
	for i = 0, mesh:CountShapes() - 1 do
		if mesh:Shape(i).fSelected then
			table.insert (selShapes, mesh:Shape(i))
		end
	end
	cSS = #selShapes
--	HS_Test_harness:diagnosePrint (thisUUT, "found", cSS, "selected shapes")


-- deselect all shapes
	for i = 0, mesh:CountShapes() - 1 do
		mesh:Shape(i).fSelected = false
	end

-- select shapes one by one
	for i = 1, cSS do
		shape = selShapes[i]
		shape.fSelected = true

-- get the shape effects channel for the selected shape
		ctChannels = layer:CountChannels()
--		HS_Test_harness:diagnosePrint (thisUUT, "Found", ctChannels, "channels in layer", layer:Name())

		selShapeEffectsChan = nil
		for j = 0, ctChannels do 

			layer:GetChannelInfo(j, chInfo)

			if chInfo.channelID == CHANNEL_SHAPEFX_SEL then
				HS_Test_harness:diagnosePrint (thisUUT, "Found Channel", chInfo.name:Buffer(), "With", chInfo.subChannelCount, "Sub Channels")
--				selShapeEffectsChan = layer:Channel(j, 0, doc)
				selShapeEffectsChan = j
				selSubChans = chInfo.subChannelCount
				break
			end
		end


-- get the gradient subchannels
		local gradCol = {}
		local gradStops = {}
		local gradChan, stopChan

		HS_Test_harness:diagnosePrint (thisUUT, "in layer", layer:Name(), "channel #", selShapeEffectsChan, "has", selSubChans, "subchannels")


		if selShapeEffectsChan then
			local stages = math.floor(doc:EndFrame()/interval)
			local stops = selSubChans / 2
			HS_Test_harness:diagnosePrint (thisUUT, "loops", stages, "stops", stops, "colours", #colours)

			for k = 1, selSubChans, 2 do
--				HS_Test_harness:diagnosePrint (thisUUT, "in layer", layer:Name(), "channel #", selShapeEffectsChan, "subchannel", k)

				gradChan = layer:Channel(selShapeEffectsChan, k, doc)
				if not gradChan then
					HS_Test_harness:tracePrint (thisUUT, "ooops" , k, "not a channel")
					break
--				else
--					HS_Test_harness:diagnosePrint (thisUUT, "sub chan is of type" , gradChan:ChannelType())
				end

				if gradChan:ChannelType() == MOHO.CHANNEL_COLOR then
					table.insert (gradCol, moho:ChannelAsAnimColor(gradChan))
--					HS_Test_harness:diagnosePrint (thisUUT, "sub chan" , k, "stored. Table has", #gradCol, "entries")
				else
					HS_Test_harness:diagnosePrint (thisUUT, "sub chan" , k, "is not a colour channel")
				end
			end


			for k = 0, selSubChans-1, 2 do
--				HS_Test_harness:diagnosePrint (thisUUT, "in layer", layer:Name(), "channel #", selShapeEffectsChan, "subchannel", k)

				stopChan = layer:Channel(selShapeEffectsChan, k, doc)
				if not stopChan then
					HS_Test_harness:tracePrint (thisUUT, "ooops" , k, "not a channel")
					break
--				else
--					HS_Test_harness:diagnosePrint (thisUUT, "sub chan is of type" , stopChan:ChannelType())
				end

				if stopChan:ChannelType() == MOHO.CHANNEL_VAL then
					table.insert (gradStops, moho:ChannelAsAnimVal(stopChan))
--					HS_Test_harness:diagnosePrint (thisUUT, "sub chan" , k, "stored. Table has", #gradStops, "entries")
				else
					HS_Test_harness:diagnosePrint (thisUUT, "sub chan" , k, "is not a value channel")
				end
			end



-- initialise the current gradient
                       	local interp = MOHO.InterpSetting:new_local()
                        interp.interpMode = MOHO.INTERP_LINEAR

			local frame = 1
			local oldCols = {}

			for k = 1, stops do
				gradCol[k]:SetValue(frame, colours[1])
				gradCol[k]:SetKeyInterp(frame, interp)
				gradCol[k]:ClearAfter(frame)
				oldCols[k] = 1

				gradStops[k]:SetValue(frame, (k-1) * (1 / (stops-1)))
				gradStops[k]:ClearAfter(frame)
--				HS_Test_harness:diagnosePrint (thisUUT, "set stop" , k, "to", (k-1) * (1 / (stops-1)))
			end


-- set the gradient values
			frame = 0
			local newCol
			local rgbCol = LM.rgb_color:new_local()

			for m = 1, stages do
				frame = m * interval
				newCol = 1 + math.fmod (m, #colours)
--				HS_Test_harness:diagnosePrint (thisUUT, "stage" , m, "frame", frame, "new colour is", newCol, colours[newCol])


				for k = stops, 1, -1 do
					if k == 1 then
						oldCols[1] = newCol
					else
						oldCols[k] = oldCols[k-1]
					end
					p = oldCols[k]
					rgbCol.r = colours[p].r
					rgbCol.g = colours[p].g
					rgbCol.b = colours[p].b
					rgbCol.a = colours[p].a
--					HS_Test_harness:diagnosePrint (thisUUT, "set colour", k, "to", rgbCol)
					gradCol[k]:SetValue(frame, rgbCol)
					gradCol[k]:SetKeyInterp(frame, interp)
				end
--[[
				s = "{"
				for n = 1, #oldCols do
					s = s .. tostring(oldCols[n]) .. ", "
				end
				s = s:sub(1, #s-2) .. "}"
				HS_Test_harness:diagnosePrint (thisUUT, "Old colours after stage", m, "are", s)
]]

			end

		end
		shape.fSelected = false

	end


-- restore selected shapes
	for i = 1, cSS do
		selShapes[i].fSelected = true
	end


end