Investigator force graph node legends now only sites appearing in the graph. This is supported by additions to GraphTagLib.

Author: eichmann

.gitattributes

@@ -38,6 +38,9 @@ src/main/java/edu/uiowa/slis/graphtaglib/NodeIterator.java -text
 src/main/java/edu/uiowa/slis/graphtaglib/NodeLabel.java -text
 src/main/java/edu/uiowa/slis/graphtaglib/NodeScore.java -text
 src/main/java/edu/uiowa/slis/graphtaglib/NodeUri.java -text
+src/main/java/edu/uiowa/slis/graphtaglib/SiteID.java -text
+src/main/java/edu/uiowa/slis/graphtaglib/SiteIDIterator.java -text
+src/main/java/edu/uiowa/slis/graphtaglib/SiteName.java -text
 src/main/java/edu/uiowa/slis/graphtaglib/filters/EdgeLookup.java -text
 src/main/java/edu/uiowa/slis/graphtaglib/filters/EdgePopulator.java -text
 src/main/java/edu/uiowa/slis/graphtaglib/filters/ImplicitEdgeLookup.java -text

src/main/java/edu/uiowa/slis/graphtaglib/SiteID.java

@@ -0,0 +1,29 @@
+package edu.uiowa.slis.graphtaglib;
+
+import java.io.IOException;
+
+import javax.servlet.jsp.JspException;
+import javax.servlet.jsp.JspTagException;
+import javax.servlet.jsp.tagext.TagSupport;
+
+public class SiteID extends TagSupport {
+    private static final long serialVersionUID = 1L;
+
+    public int doStartTag() throws JspException {
+	SiteIDIterator theIterator = (SiteIDIterator) findAncestorWithClass(this, SiteIDIterator.class);
+	try {
+	    pageContext.getOut().print(theIterator.currentID);
+	} catch (IOException e) {
+	    e.printStackTrace();
+	}
+	return SKIP_BODY;
+    }
+
+    public int doEndTag() throws JspTagException, JspException {
+	clearServiceState();
+	return super.doEndTag();
+    }
+
+    private void clearServiceState() {
+    }
+}

src/main/java/edu/uiowa/slis/graphtaglib/SiteIDIterator.java

@@ -0,0 +1,101 @@
+package edu.uiowa.slis.graphtaglib;
+
+import java.sql.Connection;
+import java.sql.PreparedStatement;
+import java.sql.ResultSet;
+import java.sql.SQLException;
+import java.util.Hashtable;
+import java.util.Iterator;
+import java.util.SortedSet;
+import java.util.TreeSet;
+
+import org.apache.commons.logging.Log;
+import org.apache.commons.logging.LogFactory;
+
+import javax.naming.InitialContext;
+import javax.naming.NamingException;
+import javax.servlet.jsp.JspException;
+import javax.servlet.jsp.JspTagException;
+import javax.servlet.jsp.tagext.BodyTagSupport;
+import javax.sql.DataSource;
+
+public class SiteIDIterator extends BodyTagSupport {
+    private static final long serialVersionUID = 1L;
+
+    private static final Log logger = LogFactory.getLog(SiteIDIterator.class);
+    static Hashtable<Integer, String> siteHash = null;
+
+    SortedSet<Integer> idSet = null;
+    Iterator<Integer> idIterator = null;
+    int currentID = 0;
+    String currentSite = null;
+
+    boolean pruneOrphans = false;
+    
+    synchronized void init() throws NamingException, SQLException {
+	if (siteHash != null)
+	    return;
+	siteHash = new Hashtable<Integer, String>();
+	DataSource theDataSource = (DataSource) new InitialContext().lookup("java:/comp/env/jdbc/VIVOTagLib");
+	Connection conn = theDataSource.getConnection();
+	PreparedStatement stmt = conn.prepareStatement("select id,site from vivo_aggregated.site order by id");
+	ResultSet rs = stmt.executeQuery();
+	while (rs.next()) {
+	    siteHash.put(rs.getInt(1), rs.getString(2));
+	}
+	stmt.close();
+	conn.close();
+    }
+
+    public int doStartTag() throws JspException {
+	logger.debug("doStartTag");
+	try {
+	    init();
+	} catch (Exception e) {
+	    logger.error("error initializing SiteIDIterator:", e);
+	}
+	idSet = new TreeSet<Integer>();
+	Graph theGraph = (Graph) findAncestorWithClass(this, Graph.class);
+	for (GraphNode node: theGraph.nodes) {
+	    idSet.add(node.getGroup("site"));
+	}
+	
+	idIterator = idSet.iterator();
+	
+	if (idIterator.hasNext()) {
+	    currentID = idIterator.next();
+	    currentSite = siteHash.get(currentID);
+	    pageContext.setAttribute("isLastID", !idIterator.hasNext());
+	    return EVAL_BODY_INCLUDE;
+	}
+
+	return SKIP_BODY;
+    }
+
+    public int doAfterBody() throws JspTagException {
+	if (idIterator.hasNext()) {
+	    currentID = idIterator.next();
+	    currentSite = siteHash.get(currentID);
+	    pageContext.setAttribute("isLastID", !idIterator.hasNext());
+	    return EVAL_BODY_AGAIN;
+	}
+
+	return SKIP_BODY;
+    }
+
+    public int doEndTag() throws JspTagException, JspException {
+	logger.debug("doEndTag");
+	clearServiceState();
+	return super.doEndTag();
+    }
+
+    private void clearServiceState() {
+	idSet = null;
+	idIterator = null;
+	currentID = 0;
+	currentSite = null;
+
+	pageContext.removeAttribute("isLastNode");
+    }
+
+}

src/main/java/edu/uiowa/slis/graphtaglib/SiteName.java

@@ -0,0 +1,29 @@
+package edu.uiowa.slis.graphtaglib;
+
+import java.io.IOException;
+
+import javax.servlet.jsp.JspException;
+import javax.servlet.jsp.JspTagException;
+import javax.servlet.jsp.tagext.TagSupport;
+
+public class SiteName extends TagSupport {
+    private static final long serialVersionUID = 1L;
+
+    public int doStartTag() throws JspException {
+	SiteIDIterator theIterator = (SiteIDIterator) findAncestorWithClass(this, SiteIDIterator.class);
+	try {
+	    pageContext.getOut().print(theIterator.currentSite);
+	} catch (IOException e) {
+	    e.printStackTrace();
+	}
+	return SKIP_BODY;
+    }
+
+    public int doEndTag() throws JspTagException, JspException {
+	clearServiceState();
+	return super.doEndTag();
+    }
+
+    private void clearServiceState() {
+    }
+}

src/main/resources/META-INF/taglib.tld

@@ -228,4 +228,22 @@
 			<rtexprvalue>true</rtexprvalue>
 		</attribute>
 	</tag>
-</taglib>
+
+	<tag>
+		<name>foreachSiteID</name>
+		<tag-class>edu.uiowa.slis.graphtaglib.SiteIDIterator</tag-class>
+		<body-content>JSP</body-content>
+	</tag>
+
+	<tag>
+		<name>SiteID</name>
+		<tag-class>edu.uiowa.slis.graphtaglib.SiteID</tag-class>
+		<body-content>JSP</body-content>
+	</tag>
+
+	<tag>
+		<name>SiteName</name>
+		<tag-class>edu.uiowa.slis.graphtaglib.SiteName</tag-class>
+		<body-content>JSP</body-content>
+	</tag>
+</taglib>