Upload New File: P_amp.py

proxtoolbox/ProxOperators/P_amp.py

+from proxoperators import ProxOperator
+import numpy as np
+
+class P_amp(ProxOperator):
+	"""
+	Projection  onto amplitude constraints    
+   	"""	
+
+	def __init__(self, config):
+    	"""
+        Initialization
+        
+        Parameters
+        ----------
+        config : dict - Dictionary containing the problem configuration. It must contain the following mapping:
+		'amplitude' : array_like - nonnegative array
+        """
+		self.amplitude = config['amplitude']
+
+	def work(self,u):
+		"""
+		Evaluates Magproj (Projection operator onto a magnitude constraint) with arguments u and the amplitude constraints
+
+        Parameters
+        ----------
+        u : array_like - complex-valued array to be projected onto the amplitude constraints
+        
+        Returns
+        -------
+        array_like - the Projection
+        """
+		a = self.amplitude
+		return eval('magproj(u,a)')