Update screensaver.lua

screensaver.lua

@@ -1,69 +1,104 @@
+local tArg = {...}
+
 local scr_x, scr_y = term.getSize()
 local mx, my = scr_x/2, scr_y/2
-local debugMode = false
+
+-- special modes for special people
+local mouseMode = tArg[1] == "mouse" or tArg[2] == "mouse"
+local fuck = tArg[1] == "fuck" or tArg[2] == "fuck"
+
+-- localize functions to increase speed, maybe, I think
+local concat, blit = table.concat, term.blit
+local sin, cos, rad, abs, sqrt, floor = math.sin, math.cos, math.rad, math.abs, math.sqrt, math.floor
 
 -- rainbow pattern
 local palette = {"e","1","4","5","d","9","b","a","2"}
 
 local distance = function(x1, y1, x2, y2)
-	return math.sqrt( (x2 - x1) ^ 2 + (y2 - y1) ^ 2 )
+	return sqrt( (x2 - x1) ^ 2 + (y2 - y1) ^ 2 )
 end
 
-local round = function(num)
-	return math.floor(num + 0.5)
+local randCase = function(str)
+	local output = ""
+	for i = 1, #str do
+		output = output .. ((math.random(0,1) == 1) and str:sub(i,i):upper() or str:sub(i,i):lower())
+	end
+	return output
 end
 
-local render = function(iterate, xscroll, yscroll)
+local render = function(iterate, xscroll1, yscroll1, xscroll2, yscroll2)
 	local buffer, cx, cy = {{},{},{}}
 	for y = 1, scr_y do
 		buffer[1][y] = {}
 		buffer[2][y] = {}
 		buffer[3][y] = {}
 		for x = 1, scr_x do
-			cx = ((x - mx) > 0 and 1 or -1) * (math.abs(x - mx) ^ 1.2) / 1.5
-			cy = ((y - my) > 0 and 1 or -1) * (math.abs(y - my) ^ 1.2)
+			cx = 0.66 * ((x - mx) > 0 and 1 or -1) * (abs(x - mx) ^ 1.2)
+			cy =        ((y - my) > 0 and 1 or -1) * (abs(y - my) ^ 1.2)
 
-			buffer[1][y][x] = "\127"
-			--buffer[1][y][x] = ("FUCK"):sub(1+(cx%4),1+(cx%4))
+			buffer[1][y][x] = fuck and randCase("fuck"):sub(1+(cx%4),1+(cx%4)) or "\127"
 
-			buffer[2][y][x] = palette[1 + round(
-				iterate + distance( cx - xscroll, cy - yscroll, 0, 0 )
+			buffer[2][y][x] = palette[1 + floor(
+				iterate + distance( cx + xscroll1, cy + yscroll1, 0, 0 )
 			) % #palette] or " "
 
-			buffer[3][y][x] = palette[1 + round(
-				iterate + distance( cx + xscroll, cy + yscroll, 0, 0 )
+			buffer[3][y][x] = palette[1 + floor(
+				iterate + distance( cx + xscroll2, cy + yscroll2, 0, 0 )
 			) % #palette] or " "
 		end
 	end
 
 	for y = 1, scr_y do
 		term.setCursorPos(1,y)
-		if debugMode then
-			term.write(#buffer[1][y]..", "..#buffer[2][y]..", "..#buffer[3][y])
-		else
-			term.blit(
-				table.concat(buffer[1][y]),
-				table.concat(buffer[2][y]),
-				table.concat(buffer[3][y])
-			)
-		end
+		-- suka
+		blit(
+			concat(buffer[1][y]),
+			concat(buffer[2][y]),
+			concat(buffer[3][y])
+		)
 	end
 end
 
 local main = function()
 	term.clear()
-	local wave = 0
-	while true do
-		render(
-			wave,
-			math.sin(math.rad(wave * 2)) * scr_x * 0.4,
-			math.cos(math.rad(wave * 3.5)) * scr_y * 0.4
-		)
-		wave = (wave + 1) % (360 * 7)
-		sleep(0.05)
+	local wave, evt = 0
+	local xscroll1, yscroll1, xscroll2, yscroll2 = 0, 0, 0, 0
+	if mouseMode then
+		parallel.waitForAny(function()
+			while true do
+				evt = {os.pullEvent()}
+				if evt[1] == "mouse_click" or evt[1] == "mouse_drag" then
+					if evt[2] == 1 then
+						xscroll1 = mx - evt[3]
+						yscroll1 = my - evt[4]
+					elseif evt[2] == 2 then
+						xscroll2 = mx - evt[3]
+						yscroll2 = my - evt[4]
+					end
+				end
+			end
+		end,
+		function()
+			while true do
+				render(wave, xscroll1, yscroll1, xscroll2, yscroll2)
+				wave = (wave + 1) % (360 * 7)
+				sleep(0.05)
+			end
+		end)
+	else
+		while true do
+			xscroll1 = -sin(rad(wave * 2)) * scr_x * 0.4
+			yscroll1 = -cos(rad(wave * 3.5)) * scr_y * 0.4
+			xscroll2 = -xscroll1
+			yscroll2 = -yscroll1
+			render(wave, xscroll1, yscroll1, xscroll2, yscroll2)
+			wave = (wave + 1) % (360 * 7)
+			sleep(0.05)
+		end
 	end
 end
 
+-- wait for keypress to exit program
 local waitForInput = function()
 	local evt
 	sleep(0.25)