Modify default options depending on the engine (featurecat#914)

kaorahi · Jan 14, 2023 · 03bfd1a · 03bfd1a
1 parent 1cacd75
commit 03bfd1a
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 5 deletions.
diff --git a/src/main/java/featurecat/lizzie/gui/ConfigDialog.java b/src/main/java/featurecat/lizzie/gui/ConfigDialog.java
@@ -1848,7 +1848,8 @@ private String getEngineLine() {
           weightFile = chooserw.getSelectedFile();
           if (weightFile != null) {
             weightPath = relativizePath(weightFile.toPath(), this.curPath);
-            EngineParameter ep = new EngineParameter(enginePath, weightPath, this);
+            EngineParameter ep =
+                new EngineParameter(enginePath, weightPath, guessedEngineType(), this);
             ep.setVisible(true);
             if (!ep.commandLine.isEmpty()) {
               engineLine = ep.commandLine;
@@ -1860,6 +1861,13 @@ private String getEngineLine() {
     return engineLine;
   }
 
+  private String guessedEngineType() {
+    String engineFileName = (new File(enginePath)).toPath().getFileName().toString();
+    // fixme: ad hoc!
+    Boolean isKataGo = engineFileName.toLowerCase().contains("katago");
+    return isKataGo ? "katago" : "leelaz";
+  }
+
   private String getImagePath() {
     String imagePath = "";
     File imageFile = null;
@@ -1903,6 +1911,9 @@ private void getCommandHelp() {
 
     List<String> commands = new ArrayList<String>();
     commands.add(enginePath);
+    if (guessedEngineType().equals("katago")) {
+      commands.add("gtp");
+    }
     commands.add("-h");
 
     try {

diff --git a/src/main/java/featurecat/lizzie/gui/EngineParameter.java b/src/main/java/featurecat/lizzie/gui/EngineParameter.java
@@ -30,7 +30,8 @@ public class EngineParameter extends JDialog {
   private Color oriColor;
 
   /** Create the dialog. */
-  public EngineParameter(String enginePath, String weightPath, ConfigDialog configDialog) {
+  public EngineParameter(
+      String enginePath, String weightPath, String engineType, ConfigDialog configDialog) {
     setTitle(configDialog.resourceBundle.getString("LizzieConfig.title.parameterConfig"));
     setModalExclusionType(ModalExclusionType.APPLICATION_EXCLUDE);
     setModal(true);
@@ -48,7 +49,8 @@ public EngineParameter(String enginePath, String weightPath, ConfigDialog config
     txtCommandLine = new JTextField();
     txtCommandLine.setEditable(false);
     txtCommandLine.setBounds(89, 12, 565, 26);
-    txtCommandLine.setText(enginePath + " --weights " + weightPath);
+    String weightOption = (engineType.equals("leelaz")) ? " --weights " : " gtp -model ";
+    txtCommandLine.setText(enginePath + weightOption + weightPath);
     contentPanel.add(txtCommandLine);
     txtCommandLine.setColumns(10);
     JLabel lblParameter =
@@ -67,7 +69,9 @@ public void focusLost(FocusEvent e) {
         });
     txtParameter.setColumns(10);
     txtParameter.setBounds(89, 44, 565, 26);
-    txtParameter.setText("-g --lagbuffer 0 ");
+    if (engineType.equals("leelaz")) {
+      txtParameter.setText("-g --lagbuffer 0 ");
+    }
     oriColor = txtParameter.getBackground();
     contentPanel.add(txtParameter);
 
@@ -95,7 +99,7 @@ public void focusLost(FocusEvent e) {
     okButton.addActionListener(
         new ActionListener() {
           public void actionPerformed(ActionEvent e) {
-            if (txtParameter.getText().isEmpty()) {
+            if (txtParameter.getText().isEmpty() && engineType.equals("leelaz")) {
               txtParameter.setBackground(Color.RED);
             } else {
               parameters = txtParameter.getText().trim();