#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
Fit model using multiple scattering.

As of this writing, multiscattering isn't integrated into sasmodels, and a
number of hacks are required to get it running as a fit.

The appropriate items need to be on the python path.  These include
sasview (for reading the data), sasmodels and bumps.  The multiscat module
(currently in the sasmodels/explore directory) is also needed, either beside
this example fit file, or by putting sasmdoels/explore on the python path.

On Unix/Mac running as developer I do::

    # Show the model without fitting
    PYTHONPATH=..:../explore:../../bumps:../../sasview/src python multiscatfit.py

    # Run the fit
    PYTHONPATH=..:../explore:../../bumps:../../sasview/src ../../bumps/run.py \
    multiscatfit.py --store=/tmp/t1

You may be able to run multiscatfit.py against the distributed sasview
application (if it is new enough, and if you have multiscat.py in the
same directory).  You probably need a command such as::

    sasview.exe bumps.cli multiscatfit.py --store=t1
"""

import sys
from bumps.names import *
from sasmodels.core import load_model
from sasmodels.bumps_model import Model, Experiment
from sasmodels.data import load_data, set_beam_stop, set_top

from multiscat import MultipleScattering

## Load the data
#data = load_data('DEC07267.DAT')
#set_beam_stop(data, 0.003, outer=0.025)
data = load_data('latex_smeared.xml', index=1)

## Define the model
kernel = load_model("ellipsoid")

model = Model(
    kernel,
    scale=0.005, background=0.05,
    radius_polar=2200, radius_equatorial=2200,
    sld=.291, sld_solvent=7.105,
    #theta=90, theta_pd=0, theta_pd_n=0, theta_pd_nsigma=3,
    #phi=90, phi_pd=0, phi_pd_n=20, phi_pd_nsigma=3,
    radius_polar_pd=0.222296, radius_polar_pd_n=1, radius_polar_pd_nsigma=0,
    radius_equatorial_pd=.000128, radius_equatorial_pd_n=1, radius_equatorial_pd_nsigma=0,
    )

# SET THE FITTING PARAMETERS
model.radius_polar.range(15, 3000)
model.radius_equatorial.range(15, 3000)
#model.theta.range(0, 90)
#model.theta_pd.range(0,10)
#model.phi_pd.range(0,20)
#model.phi.range(0, 180)
model.background.range(0,1000)
model.scale.range(0, 0.1)

# Mulitple scattering probability parameter
# HACK: the parameter is assigned to model.theta, which is otherwise unused
# since the dataset is 1D; this won't work for 2D data
probability = Parameter(name="probability", value=0.0)
probability.range(0.0, 0.9)
model.phi = probability

M = Experiment(data=data, model=model)

# Stack mulitple scattering on top of the existing resolution function.
# Because resolution functions in sasview don't have fitting parameters,
# we instead allow the multiple scattering calculator to take a function
# instead of a probability.  This function returns the current value of
# the parameter. ** THIS IS TEMPORARY ** when multiple scattering is
# properly integrated into sasmodels and sasview, its fittable parameter
# will be treated like the model parameters.
M.resolution = MultipleScattering(resolution=M.resolution,
                                  probability=lambda: probability.value,
                                  )
M._kernel_inputs = M.resolution.q_calc
problem = FitProblem(M)

if __name__ == "__main__":
    #M.theory()
    M.plot()
    import pylab; pylab.show()