dice_inputs.py

Go to the documentation of this file.

 
 
# Import python functions
from DicePy.dice_write import *
from DicePy.dice_check import *
 
#-------------------------
# MAIN ROUTINE FOR INPUTS-
#-------------------------
 
 
def dice_inputs():
 
    # Welcome Message
    print('')
    print('------------')
    print('Dice QMC')
    print('------------')
    print('')
    print('Hello! I\'m here to help you with the input file for your Dice QMC simulation.')
 
    # Query the Task
    print('')
    print('Would you like me to write (w) or check (c) your input file?')
    task = input('Your response ("w" or "c", without quotes): ')
 
    # Check
    while task!='w' and task!='c':
        print('')
        print('Error: Oops, your response is invalid, please enter "w" or "c", without quotes.')
        task = input('Your response: ')
 
    # Report
    if task=='w':
        print('')
        print('Alright! I shall write out a new input file for you.')
 
        # Query the user for data
        control, params = query_params()
 
        # Write the params.txt file
        write_params(control,params)
 
        # Notify user
        print('')
        print('Your params.txt file is ready! You are all set to run the next step on Fortran :-) ')
 
    else:
        print('')
        print('Alright! I shall check the input file in this directory.')
 
        # Check that params.txt exits
        try: 
            f = open('params.txt')
        except FileNotFoundError:
            print('')
            print('Error: Oops, there is no param.txt file in the current working directory.')
            print('       Please copy your params.txt file into this directory, and then run dice_inputs() again.')
            print('       Exiting the program.')
            print(''); exit()
        f.close()
 
        # Reading Namelists
        import f90nml
        nml = f90nml.read('params.txt')
 
        # Check content of params.txt
        check_inputs(nml)
 
        # Notify user
        print('')
        print('Your params.txt file is correct! You are all set to run the next step on Fortran :-) ')
 
# For users running the program directly on the terminal,
# call the input routine defined above
if __name__ == "__main__":
    dice_inputs()