casa6

import os

import sys

import time

import argparse

import multiprocessing

from IPython.terminal.prompts import Prompts, Token

​

try:

    from casac import casac

except ImportError, e:

    print "failed to load casa:\n", e

    os._exit(1)

​

try:

    import matplotlib

except ImportError, e:

    print "failed to load matplotlib:\n", e

    print "sys.path =", "\n\t".join(sys.path)

​

from casa_system import casa

​

if not os.environ.has_key('OMP_NUM_THREADS'):

    # if OMP_NUM_THREADS is not set, set it to max(1,N_CPU-2)

    os.environ['OMP_NUM_THREADS'] = str(max(1,multiprocessing.cpu_count()-2))

​

class _Prompt(Prompts):

     def in_prompt_tokens(self, cli=None):

         return [(Token.Prompt, 'CASA <'),

                 (Token.PromptNum, str(self.shell.execution_count)),

                 (Token.Prompt, '>: ')]

​

_ip = get_ipython()

_ip.prompts = _Prompt(_ip)

​

###

### provide extra context for T/F errors...

###

def true_false_handler(self, etype, value, tb, tb_offset=None):

    if type(etype) is type(NameError):

        if str(value) == "name 'T' is not defined" or \

           str(value) == "name 'F' is not defined" or \

           str(value) == "name 'true' is not defined" or \

           str(value) == "name 'false' is not defined" :

            print "------------------------------------------------------------------------------"

            print "Warning: CASA no longer defines T/true and F/false as synonyms for True/False"

            print "------------------------------------------------------------------------------"

    ###

    ### without incrementing the execution_count, you get:

    ###

    ###    ERROR! Session/line number was not unique in database. History logging moved to new session 51

    ###

    ### on the second syntax error in a row... presumably this will eventually

    ### be fixed in ipython and then this will need to be removed... but such

    ### is the current state of affairs in ipython 5.1.0...

    ###

    self.execution_count += 1

    self.showtraceback((etype, value, tb), tb_offset=tb_offset)

​

_ip.set_custom_exc((BaseException,), true_false_handler)

​

​

##

## toplevel frame marker

##

_casa_top_frame_ = True

​

## ---- ---- ---- ---- ---- ---- ---- ---- ---- ---- ---- ----

## set up casa root

## ---- ---- ---- ---- ---- ---- ---- ---- ---- ---- ---- ----

if os.environ.has_key('CASAPATH') :

    __casapath__ = os.environ['CASAPATH'].split(' ')[0]

    __casaarch__ = os.environ['CASAPATH'].split(' ')[1]

    if not os.path.exists(__casapath__ + "/data") :

        print "DEBUG: CASAPATH = %s" % (__casapath__)

        print "Unable to find the data repository directory in your CASAPATH. Please fix."

        os._exit(1)

    else :

        casa['dirs']['root'] = __casapath__

        casa['dirs']['data'] = __casapath__ + "/data"

​

        if os.path.exists(__casapath__ + "/lib/python2.7/start_casa.py"):

            casa['dirs']['python'] = __casapath__ + "/lib/python2.7"

        elif os.path.exists(__casapath__ + "/Resources/python/start_casa.py"):

            casa['dirs']['python'] = __casapath__ + "/Resources/python"

        elif os.path.exists(__casapath__ + "/" + __casaarch__ + "/lib/python2.7/start_casa.py"):

            casa['dirs']['python'] = __casapath__ + "/" + __casaarch__ + "/lib/python2.7"

​

        if casa['dirs']['python'] is not None:

            casa['dirs']['recipes'] = casa['dirs']['python'] + "/recipes"

​

        if os.path.exists(__casapath__ + "/" + __casaarch__ + "/xml"):