plotter.py

#! /usr/bin/env python3
import svgwrite as svg
import xml.etree.ElementTree as et
import os, configparser, argparse, sys, platform, traceback
from PIL import Image

settings = {}

def cleanQuotes(path):
    if (path.startswith('\'') and path.endswith('\'')) or (path.startswith('\"') and path.endswith('\"')):
       return path[1:-1]
    else:
        return path

def parsearg():
    parser = argparse.ArgumentParser(description='Plot images in image-label networks generated by Memespector script')
    parser.add_argument('--configfile',
                        metavar="CONFIG_PATH",
                        nargs=1,
                        default='config.txt',
                        help='Receives string for path to config file. Use this in case you wish to override the default config file.'
                        )
    return parser.parse_args()

def parseconfigfile(path="config.txt"):
    global settings
    configfile = configparser.ConfigParser()
    try:
        configfile.read(path)
    except Exception:
        raise

    try:
        settings['input']       = cleanQuotes(configfile.get('Input', 'InputGraph'))
        settings['inimgdir']    = cleanQuotes(configfile.get('Input', 'InputImageFolder'))

        settings['copyresized'] = configfile.getboolean('Output', 'CopyImagesResized')
        settings['outimgdir']   = configfile.get('Output', 'ResizedImageFolderName')
        settings['resizew']     = configfile.getint('Output', 'ResizeMaxWidth')
        settings['resizeh']     = configfile.getint('Output', 'ResizeMaxHeight')
        settings['dispw']       = configfile.getint('Output', 'ImageMaxDispWidth')
        settings['disph']       = configfile.getint('Output', 'ImageMaxDispHeight')
        settings['restrpage']   = configfile.getboolean('Output', 'RestricttoPage')
        settings['outw']        = configfile.getint('Output', 'OutputWidth')
        settings['outh']        = configfile.getint('Output', 'OutputHeight')
    except Exception as exc:
        print(exc)
        sys.exit("\n**ERROR**\nCould not parse at least one of the settings from the config file. Please verify its contents carefully.")

def loadSettings():
    args = parsearg()
    parseconfigfile(args.configfile)

def main():
    try:

        print("\n-------------------------\nImage Network Plotter\n-------------------------")
        loadSettings()

        # ------------------------------------------
        # Set internal variables
        #-------------------------------------------

        outputfilename = os.path.join(os.path.dirname(settings['input']), "visual_" + os.path.basename(settings['input']).split(".")[0] + ".svg")
        imgresizedim = settings['resizew'], settings['resizeh']
        imgdrawdim = settings['dispw'], settings['disph']

        print("Input file:", settings['input'])
        ingexf = et.parse(settings['input'])

        # ------------------------------------------
        # Create output dir
        #-------------------------------------------

        if settings['copyresized']:
            if os.path.isabs(settings['outimgdir']):
                outimgdir = settings['outimgdir']
            else:
                outimgdir = os.path.join(os.path.dirname(settings['input']), settings['outimgdir'])
            if not os.path.exists(outimgdir):
                os.makedirs(outimgdir)

        # ------------------------------------------
        # Parse GEXF and generate SVG
        #-------------------------------------------

        try:
            inroot = ingexf.getroot()
            ns = {'gexf' : "http://www.gexf.net/1.3" }
            viz = {'viz' : "http://www.gexf.net/1.3/viz"}
        except Exception as exc:
            print(exc)
            print("**ERROR**\nCould not parse GEXF.")

        typeAttId = -1
        linkAttId = -1
        fileAttId = -1

        graph = inroot.find("gexf:graph", ns)
        if not graph:
            sys.exit("\n**ERROR**\nCould not parse graph file.\n")

        attributes = graph.find(".gexf:attributes",ns)

        for att in attributes:
            if att.get('title') == 'type':
                typeAttId = att.get('id')
            elif att.get('title') == 'link':
                linkAttId = att.get('id')
            elif att.get('title') == 'file':
                fileAttId = att.get('id')

        nodes = graph.find("gexf:nodes", ns)

        # Find graph bounding box and count images
        numnodes = 0
        numimages = 0

        minx = 0
        maxx = 0
        miny = 0
        maxy = 0

        for node in nodes:
            numnodes += 1
            typeAtt = node.find("gexf:attvalues/gexf:attvalue[@for=\'" + str(typeAttId) +"\']",ns)
            if typeAtt.get('value') == "image":
                numimages += 1
                try:
                    inimgx = float(node.find("viz:position", viz).get('x'))
                    inimgy = float(node.find("viz:position", viz).get('y'))
                except AttributeError:
                    sys.exit("\n\n**ERROR**\nGraph has not been spatialized. Could not find position data for the nodes\nOpen it in Gephi, apply spatialization algorithm and export to another file.\n")
                except Exception:
                    raise
                if inimgx < minx:
                    minx = inimgx
                if inimgx > maxx:
                    maxx = inimgx
                if inimgy < miny:
                    miny = inimgy
                if inimgy > maxy:
                    maxy = inimgy

        print("Graph contains", numnodes, "nodes.")
        print("Plotting", numimages, "images.\n")
        print("Minimum X:", minx)
        print("Maximum X:", maxx)
        print("Minimum Y:", miny)
        print("Maximum Y:", maxy)

        # --------
        # Configure output conversion

        inw = maxx - minx
        inh = maxy - miny

        if (inw/inh) >= (settings['outw']/settings['outh']):
            # match width
            outfactor = settings['outw'] / inw
        else:
            # match height
            outfactor = settings['outh'] / inh

        outw = inw * outfactor
        outh = inh * outfactor
        outx = (settings['outw'] - outw)/2
        outy = (settings['outh'] - outh)/2

        # --------
        # Draw output

        outsvg = svg.Drawing(outputfilename, (settings['outw'], settings['outh']), debug=True)
        curimg = 1

        for node in nodes:
            typeAtt = node.find("gexf:attvalues/gexf:attvalue[@for=\'" + str(typeAttId) +"\']",ns)
            if typeAtt.get('value') == "image":
                print("\n\nDrawing image", curimg, "in", numimages)
                curimg += 1

                innodex = (float(node.find("viz:position", viz).get('x'))-minx)/inw
                innodey = (float(node.find("viz:position", viz).get('y'))-miny)/inh

                if settings['restrpage']:
                    outnodex = (innodex * outw) + outx
                    outnodey = (innodey * outh) + outy
                else:
                    outnodex = innodex
                    outnodey = innodey

                nodeid = node.get('id')
                imgfile = node.find("gexf:attvalues/gexf:attvalue[@for=\'" + str(fileAttId) +"\']",ns).get('value')
                linkUrl = node.find("gexf:attvalues/gexf:attvalue[@for=\'" + str(linkAttId) +"\']",ns).get('value')
                print("\tImage file:", imgfile)
                infile = os.path.join(settings['inimgdir'], imgfile)


                try:
                    curimage = Image.open(infile)
                except Exception:
                    print("\t**ATTENTION** Image could not be loaded.\n")
                    continue

                if settings['copyresized']:
                    imgfp = os.path.join(outimgdir, imgfile)
                    print("\tResizing image...")
                    try:
                        curimage.thumbnail(imgresizedim, Image.ANTIALIAS)
                        curimage.save(imgfp)
                    except Exception as exc:
                        print("\t**ATTENTION** Problem resizing image.\n")
                        print(exc)
                        continue
                else:
                    imgfp = infile

                print("\tPlotting image...\n")

                link = outsvg.add(outsvg.a(linkUrl,id=nodeid))
                image = link.add(outsvg.image(imgfp, insert=(outnodex, outnodey), size=imgdrawdim))

        outsvg.save(pretty=True)
    except KeyboardInterrupt:
        if outsvg:
            outsvg.save(pretty=True)
        print("\n\n**Script interrupted by user**\n\n")
    except Exception:
        traceback.print_exc(file=sys.stdout)
    sys.exit(0)

if __name__ == "__main__":
    main()