from herschel.pacs.spg.phot import PhotHelper

"""
  Get the calibration tree
"""
calTree = getCalTree(obs=getObservation(obsids[0], useHsa=True, instrument="PACS"))

"""
  Create the wcs
"""
fitsFile = dataDir + str(obsids[0]) +  "-" + str(obsids[1])  +  "-red-aligned.fits"
mergedFrames = FitsArchive().load(fitsFile)
wcs = wcs4map(mergedFrames, outputPixelsize=outputPixelSize, crota=0, calTree=calTree)

"""
  Merge all the aligned frames from the red camera
"""
camera = "red"
cameraName = "red"

fitsFile = dataDir + str(obsids[0]) +  "-" + str(obsids[1])  +  "-" + cameraName + "-aligned.fits"
mergedFrames = FitsArchive().load(fitsFile)

for i in range(len(borderObsids)/2):
   fitsFile = dataDir + str(borderObsids[2*i]) +  "-" + str(borderObsids[2*i + 1])  +  "-" + cameraName + "-aligned.fits"
   mergedFrames.join(FitsArchive().load(fitsFile))

"""
  Create the red map including all data
"""
redMapExt, mi = photProject(mergedFrames, outputPixelsize=outputPixelSize, pixfrac=pixfrac, calTree=calTree, wcs=wcs)

"""
  Create the red map without the borders
"""
mergedFrames.setActive("Scanamorphos_NonOverlapMask", True)
redMap, mi = photProject(mergedFrames, outputPixelsize=outputPixelSize, pixfrac=pixfrac, calTree=calTree, wcs=wcs)

"""
  Combine the two maps
"""
redMapCombined =  PhotHelper.combineMaps(redMap, redMapExt)

"""
  Remove the map offset
"""
PhotHelper.removeOffset(redMapCombined, redOffset)

"""
  Save the map
"""
outputFitsFile = dataDir + "map-" + str(obsids[0]) +  "-" + str(obsids[1]) + "-" + cameraName + ".fits"
FitsArchive().save(outputFitsFile, redMapCombined)


"""
  Merge all the aligned frames from the blue camera
"""
camera = "blue"
cameraName = "blue"

try:
   fitsFile = dataDir + str(obsids[0]) +  "-" + str(obsids[1])  +  "-" + cameraName + "-aligned.fits"
   mergedFrames = FitsArchive().load(fitsFile)
except:
   cameraName = "green"
   fitsFile = dataDir + str(obsids[0]) +  "-" + str(obsids[1])  +  "-" + cameraName + "-aligned.fits"
   mergedFrames = FitsArchive().load(fitsFile)

for i in range(len(borderObsids)/2):
   fitsFile = dataDir + str(borderObsids[2*i]) +  "-" + str(borderObsids[2*i + 1])  +  "-" + cameraName + "-aligned.fits"
   mergedFrames.join(FitsArchive().load(fitsFile))

"""
  Create the blue map including all data
"""
blueMapExt, mi = photProject(mergedFrames, outputPixelsize=outputPixelSize, pixfrac=pixfrac, calTree=calTree, wcs=wcs)

"""
  Create the blue map without the borders
"""
mergedFrames.setActive("Scanamorphos_NonOverlapMask", True)
blueMap, mi = photProject(mergedFrames, outputPixelsize=outputPixelSize, pixfrac=pixfrac, calTree=calTree, wcs=wcs)

"""
  Combine the two maps
"""
blueMapCombined =  PhotHelper.combineMaps(blueMap, blueMapExt)

"""
  Remove the map offset
"""
PhotHelper.removeOffset(blueMapCombined, blueOffset)

"""
  Save the map
"""
outputFitsFile = dataDir + "map-" + str(obsids[0]) +  "-" + str(obsids[1]) + "-" + cameraName + ".fits"
FitsArchive().save(outputFitsFile, blueMapCombined)