casa6

import os

import math

import shutil

import string

import time

import re;

import copy

import pdb

​

from casatools import synthesisimager, synthesisnormalizer

from casatasks import casalog

​

from .imager_base import PySynthesisImager

from .parallel_imager_helper import PyParallelImagerHelper

synth_imager_name = 'synthesisimager'

synth_imager_import = 'from casatools import synthesisimager'

​

​

'''

An implementation of parallel continuum imaging, using synthesisxxxx tools

​

Datasets are partitioned by row and major cycles are parallelized. 

Gathers and normalization are done before passing the images to a

non-parallel minor cycle. The output model image is them scattered to

all the nodes for the next parallel major cycle.

​

There are N synthesisimager objects.

There is 1 instance per image field, of the normalizer and deconvolver.

There is 1 iterbot. 

'''

​

#############################################

#############################################

## Parallelize only major cycle.

#############################################

class PyParallelContSynthesisImager(PySynthesisImager):

​

    def __init__(self,params=None):

​

         PySynthesisImager.__init__(self,params)

​

         self.PH = PyParallelImagerHelper()

         self.NN = self.PH.NN

         self.selpars = self.allselpars;

         self.allselpars = self.PH.partitionContDataSelection(self.allselpars)

         # self.allcflist = self.PH.partitionCFCacheList(self.cfcachepars['cflist']);

         # self.allcflist = self.PH.partitionCFCacheList(self.allgridpars['0']);

         self.listOfNodes = self.PH.getNodeList();

         self.coordsyspars = {};

         self.toolsi=None

​

#############################################

    def resetSaveModelParams(self, params=None):

         mainparams = params.getSelPars()

         for n in self.allselpars: # for all nodes

             for v in self.allselpars[n]: # for all MSes

                 self.allselpars[n][v]['readonly']=mainparams[v]['readonly']

                 self.allselpars[n][v]['usescratch']=mainparams[v]['usescratch']

​

#############################################

#    def initializeImagers(self):

#        ### Drygridding, and Coordsys comes from a single imager on MAIN node.