Upload New File

proxtoolbox/P_S.py

+from proxoperators import ProxOperator
+import numpy as np
+
+class P_S(ProxOperator):
+	"""
+	Projection onto support constraints
+	"""
+
+	def __init__(self,config):
+		"""
+        Initialization
+        
+        Parameters
+        ----------
+        config : dict - Dictionary containing the problem configuration. It must contain the following mapping:
+		'support :idx' : array_like - a vector of indeces of the nonzero elements of the array.
+        """
+		self.n = config['support_idx']
+
+	def work(self,u):
+		"""
+        Applies the proxoperator P_S
+        
+        Parameters
+        ----------
+        u : array_like - array to be projected
+        
+        Returns
+        -------
+        p_S : array_like - the Projection
+        """
+		n = self.n
+
+		p_S= np.zeros(u.shape,dtype=u.dtype)
+		for x in n:
+			p_S[x] = u[x]
+		return p_S
+