Lyon changes

exopy_pulses/pulses/manifest.enaml

@@ -432,6 +432,12 @@ enamldef PulsesManagerManifest(PluginManifest):
             Shape:
                 shape = 'slope_shape:SlopeShape'
                 view = 'views.slope_shape_view:SlopeShapeView'
+            Shape:
+                shape = 'gaussian_shape:GaussianShape'
+                view = 'views.gaussian_shape_view:GaussianShapeView'
+            Shape:
+                shape = 'tanh_shape:TanhShape'
+                view = 'views.tanh_shape_view:TanhShapeView'
 
     PulsesBuildingDependenciesExtension:
         pass

exopy_pulses/pulses/shapes/gaussian_shape.py

@@ -0,0 +1,88 @@
+# -*- coding: utf-8 -*-
+# -----------------------------------------------------------------------------
+# Copyright 2021 by ExopyPulses Authors, see AUTHORS for more details.
+#
+# Distributed under the terms of the BSD license.
+#
+# The full license is in the file LICENCE, distributed with this software.
+# -----------------------------------------------------------------------------
+"""Most basic shape for an analogical pulse.
+
+"""
+from numbers import Real
+
+import numpy as np
+from atom.api import Str
+
+from ..utils.validators import Feval
+
+from .base_shape import AbstractShape
+
+
+class GaussianShape(AbstractShape):
+    """ Gaussian pulse with a variable amplitude and sigma.
+
+    """
+    #: Amplitude of the pulse this should be a number between -1.0 and 1.0
+    amplitude = Str('1.0').tag(pref=True, feval=Feval(types=Real))
+
+    #: Sigma of gaussian pulse, units are AWG context units
+    sigma = Str('10.0').tag(pref=True, feval=Feval(types=Real))
+
+    def eval_entries(self, root_vars, sequence_locals, missing, errors):
+        """ Evaluate the amplitude of the pulse.
+
+        Parameters
+        ----------
+        root_vars : dict
+            Global variables. As shapes and modulation cannot update them an
+            empty dict is passed.
+
+        sequence_locals : dict
+            Known locals variables for the pulse sequence.
+
+        missing : set
+            Set of variables missing to evaluate some entries in the sequence.
+
+        errors : dict
+            Errors which occurred when trying to compile the pulse sequence.
+
+        Returns
+        -------
+        result : bool
+            Flag indicating whether or not the evaluation succeeded.
+
+        """
+        res = super(GaussianShape, self).eval_entries(root_vars, sequence_locals,
+                                                    missing, errors)
+
+        if res:
+            if not -1.0 <= self._cache['amplitude'] <= 1.0:
+                msg = 'Shape amplitude must be between -1 and 1.'
+                errors[self.format_error_id('amplitude')] = msg
+                res = False
+
+        return res
+
+    def compute(self, time, unit):
+        """ Computes the shape of the pulse at a given time.
+
+        Parameters
+        ----------
+        time : ndarray
+            Times at which to compute the modulation.
+
+        unit : str
+            Unit in which the time is expressed.
+
+        Returns
+        -------
+        shape : ndarray
+            Amplitude of the pulse.
+
+        """
+        amp = self._cache['amplitude']
+        sigma = self._cache['sigma']
+        t0 = (time[0]+time[-1])/2
+        pulse_shape = [amp*np.exp(-(t-t0)**2/2/sigma**2) for t in time]
+        return np.asarray(pulse_shape)

exopy_pulses/pulses/shapes/tanh_shape.py

@@ -0,0 +1,90 @@
+# -*- coding: utf-8 -*-
+# -----------------------------------------------------------------------------
+# Copyright 2015-2021 by ExopyPulses Authors, see AUTHORS for more details.
+#
+# Distributed under the terms of the BSD license.
+#
+# The full license is in the file LICENCE, distributed with this software.
+# -----------------------------------------------------------------------------
+"""Most basic shape for an analogical pulse.
+
+"""
+from numbers import Real
+
+import numpy as np
+from atom.api import Str
+
+from ..utils.validators import Feval
+
+from .base_shape import AbstractShape
+
+
+class TanhShape(AbstractShape):
+    """ Tanh pulse with a variable amplitude and sigma.
+
+    """
+    #: Amplitude of the pulse this should be a number between -1.0 and 1.0
+    amplitude = Str('1.0').tag(pref=True, feval=Feval(types=Real))
+
+    #: Sigma of tanh pulse, basically raising time, units are AWG context units
+    sigma = Str('10.0').tag(pref=True, feval=Feval(types=Real))
+
+    def eval_entries(self, root_vars, sequence_locals, missing, errors):
+        """ Evaluate the amplitude of the pulse.
+
+        Parameters
+        ----------
+        root_vars : dict
+            Global variables. As shapes and modulation cannot update them an
+            empty dict is passed.
+
+        sequence_locals : dict
+            Known locals variables for the pulse sequence.
+
+        missing : set
+            Set of variables missing to evaluate some entries in the sequence.
+
+        errors : dict
+            Errors which occurred when trying to compile the pulse sequence.
+
+        Returns
+        -------
+        result : bool
+            Flag indicating whether or not the evaluation succeeded.
+
+        """
+        res = super(TanhShape, self).eval_entries(root_vars, sequence_locals,
+                                                    missing, errors)
+
+        if res:
+            if not -1.0 <= self._cache['amplitude'] <= 1.0:
+                msg = 'Shape amplitude must be between -1 and 1.'
+                errors[self.format_error_id('amplitude')] = msg
+                res = False
+
+        return res
+
+    def compute(self, time, unit):
+        """ Computes the shape of the pulse at a given time.
+
+        Parameters
+        ----------
+        time : ndarray
+            Times at which to compute the modulation.
+
+        unit : str
+            Unit in which the time is expressed.
+
+        Returns
+        -------
+        shape : ndarray
+            Amplitude of the pulse.
+
+        """
+        amp = self._cache['amplitude']
+        sigma = self._cache['sigma']
+        t0 = (time[0]+time[-1])/2
+        duration = time[-1]-time[0]
+        func = lambda t:(0.5+0.5*np.tanh((t+duration/2)/sigma*2*np.pi-np.pi))
+        pulse_shape = [amp*func(t-t0)*func(-(t-t0)) for t in time]
+        return np.asarray(pulse_shape)

exopy_pulses/pulses/shapes/views/gaussian_shape_view.enaml

@@ -0,0 +1,42 @@
+# -*- coding: utf-8 -*-
+# -----------------------------------------------------------------------------
+# Copyright 2015-2021 by ExopyPulses Authors, see AUTHORS for more details.
+#
+# Distributed under the terms of the BSD license.
+#
+# The full license is in the file LICENCE, distributed with this software.
+# -----------------------------------------------------------------------------
+"""View for the Gaussian Shape.
+
+"""
+from enaml.layout.api import hbox, vbox, align
+from enaml.widgets.api import (Label, GroupBox)
+
+from exopy.utils.widgets.qt_completers import QtLineCompleter
+from ...utils.entry_eval import EVALUATER_TOOLTIP
+
+from .base_shape_view import AbstractShapeView
+
+enamldef GaussianShapeView(AbstractShapeView): view:
+    """ View for a Gaussian pulse.
+
+    """
+    GroupBox:
+        title = 'Gaussian'
+        constraints = [hbox(amp_lab, amp_val, sigma_lab, sigma_val),
+                       align('v_center', amp_lab, amp_val, sigma_lab, sigma_val)]
+
+        Label: amp_lab:
+            text = 'Amplitude'
+        QtLineCompleter: amp_val:
+            text := shape.amplitude
+            entries_updater = item.parent.get_accessible_vars
+            tool_tip = ('Relative amplitude of the pulse (should be between '
+                        '-1.0 and 1.0)')
+
+        Label: sigma_lab:
+            text = 'Sigma'
+        QtLineCompleter: sigma_val:
+            text := shape.sigma
+            entries_updater = item.parent.get_accessible_vars
+            tool_tip = ('Sigma of gaussian pulse, units are AWG context units')

exopy_pulses/pulses/shapes/views/tanh_shape_view.enaml

@@ -0,0 +1,42 @@
+# -*- coding: utf-8 -*-
+# -----------------------------------------------------------------------------
+# Copyright 2015-2021 by ExopyPulses Authors, see AUTHORS for more details.
+#
+# Distributed under the terms of the BSD license.
+#
+# The full license is in the file LICENCE, distributed with this software.
+# -----------------------------------------------------------------------------
+"""View for the Gaussian Shape.
+
+"""
+from enaml.layout.api import hbox, vbox, align
+from enaml.widgets.api import (Label, GroupBox)
+
+from exopy.utils.widgets.qt_completers import QtLineCompleter
+from ...utils.entry_eval import EVALUATER_TOOLTIP
+
+from .base_shape_view import AbstractShapeView
+
+enamldef TanhShapeView(AbstractShapeView): view:
+    """ View for a Tanh pulse.
+
+    """
+    GroupBox:
+        title = 'Tanh'
+        constraints = [hbox(amp_lab, amp_val, sigma_lab, sigma_val),
+                       align('v_center', amp_lab, amp_val, sigma_lab, sigma_val)]
+
+        Label: amp_lab:
+            text = 'Amplitude'
+        QtLineCompleter: amp_val:
+            text := shape.amplitude
+            entries_updater = item.parent.get_accessible_vars
+            tool_tip = ('Relative amplitude of the pulse (should be between '
+                        '-1.0 and 1.0)')
+
+        Label: sigma_lab:
+            text = 'Sigma'
+        QtLineCompleter: sigma_val:
+            text := shape.sigma
+            entries_updater = item.parent.get_accessible_vars
+            tool_tip = ('Sigma of tanh pulse, basically raising time, units are AWG context units')

exopy_pulses/tasks/manifest.enaml

@@ -31,3 +31,11 @@ enamldef PulsesTasksManifest(PluginManifest):
                 # Way to declare instrument dependencies without specifying
                 # any instrument.
                 instruments = [None]
+            path = 'exopy_pulses.tasks.tasks.instrs'
+            Task:
+                task = 'transfer_pulse_loop_task:TransferPulseLoopTask'
+                view = ('views.transfer_pulse_loop_task_view:'
+                        'TransferPulseLoopView')
+                # Way to declare instrument dependencies without specifying
+                # any instrument.
+                instruments = [None]