Replaced by :Lib:cfmfile.py.

Mac/scripts/cfmfile.py

-"""cfmfile - Interface to code fragments on file"""
-import struct
-import Res
-import macfs
-import string
-
-Error = 'cfmfile.Error'
-
-READ = 1
-WRITE = 2
-smAllScripts = -3
-BUFSIZE = 0x100000
-
-class FragmentInfo:
-	"""Information on a single fragment"""
-	def __init__(self):
-		self.arch = 'pwpc'
-		self.current_version = 0
-		self.oldest_version = 0
-		self.stacksize = 0
-		self.libdir = 0
-		self.fragtype = 1
-		self.location = 1
-		self.offset = 0
-		self.length = 0
-		self.res_0 = 0
-		self.res_1 = 0
-		self.name = ''
-		self.ifp = None
-		
-	def load(self, data):
-		if len(data) < 43:
-			raise Error, 'Not enough data in cfrg resource'
-		self.arch = data[:4]
-		self.update, self.current_version, self.oldest_version, \
-			self.stacksize, self.libdir, self.fragtype, self.location, \
-			self.offset, self.length, self.res_0, self.res_1, length = \
-			struct.unpack("llllhbbllllh", data[4:42])
-		namelen = ord(data[42])
-		self.name = data[43:43+namelen]
-		if len(self.name) != namelen:
-			raise Error, 'Not enough data in cfrg resource'
-		return length
-		
-	def save(self):
-		length = (43+len(self.name)+3) & ~3
-		data = self.arch + struct.pack("llllhbbllllh", self.update, \
-			self.current_version, self.oldest_version, self.stacksize, \
-			self.libdir, self.fragtype, self.location, self.offset, \
-			self.length, self.res_0, self.res_1, length)
-		data = data + chr(len(self.name)) + self.name
-		data = data + ('\0'*(length-len(data)))
-		return data
-		
-	def copydata(self, ofp):
-		"""Copy fragment data to a new file, updating self.offset"""
-		if self.location != 1:
-			raise Error, 'Can only copy kOnDiskFlat (data fork) fragments'
-		if not self.ifp:
-			raise Error, 'No source file for fragment'
-		# Find out real length (if zero)
-		if self.length == 0:
-			self.ifp.seek(0, 2)
-			self.length = self.ifp.tell()
-		# Position input file and record new offset from output file
-		self.ifp.seek(self.offset)
-		self.offset = ofp.tell()
-		l = self.length
-		while l:
-			if l > BUFSIZE:
-				ofp.write(self.ifp.read(BUFSIZE))
-				l = l - BUFSIZE
-			else:
-				ofp.write(self.ifp.read(l))
-				l = 0
-		self.ifp = ofp
-		
-	def setfile(self, ifp):
-		self.ifp = ifp
-		
-class FragmentResource:
-
-	def __init__(self, data):
-		self.load(data)
-
-	def load(self, data):
-		r0, r1, version, r3, r4, r5, r6, nfrag = struct.unpack("llllllll", data[:32])
-		if version != 1:
-			raise Error, 'Unsupported cfrg version number %d'%version
-		data = data[32:]
-		self.fragments = []
-		for i in range(nfrag):
-			f = FragmentInfo()
-			len = f.load(data)
-			data = data[len:]
-			self.fragments.append(f)
-		if data:
-			raise Error, 'Spurious data after fragment descriptions'
-			
-	def save(self):
-		data = struct.pack("llllllll", 0, 0, 1, 0, 0, 0, 0, len(self.fragments))
-		for f in self.fragments:
-			data = data+f.save()
-		return data
-			
-	def setfile(self, ifp):
-		for f in self.fragments:
-			f.setfile(ifp)
-			
-	def copydata(self, ofp):
-		for f in self.fragments:
-			f.copydata(ofp)
-			
-	def getfragments(self):
-		return self.fragments
-		
-	def addfragments(self, fragments):
-		self.fragments = self.fragments + fragments
-
-class ResourceCollection:
-	def __init__(self, fhandle):
-		self.reslist = []
-		self.fhandle = fhandle
-		oldresfile = Res.CurResFile()
-		Res.UseResFile(fhandle)
-		Res.SetResLoad(0)
-		ntypes = Res.Count1Types()
-		for itype in range(1, 1+ntypes):
-			type = Res.Get1IndType(itype)
-			nresources = Res.Count1Resources(type)
-			for ires in range(1, 1+nresources):
-				res = Res.Get1IndResource(type, ires)
-				id, type, name = res.GetResInfo()
-				self.reslist.append((type, id))
-		Res.SetResLoad(1)
-		Res.UseResFile(oldresfile)
-			
-	def contains(self, type, id):
-		return (type, id) in self.reslist
-		
-	def getresource(self, type, id):
-		oldresfile = Res.CurResFile()
-		Res.UseResFile(self.fhandle)
-		Res.SetResLoad(1)
-		resource = Res.Get1Resource(type, id)
-		Res.UseResFile(oldresfile)
-		return resource
-		
-	def saveresto(self, type, id, fhandle):
-		oldresfile = Res.CurResFile()
-		resource = self.getresource(type, id)
-		id, type, name = resource.GetResInfo()
-		resource.DetachResource()
-		Res.UseResFile(fhandle)
-		resource.AddResource(type, id, name)
-		Res.UseResFile(oldresfile)
-		
-	def getreslist(self):
-		return self.reslist
-		
-class CfmFile(ResourceCollection, FragmentResource):
-	
-	def __init__(self, fsspec):
-		rfork = Res.FSpOpenResFile(fsspec, READ)
-		dfork = open(fsspec.as_pathname(), 'rb')
-		ResourceCollection.__init__(self, rfork)
-		cfrg_resource = self.getresource('cfrg', 0)
-		FragmentResource.__init__(self, cfrg_resource.data)
-		self.setfile(dfork)
-
-def mergecfmfiles(inputs, output):
-	# Convert inputs/outputs to fsspecs
-	inputs = map(None, inputs)
-	for i in range(len(inputs)):
-		if type(inputs[i]) == type(''):
-			inputs[i] = macfs.FSSpec(inputs[i])
-	if type(output) == type(''):
-		output = macfs.FSSpec(output)
-		
-	input_list = []
-	for i in inputs:
-		input_list.append(CfmFile(i))
-		
-	# Create output file, if needed
-	creator, tp = inputs[0].GetCreatorType()
-	try:
-		Res.FSpCreateResFile(output, creator, tp, smAllScripts)
-	except Res.Error:
-		pass
-		
-	# Copy fragments
-	dfork = open(output.as_pathname(), 'wb')
-	for i in input_list:
-		i.copydata(dfork)
-	dfork.close()
-		
-	# Merge cfrg's
-	for i in input_list[1:]:
-		input_list[0].addfragments(i.getfragments())
-		
-	old_res_file = Res.CurResFile()
-	rfork = Res.FSpOpenResFile(output, WRITE)
-	Res.UseResFile(rfork)
-	
-	# Write cfrg
-	data = input_list[0].save()
-	cfrg_resource = Res.Resource(data)
-	cfrg_resource.AddResource('cfrg', 0, '')
-	resources_done = [('cfrg', 0)]
-	
-	# Write other resources
-	for i in input_list:
-		todo = i.getreslist()
-		for tp, id in todo:
-			if (tp, id) in resources_done:
-				continue
-			i.saveresto(tp, id, rfork)
-			resources_done.append((tp, id))
-			
-def main():
-	list = []
-	while 1:
-		fss, ok = macfs.PromptGetFile("Next input file:", "shlb", "APPL")
-		if not ok: break
-		list.append(fss)
-	if not list:
-		sys.exit(0)
-	output, ok = macfs.StandardPutFile("Output file:")
-	if not ok:
-		sys.exit(0)
-	mergecfmfiles(list, output)
-	
-if __name__ == '__main__':
-	main()
-	
-