def main(args):
if args.out_dir:
os.makedirs(args.out_dir, exist_ok=True)
if not args.tikz:
import matplotlib
matplotlib.use('Agg')
to_stdout = (args.tikz or args.standoff) and not args.out_dir
t = args.passages
t = get_passages(t) if to_stdout else get_passages_with_progress_bar(
t, desc="Visualizing")
if args.sentences:
t = (sentence for passage in t
for sentence in split2sentences(passage))
for passage in t:
if args.tikz:
print_text(args, visualization.tikz(passage),
passage.ID + ".tikz.txt")
elif args.standoff:
print_text(args, visualization.standoff(passage),
passage.ID + ".ann")
else:
import matplotlib.pyplot as plt
width = len(passage.layer(layer0.LAYER_ID).all) * 19 / 27
plt.figure(passage.ID, figsize=(width, width * 10 / 19))
visualization.draw(passage, node_ids=args.node_ids)
if args.out_dir:
plt.savefig(
os.path.join(args.out_dir, passage.ID + "." + args.format))
plt.close()
else:
plt.show()
def visualize():
xml = request.get_data()
passage = from_standard(fromstring(xml))
print("Visualizing passage %s: %s" % (passage.ID, passage.layer(layer1.LAYER_ID).heads[0]))
canvas = FigureCanvasAgg(plt.figure())
draw(passage)
image = BytesIO()
canvas.print_png(image)
data = b64encode(image.getvalue()).decode()
return Response(quote(data.rstrip("\n")))
def visualize():
xml = request.get_data()
passage = from_standard(fromstring(xml))
print("Visualizing passage %s: %s" %
(passage.ID, passage.layer("1").top_node))
canvas = FigureCanvasAgg(plt.figure())
draw(passage)
image = BytesIO()
canvas.print_png(image)
data = b64encode(image.getvalue()).decode()
return Response(quote(data.rstrip("\n")))
def visualize():
xml = request.get_data()
passage = from_standard(fromstring(xml))
print("Visualizing passage %s: %s" %
(passage.ID, passage.layer(layer1.LAYER_ID).heads[0]))
fig = plt.figure()
canvas = FigureCanvasAgg(fig)
draw(passage)
image = BytesIO()
canvas.print_png(image)
data = b64encode(image.getvalue()).decode()
plt.close()
return Response(quote(data.rstrip("\n")))
help="image format")
args = argparser.parse_args()
if args.out_dir:
os.makedirs(args.out_dir, exist_ok=True)
if not args.tikz:
import matplotlib
matplotlib.use('Agg')
for passage in get_passages_with_progress_bar(args.passages,
desc="Visualizing"):
if args.tikz:
tikz = visualization.tikz(passage)
if args.out_dir:
with open(os.path.join(args.out_dir, passage.ID + ".tikz.txt"),
"w") as f:
print(tikz, file=f)
else:
with external_write_mode():
print(tikz)
else:
import matplotlib.pyplot as plt
width = len(passage.layer(layer0.LAYER_ID).all) * 19 / 27
plt.figure(figsize=(width, width * 10 / 19))
visualization.draw(passage, node_ids=args.node_ids)
if args.out_dir:
plt.savefig(
os.path.join(args.out_dir, passage.ID + "." + args.format))
plt.close()
else:
plt.show()
from argparse import ArgumentParser
import matplotlib.pyplot as plt
from ucca import visualization
from ucca.ioutil import read_files_and_dirs
if __name__ == "__main__":
argparser = ArgumentParser(
description="Visualize the given passages as graphs.")
argparser.add_argument(
"passages",
nargs="+",
help="UCCA passages, given as xml/pickle file names")
args = argparser.parse_args()
for passage in read_files_and_dirs(args.passages):
visualization.draw(passage)
plt.show()
def test_draw(create):
import matplotlib
matplotlib.use('Agg')
draw(create())