Mercurial > hg > Members > nobuyasu > TPPageRank
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); } }