SFGenUtils Namespace Reference

Classes
class	SFGenConfig

Functions
def	writeInputDAT (Init)
def	run_command (command, stdin=None)
def	SFGenInitialize (Init, stdin=None)
def	SFGenExecute (Init)
def	SFGenRun (Init, genSeq)

Variables
	logger = Logging.logging.getLogger("SFGen_i")
	Get handle to Athena logging. More...

Function Documentation

run_command()

def SFGenUtils.run_command	(		command,
			stdin = None
	)

Run a command and print output continuously

Definition at line 159 of file SFGenUtils.py.

def run_command(command, stdin = None ):
    """
    Run a command and print output continuously
    """
    process = subprocess.Popen(shlex.split(command), stdout=subprocess.PIPE, stdin=stdin)
    while True:
        output = process.stdout.readline().decode("utf-8")
        if output == '' and process.poll() is not None:
            break
        if output:
            # remove ANSI escape formatting characters
            reaesc = re.compile(r'(\x9B|\x1B\[)[0-?]*[ -\/]*[@-~]')
            text = reaesc.sub('', output.strip())
            logger.info(text)
 
    rc = process.poll()
    return rc
 
# SFGen initialization
 

SFGenExecute()

def SFGenUtils.SFGenExecute

(

Init

)

Definition at line 212 of file SFGenUtils.py.

def SFGenExecute(Init):
 
    logger.info("Starting SFGen Itself")
 
    if not os.path.exists('evrecs'):
        os.makedirs('evrecs')
    if not os.path.exists('outputs'):
        os.makedirs('outputs')
 
    
    try:
        inputDAT = open('input.DAT')
    except IOError:
        raise  Exception ("Problem with IO; potentially input.DAT not created correctly")
    else: 
 
        try:
            
            rc = run_command(Init.sfgenpath+'/bin/SFGen', stdin=inputDAT)
 
        except OSError:
            raise Exception("SFGen executable or file not found")
 
        except Exception:
            raise Exception("Non-OSError or IOError in SFGen execution block")
 
    if rc:
        raise Exception('Unexpected error in sfgen execution in SFGenExecute')
 
    return
 
# SFGen running
 

SFGenInitialize()

def SFGenUtils.SFGenInitialize	(		Init,
			stdin = None
	)

Definition at line 179 of file SFGenUtils.py.

def SFGenInitialize(Init, stdin=None):
 
    logger.info("Starting SFGen Initialization")
 
    if not os.path.exists('evrecs'):
        os.makedirs('evrecs')
    if not os.path.exists('outputs'):
        os.makedirs('outputs')
 
 
    try:
        inputDAT = open('input.DAT')
 
    except IOError:
        raise Exception("Problem with file IO; potentially input.DAT not created correctly")
    else:
 
        try: 
            rc = run_command(Init.sfgenpath+"/bin/SFGen", inputDAT)
 
        except OSError:
            raise Exception("File not found")
 
        except Exception:
            raise Exception("Non-OSError or IOError in execution block")
 
    if rc:
        raise Exception('Unexpected error in sfgen execution in SFGenInitialize')
       
    return
 
# SFGen execution
 

SFGenRun()

def SFGenUtils.SFGenRun	(		Init,
			genSeq
	)

Definition at line 245 of file SFGenUtils.py.

def SFGenRun(Init, genSeq):
 
    # dump the job configuration for fortran code
    print(Init.toFortran())
 
    # attach config to genSequence for later using in showering
    genSeq.SFGenConfig = Init
    
    writeInputDAT(Init)
    SFGenInitialize(Init)
    SFGenExecute(Init)
    
    return

writeInputDAT()

def SFGenUtils.writeInputDAT

(

Init

)

Definition at line 151 of file SFGenUtils.py.

def writeInputDAT(Init):
    
    with open("input.DAT", "w") as outF:
        outF.write(Init.toFortran())
    
    return
 
 

Variable Documentation

logger

SFGenUtils.logger = Logging.logging.getLogger("SFGen_i")

Get handle to Athena logging.

Definition at line 9 of file SFGenUtils.py.