2023-06-18 16:35:46 -07:00
|
|
|
package com.rusefi.tools.tune;
|
|
|
|
|
2023-11-25 14:24:23 -08:00
|
|
|
import com.devexperts.logging.Logging;
|
2023-06-18 16:35:46 -07:00
|
|
|
import com.opensr5.ini.DialogModel;
|
|
|
|
import com.opensr5.ini.IniFileModel;
|
|
|
|
import com.rusefi.*;
|
|
|
|
import com.rusefi.core.preferences.storage.Node;
|
|
|
|
import com.rusefi.output.ConfigStructure;
|
|
|
|
import com.rusefi.tune.xml.Constant;
|
|
|
|
import com.rusefi.tune.xml.Msq;
|
|
|
|
import org.jetbrains.annotations.NotNull;
|
|
|
|
|
2023-06-19 07:47:48 -07:00
|
|
|
import javax.xml.bind.JAXBException;
|
2023-06-18 16:35:46 -07:00
|
|
|
import java.io.File;
|
2023-06-19 07:47:48 -07:00
|
|
|
import java.io.FileWriter;
|
2023-06-18 16:35:46 -07:00
|
|
|
import java.io.IOException;
|
2023-07-03 11:44:18 -07:00
|
|
|
import java.io.InputStream;
|
|
|
|
import java.net.URL;
|
2023-06-18 16:35:46 -07:00
|
|
|
import java.nio.file.Files;
|
|
|
|
import java.nio.file.Paths;
|
2023-07-03 11:44:18 -07:00
|
|
|
import java.nio.file.StandardCopyOption;
|
2023-06-18 16:35:46 -07:00
|
|
|
import java.util.List;
|
2023-06-18 17:01:56 -07:00
|
|
|
import java.util.Objects;
|
2023-06-18 16:35:46 -07:00
|
|
|
|
2023-11-25 14:24:23 -08:00
|
|
|
import static com.devexperts.logging.Logging.getLogging;
|
2023-06-18 16:35:46 -07:00
|
|
|
import static com.rusefi.ConfigFieldImpl.unquote;
|
2023-06-20 06:02:49 -07:00
|
|
|
import static com.rusefi.config.Field.niceToString;
|
2023-06-18 16:35:46 -07:00
|
|
|
import static com.rusefi.tools.tune.WriteSimulatorConfiguration.INI_FILE_FOR_SIMULATOR;
|
|
|
|
|
|
|
|
public class TuneCanTool {
|
2023-11-25 14:24:23 -08:00
|
|
|
private static final Logging log = getLogging(TuneCanTool.class);
|
|
|
|
|
2023-06-18 16:35:46 -07:00
|
|
|
public static final String SRC_TEST_RESOURCES = "src/test/resources/";
|
|
|
|
private static final String FOLDER = "generated";
|
2023-11-25 16:02:56 -08:00
|
|
|
public static final String SIMULATED_PREFIX = FOLDER + File.separator + "simulator_tune";
|
|
|
|
public static final String SIMULATED_SUFFIX = ".xml";
|
|
|
|
public static final String DEFAULT_TUNE = SIMULATED_PREFIX + SIMULATED_SUFFIX;
|
2023-07-03 11:44:18 -07:00
|
|
|
private static final String workingFolder = "downloaded_tunes";
|
2023-06-18 16:35:46 -07:00
|
|
|
|
2023-06-19 07:47:48 -07:00
|
|
|
private static Msq simulatorDefaultTune;
|
|
|
|
private static IniFileModel ini;
|
|
|
|
|
|
|
|
|
2023-06-18 16:35:46 -07:00
|
|
|
public static void main(String[] args) throws Exception {
|
2023-06-19 07:47:48 -07:00
|
|
|
simulatorDefaultTune = Msq.readTune(TuneCanTool.DEFAULT_TUNE);
|
|
|
|
ini = new IniFileModel().readIniFile(INI_FILE_FOR_SIMULATOR);
|
2023-06-18 16:35:46 -07:00
|
|
|
if (ini == null)
|
|
|
|
throw new IllegalStateException("Not found " + INI_FILE_FOR_SIMULATOR);
|
|
|
|
|
|
|
|
RootHolder.ROOT = "../firmware/";
|
|
|
|
|
2023-11-25 19:55:20 -08:00
|
|
|
// handle("BK2", 1507, simulatorDefaultTune);
|
|
|
|
handle("BK2-diff", 1507, Msq.readTune(SIMULATED_PREFIX + "_95" + SIMULATED_SUFFIX));
|
2023-11-25 14:24:23 -08:00
|
|
|
// handle("PB", 1502);
|
|
|
|
// handle("Mitsubicha", 1258);
|
|
|
|
// handle("Scion-1NZ-FE", 1448);
|
|
|
|
// handle("4g93", 1425);
|
|
|
|
// handle("BMW-mtmotorsport", 1479);
|
|
|
|
// handle("m111-alex", 1490);
|
2023-06-19 07:47:48 -07:00
|
|
|
}
|
2023-06-18 16:35:46 -07:00
|
|
|
|
2023-11-25 19:55:20 -08:00
|
|
|
private static void handle(String vehicleName, int tuneId, Msq currentTune) throws JAXBException, IOException {
|
2023-07-03 11:44:18 -07:00
|
|
|
String localFileName = workingFolder + File.separator + tuneId + ".msq";
|
2023-06-19 07:47:48 -07:00
|
|
|
String url = "https://rusefi.com/online/view.php?msq=" + tuneId;
|
2023-06-18 20:02:38 -07:00
|
|
|
|
2023-07-03 11:44:18 -07:00
|
|
|
downloadTune(tuneId, localFileName);
|
2023-06-18 16:35:46 -07:00
|
|
|
|
2023-07-03 11:44:18 -07:00
|
|
|
String reportsOutputFolder = "tune_reports";
|
|
|
|
new File(reportsOutputFolder).mkdir();
|
|
|
|
|
|
|
|
Msq custom = Msq.readTune(localFileName);
|
2023-11-25 19:55:20 -08:00
|
|
|
StringBuilder sb = TuneCanTool.getTunePatch(currentTune, custom, ini);
|
2023-06-18 16:35:46 -07:00
|
|
|
|
2023-11-25 19:55:20 -08:00
|
|
|
String fileName = reportsOutputFolder + "/" + vehicleName + ".md";
|
|
|
|
log.info("Writing to " + fileName);
|
|
|
|
|
|
|
|
FileWriter w = new FileWriter(fileName);
|
2023-06-19 11:02:15 -07:00
|
|
|
w.append("# " + vehicleName + "\n\n");
|
2023-11-25 19:55:20 -08:00
|
|
|
w.append("// canned tune " + url + "\n\n");
|
2023-06-18 16:35:46 -07:00
|
|
|
|
2023-06-19 07:47:48 -07:00
|
|
|
w.append("```\n");
|
|
|
|
w.append(sb);
|
|
|
|
w.append("```\n");
|
|
|
|
w.close();
|
2023-06-18 16:35:46 -07:00
|
|
|
}
|
|
|
|
|
2023-07-03 11:44:18 -07:00
|
|
|
private static void downloadTune(int tuneId, String localFileName) throws IOException {
|
|
|
|
new File(workingFolder).mkdirs();
|
|
|
|
String downloadUrl = "https://rusefi.com/online/download.php?msq=" + tuneId;
|
|
|
|
InputStream in = new URL(downloadUrl).openStream();
|
|
|
|
Files.copy(in, Paths.get(localFileName), StandardCopyOption.REPLACE_EXISTING);
|
|
|
|
}
|
|
|
|
|
2023-06-18 16:35:46 -07:00
|
|
|
private static boolean isHardwareEnum(String type) {
|
|
|
|
switch (type) {
|
|
|
|
case "output_pin_e":
|
|
|
|
case "brain_input_pin_e":
|
|
|
|
case "adc_channel_e":
|
|
|
|
case "Gpio":
|
2023-06-19 07:18:40 -07:00
|
|
|
case "spi_device_e":
|
|
|
|
case "pin_input_mode_e":
|
2023-06-19 07:47:48 -07:00
|
|
|
case "pin_output_mode_e":
|
2023-06-18 16:35:46 -07:00
|
|
|
return true;
|
|
|
|
}
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
|
|
|
private static Object simplerSpaces(String value) {
|
|
|
|
if (value == null)
|
|
|
|
return value;
|
|
|
|
return value.replaceAll("\\s+", " ").trim();
|
|
|
|
}
|
|
|
|
|
|
|
|
@NotNull
|
2023-06-19 07:47:48 -07:00
|
|
|
public static StringBuilder getTunePatch(Msq defaultTune, Msq customTune, IniFileModel ini) throws IOException {
|
2023-06-18 16:35:46 -07:00
|
|
|
List<String> options = Files.readAllLines(Paths.get(RootHolder.ROOT + "../" + ConfigDefinition.CONFIG_PATH));
|
|
|
|
String[] totalArgs = options.toArray(new String[0]);
|
|
|
|
|
|
|
|
ReaderStateImpl state = new ReaderStateImpl();
|
|
|
|
ConfigDefinition.doJob(totalArgs, state);
|
|
|
|
|
|
|
|
StringBuilder sb = new StringBuilder();
|
2023-06-19 07:18:40 -07:00
|
|
|
for (DialogModel.Field f : ini.fieldsInUiOrder.values()) {
|
2023-06-18 16:35:46 -07:00
|
|
|
String name = f.getKey();
|
2023-06-18 17:01:56 -07:00
|
|
|
Constant customValue = customTune.getConstantsAsMap().get(name);
|
|
|
|
Constant defaultValue = defaultTune.getConstantsAsMap().get(name);
|
2023-06-18 16:35:46 -07:00
|
|
|
if (defaultValue == null) {
|
|
|
|
// no longer present
|
|
|
|
continue;
|
|
|
|
}
|
2023-06-18 17:01:56 -07:00
|
|
|
Objects.requireNonNull(defaultValue.getValue(), "d value");
|
|
|
|
if (customValue == null) {
|
2023-11-25 14:24:23 -08:00
|
|
|
log.info("Skipping " + name + " TODO");
|
2023-06-18 17:01:56 -07:00
|
|
|
continue;
|
|
|
|
}
|
|
|
|
Objects.requireNonNull(customValue.getValue(), "c value");
|
2023-06-18 16:35:46 -07:00
|
|
|
|
|
|
|
boolean isSameValue = simplerSpaces(defaultValue.getValue()).equals(simplerSpaces(customValue.getValue()));
|
|
|
|
if (!isSameValue) {
|
2023-11-25 14:24:23 -08:00
|
|
|
// todo: what about stuff outside of engine_configuration_s?
|
2023-11-25 19:55:20 -08:00
|
|
|
StringBuffer context = new StringBuffer();
|
|
|
|
|
|
|
|
ConfigField cf = findField(state, name, context);
|
2023-06-18 16:35:46 -07:00
|
|
|
if (cf == null) {
|
2023-11-25 14:24:23 -08:00
|
|
|
log.info("Not found " + name);
|
2023-06-18 16:35:46 -07:00
|
|
|
continue;
|
|
|
|
}
|
2023-11-25 19:55:20 -08:00
|
|
|
String cName = context + cf.getOriginalArrayName();
|
2023-06-18 16:35:46 -07:00
|
|
|
|
|
|
|
if (cf.getType().equals("boolean")) {
|
2023-06-18 18:21:36 -07:00
|
|
|
sb.append(TuneTools.getAssignmentCode(defaultValue, customValue.getName(), unquote(customValue.getValue())));
|
2023-06-18 16:35:46 -07:00
|
|
|
continue;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
if (!Node.isNumeric(customValue.getValue())) {
|
|
|
|
// todo: smarter logic for enums
|
|
|
|
|
|
|
|
String type = cf.getType();
|
|
|
|
if (isHardwareEnum(type)) {
|
|
|
|
continue;
|
|
|
|
}
|
|
|
|
EnumsReader.EnumState sourceCodeEnum = state.getEnumsReader().getEnums().get(type);
|
|
|
|
if (sourceCodeEnum == null) {
|
2023-11-25 14:24:23 -08:00
|
|
|
log.info("No info for " + type);
|
2023-06-18 16:35:46 -07:00
|
|
|
continue;
|
|
|
|
}
|
|
|
|
String customEnum = state.getTsCustomLine().get(type);
|
|
|
|
|
|
|
|
int ordinal;
|
|
|
|
try {
|
|
|
|
ordinal = TuneTools.resolveEnumByName(customEnum, unquote(customValue.getValue()));
|
|
|
|
} catch (IllegalStateException e) {
|
2023-11-25 14:24:23 -08:00
|
|
|
log.info("Looks like things were renamed: " + customValue.getValue() + " not found in " + customEnum);
|
2023-06-18 16:35:46 -07:00
|
|
|
continue;
|
|
|
|
}
|
|
|
|
|
2023-11-25 14:24:23 -08:00
|
|
|
log.info(cf + " " + sourceCodeEnum + " " + customEnum + " " + ordinal);
|
2023-06-18 16:35:46 -07:00
|
|
|
|
|
|
|
String sourceCodeValue = sourceCodeEnum.findByValue(ordinal);
|
2023-11-25 19:55:20 -08:00
|
|
|
sb.append(TuneTools.getAssignmentCode(defaultValue, cName, sourceCodeValue));
|
2023-06-18 16:35:46 -07:00
|
|
|
|
|
|
|
continue;
|
|
|
|
}
|
2023-07-09 18:54:54 -07:00
|
|
|
double doubleValue = Double.valueOf(customValue.getValue());
|
|
|
|
int intValue = (int) doubleValue;
|
|
|
|
boolean isInteger = intValue == doubleValue;
|
2023-06-20 06:02:49 -07:00
|
|
|
if (isInteger) {
|
2023-07-09 18:54:54 -07:00
|
|
|
sb.append(TuneTools.getAssignmentCode(defaultValue, cName, Integer.toString(intValue)));
|
2023-06-20 06:02:49 -07:00
|
|
|
} else {
|
2023-07-09 18:54:54 -07:00
|
|
|
sb.append(TuneTools.getAssignmentCode(defaultValue, cName, niceToString(doubleValue)));
|
2023-06-20 06:02:49 -07:00
|
|
|
}
|
2023-06-18 16:35:46 -07:00
|
|
|
}
|
|
|
|
}
|
|
|
|
return sb;
|
|
|
|
}
|
2023-11-25 19:55:20 -08:00
|
|
|
|
|
|
|
private static ConfigField findField(ReaderStateImpl state, String name, StringBuffer context) {
|
|
|
|
ConfigStructure s = state.getStructures().get("engine_configuration_s");
|
|
|
|
// log.info("We have a custom value " + name);
|
|
|
|
ConfigField cf = s.getTsFieldByName(name);
|
|
|
|
if (cf != null) {
|
|
|
|
return cf; // it was easy!
|
|
|
|
}
|
|
|
|
int fromIndex = 0;
|
|
|
|
while (true) {
|
|
|
|
fromIndex = name.indexOf('_', fromIndex);
|
|
|
|
if (fromIndex == -1) {
|
|
|
|
// no struct names
|
|
|
|
return null;
|
|
|
|
}
|
|
|
|
String parentName = name.substring(0, fromIndex);
|
|
|
|
cf = s.getTsFieldByName(parentName);
|
|
|
|
String type = cf.getType();
|
|
|
|
s = state.getStructures().get(type);
|
|
|
|
|
|
|
|
if (s != null) {
|
|
|
|
String substring = name.substring(fromIndex + 1);
|
|
|
|
ConfigField tsFieldByName = s.getTsFieldByName(substring);
|
|
|
|
if (tsFieldByName == null) {
|
|
|
|
log.info("Not located " + substring + " in " + s);
|
|
|
|
} else {
|
|
|
|
context.append(cf.getOriginalArrayName()).append(".");
|
|
|
|
log.info("Located " + tsFieldByName + " in " + s);
|
|
|
|
}
|
|
|
|
return tsFieldByName;
|
|
|
|
}
|
|
|
|
fromIndex++; // skip underscore
|
|
|
|
}
|
|
|
|
}
|
2023-06-18 16:35:46 -07:00
|
|
|
}
|