/usr/share/gwyddion/pygwy/gwyutils.py is in gwyddion-common 2.44-1.
This file is owned by root:root, with mode 0o644.
The actual contents of the file can be viewed below.
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 | import gwy
def save_dfield_to_png(container, datafield_name, filename, run_type):
"""
Save desired datafield given by name stored in container to file.
@param container: gwy.Container which has datafield of given name
@param datafield_name: datafield name in string representation (like '/0/data')
@param filename: expected filename
@param run_type: select of interactive (RUN_INTERACTIVE) or noninteractive mode (RUN_NONINTERACTIVE)
"""
gwy.gwy_app_data_browser_reset_visibility(container, gwy.VISIBILITY_RESET_SHOW_ALL)
datafield_num = int(datafield_name.split('/')[1])
gwy.gwy_app_data_browser_select_data_field(container, datafield_num)
gwy.gwy_file_save(container, filename, run_type)
def get_data_fields(container):
"""
Get list of available datafield stored in given container
@param container: gwy.Container with datafields
@return: a list of datafields
"""
l = []
for key in container.keys():
x = container.get_value(key)
if type(x) is gwy.DataField:
l.append(x)
return l
def get_data_fields_dir(container):
"""
Get list of available datafield stored in given container as directory where key is key name and value is DataField object.
@param container: gwy.Container with datafields
@return: a directory of datafields
"""
d = {}
for key in container.keys_by_name():
x = container.get_value_by_name(key)
if type(x) is gwy.DataField:
d[key] = x
return d
def get_current_datafield():
"""
Short version of function L{gwy.gwy_app_data_browser_get_current}(gwy.APP_DATA_FIELD)
@return: current datafield
"""
return gwy.gwy_app_data_browser_get_current(gwy.APP_DATA_FIELD)
def get_current_container():
"""
Short version of function L{gwy.gwy_app_data_browser_get_current}(gwy.APP_CONTAINER)
@return: current container
"""
return gwy.gwy_app_data_browser_get_current(gwy.APP_CONTAINER)
try:
import numpy as np
except ImportError:
pass
else:
def data_field_data_as_array(field):
"""Create a view the DataField's data as numpy array.
The array can point to an invalid memory location after using other
gwyddion functions and lead to application crash. Use with care.
@param field: the L{gwy.DataField} to view
@return: array viewing the data
"""
class gwydfdatap():
def __init__(self,addr,shape):
data = (addr,False)
stride = (8,shape[0]*8)
self.__array_interface__= {
'strides' : stride,
'shape' : shape,
'data' : data,
'typestr' : "|f8",
'version' : 3}
addr = field.get_data_pointer()
shape = (field.get_xres(),field.get_yres())
return np.array(gwydfdatap(addr,shape),copy=False)
def brick_data_as_array(brick):
"""Create a view the Brick's data as numpy array.
The array can point to an invalid memory location after using other
gwyddion functions and lead to application crash. Use with care.
@param field: the L{gwy.Brick} to view
@return: array viewing the data
"""
class gwydfdatap():
def __init__(self,addr,shape):
data = (addr,False)
stride = (8,shape[0]*8,shape[0]*shape[1]*8)
self.__array_interface__= {
'strides' : stride,
'shape' : shape,
'data' : data,
'typestr' : "|f8",
'version' : 3}
addr = brick.get_data_pointer()
shape = (brick.get_xres(),brick.get_yres(),brick.get_zres())
return np.array(gwydfdatap(addr,shape),copy=False)
def data_field_get_data(datafield):
"""Gets the data from a data field.
The returned array is a copy of the data.
But it can be safely stored without ever referring to invalid memory.
"""
return data_field_data_as_array(datafield).copy()
def data_field_set_data(datafield, data):
"""Sets the data of a data field.
The data shape must correspond to the data field shape.
"""
dest = data_field_data_as_array(datafield)
if dest.shape != data.shape:
raise ValueError("Data needs same size as the DataField.")
dest[:] = data
def brick_get_data(brick):
"""Gets the data from a brick.
The returned array is a copy of the data.
But it can be safely stored without ever referring to invalid memory.
"""
return brick_data_as_array(brick).copy()
def brick_set_data(brick, data):
"""Sets the data of a brick.
The data shape must correspond to the brick shape.
"""
dest = brick_data_as_array(brick)
if dest.shape != data.shape:
raise ValueError("Data needs same size as the Brick.")
dest[:] = data
|