ModGrab.py

import TermUI as tui
import TermIntr as ti
import Befunge as bf
import Funge as fng

def selectingMod(char,value,subPos,matrix,corners):
	global state
	mx,my=(subPos[1]+corners[1],subPos[0]+corners[0])
	if state=="Selecting":
		start=corner.copy()
		end=cursor.cursor.copy()
		if start.x>end.x:
			start.x,end.x=(end.x,start.x)
		if start.y>end.y:
			start.y,end.y=(end.y,start.y)
		if end.y>=my>=start.y and end.x>=mx>=start.x:
			char.bcolor="27"
		# raise ValueError((mx,my),end,start,subPos,corners)
		# raise ValueError

def selectedMarkers():
	global state,cursor
	if state=="Selected":
		root=cursor.cursor
		for y in range(len(data)):
			for x in range(len(data[0])):
				if data[y][x] is not None:
					def changer(char):
						char.char=chr(data[y][x])
						char.bcolor="27"
						char.fcolor="232"
						# raise TypeError
					yield (root.x+x,root.y+y,changer)

state="None"
corner=None
data=None
origData=None
rotation=0
def modInit(m,config,lock):
	global cursor
	cursor=m.cursor
	statusText=m.statustext.queueText
	m.fungescreen.display.modifiers.append(selectingMod)
	m.fungescreen.display.markers.append(selectedMarkers)
	listener=ti.Listener()
	@listener.handle
	def handle(key):
		global state,corner,data,rotation,origData
		if key==config["GrabKey"]:
			# start selecting
			if state=="None":
				corner=cursor.cursor.copy()
				state="Selecting"
				statusText(f"Grabber: *Selecting*")
			# select
			elif state=="Selecting":
				matrix=m.load.funge.plane.matrix
				start=corner.copy()
				end=cursor.cursor.copy()
				if start.x>end.x:
					start.x,end.x=(end.x,start.x)
				if start.y>end.y:
					start.y,end.y=(end.y,start.y)
				data=[[None for _ in range(end.x-start.x+1)] for _ in range(end.y-start.y+1)]
				toClear=[]
				for y in matrix:
					if end.y>=y>=start.y:
						row=matrix[y]
						for x in row:
							if end.x>=x>=start.x:
								data[y-start.y][x-start.x]=row[x]
								toClear.append((x,y))
				for coord in toClear:
					del m.load.funge.plane[fng.Vect2d(*coord)]
				state="Selected"
				origData=data
				statusText(f"Grabber: *Selected* - use *{config['PlaceKey']}* to place or *{config['DropKey']}* to drop")
			elif state=="Selected":
				# rotate
				rotation+=1
				rotation=rotation%4
				if rotation==0:
					data=origData
					statusText(f"Grabber: *Rotated*")
					return
				else:
					data=origData
					for _ in range(rotation):
						data=list(zip(*data[::-1]))
					for row in range(len(data)):
						data[row]=list(data[row])
				functional=True
				for row in range(len(data)):
					for col in range(len(data[0])):
						val=data[row][col]
						if val is not None:
							table=None
							for line in bf.rotations:
								if chr(val) in line:
									table=line
							if table:
								transformed=table[(table.index(chr(val))+rotation)%4]
								if transformed is None:
									functional=False
								else:
									data[row][col]=ord(transformed)
				statusText(f"Grabber: *Rotated*"+ (" Warn - Could not rotate function" if not functional else ""))
		elif key==config["PlaceKey"] or key==config["DropKey"]:
			if state=="Selected":
				# place & drop
				plane=m.load.funge.plane
				for y in range(len(data)):
					for x in range(len(data[0])):
						if data[y][x] is not None:
							plane[fng.Vect2d(cursor.cursor.x+x,cursor.cursor.y+y)]=data[y][x]
				statusText(f"Grabber: *Placed*")
				if key==config["DropKey"]:
					state="None"
					data=None
					statusText(f"Grabber: *Dropped*")
			elif state=="Selecting" and key==config["DropKey"]:
				# clear
				matrix=m.load.funge.plane.matrix
				start=corner.copy()
				end=cursor.cursor.copy()
				if start.x>end.x:
					start.x,end.x=(end.x,start.x)
				if start.y>end.y:
					start.y,end.y=(end.y,start.y)
				toClear=[]
				for y in matrix:
					if end.y>=y>=start.y:
						row=matrix[y]
						for x in row:
							if end.x>=x>=start.x:
								toClear.append((x,y))
				for coord in toClear:
					del m.load.funge.plane[fng.Vect2d(*coord)]
				state="None"
				data=None
				statusText("Grabber: *Erased*")
			else:
				statusText("Select something with *ctrl c* first!")

	m.ui.addIntr(listener)