function energy = uEnergy(x,prm)

  % The potential gets different values if x is smaller or greater than 0
  right = prm*(x-1).*(x-1);
  left = prm*(x+1).*(x+1);

  % Calculate the energy for x of any size array
  energy = zeros(size(x));     % Zero out the energy
  energy(x<0) = left(x<0);     % Set energy to be the 'left' array for negative x values
  energy(x>=0) = right(x>=0);  % Set energy to be the 'right' array for positive x values
end

Published with MATLAB® R2012b