segmentation / graph_gui

  #!/usr/bin/env python3
  # -*- coding: utf-8 -*-
  """
  Created on Thu Jan 31 15:29:40 2019
  
  @author: pavel
  `"""
  
  from vispy import app

  import vispy

  
  import network_dep as nwt
  

  #from pyqtgraph.Qt import QtCore, QtGui, QtWidgets
  from PyQt5 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

  #vispy.use(app='egl')

  DEBUG = False
  

  #set the backend. for different versions of PyQt
  app.use_app(backend_name='PyQt5')
  

  app.set_interactive(True)
  

  #Define a top level application

  appMain = QtWidgets.QApplication([])
  #QtWidgets.QApplication.setAttribute(QtCore.Qt.AA_ShareOpenGLContexts, True)

  
  ##Define a toplevel Widget

  top = QtWidgets.QWidget()

  top.resize(900, 900)
  
  

  #fibers = FiberView()
  fibers = TubeWidget()
  fibers.canvas.create_native()
  #fibers = gl.GLViewWidget()
  
  #plt = hist.addPlot()
  

  layout = QtWidgets.QGridLayout()

  graph = GraphWidget()
  graph.canvas.create_native()
  
  graph.connect(fibers)

  fibers.connect(graph)

  
  #initialize the layout

  layout.addWidget(graph.canvas.native, 0, 0, 2, 3)
  layout.addWidget(fibers.canvas.native, 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_()
      app.set_interactive(enabled=True)
      app.run()