How to make a batch fit program run on multiple cores?

DisplayHelpTopic "ThreadSafe Functions and Multitasking"

ThreadSafe Function lingaussFit(coefwave,data)
wave data, coefwave

    FuncFit/NTHR=0 linguass coefwave  data /D

End

function BatchlinGauss(Nmax)
    variable Nmax
    variable ii,jj
    wave W_coef1,W_sigma1
    wave W_coef2,W_sigma2
    Variable Nmid=Nmax/2   //split total number in half to run on separate cores
   
    Variable tgID= ThreadGroupCreate(2)
 
     //Create batch coeficent waves
    make /N=(Nmax) W_A
    make /N=(Nmax) W_Aerr
    make /N=(Nmax) W_width
    make /N=(Nmax) W_widtherr
    make /N=(Nmax) W_y0
    make /N=(Nmax) W_y0err
    make /N=(Nmax) W_B
    make /N=(Nmax) W_Berr
    make /N=(Nmax) W_c
    make /N=(Nmax) W_cerr
 
    for(ii=0;ii<Nmid;ii+=1)
   
    jj=ii+Nmid
   
    string loopwave1="wave"+num2str(ii)
    string loopwave2="wave"+num2str(jj)

    //Parallel on different cores
    ThreadStart tgID, 0, lingaussFit(W_coef1,$loopwave1)
    ThreadStart tgID, 1, lingaussFit(W_coef2,$loopwave2)
   
    //Write results from first thread
    W_A[ii]=W_coef1[0]
    W_Aerr[ii]=W_sigma1[0]
    W_width[ii]=W_coef1[3]
    W_widtherr[ii]=W_sigma1[3]
    W_y0[ii]=W_coef1[4]
    W_y0err[ii]=W_sigma1[4]
    W_B[ii]=W_coef1[1]
    W_Berr[ii]=W_sigma1[1]
    W_c[ii]=W_coef1[2]
    W_cerr[ii]=W_sigma1[2]
   
    //write results from second thread.
    W_A[jj]=W_coef2[0]
    W_Aerr[jj]=W_sigma2[0]
    W_width[jj]=W_coef2[3]
    W_widtherr[jj]=W_sigma2[3]
    W_y0[jj]=W_coef2[4]
    W_y0err[jj]=W_sigma2[4]
    W_B[jj]=W_coef2[1]
    W_Berr[jj]=W_sigma2[1]
    W_c[jj]=W_coef2[2]
    W_cerr[jj]=W_sigma2[2]
   
    endfor

end



Function linguass(w,x) : FitFunc
    Wave w
    Variable x

    //CurveFitDialog/ These comments were created by the Curve Fitting dialog. Altering them will
    //CurveFitDialog/ make the function less convenient to work with in the Curve Fitting dialog.
    //CurveFitDialog/ Equation:
    //CurveFitDialog/ f(x) = a*x+b*exp(-(x-c)^2/d^2)+y_0
    //CurveFitDialog/ End of Equation
    //CurveFitDialog/ Independent Variables 1
    //CurveFitDialog/ x
    //CurveFitDialog/ Coefficients 5
    //CurveFitDialog/ w[0] = a
    //CurveFitDialog/ w[1] = b
    //CurveFitDialog/ w[2] = c
    //CurveFitDialog/ w[3] = d
    //CurveFitDialog/ w[4] = y_0

    return w[0]*x+w[1]*exp(-(x-w[2])^2/w[3]^2)+w[4]
End

<pre><code class="language-igor">

    // Wait for all threads to finish before starting another batch of threads
    do
        Variable threadGroupStatus = ThreadGroupWait(tgID,100)
    while (threadGroupStatus != 0)

Function MultiThreadIt(Nmax)
    variable nMax
 
    Make /N=(Nmax) /D /O W_A, W_Aerr, W_width, W_widtherr, W_y0, w_y0err
 
    Make /N=(Nmax) /FREE W_Dummy
    // W_Dummy exists only so the MultiThread keyword can be used.
    // since MultiThread expects a wave assignment, we give it one,
    // even though its contents will be discarded
 
 
    // pack the data waves into a 'wave wave'. I use a wave wave simply
    // because I like it, but it's entirely optional
    Make /WAVE /FREE W_DataWaves
    variable i
    for (i = 0; i < Nmax; i+=1)
        W_DataWaves[i] = $("wave" + num2istr(i))
    endfor
 
    MultiThread W_Dummy = RunAFit(W_DataWaves, W_A, W_Aerr, W_width, W_widtherr, W_y0, w_y0err, p)
End
 
 
ThreadSafe Function RunAFit(dataWaves, W_A, W_Aerr, W_width, W_widtherr, W_y0, w_y0err, index)
    wave /WAVE dataWaves
    wave W_A, W_Aerr, W_width, W_widtherr, W_y0, w_y0err
    variable index
 
    wave thisData = dataWaves[index]
    CurveFit/M=2/W=0 gauss, thisData /D
 
    wave W_coef
    wave W_sigma
 
    W_A[index]=W_coef[1]
    W_Aerr[index]=W_sigma[1]
    W_width[index]=W_coef[0]
    W_widtherr[index]=W_sigma[0]
    W_y0[index]=W_coef[3]
    W_y0err[index]=W_sigma[3]
 
    return 0    // make the multithread wave assignment happy
End