def constraint(self, xx, bb): ''' include TVconstraint and others ''' cons = CsSolver.TVconstraint(xx[0:2], bb[0:2]) * self.LMBD / 40.0 #cons = CsSolver.TVconstraint(xx[0:2],bb[0:2]) * self.LMBD/100.0 #cons = cons + CsSolver.TVconstraint(xx[2:3],bb[2:3]) * self.LMBD cons = cons + scipy.fftpack.ifftn(xx[3] - bb[3], axes=(2, )) * self.gamma cons = cons + (xx[4] - bb[4]) * self.gamma #cons = cons + xx[2]-bb[2] #print('inside constraint, cons.shpae',cons.shape) # cons = cons + freq_gradient_H(xx[3]-bb[3]) #print('inside constraint 1117, cons.shpae',cons.shape) return cons