Initial revision

intern/python/modules/mcf/utils/copy_extend.py

@@ -0,0 +1,83 @@
+'''
+Module to allow for "copying" Numeric arrays,
+(and thereby also matrices and userarrays)
+standard arrays, classes and modules
+(last two are not actually copied, but hey :) ).
+
+Could do approximately the same thing with
+copy_reg, but would be inefficient because
+of passing the data into and out of strings.
+
+To use, just import this module.
+'''
+# altered 98.11.05, moved copy out of NUMPY test
+import copy
+try: # in case numpy not installed
+	import Numeric
+	def _numpyarray_copy(somearray, memo=None):
+		'''
+		Simple function for getting a copy of a NUMPY array
+		'''
+		if memo == None:
+			memo = {} # yeah, I know, not _really_ necessary
+		# see if already done this item, return the copy if we have...
+		d = id(somearray)
+		try:
+			return memo[d]
+		except KeyError:
+			pass
+		temp = Numeric.array(somearray, copy=1)
+		memo[d] = temp
+		return temp
+	# now make it available to the copying functions
+	copy._copy_dispatch[Numeric.ArrayType] = _numpyarray_copy
+	copy._deepcopy_dispatch[Numeric.ArrayType] = _numpyarray_copy
+except ImportError: # Numeric not installed...
+	pass
+
+try: # in case array not installed
+	import array
+	def _array_copy(somearray, memo = None):
+		'''
+		Simple function for getting a copy of a standard array.
+		'''
+		if memo == None:
+			memo = {} # yeah, I know, not _really_ necessary
+		# see if already done this item, return the copy if we have...
+		d = id(somearray)
+		try:
+			return memo[d]
+		except KeyError:
+			pass
+		newarray = somearay[:]
+		memo[d] = newarray
+		return newarray
+		
+	# now make it available to the copying functions
+	copy._copy_dispatch[ array.ArrayType ] = _array_copy
+	copy._deepcopy_dispatch[ array.ArrayType ] = _array_copy
+except ImportError:
+	pass
+
+import types
+
+def _module_copy(somemodule, memo = None):
+	'''
+	Modules we will always treat as themselves during copying???
+	'''
+	return somemodule
+
+# now make it available to the copying functions
+copy._copy_dispatch[ types.ModuleType ] = _module_copy
+copy._deepcopy_dispatch[ types.ModuleType ] = _module_copy
+
+def _class_copy(someclass, memo=None):
+	'''
+	Again, classes are considered immutable, they are
+	just returned as themselves, not as new objects.
+	'''
+	return someclass
+
+# now make it available to the copying functions
+#copy._copy_dispatch[ types.ClassType ] = _class_copy
+copy._deepcopy_dispatch[ types.ClassType ] = _class_copy