Как написать подпрограмму для начального условия в нелинейном уравнении Шредингера, которое зависит от x

Question

Я решаю нелинейное уравнение Шредингера методом Фурье с разделением шагов: i df/dz+1/2* d^2f/dX^2+|f|^2*f=0

используя начальное условие: f=q*exp(-(X/X0)^24).

Но я должен использовать условие, что q = 1 для |x|<1, иначе q=0. Поэтому я пишу следующую подпрограмму (отрывок из кода с поперечной переменной) для поперечной переменной x:

   fs=120;
   N_fx=2^11; %number of points in frequency domain
   dX=1/fs;
   N_X=N_fx;
   X=(-(N_X-1)/2:(N_X-1)/2)*dX;
   X0=1;
   Xn=length(X);

   for m=1:Xn
   Xnn=Xn/8;
   pp=m;
   if pp>3*Xnn && pp<5*Xnn
   q=1.0;
   f=q*exp(-(X/X0).^24);
   else  
   f=0;
   end
   end

Но похоже, что 'f' ошибается, и это вектор 1 на 2048 со всеми нулевыми записями. Я не получаю ожидаемого результата. Если мое начальное условие только f=q*exp(-(X/X0).^24), q=1, это просто, но с указанным выше условием (q=1 для |x|<1, в противном случае q=0) что мне делать? Любая помощь будет высоко ценится. Заранее спасибо.

Answer 1

MWE, это 0 < [f(867) : f(1162)] <= 1:

fs=120;
N_fx=2^11; %number of points in frequency domain
dX=1/fs;
N_X=N_fx;
X=(-(N_X-1)/2:(N_X-1)/2)*dX;
X0=1;
Xn=length(X);
f = zeros(1,Xn);  % new: preallocate f size and initialise it to 0

for m=1:Xn
    Xnn=Xn/8;
    if m>3*Xnn && m<5*Xnn
   %if abs(X(m)) < 1   %alternative to line above
        q=1.0;
        % error was here below: you overwrote a 1x1 f at each iteration
        f(m)=q*exp(-(X(m)/X0).^24);   
    else
        f(m)=0;
    end
end