fixed bug when swc file only contains one edge

Jiaming Guo
1 parent 26e84a59
Showing 2 changed files with 31 additions and 4 deletions Show diff stats
stim/biomodels/network.h
stim/visualization/swc.h
@@ -170,6 +170,26 @@ public:
 		return V.size();
 	}
  
+	///Returns the radius at specific point in the edge
+	T get_R(unsigned e, unsigned i) {
+		return E[e].r(i);
+	}
+
+	///Returns the length of current edge
+	T get_L(unsigned e) {
+		return E[e].length();
+	}
+
+	///Returns the start vertex of current edge
+	size_t get_SP(unsigned e) {
+		return E[e].v[0];
+	}
+
+	///Returns the end vertex of current edge
+	size_t get_EP(unsigned e) {
+		return E[e].v[1];
+	}
+
 	//scale the network by some constant value
 	//	I don't think these work??????
 	/*std::vector<vertex> operator*(T s){
@@ -195,10 +195,10 @@ namespace stim {
  
 		// resample the tree-like SWC
 		void resample() {
-			unsigned n = node.size();
+			unsigned nn = node.size();
  
 			std::vector<int> joint_node;
-			for (unsigned i = 1; i < n; i++) {			// search all nodes(except the first one) to find joint nodes
+			for (unsigned i = 1; i < nn; i++) {			// search all nodes(except the first one) to find joint nodes
 				if (node[i].son_idx.size() > 1) {
 					joint_node.push_back(node[i].idx);	// store the original index
 				}
@@ -206,10 +206,10 @@ namespace stim {
  
 			std::vector<int>  new_edge;					// new edge in the network
  
-			n = joint_node.size();
+			unsigned n = joint_node.size();
  
 			for (unsigned i = 0; i < n; i++) {			// for every joint nodes
-				std::vector<int>  new_edge;									// new edge in the network
+
 				swc_tree::swc_node<T> cur_node = node[joint_node[i] - 1];	// store current node
  
 				// go up
@@ -236,6 +236,13 @@ namespace stim {
 					}
 				}
 			}
+
+			if (n == 0) {		// just one edge
+				new_edge.clear();
+				for (unsigned i = 0; i < nn; i++)
+					new_edge.push_back(node[i].idx);
+				E.push_back(new_edge);
+			}
 		}
  
 		// get points in one edge