from UnbinnedAnalysis import *
from BinnedAnalysis import *
from Composite2 import Composite2
from CompositeLikelihood import CompositeLikelihood

optimizer = 'MINUIT'

blike = binnedAnalysis(irfs='DC1A',
                       expcube='expcube_1_day.fits',
                       srcmdl='srcModel_galprop.xml',
                       optimizer=optimizer,
                       cmap='sourceMaps.fits',
                       bexpmap='binned_exposure.fits')

ulike = unbinnedAnalysis(evfile='filtered_events_0000.fits',
                         scfile='orbSim_scData_0000.fits',
                         expmap='expMap.fits',
                         expcube='expcube_1_day.fits',
                         irfs='DC1A',
                         srcmdl='srcModel_galprop.xml',
                         optimizer=optimizer)

like = Composite2(optimizer=optimizer)
like.addComponent(blike)
like.addComponent(ulike)

target = 'my_3EG_J0530p1323'

tiedParGroup = ((ulike, target, 'Index'),
                (blike, target, 'Index'))
like.tieParameters(tiedParGroup)

# Fit the composite likelihood model.
like.fit(0)

print ulike.model

blike1 = binnedAnalysis(irfs='DC1A',
                        expcube='expcube_1_day.fits',
                        srcmdl='srcModel_galprop.xml',
                        optimizer=optimizer,
                        cmap='sourceMaps.fits',
                        bexpmap='binned_exposure.fits')

ulike1 = unbinnedAnalysis(evfile='filtered_events_0000.fits',
                          scfile='orbSim_scData_0000.fits',
                          expmap='expMap.fits',
                          expcube='expcube_1_day.fits',
                          irfs='DC1A',
                          srcmdl='srcModel_galprop.xml',
                          optimizer=optimizer)

composite_like = CompositeLikelihood(optimizer=optimizer)

composite_like.addComponent(target, blike1)
composite_like.addComponent(target, ulike1)

composite_like.fit(0)

print ulike1.model