import java.io.*;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
/**
* 输入：holeç”±æ£ç¡®ç”案填充åŽçš„groum，以åŠtop10结果集åˆï¼Œä¸”æ£ç¡®ç”案为结果集åˆä¸çš„第一个元ç´
*/
public class CreateDataSet {
static HashMap pathCounter;
static Map API2Index;
/**
* 读å–api2indexçš„Map
* @param file
* @throws IOException
*/
private static void getPathCounter(File file) throws IOException {
pathCounter = new HashMap<>();
FileReader fileReader = new FileReader(file);
BufferedReader bufferedReader = new BufferedReader(fileReader);
String row;
String path; Integer num;
while ((row = bufferedReader.readLine()) != null) {
path = row.split(" ")[0];
num = new Integer(row.split(" ")[1]);
pathCounter.put(path, num);
}
bufferedReader.close();
fileReader.close();
}
/**
* 创建适用于SVM_Rankçš„æ•°æ®é›†
* 对å•ä¸ªGroum进行处ç†
* 在åštop-10结果替æ¢çš„时候，åšpath数目的查询
* @param groum holeç”±æ£ç¡®ç”案填充åŽçš„groum
* @param results top10结果集åˆ, results第一个为æ£ç¡®ç”案
*/
public static void create(Groum groum, List results, int qId, File file, PrintWriter pW, Map A2I) throws IOException {
getPathCounter(file);
API2Index = A2I;
Map nodeMap = groum.getNodeMap();
GroumNode groumNode = null;
List startList = null;
String rows = null;
for (String id : nodeMap.keySet()) {
groumNode = nodeMap.get(id);
if (groumNode.getOriginalApi().equals(results.get(0))) {
startList = new ArrayList<>();
startList.add(id);
List> outList = GetPath.getAllPath(groum, startList, 4); // æ¤å¤„所得到的path表示是使用Mapä¸çš„APIçš„indexæž„æˆ
rows = createRows(outList, results, groumNode.getApi(), qId);
writeFile(pW, rows);
}
}
}
/**
* å°†Stringå˜ä¸ºList，并得到填充holeçš„Path（aPath）以åŠé™¤åŽ»hole以外的剩余API组æˆçš„Path（bPath）
* @param list
* @param id
* @param result
* @return
*/
private static List convertListToStringAndGetAPathBPath(List list, String id, String result) {
List ret = new ArrayList();
StringBuilder aPath = new StringBuilder();
StringBuilder bPath = new StringBuilder();
boolean ifStarted = false;
boolean ifMeetHole = false;
for (String item : list) {
if (item.equals(id)) {
aPath.append(result);
aPath.append(",");
}
else {
aPath.append(item);
bPath.append(item);
aPath.append(",");
bPath.append(",");
}
}
if (aPath.charAt(aPath.length() - 1) == ',') aPath.deleteCharAt(aPath.length() - 1);
if (bPath.charAt(bPath.length() - 1) == ',') bPath.deleteCharAt(bPath.length() - 1);
ret.add(aPath.toString());
ret.add(bPath.toString());
return ret;
}
/**
* 构建形å¼ä¸º"3 qid:1 1:1 2:1 3:0 4:0.2 5:0"æ¯è¡Œæ•°æ®å…ƒ
* @param outList 使用hole抽å–çš„Path
* @param results top10结果集
* @param id æ£ç¡®ç”案在Mapä¸çš„Id
* @param qId æ•°æ®å…ƒä¸çš„qid
*/
private static String createRows(List> outList, List results, String id, int qId) {
StringBuilder rows = new StringBuilder();
List midRes = null;
double feature = 0;
int featureId = 1;
boolean ifFirst = true;
String resultId;
for (String result : results) {
if (ifFirst) {
rows.append("2"); // æ£ç¡®ç”案的rank值为2
ifFirst = false;
}
else {
rows.append("1"); // éžæ£ç¡®ç”案的rank值å‡ä¸º1
}
rows.append(" ");rows.append("qid:");rows.append(qId);rows.append(" ");
featureId = 1;
boolean ifFirstFeature = true;
for (List path : outList) {
resultId = API2Index.get(result);
midRes = convertListToStringAndGetAPathBPath(path, id, resultId);
if (midRes.get(1).length() == 1) continue;
System.out.println(midRes.get(1));
System.out.println(midRes.get(0));
feature = (double)pathCounter.getOrDefault(midRes.get(1), 0) / pathCounter.getOrDefault(midRes.get(0), 1); // æ¤å¤„对a值设定最å°ä¸º1
if (ifFirstFeature) ifFirstFeature = false;
else rows.append(" ");
rows.append(featureId);rows.append(":");rows.append(feature);
++featureId;
}
rows.append("\n");
}
return rows.toString();
}
/**
* 将一个Groumçš„æ•°æ®é›†å†™å…¥æ–‡ä»¶
* @param pW
* @param rows
*/
private static void writeFile(PrintWriter pW, String rows) {
pW.write(rows);
}
}