package hep.aida.ref.test.jaida;

import hep.aida.*;
import java.util.Random;
import junit.framework.TestCase;


/**
 * 
 * @author Tony Johnson
 * @version $Id: TestFitting.java,v 1.4 2004/07/23 21:53:48 turri Exp $
 */
public class TestFitting extends TestCase
{
   private IAnalysisFactory af;
   private ITree tree;
   private IHistogram1D hist;
   
   /** Creates a new instance of FitTest */
   public TestFitting(String name)
   {
      super(name);
   }
   public void testMinuitScriptedFunction()
   {
      testScriptedFunction("minuit");
   }
   public void testUncminScriptedFunction()
   {
      testScriptedFunction("uncmin");
   }
   public void testMinuitBuiltinFunction()
   {
      testBuiltinFunction("minuit");
   }
   public void testUncminBuiltinFunction()
   {
      testBuiltinFunction("uncmin");
   }
   private void testScriptedFunction(String fitter)
   {
      IFunctionFactory functionfact = af.createFunctionFactory(tree);
      IFunction f = functionfact.createFunctionFromScript("gauss",1,"b*lambda*exp(-x[0]*lambda)+a*exp(-(x[0]-mean)*(x[0]-mean)/sigma/sigma)","b,lambda,a,mean,sigma","Gaussian+Exponential");
      assertEquals(1,f.dimension());
      assertEquals(5,f.parameters().length);
      f.setParameter("b",1800);
      f.setParameter("lambda",.3);
      f.setParameter("a",70);
      f.setParameter("mean",10);
      f.setParameter("sigma",1.5);
      
      testFunction(fitter,f);
   }
   private void testBuiltinFunction(String fitter)
   {
      IFunctionFactory functionfact = af.createFunctionFactory(tree);
      IFunction f = functionfact.createFunctionByName("e+g","e+g");
      assertEquals(1,f.dimension());
      assertEquals(5,f.parameters().length);
      f.setParameter("amplitude",600);
      f.setParameter("exponent",-.3);
      f.setParameter("amplitude_1",80);
      f.setParameter("mean",10);
      f.setParameter("sigma",1);
      
      testFunction(fitter,f);
   }
   private void testFunction(String fitter, IFunction f)
   {
      IFitFactory ff = af.createFitFactory();
      IFitter minuit = ff.createFitter("Chi2",fitter);
      IFitResult result = minuit.fit(hist,f);
      
      assertTrue(result.isValid());
      assertEquals(5,result.fittedParameterNames().length);
      //assertEquals(95,result.ndf());
      int status = result.fitStatus();
      assertTrue(status>0 && status<6);
      assertTrue(result.quality()<1.5);
   }
   
   protected void setUp() throws java.lang.Exception
   {
      af = IAnalysisFactory.create();
      tree = af.createTreeFactory().create();
      IHistogramFactory hf = af.createHistogramFactory(tree);

      Random r = new Random();
      
      hist = hf.createHistogram1D("hist",100,0,20);

      for (int i=0; i<10000; i++)
      {
         hist.fill(-Math.log(r.nextDouble()) / .3);
      }
      for (int i=0; i<1000; i++)
      {
         hist.fill(r.nextGaussian()+10);
      }
   }   
}