setup.py

import os
import sys
import shutil

try:
  os.environ['PLUTO_DIR']
except KeyError:
  print('PLUTO_DIR not defined. Setting it to the Current Directory')
  pluto_directory = os.getcwd()
  pass
else:
  pluto_directory = os.environ['PLUTO_DIR']

sys.path.append(pluto_directory + '/Tools/Python/')
import menu
import configure
from make_problem import MakeProblem


def PlutoInterface(pluto_dir, do_auto_update = False):
  work_dir = os.getcwd()
  interface_optval = ''
  interface_opts = ['Setup problem', 'Change makefile', 
                    'Auto-update','Save Setup','Quit']
    
  if do_auto_update:
    MakeProblem(work_dir, pluto_dir, 1, 1)
    interface_optval = interface_opts[-1]   # set to "Quit" so it'll skip next loop

  while interface_optval != interface_opts[-1]:
    menu.SetTitle("Python setup (May 2018)","Working dir: "+work_dir+"\nPLUTO dir  : "+pluto_dir)

    interface_optval = menu.Browse(interface_opts)

    if interface_optval == interface_opts[0]:
      if not os.path.exists(work_dir+'/init.c'):
        shutil.copy(pluto_dir+'/Src/Templates/init.c',work_dir+'/init.c')
    
      if not os.path.exists(work_dir+'/pluto.ini'):
        shutil.copy(pluto_dir+'/Src/Templates/pluto.ini',work_dir+'/pluto.ini')

      MakeProblem(work_dir, pluto_dir, 0, 1)
    
    if interface_optval == interface_opts[1]:
      MakeProblem(work_dir, pluto_dir, 1, 0)

    if interface_optval == interface_opts[2]:
      menu.Prompt('Press Enter to Update '+work_dir)
      MakeProblem(work_dir, pluto_dir, 1, 1)
      menu.Print ("Configuration up to date",sleep=0.75)
      break

    if interface_optval == interface_opts[3]: #Save Setup
      sys.exit()

  if (menu.CursesIsActive()): menu.RestoreScreen()
  print("\n> Done.")
  sys.exit()

if __name__ == "__main__":   # starts from here
  auto_update = 0
  print("\n> Checking system architecture\n")
  configure.check(pluto_directory, 1)
  for x in sys.argv[1:]:     # check argument list

    if (x == "--auto-update"):
      auto_update = 1	

    elif (x == "--get-arch"):
      sys.exit(1)
      break

    elif (x == "--help" or x == "-help"):
      print ("Usage: python $PLUTO_DIR/setup.py [options]\n")
      print ("Here [options] can be:\n")
      print (" --auto-update     Run the python script in background.")
      print (" --no-curses       Disable ncurses library and use a")
      print ("                   simpler text-based menu.")
      print (" --with-chombo     Enable support for adaptive mesh refinement.")
      print ("                   (AMR) module using the Chombo library.")
      print (" --with-fargo      Enable the FARGO-MHD module.")
      print (" --with-fd         Enable the finite difference module.")
      print (" --with-particles  Enable the particle module.")
      print (" --with-sb         Enable the shearing box module.")
      sys.exit(1)

    elif (x == "--no-curses"):
      print("")

    elif (x == "--with-chombo" or x == "--with-chombo:"): 
      print ("Enabling Chombo support for AMR")
      cmset = set(['--with-fd','--with-sb','--with-fargo','--with-particles']) & set(sys.argv)
      if len(cmset) != 0:
        print('! Incompatible modules, ',x,' + '.join(y for y in cmset))
        sys.exit(1)
      break

    elif (x == "--with-fargo"): 
      print ("Enabling support for FARGO scheme")

    elif (x == "--with-fd"): 
      print ("Enabling support for finite difference module")

    elif (x == "--with-particles"): 
      print ("Enabling support for particles")

    elif (x == "--with-sb"): 
      print ("Enabling support for shearing box module")
      if '--with-fd' in sys.argv:
        print('! Incompatible modules, ',x,' +  --with-fd')
        sys.exit(1)

    elif (x == "--with-cr_transport"): 
      print ("Enabling support for cr_transport module")
  
    else:
      print ("! Unrecognized option '",x,"'")
      sys.exit(1)

  print('\n> Loading PLUTO Interface...')
  
  if auto_update == 1:
    PlutoInterface(pluto_directory,do_auto_update=True)
  else:
    PlutoInterface(pluto_directory)