XMLUtil.java

//package javaToolkit.lib.utils;
//
//import java.io.File;
//import javax.xml.transform.*;
//import javax.xml.transform.dom.DOMSource;
//import javax.xml.transform.stream.StreamResult;
//import javax.xml.parsers.*;
//import org.w3c.dom.Attr;
//import org.w3c.dom.Document;
//import org.w3c.dom.Element;
//
//import smu.bowen.ct4j.datapreparation.Group;
//import smu.bowen.ct4j.datapreparation.Constants;
//
//public class XMLUtil {
//
//	public static Boolean writeProjectGroup(Group group, String xmlPath) {
//
//		try {
//			DocumentBuilderFactory documentFactory = DocumentBuilderFactory.newInstance();
//			DocumentBuilder documentBuilder = documentFactory.newDocumentBuilder();
//			Document document = documentBuilder.newDocument();
//
//			// root element
//			Element root = document.createElement("list");
//			document.appendChild(root);
//
//			// projectGroup element
//			Element projectGroup = document.createElement("projectGroup");
//			root.appendChild(projectGroup);
//
//			// set an attribute to staff element
//			Attr groupAttr = document.createAttribute("id");
//			groupAttr.setValue(String.valueOf(group.groupId));
//			projectGroup.setAttributeNode(groupAttr);
//
//			// bowen define the #train
//			System.out.println(group.methodList.size());
//			if (group.methodList.size() < 10) {
//				return null;
//			}
//			int trainNum;
//			if ((int) (group.methodList.size() * Constants.trainRatio) > 40) {
//				trainNum = 40;
//			} else {
//				trainNum = (int) (group.methodList.size() * Constants.trainRatio);
//			}
//
//			for (int i = 0; i < trainNum; i++) {
//
//				System.out.println(group.methodList.get(i).oldProjectName);
//
//				Method method = group.methodList.get(i);
//				method.id = i + 1;
//				// method element
//				Element projectMethod = document.createElement("projectMethod");
//				projectGroup.appendChild(projectMethod);
//
//				// set an attribute to staff element
//				Attr methodAttr = document.createAttribute("id");
//				methodAttr.setValue(String.valueOf(method.id));
//				projectMethod.setAttributeNode(methodAttr);
//
//				// src element
//				Element src = document.createElement("src");
//				src.appendChild(document.createTextNode("SOMETHING"));
//				projectMethod.appendChild(src);
//
//				// oldProjectName element
//				Element oldProjectName = document.createElement("oldProjectName");
//				oldProjectName.appendChild(document.createTextNode(method.oldProjectName));
//				projectMethod.appendChild(oldProjectName);
//
//				// newProjectName element
//				Element newProjectName = document.createElement("newProjectName");
//				newProjectName.appendChild(document.createTextNode(method.newProjectName));
//				projectMethod.appendChild(newProjectName);
//
//				// oldClassName element
//				Element oldClassName = document.createElement("oldClassName");
//				oldClassName.appendChild(document.createTextNode(method.oldClassName));
//				projectMethod.appendChild(oldClassName);
//
//				// newClassName element
//				Element newClassName = document.createElement("newClassName");
//				newClassName.appendChild(document.createTextNode(method.newClassName));
//				projectMethod.appendChild(newClassName);
//
//				// oldFilePath element
//				Element oldFilePath = document.createElement("oldFilePath");
//				oldFilePath.appendChild(document.createTextNode(method.oldFilePath));
//				projectMethod.appendChild(oldFilePath);
//
//				// newFilePath element
//				Element newFilePath = document.createElement("newFilePath");
//				newFilePath.appendChild(document.createTextNode(method.newFilePath));
//				projectMethod.appendChild(newFilePath);
//
//				// oldMethodName element
//				Element oldMethodName = document.createElement("oldMethodName");
//				oldMethodName.appendChild(document.createTextNode(method.oldMethodName));
//				projectMethod.appendChild(oldMethodName);
//
//				// newMethodName element
//				Element newMethodName = document.createElement("newMethodName");
//				newMethodName.appendChild(document.createTextNode(method.newMethodName));
//				projectMethod.appendChild(newMethodName);
//			}
//
//			// candidate element
//			int candidateNum = 1;
//			Element candidateProjects = document.createElement("candidateProjects");
//			projectGroup.appendChild(candidateProjects);
//
//			for (int i = trainNum; i < group.methodList.size(); i++) {
//
//				Method method = group.methodList.get(i);
//
//				// newMethodName element
//				Element candidateProject = document.createElement("candidateProject");
//
//				// set an attribute to staff element
//				Attr candidateProjectAttr = document.createAttribute("id");
//				candidateProjectAttr.setValue(String.valueOf(candidateNum));
//				candidateProject.setAttributeNode(candidateProjectAttr);
//
//				candidateNum++;
//
//				candidateProject.appendChild(document.createTextNode(method.oldProjectName));
//				candidateProjects.appendChild(candidateProject);
//			}
//
//			// write the content into xml file
//			TransformerFactory transformerFactory = TransformerFactory.newInstance();
//			Transformer transformer = transformerFactory.newTransformer();
//			transformer.setOutputProperty(OutputKeys.INDENT, "yes");
//			transformer.setOutputProperty("{http://xml.apache.org/xslt}indent-amount", "2");
//			DOMSource source = new DOMSource(document);
//
//			// Output to console for testing
//			StreamResult resultConsole = new StreamResult(System.out);
//			transformer.transform(source, resultConsole);
//
//			StreamResult resultFile = new StreamResult(new File(xmlPath));
//			transformer.transform(source, resultFile);
//			System.out.println("File saved!");
//
//		} catch (ParserConfigurationException pce) {
//			pce.printStackTrace();
//		} catch (TransformerException tfe) {
//			tfe.printStackTrace();
//		}
//
//		return true;
//	}
//
//}