GuiVisPy_tube.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
Created on Thu Jan 31 15:29:40 2019
@author: pavel
`"""
from vispy import app
import network_dep as nwt
from pyqtgraph.Qt import QtCore, QtGui, QtWidgets
import pyqtgraph as pg
import numpy as np
import math
from mpl_toolkits.mplot3d import Axes3D
import matplotlib
import matplotlib.pyplot as plt
matplotlib.use('Qt5Agg')
from GraphWidget import GraphWidget
from TubeWidget import TubeWidget
DEBUG = False
#set the backend. for different versions of PyQt
app.use_app(backend_name='PyQt5')
#Define a top level application
appMain = QtGui.QApplication([])
##Define a toplevel Widget
top = QtGui.QWidget()
top.resize(900, 900)
#fibers = FiberView()
fibers = TubeWidget()
fibers.canvas.create_native()
#fibers = gl.GLViewWidget()
#plt = hist.addPlot()
layout = QtGui.QGridLayout()
graph = GraphWidget()
graph.canvas.create_native()
graph.connect(fibers)
fibers.connect(graph)
#initialize the layout
layout.addWidget(graph, 0, 0, 2, 3)
layout.addWidget(fibers, 0, 2, 2, 3)
layout.setColumnStretch(0, 2)
layout.setRowStretch(0, 2)
layout.setColumnStretch(2, 1)
layout.setRowStretch(0, 1)
top.setLayout(layout)
top.show()
#def draw_histogram(G, value):
#    vals = G.edge_properties[value].get_array()
#    y, x = np.histogram(vals,40)
#    hist.plot(x,y, stepMode=True, fillLevel=0, brush=(0,0,255,150))
def load_nwt(filepath):
    net = nwt.Network(filepath)
    G = net.createFullGraph_gt()
    G = net.filterDisconnected(G)
    #G = net.gen_spring_onion_layout(G, 'degree', 1.0, 1000, 0.01, 1.0)
    #G = net.filterFullGraph_gt(G, erode=True)
    #G = net.filterFullGraph_gt(G, erode=True)
    #G = net.gen_new_fd_layout(G)
    #G = net.filterBorder(G)
    #nwt.Network.saveGraph_gt(nwt, G, "FilteredNetwork_JACK_3.nwt")
    color = np.zeros(4, dtype = np.double)
    color = [0.0, 1.0, 0.0, 1.0]
    G.edge_properties["RGBA"] = G.new_edge_property("vector<double>", val=color)
    color = [1.0, 0.0, 0.0, 0.9]
    G.vertex_properties["RGBA"] = G.new_vertex_property("vector<double>", val=color)
    bbl, bbu = net.aabb()
    return G, bbl, bbu
G, bbl, bbu = load_nwt("/home/pavel/Documents/Python/GraphGuiQt/network_4.nwt")
#nwt.Network.write_vtk(G)
#G, bbl, bbu = load_nwt("/home/pavel/Documents/Python/GraphGuiQt/net`````````work_test_251_1kx3_short_NEW.nwt")
#ret = nwt.Network.get_affinity_matrix(G, "length")
node_image = QtGui.QImage("/home/pavel/Documents/Python/GraphGuiQt/node_tex.jpg")
node_tex = QtGui.QBitmap.fromImage(node_image)
if DEBUG:
    print(node_tex.depth())
center = (bbu-bbl)/2.0
#fibers.opts['distance'] = 5
#    item = NodeItem(G)
#    graph.addItem(item)
#draw_histogram(G, "length")
graph.canvas.set_data(G, bbl, bbu)
fibers.canvas.set_data(G, bbl, bbu, 16)
#fibers.draw_all(G, center, fibers, graph, node_tex)
graph.set_g_view(fibers)
## Start Qt event loop unless running in interactive mode.
if __name__ == '__main__':
    import sys
    if (sys.flags.interactive != 1) or not hasattr(QtCore, 'PYQT_VERSION'):
        QtGui.QApplication.instance().exec_()