84 lines
2.3 KiB
Python
84 lines
2.3 KiB
Python
|
'''
|
||
|
|
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
|