# Reports System
from openmdao.utils.reports_system import register_report, get_reports_dir, list_reports, \
clear_reports
import os
import builtins
from openmdao.utils.general_utils import wing_dbg, env_truthy
from openmdao.utils.array_utils import shape_to_len
wing_dbg()
# set up tracing or memory profiling if env vars are set.
if env_truthy('OPENMDAO_TRACE'):
from openmdao.devtools.itrace import setup, start
setup(os.environ['OPENMDAO_TRACE'])
start()
elif env_truthy('OPENMDAO_PROF_MEM'):
from openmdao.devtools.iprof_mem import setup, start
setup(os.environ['OPENMDAO_PROF_MEM'])
start()
if env_truthy('FLUSH_PRINT'):
_oldprint = builtins.print
def _flushprint(*args, **kwargs):
kwargs['flush'] = True
_oldprint(*args, **kwargs)
builtins.print = _flushprint
from openmdao.jacobians.assembled_jacobian import AssembledJacobian, \
DenseJacobian, COOJacobian, CSRJacobian, CSCJacobian
# Drivers
try:
from openmdao.drivers.pyoptsparse_driver import pyOptSparseDriver
except ImportError:
pass
from openmdao.drivers.scipy_optimizer import ScipyOptimizer, ScipyOptimizeDriver
from openmdao.drivers.genetic_algorithm_driver import SimpleGADriver
# System-Building Tools
from openmdao.utils.options_dictionary import OptionsDictionary
# Recorders
from openmdao.recorders.sqlite_recorder import SqliteRecorder
from openmdao.recorders.web_recorder import WebRecorder
from openmdao.recorders.upload_data import upload
# set up tracing or memory profiling if env vars are set.
import os
if os.environ.get('OPENMDAO_TRACE'):
from openmdao.devtools.itrace import setup, start
ret = bool(os.environ.get('OPENMDAO_TRACE_RETURN'))
setup(os.environ['OPENMDAO_TRACE'], show_return=ret)
start()
elif os.environ.get('OPENMDAO_PROF_MEM'):
from openmdao.devtools.iprof_mem import setup, start
setup(os.environ['OPENMDAO_PROF_MEM'])
start()
