casa6

from __future__ import absolute_import

from __future__ import print_function

​

import time

import numpy

import os

import shutil

import re

​

# get is_CASA6 and is_python3, and import other classes

try:

    from casatasks.private.casa_transition import *

except:

    from sys import version_info

    is_python3 = version_info > (3,)

    is_CASA6 = is_python3

if is_CASA6:

    from casatasks import casalog

​

    from casatools import image

    from casatasks.private.imagerhelpers.imager_deconvolver import PyDeconvolver

    from casatasks.private.imagerhelpers.input_parameters import ImagerParameters

    from casatasks.private.parallel.parallel_task_helper import ParallelTaskHelper

    from .cleanhelper import write_tclean_history, get_func_params

    from casatools import synthesisimager

    ia = image( )

else:

    from taskinit import *

​

    from imagerhelpers.imager_deconvolver import PyDeconvolver

    from imagerhelpers.input_parameters import ImagerParameters

    from imregrid import imregrid

    from parallel.parallel_task_helper import ParallelTaskHelper

    from cleanhelper import write_tclean_history, get_func_params

    synthesisimager=casac.synthesisimager

    ia = iatool( )

​

try:

    if is_CASA6:

        from casampi.MPIEnvironment import MPIEnvironment

        from casampi import MPIInterface

    else:

        from mpi4casa.MPIEnvironment import MPIEnvironment

        from mpi4casa import MPIInterface

    mpi_available = True

except ImportError:

    mpi_available = False

​

def check_requiredmask_exists(usemask, mask):

    if usemask != "user":        # don't use the mask parameter

        return

    if type(mask) is type([]):   # mask is an array of values

        return

    if mask == "":               # mask is an empty string -> no file specified

        return

    if "[" in mask:              # mask is a region string

        return

    if not os.path.exists(mask): # mask is a filename string <- only option left

        raise RuntimeError("Internal Error: 'mask' parameter specified as a filename '"+mask+"', but no such file exists")

​

def check_requiredimgs_exist(imagename, inp):

    # get the list of images to check for

    reqims = []

    if inp['deconvolver'] == 'mtmfs':

        nterms = inp['nterms']

        end = nterms*2-1

        for ttn in range(0, nterms*2-1):

            reqims.append(imagename + ".psf" + ".tt" + str(ttn))

        for ttn in range(0, nterms):

            reqims.append(imagename + ".residual" + ".tt" + str(ttn))

    else:

        reqims.append(imagename + ".residual")

        reqims.append(imagename + ".psf")

​

    # find images that exist on disk

    allfiles = os.listdir(os.path.dirname(os.path.abspath(imagename)))

    extims = list(filter(lambda im: os.path.exists(im), reqims)) # TODO replace with allfiles

​

    # verify required images are available

    if len(extims) != len(reqims):

        diffims = list(filter(lambda im: im not in extims, reqims))

        raise RuntimeError("Internal Error: missing one or more of the required images: " + str(diffims))

​

    # check for .pb image in the case that nsigma > 0

    # see comments on CAS-13144 about casa crashing as to why this check is here

    if (inp['nsigma'] > 0):

        reqpb = ".pb" if (inp['deconvolver'] != 'mtmfs') else ".pb.tt0"

        if (imagename+reqpb not in allfiles):

            raise RuntimeError("The parameter nsigma>0 ("+str(inp['nsigma'])+") requires a "+reqpb+" image to be available.")

​

def check_starmodel_model_collisions(startmodel, imagename, deconvolver):