61c44f112df42595127056e46af931c1a795bb23
[pyafm.git] / pyafm / afm.py
1 # Copyright (C) 2009-2011 W. Trevor King <wking@drexel.edu>
2 #
3 # This file is part of pyafm.
4 #
5 # pyafm is free software; you can redistribute it and/or modify it
6 # under the terms of the GNU General Public License as published by the
7 # Free Software Foundation, either version 3 of the License, or (at your
8 # option) any later version.
9 #
10 # pyafm is distributed in the hope that it will be useful, but
11 # WITHOUT ANY WARRANTY; without even the implied warranty of
12 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
13 # General Public License for more details.
14 #
15 # You should have received a copy of the GNU General Public License
16 # along with pyafm.  If not, see <http://www.gnu.org/licenses/>.
17
18 """Tools for controlling atomic force microscopes.
19
20 Provides control of AFM postition using both short-range (piezo) and
21 long range (stepper) vertical positioning.  There are separate modules
22 for controlling the piezo (`pypiezo`) and stepper (`stepper`), this
23 module only contains methods that require the capabilities of both.
24 """
25
26 from pypiezo.base import convert_bits_to_meters as _convert_bits_to_meters
27 from pypiezo.base import convert_meters_to_bits as _convert_meters_to_bits
28 from pypiezo.base import convert_volts_to_bits as _convert_volts_to_bits
29 from pypiezo.surface import FlatFit as _FlatFit
30 from pypiezo.surface import SurfaceError as _SurfaceError
31
32 from . import LOG as _LOG
33
34
35 class AFM (object):
36     """Atomic force microscope positioning.
37
38     Uses a short range `piezo` and a long range `stepper` to position
39     an AFM tip relative to the surface.
40
41     Parameters
42     ----------
43     piezo | pypiezo.afm.AFMpiezo instance
44         Fine positioning and deflection measurements.
45     stepper | stepper.Stepper instance
46         Coarse positioning.
47     temperature | temperature.Controller instance or None
48         Optional temperature monitoring and control.
49     """
50     def __init__(self, piezo, stepper, temperature=None, axis_name='z'):
51         self.piezo = piezo
52         self.stepper = stepper
53         self.temperature = temperature
54         self.axis_name = axis_name
55
56     def get_temperature(self):
57         """Measure the sample temperature.
58
59         Return the sample temperature in Kelvin or `None` if such a
60         measurement is not possible.
61         """
62         if hasattr(self.temperature, 'get'):
63             return self.temperature.get_temperature()
64
65     def move_just_onto_surface(self, depth=-50e-9, setpoint=2, far=200):
66         """Position the AFM tip close to the surface.
67
68         Uses `.piezo.get_surface_position()` to pinpoint the position
69         of the surface.  Adjusts the stepper position as required via
70         `.stepper.step_relative()` to get within
71         `2*.stepper.step_size` meters of the surface.  Then adjusts
72         the piezo to place the cantilever `depth` meters onto the
73         surface.  Negative `depth`\s place the tip off the surface
74
75         If `.piezo.get_surface_position()` fails to find the surface,
76         backs off `far` half steps (for safety) and steps in (without
77         moving the zpiezo) until deflection voltage is greater than
78         `setpoint`.
79         """
80         _LOG.info('moving to %g onto the surface' % depth)
81
82         stepper_tolerance = 2*self.stepper.step_size
83
84         axis = self.piezo.axis_by_name(self.axis_name)
85
86         zero = _convert_volts_to_bits(axis.config['channel'], 0)
87         target_def = _convert_volts_to_bits(axis.config['channel'], setpoint)
88
89         _LOG.debug('zero the %s piezo output' % self.axis_name)
90         self.piezo.jump(axis_name=self.axis_name, position=zero)
91
92         _LOG.debug("see if we're starting near the surface")
93         try:
94             pos = self.piezo.get_surface_position(
95                 axis_name=self.axis_name, max_deflection=target_def)
96         except _FlatFit, e:
97             _LOG.info(e)
98             pos = self._stepper_approach_again(
99                 target_deflection=target_def, far=far)
100         except _SurfaceError, e:
101             _LOG.info(e)
102             pos = self._stepper_approach_again(
103                 target_deflection=target_def, far=far)
104
105         pos_m = _convert_bits_to_meters(axis.config, pos)
106         _LOG.debug('located surface at stepper %d, piezo %d (%g m)'
107                   % (self.stepper.position, pos, pos_m))
108
109         _LOG.debug('fine tune the stepper position')
110         while pos_m < -stepper_tolerance:  # step back if we need to
111             _LOG.debug('step back')
112             self.stepper.step_relative(-1, backlash_safe=True)
113             try:
114                 pos = self.piezo.get_surface_position(
115                     axis_name=self.axis_name, max_deflection=target_def)
116             except _FlatFit, e:
117                 _LOG.debug(e)
118                 continue
119             pos_m = _convert_bits_to_meters(axis.config, pos)
120             _LOG.debug('located surface at stepper %d, piezo %d (%g m)'
121                       % (self.stepper.position, pos, pos_m))
122         while pos_m > stepper_tolerance:  # step forward if we need to
123             _LOG.debug('step forward')
124             self.stepper.step_relative(1)
125             try:
126                 pos = self.piezo.get_surface_position(
127                     axis_name=self.axis_name, max_deflection=target_def)
128             except _FlatFit, e:
129                 _LOG.debug(e)
130                 continue
131             pos_m = _convert_bits_to_meters(axis.config, pos)
132             _LOG.debug('located surface at stepper %d, piezo %d (%g m)'
133                       % (self.stepper.position, pos, pos_m))
134
135         _LOG.debug('adjust the %s piezo to place us just onto the surface'
136                   % self.axis_name)
137         target_m = pos_m + depth
138         target = _convert_meters_to_bits(axis.config, target_m)
139         self.piezo.jump(self.axis_name, target)
140
141         _LOG.debug(
142             'positioned %g m into the surface at stepper %d, piezo %d (%g m)'
143             % (depth, self.stepper.position, target, target_m))
144
145
146     def _stepper_approach_again(self, target_deflection, far):
147         _LOG.info('back off %d half steps and approach until deflection > %g'
148                  % (far, target_deflection))
149
150         # back away
151         self.stepper.step_relative(-far, backlash_safe=True)
152
153         cd = self.piezo.read_deflection()  # cd = current deflection in bits
154         _LOG.debug('single stepping approach')
155         while cd < target_deflection:
156             _LOG.debug('deflection %g < setpoint %g.  step closer'
157                       % (cd, target_deflection))
158             self.stepper.single_step(1)  # step in
159             cd = self.piezo.read_deflection()
160
161         for i in range(2*max(1, self.stepper.backlash)):
162             _LOG.debug(
163                 'additional surface location attempt (stepping backwards)')
164             try:
165                 pos = self.piezo.get_surface_position(
166                     axis_name=self.axis_name, max_deflection=target_deflection)
167                 return pos
168             except _SurfaceError, e:
169                 _LOG.info(e)
170             self.stepper.single_step(-1)  # step out
171         _LOG.debug('giving up on finding the surface')
172         _LOG.warn(e)
173         raise e