import org.apache.poi.ss.usermodel.Row; //导入方法依赖的package包/类

/**

* generates the IDCG and nDCG

* @param prec_recall

* @param rowPR

* @param ratings

*/

private static void computeIDCG_nDCG(Sheet prec_recall, Row rowPR, ArrayList ratings) {

CellReference cellRefDCGFirst = new CellReference(1, COLUMN_DCG);

CellReference cellRefDCGLast = new CellReference(50, COLUMN_DCG);

Cell cellPR = rowPR.createCell(COLUMN_DCG);

cellPR.setCellFormula("SUM("+cellRefDCGFirst.formatAsString()

+":"+cellRefDCGLast.formatAsString()+")");

//Get the List of ratings and sort them by relevance (rating)

Collections.sort(ratings, new Comparator() {

@Override

public int compare(Rating o1, Rating o2) {

return Float.compare(o2.rating, o1.rating);

}

});

//go through the sheet again and add the ideal rank

int current_position = 1;

for (Iterator it = ratings.iterator(); it.hasNext();) {

int rank = it.next().getCompetenceRank();

Row rowPR2 = prec_recall.getRow(rank);

//CellReference cellRef = new CellReference(row.getRowNum(), cell.getColumnIndex());

cellPR = rowPR2.createCell(COLUMN_IDEAL_RANK, Cell.CELL_TYPE_NUMERIC);

cellPR.setCellValue(current_position);

CellReference cellRefNovice = new CellReference(rowPR2.getRowNum(), COLUMN_NOVICE);

CellReference cellRefIrrelevant = new CellReference(rowPR2.getRowNum(), COLUMN_IRRELEVANT);

CellReference cellRefNoviceWeight = new CellReference(0, COLUMN_NOVICE);

CellReference cellRefIrrelevantWeight = new CellReference(0, COLUMN_IRRELEVANT);

CellReference cellRefIdealRank = new CellReference(rowPR2.getRowNum(), COLUMN_IDEAL_RANK);

cellPR = rowPR2.createCell(COLUMN_IDCG, Cell.CELL_TYPE_NUMERIC);

//LOG 1 is not defined

if(current_position==1)

cellPR.setCellFormula("SUMIF("+cellRefNovice.formatAsString()+":"+cellRefIrrelevant.formatAsString()+",1,"+cellRefNoviceWeight.formatAsString()+":"+cellRefIrrelevantWeight.formatAsString()+")");

else

cellPR.setCellFormula("SUMIF("+cellRefNovice.formatAsString()+":"+cellRefIrrelevant.formatAsString()+",1,"+cellRefNoviceWeight.formatAsString()+":"+cellRefIrrelevantWeight.formatAsString()+")/LOG("+cellRefIdealRank.formatAsString()+",2)");

current_position++;

}

//sum of iDCG

cellPR = rowPR.createCell(COLUMN_IDCG, Cell.CELL_TYPE_NUMERIC);

CellReference cellRefIDCGFirst = new CellReference(1, COLUMN_IDCG);

CellReference cellRefIDCGLast = new CellReference(50, COLUMN_IDCG);

cellPR.setCellFormula("SUM("+cellRefIDCGFirst.formatAsString()+":"+cellRefIDCGLast.formatAsString()+")");

//nDCG

CellReference cellRefDCG = new CellReference(rowPR.getRowNum(), COLUMN_DCG);

CellReference cellRefIDCG = new CellReference(rowPR.getRowNum(), COLUMN_IDCG);

cellPR = rowPR.createCell(COLUMN_nDCG, Cell.CELL_TYPE_NUMERIC);

cellPR.setCellFormula("SUM("+cellRefDCG.formatAsString()+"/"+cellRefIDCG.formatAsString()+")");

}