Restore ion imaging current, add missing datetime import (#29)

DeMarcoLab · Dec 4, 2020 · 980ec61 · 980ec61
1 parent b258224
commit 980ec61
Show file tree

Hide file tree

Showing 4 changed files with 7 additions and 3 deletions.
diff --git a/autolamella/_version.py b/autolamella/_version.py
@@ -1 +1 @@
-__version__ = "0.1.4"
+__version__ = "0.1.5"
diff --git a/autolamella/acquire.py b/autolamella/acquire.py
@@ -204,12 +204,15 @@ def save_final_images(microscope, settings, lamella_number):
         microscope.imaging.set_active_view(2)  # the ion beam view
         microscope.auto_functions.run_auto_cb()
     if settings["imaging"]["full_field_ib_images"]:
+        original_ion_current = microscope.beams.ion_beam.beam_current.value
+        microscope.beams.ion_beam.beam_current.value = 20e-12  # ion imaging
         image = grab_ion_image(microscope, fullfield_cam_settings)
         filename = os.path.join(
             output_dir, "IB_lamella{}-milling-complete.tif".format(
                 lamella_number + 1)
         )
         image.save(filename)
+        microscope.beams.ion_beam.beam_current.value = original_ion_current
     sem_adorned_image = grab_sem_image(microscope, fullfield_cam_settings)
     sem_fname = "SEM_lamella{}-milling-complete.tif".format(lamella_number + 1)
     sem_filename = os.path.join(output_dir, sem_fname)

diff --git a/autolamella/main.py b/autolamella/main.py
@@ -1,3 +1,4 @@
+from datetime import datetime
 import logging
 import os
 

diff --git a/autolamella/milling.py b/autolamella/milling.py
@@ -313,5 +313,5 @@ def mill_all_stages(
             if stage_number + 1 == len(protocol_stages):
                 save_final_images(microscope, settings, lamella_number)
             reset_state(microscope, settings)
-            # Return ion beam current to imaging current (20 pico-Amps)
-            microscope.beams.ion_beam.beam_current.value = 20e-12
+    # Return ion beam current to imaging current (20 pico-Amps)
+    microscope.beams.ion_beam.beam_current.value = 20e-12