def list_strategies(showdocs, output=sys.stdout):
def data_docs(name):
s = STRATEGIES[name]
sdoc = s.__doc__ or '<not documented>'
ev = s.SOLUTION_EVALUATOR
evdoc = ev.__doc__ or '<not documented>'
return (name, sdoc, evdoc)
def data_short(name):
return (name, )
if showdocs:
fmt = "{}: {}\n Tree evaluator: {}"
data = data_docs
else:
fmt = "{}"
data = data_short
for name in sorted(STRATEGIES.keys()):
print >>output, fmt.format(*data(name))
def main():
p = parser()
args = p.parse_args()
if args.list_strategies or args.list_strategies_long:
list_strategies(args.list_strategies_long)
sys.exit(0)
if not args.strategy:
p.print_help(sys.stderr)
print >>sys.stderr, "Strategy name not specified."
sys.exit(1)
s = STRATEGIES.get(args.strategy)
if not s:
p.print_help(sys.stderr)
print >>sys.stderr, "Unknown strategy name: '{}', available names are:".format(args.strategy)
list_strategies(False, sys.stderr)
sys.exit(1)
if args.maxdepth <= 0:
print >>sys.stderr, "Tree height constraint must be a positive integer."
sys.exit(1)
if args.root is not None and args.root not in CODETABLE.ALL_SET:
print >>sys.stderr, "Initial guess must be in the range [{}, {}]; input: {}".format(
0, CODETABLE.NCODES-1, args.root)
sys.exit(1)
initialize()
t = s.build_tree(CODETABLE.ALL, args.maxdepth, args.root, progress=args.progress)
if args.output:
t.to_json_file(args.output)
else:
print t.as_json_string()
