PR fixes

Author: alelievr

Assets/com.alelievr.NodeGraphProcessor/Editor/Logic/EdgeConnectorListener.cs

@@ -12,7 +12,7 @@ namespace GraphProcessor
     /// </summary>
     public class BaseEdgeConnectorListener : IEdgeConnectorListener
     {
-        protected readonly BaseGraphView graphView;
+        public readonly BaseGraphView graphView;
 
         Dictionary< Edge, PortView >    edgeInputPorts = new Dictionary< Edge, PortView >();
         Dictionary< Edge, PortView >    edgeOutputPorts = new Dictionary< Edge, PortView >();

Assets/com.alelievr.NodeGraphProcessor/Editor/Utils/BaseEdgeConnector.cs

@@ -13,16 +13,18 @@ public class BaseEdgeConnector : EdgeConnector
         Edge edgeCandidate;
         protected bool active;
         Vector2 mouseDownPosition;
+        protected BaseGraphView graphView;
 
         internal const float k_ConnectionDistanceTreshold = 10f;
 
 		public BaseEdgeConnector(IEdgeConnectorListener listener) : base()
 		{
+            graphView = (listener as BaseEdgeConnectorListener)?.graphView;
             active = false;
-            initEdgeConnector(listener);
+            InitEdgeConnector(listener);
         }
 
-        protected virtual void initEdgeConnector(IEdgeConnectorListener listener)
+        protected virtual void InitEdgeConnector(IEdgeConnectorListener listener)
         {
             dragHelper = new BaseEdgeDragHelper(listener);
             activators.Add(new ManipulatorActivationFilter { button = MouseButton.LeftMouse });
@@ -68,7 +70,7 @@ protected virtual void OnMouseDown(MouseDownEvent e)
 
             mouseDownPosition = e.localMousePosition;
 
-            edgeCandidate = CreateEdgeView();
+            edgeCandidate = graphView != null ? graphView.CreateEdgeView() : new EdgeView();
             edgeDragHelper.draggedPort = graphElement;
             edgeDragHelper.edgeCandidate = edgeCandidate;
 
@@ -86,11 +88,6 @@ protected virtual void OnMouseDown(MouseDownEvent e)
             }
         }
 
-        protected virtual EdgeView CreateEdgeView()
-        {
-            return new EdgeView();
-        }
-
         void OnCaptureOut(MouseCaptureOutEvent e)
         {
             active = false;

Assets/com.alelievr.NodeGraphProcessor/Editor/Views/BaseGraphView.cs

@@ -299,7 +299,7 @@ void UnserializeAndPasteCallback(string operationName, string serializedData)
 			}
 		}
 
-        protected virtual EdgeView CreateEdgeView()
+        public virtual EdgeView CreateEdgeView()
         {
 			return new EdgeView();
         }

Assets/com.alelievr.NodeGraphProcessor/Editor/Views/BaseNodeView.cs

@@ -318,10 +318,8 @@ public PortView AddPort(FieldInfo fieldInfo, Direction direction, BaseEdgeConnec
 			return p;
 		}
 
-        protected virtual PortView CreatePortView(Orientation horizontal, Direction direction, FieldInfo fieldInfo, PortData portData, BaseEdgeConnectorListener listener)
-        {
-			return PortView.CreatePV(horizontal, direction, fieldInfo, portData, listener);
-        }
+        protected virtual PortView CreatePortView(Direction direction, FieldInfo fieldInfo, PortData portData, BaseEdgeConnectorListener listener)
+        	=> PortView.CreatePortView(direction, fieldInfo, portData, listener);
 
         public void InsertPort(PortView portView, int index)
 		{

Assets/com.alelievr.NodeGraphProcessor/Editor/Views/PortView.cs

@@ -53,7 +53,7 @@ protected PortView(Direction direction, FieldInfo fieldInfo, PortData portData,
 			this.tooltip = portData.tooltip;
 		}
 
-		public static PortView CreatePV(Direction direction, FieldInfo fieldInfo, PortData portData, BaseEdgeConnectorListener edgeConnectorListener)
+		public static PortView CreatePortView(Direction direction, FieldInfo fieldInfo, PortData portData, BaseEdgeConnectorListener edgeConnectorListener)
 		{
 			var pv = new PortView(direction, fieldInfo, portData, edgeConnectorListener);
 			pv.m_EdgeConnector = new BaseEdgeConnector(edgeConnectorListener);