diff src/pagerank/LinkConvertGraph.java @ 5:140272228818 draft

fix LinkToVertex.java
author one
date Wed, 05 Sep 2012 15:38:57 +0900
parents dcd59917a2dd
children c7b139ff27e2
line wrap: on
line diff
--- a/src/pagerank/LinkConvertGraph.java	Wed Sep 05 12:44:08 2012 +0900
+++ b/src/pagerank/LinkConvertGraph.java	Wed Sep 05 15:38:57 2012 +0900
@@ -90,8 +90,8 @@
 	
 	
 	public static void main(String[] args) {
-//		final String filename = "./resources/article.xml";
-		final String filename = "/Users/aotokage/testProgram/wiki/ja-pages_current.xml";
+		final String filename = "./resources/article.xml";
+//		final String filename = "/Users/aotokage/testProgram/wiki/ja-pages_current.xml";
 		
 		LinkConvertGraph lcg;
 
@@ -112,29 +112,32 @@
 			
 			Graph graph = new TinkerGraph();
 			FileOutputStream out = new FileOutputStream(new File(fileDB));
-			LinkToVertex ltn = new LinkToVertex(graph);
-			
+			LinkToVertex ltv = new LinkToVertex(graph);
+
+			long countId = 1;
 			for (Map.Entry<String, HashSet<String>> map : hash.entrySet()) {
 				String pageTitle = map.getKey();
 				
 				Vertex v;// = graph.addVertex(null);
 				
-				if ( ltn.getId(pageTitle) == null ) {
-					v = ltn.createVertexWithPageTitle(pageTitle);
-					
+				if ( ltv.getId(pageTitle) == null ) {
+					v = ltv.createVertexWithPageTitle(pageTitle,countId);
+					ltv.setPageRank(v, (Double)0.0);
+					countId++;
 				} else {
-					v = ltn.getVertex(pageTitle);
+					v = ltv.getVertex(pageTitle);
 				}
 
 				for (String linkPageTitle : map.getValue()) {
 					Vertex linkV;
-					if ( ltn.getId(linkPageTitle) == null) {
-						linkV = ltn.createVertexWithPageTitle(linkPageTitle);
-						ltn.setPageRank(linkV, (Double)0.0);
+					if ( ltv.getId(linkPageTitle) == null) {
+						linkV = ltv.createVertexWithPageTitle(linkPageTitle,countId);
+						countId++;
+						ltv.setPageRank(linkV, (Double)0.0);
 					} else {
-						linkV = ltn.getVertex(linkPageTitle);
+						linkV = ltv.getVertex(linkPageTitle);
 					}
-					ltn.setHasLink(v, linkV);
+					ltv.setHasLink(v, linkV);
 				}
 				
 			}