405 lines
15 KiB
Java
405 lines
15 KiB
Java
package Visual_DVM_2021.Passes.All;
|
||
import Common.Current;
|
||
import Common.Global;
|
||
import Common.UI.UI;
|
||
import Common.Utils.Files.VDirectoryChooser;
|
||
import Common.Utils.Index;
|
||
import Common.Utils.Utils;
|
||
import GlobalData.Tasks.TaskState;
|
||
import TestingSystem.DVM.DVMPackage.DVMPackage;
|
||
import TestingSystem.DVM.DVMTasks.DVMRunTask;
|
||
import TestingSystem.DVM.TasksPackage.TasksPackageState;
|
||
import Visual_DVM_2021.Passes.Pass_2021;
|
||
import org.apache.commons.io.FileUtils;
|
||
import org.apache.poi.hssf.usermodel.HSSFWorkbook;
|
||
import org.apache.poi.hssf.util.HSSFColor;
|
||
import org.apache.poi.ss.usermodel.Font;
|
||
import org.apache.poi.ss.usermodel.*;
|
||
import org.apache.poi.ss.util.CellRangeAddress;
|
||
|
||
import java.awt.*;
|
||
import java.io.File;
|
||
import java.io.FileOutputStream;
|
||
import java.util.LinkedHashMap;
|
||
import java.util.Vector;
|
||
public class ExportTasksPackageToExcel extends Pass_2021<Vector<DVMPackage>> {
|
||
//https://tproger.ru/translations/how-to-read-write-excel-file-java-poi-example
|
||
File dir;
|
||
File res;
|
||
VDirectoryChooser directoryChooser = new VDirectoryChooser("Выбор папки для сохранения таблицы Excel");
|
||
//---
|
||
final int max_columns = 16;
|
||
Vector<CellStyle> styles;
|
||
//--
|
||
Workbook book = null;
|
||
Sheet sheet = null;
|
||
//--
|
||
int total_tasks_count = 0;
|
||
long total_time = 0;
|
||
//--
|
||
LinkedHashMap<Integer, Vector<DVMRunTask>> packages_tasks = null;
|
||
//--
|
||
@Override
|
||
public String getIconPath() {
|
||
return "/icons/Excel.png";
|
||
}
|
||
@Override
|
||
public String getButtonText() {
|
||
return "";
|
||
}
|
||
@Override
|
||
protected boolean needsAnimation() {
|
||
return true;
|
||
}
|
||
@Override
|
||
protected boolean canStart(Object... args) throws Exception {
|
||
book = null;
|
||
sheet = null;
|
||
styles = null;
|
||
packages_tasks = new LinkedHashMap<>();
|
||
target = null;
|
||
//--
|
||
if (Global.testingServer.db.dvmPackages.getCheckedCount() > 0) {
|
||
target = Global.testingServer.db.dvmPackages.getCheckedItems();
|
||
} else {
|
||
if (Current.Check(Log, Current.DVMPackage)) {
|
||
target = new Vector<>();
|
||
target.add(Current.getDVMPackage());
|
||
} else return false;
|
||
}
|
||
for (DVMPackage dvmPackage : target) {
|
||
if (!dvmPackage.state.equals(TasksPackageState.Done)) {
|
||
Log.Writeln_("Пакет " + dvmPackage.id + " не завершен.");
|
||
} else {
|
||
Vector<DVMRunTask> tasks = new Vector<>();
|
||
/* todo Загрузить и распаковать все выделенные пакеты
|
||
for (DVMRunTask task : Global.testingServer.db.testRunTasks.Data.values()) {
|
||
if (task.taskspackage_id == dvmPackage.id)
|
||
tasks.add(task);
|
||
}
|
||
*/
|
||
packages_tasks.put(dvmPackage.id, tasks);
|
||
}
|
||
}
|
||
if (!Log.isEmpty())
|
||
return false;
|
||
dir = directoryChooser.ShowDialog();
|
||
if (dir == null) {
|
||
Log.Writeln_("Папка не выбрана.");
|
||
return false;
|
||
}
|
||
res = new File(dir, Utils.getDateName("packages") + ".xls");
|
||
return true;
|
||
}
|
||
//---
|
||
String getPackageInfo(DVMPackage object) {
|
||
Vector<String> lines = new Vector<>();
|
||
lines.add("задач " + object.tasksCount);
|
||
lines.add("машина " + object.machine_address + ":" + object.machine_port);
|
||
lines.add("ядер " + object.kernels);
|
||
lines.add("DVM-система " + object.version);
|
||
//--
|
||
total_tasks_count += object.tasksCount;
|
||
//--
|
||
long milliseconds = object.ChangeDate - object.StartDate;
|
||
total_time += milliseconds;
|
||
//--
|
||
lines.add("время выполнения " + Utils.printSplittedDateInterval(milliseconds));
|
||
return String.join("; ", lines);
|
||
}
|
||
//---
|
||
protected void addInfoRow(int row_num, String value) {
|
||
Row row = sheet.createRow(row_num);
|
||
String cell_text = value.toString();
|
||
Cell cell = row.createCell(0);
|
||
cell.setCellValue(cell_text);
|
||
cell.setCellStyle(styles.get(1));
|
||
//https://translated.turbopages.org/proxy_u/en-ru.ru.f85b6508-657377d3-df23bc2d-74722d776562/https/stackoverflow.com/questions/23533381/creating-the-border-for-the-merged-cells-in-poi
|
||
sheet.addMergedRegion(new CellRangeAddress(row_num, row_num, 0, max_columns - 1));
|
||
for (int i = 1; i < max_columns; ++i) {
|
||
Cell blankCell = row.createCell(i);
|
||
blankCell.setCellStyle(styles.get(1));
|
||
}
|
||
}
|
||
protected void addHeaderRow(int row_num, Object... values) {
|
||
Row row = sheet.createRow(row_num);
|
||
int i = 0;
|
||
for (Object value : values) {
|
||
String cell_text = value.toString();
|
||
Cell cell = row.createCell(i);
|
||
cell.setCellStyle(styles.get(0));
|
||
cell.setCellValue(cell_text);
|
||
//--
|
||
++i;
|
||
}
|
||
}
|
||
protected void addTaskRow(int row_num, Object... values) {
|
||
Row row = sheet.createRow(row_num);
|
||
int i = 0;
|
||
for (Object value : values) {
|
||
// String cell_text = value.toString();
|
||
Cell cell = row.createCell(i);
|
||
int style_index = 1;
|
||
switch (i) {
|
||
/*
|
||
addTaskRow(offset.getValue(),
|
||
task.group_description, //0
|
||
task.test_description,//1
|
||
task.language.getDescription(), //2
|
||
task.flags, //3
|
||
task.compilation_state, //4
|
||
task.state, //5
|
||
num_threads, //6
|
||
num_cudas, //7
|
||
total_threads, //8
|
||
formatMatrix(task.matrix), //9
|
||
task.environments, //10
|
||
task.usr_par, //11
|
||
task.compilation_time, //12
|
||
task.Time,//13
|
||
task.CleanTime, //14
|
||
task.progress //15
|
||
);
|
||
*/
|
||
case 4:
|
||
case 5:
|
||
TaskState state = (TaskState) value;
|
||
switch (state) {
|
||
case FailedToQueue:
|
||
case NoSuchTask:
|
||
case AbortedByUser:
|
||
case AbortedByTimeout:
|
||
case DoneWithErrors:
|
||
case WrongTestFormat:
|
||
case InternalError:
|
||
case Canceled:
|
||
style_index = 3;
|
||
break;
|
||
case Queued:
|
||
case Running:
|
||
style_index = 5;
|
||
break;
|
||
case Done:
|
||
style_index = 2;
|
||
break;
|
||
case Crushed:
|
||
style_index = 4;
|
||
break;
|
||
case Finished:
|
||
style_index = 6;
|
||
break;
|
||
}
|
||
cell.setCellValue(state.getDescription());
|
||
break;
|
||
case 6:
|
||
case 7:
|
||
case 8:
|
||
style_index = 7;
|
||
cell.setCellValue((int) value);
|
||
break;
|
||
case 9:
|
||
style_index = 7;
|
||
cell.setCellValue(value.toString());
|
||
break;
|
||
case 15:
|
||
cell.setCellValue((int) value);
|
||
break;
|
||
case 12:
|
||
case 13:
|
||
case 14:
|
||
cell.setCellValue((double) value);
|
||
break;
|
||
default:
|
||
cell.setCellValue(value.toString());
|
||
break;
|
||
}
|
||
cell.setCellStyle(styles.get(style_index));
|
||
//--
|
||
++i;
|
||
}
|
||
}
|
||
protected void setBorder(CellStyle style, short border) {
|
||
style.setBorderBottom(border);
|
||
style.setBorderLeft(border);
|
||
style.setBorderRight(border);
|
||
style.setBorderTop(border);
|
||
}
|
||
protected String extractEnvironmentValue(String environmentsSet, String name) {
|
||
if (environmentsSet.contains(name)) {
|
||
int index = environmentsSet.lastIndexOf(name);
|
||
if (index >= 0) {
|
||
String s1 = environmentsSet.substring(index + name.length() + 1);
|
||
String[] data = s1.split("\"");
|
||
if (data.length > 0) {
|
||
return data[1];
|
||
}
|
||
}
|
||
}
|
||
return null;
|
||
}
|
||
protected Integer extractIntegerEnvironmentValue(String environmentsSet, String name) {
|
||
Integer res = null;
|
||
String s = extractEnvironmentValue(environmentsSet, name);
|
||
if (s != null) {
|
||
try {
|
||
res = Integer.parseInt(s);
|
||
} catch (NumberFormatException ex) {
|
||
ex.printStackTrace();
|
||
}
|
||
}
|
||
return res;
|
||
}
|
||
String formatMatrix(String matrix) {
|
||
return Utils.DQuotes(matrix.trim());
|
||
}
|
||
void createStyles() {
|
||
styles = new Vector<>();
|
||
//0 - заголовок
|
||
//1 - черный
|
||
//2 - зеленый
|
||
//3 - красный
|
||
//4 - темно красный
|
||
//5 - оранжевый
|
||
//6 - серый
|
||
//7 - черный выровненный по центру.
|
||
for (int i = 0; i < 8; ++i) {
|
||
CellStyle style = book.createCellStyle();
|
||
Font font = book.createFont();
|
||
switch (i) {
|
||
case 0:
|
||
font.setBold(true);
|
||
setBorder(style, CellStyle.BORDER_THICK);
|
||
break;
|
||
default:
|
||
setBorder(style, CellStyle.BORDER_THIN);
|
||
switch (i) {
|
||
case 1:
|
||
font.setColor(HSSFColor.BLACK.index);
|
||
break;
|
||
case 2:
|
||
font.setColor(HSSFColor.GREEN.index);
|
||
break;
|
||
case 3:
|
||
font.setColor(HSSFColor.RED.index);
|
||
break;
|
||
case 4:
|
||
font.setColor(HSSFColor.DARK_RED.index);
|
||
break;
|
||
case 5:
|
||
font.setColor(HSSFColor.ORANGE.index);
|
||
break;
|
||
case 6:
|
||
font.setColor(HSSFColor.GREY_40_PERCENT.index);
|
||
break;
|
||
case 7:
|
||
font.setColor(HSSFColor.BLACK.index);
|
||
style.setAlignment(CellStyle.ALIGN_CENTER);
|
||
break;
|
||
}
|
||
break;
|
||
}
|
||
style.setFont(font);
|
||
styles.add(style);
|
||
}
|
||
}
|
||
void exportPackageTasks(DVMPackage dvmPackage, Index offset) {
|
||
Vector<DVMRunTask> tasks = packages_tasks.get(dvmPackage.id);
|
||
for (DVMRunTask task : tasks) {
|
||
ShowMessage2(task.test_description);
|
||
//---
|
||
Integer NUM_THREADS = extractIntegerEnvironmentValue(task.environments, "DVMH_NUM_THREADS");
|
||
if (NUM_THREADS != null && NUM_THREADS == 0) NUM_THREADS = 1;
|
||
Integer NUM_CUDAS = extractIntegerEnvironmentValue(task.environments, "DVMH_NUM_CUDAS");
|
||
//--
|
||
int num_threads = (NUM_THREADS != null) ? NUM_THREADS : 1;
|
||
int num_cudas = (NUM_CUDAS != null) ? NUM_CUDAS : 0;
|
||
int total_threads = num_threads * Utils.getMatrixProcessors(task.matrix);
|
||
//--
|
||
addTaskRow(offset.getValue(),
|
||
task.group_description, //0
|
||
task.test_description,//1
|
||
task.language.getDescription(), //2
|
||
task.flags, //3
|
||
task.compilation_state, //4
|
||
task.state, //5
|
||
num_threads, //6
|
||
num_cudas, //7
|
||
total_threads, //8
|
||
formatMatrix(task.matrix), //9
|
||
task.environments, //10
|
||
task.usr_par, //11
|
||
task.compilation_time, //12
|
||
task.Time,//13
|
||
task.CleanTime, //14
|
||
task.progress //15
|
||
);
|
||
offset.Inc();
|
||
}
|
||
}
|
||
@Override
|
||
protected void body() throws Exception {
|
||
File file = Utils.getTempFileName("table");
|
||
//--
|
||
book = new HSSFWorkbook();
|
||
sheet = book.createSheet("Результаты тестирования");
|
||
// format = book.createDataFormat();
|
||
//--
|
||
createStyles();
|
||
//--
|
||
Index offset = new Index();
|
||
for (DVMPackage dvmPackage : target) {
|
||
addInfoRow(offset.getValue(), getPackageInfo(dvmPackage));
|
||
offset.Inc();
|
||
}
|
||
if (target.size() > 1) {
|
||
addInfoRow(offset.getValue(), "всего задач " + total_tasks_count + "; общее время выполнения " + Utils.printSplittedDateInterval(total_time));
|
||
offset.Inc();
|
||
}
|
||
addHeaderRow(offset.getValue(),
|
||
"Группа",
|
||
"Тест",
|
||
"Язык",
|
||
"Флаги",
|
||
"Компиляция",
|
||
"Запуск",
|
||
"NUM_THREADS",
|
||
"NUM_CUDAS",
|
||
"TOTAL_THREADS",
|
||
"Матрица",
|
||
"Окружение",
|
||
"usr.par",
|
||
"Время компиляции (с)",
|
||
"Время запуска (с)",
|
||
"Чистое время (с)",
|
||
"Прогресс (%)");
|
||
//--
|
||
offset.Inc();
|
||
for (DVMPackage dvmPackage : target) {
|
||
exportPackageTasks(dvmPackage, offset);
|
||
}
|
||
//--
|
||
for (int i = 0; i < max_columns; ++i)
|
||
sheet.autoSizeColumn(i);
|
||
// Записываем всё в файл
|
||
FileOutputStream stream = new FileOutputStream(file);
|
||
book.write(stream);
|
||
book.close();
|
||
stream.close();
|
||
///--
|
||
FileUtils.copyFile(file, res);
|
||
book = null;
|
||
stream = null;
|
||
file = null;
|
||
}
|
||
@Override
|
||
protected void performDone() throws Exception {
|
||
super.performDone();
|
||
if (Desktop.isDesktopSupported()) {
|
||
if (UI.Question("Таблица сформирована в файле\n" + Utils.DQuotes(res.getAbsolutePath()) + ".\nОткрыть её"
|
||
)) {
|
||
Desktop.getDesktop().open(res);
|
||
}
|
||
}
|
||
}
|
||
}
|