// ******************************************************************************
   //
   // Title:       Force Field X.
   // Description: Force Field X - Software for Molecular Biophysics.
   // Copyright:   Copyright (c) Michael J. Schnieders 2001-2024.
   //
   // This file is part of Force Field X.
   //
   // Force Field X is free software; you can redistribute it and/or modify it
  // under the terms of the GNU General Public License version 3 as published by
  // the Free Software Foundation.
  //
  // Force Field X is distributed in the hope that it will be useful, but WITHOUT
  // ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
  // FOR A PARTICULAR PURPOSE. See the GNU General Public License for more
  // details.
  //
  // You should have received a copy of the GNU General Public License along with
  // Force Field X; if not, write to the Free Software Foundation, Inc., 59 Temple
  // Place, Suite 330, Boston, MA 02111-1307 USA
  //
  // Linking this library statically or dynamically with other modules is making a
  // combined work based on this library. Thus, the terms and conditions of the
  // GNU General Public License cover the whole combination.
  //
  // As a special exception, the copyright holders of this library give you
  // permission to link this library with independent modules to produce an
  // executable, regardless of the license terms of these independent modules, and
  // to copy and distribute the resulting executable under terms of your choice,
  // provided that you also meet, for each linked independent module, the terms
  // and conditions of the license of that module. An independent module is a
  // module which is not derived from or based on this library. If you modify this
  // library, you may extend this exception to your version of the library, but
  // you are not obligated to do so. If you do not wish to do so, delete this
  // exception statement from your version.
  //
  // ******************************************************************************
  package ffx;
  
  import edu.rit.pj.Comm;
  import edu.rit.pj.cluster.Configuration;
  import ffx.potential.Utilities;
  import ffx.ui.LogHandler;
  import ffx.ui.MainPanel;
  import ffx.ui.ModelingShell;
  import ffx.ui.OSXAdapter;
  import ffx.utilities.FFXScript;
  import groovy.lang.Script;
  import org.apache.commons.io.FilenameUtils;
  import org.apache.commons.lang3.SystemUtils;
  import org.apache.commons.lang3.builder.ToStringBuilder;
  import org.apache.commons.lang3.time.StopWatch;
  import org.apache.log4j.PropertyConfigurator;
  
  import javax.annotation.Nullable;
  import javax.swing.ImageIcon;
  import javax.swing.JFrame;
  import javax.swing.SwingUtilities;
  import javax.swing.UIManager;
  import java.awt.GraphicsEnvironment;
  import java.awt.Toolkit;
  import java.awt.event.WindowAdapter;
  import java.awt.event.WindowEvent;
  import java.io.File;
  import java.io.IOException;
  import java.net.InetAddress;
  import java.net.URL;
  import java.net.UnknownHostException;
  import java.util.ArrayList;
  import java.util.Arrays;
  import java.util.Date;
  import java.util.List;
  import java.util.Objects;
  import java.util.Properties;
  import java.util.logging.Handler;
  import java.util.logging.Level;
  import java.util.logging.LogManager;
  import java.util.logging.Logger;
  
  import static java.lang.String.format;
  
  /**
   * The Main class is the entry point to the graphical user interface version of Force Field X.
   *
   * @author Michael J. Schnieders
   * @since 1.0
   */
  public final class Main extends JFrame {
  
    private static final Logger logger = Logger.getLogger(Main.class.getName());
    /** This is the main application wrapper. */
    public static MainPanel mainPanel;
    /** Constant <code>stopWatch</code> */
    private static StopWatch stopWatch = new StopWatch();
    /** Handle FFX logging. */
    private static LogHandler logHandler;
    /** The configured Scheduler port. */
    private static int schedulerPort;
    /** Parallel Java Configuration instance. */
   private static Configuration configuration = null;
   /** Parallel Java World Communicator. */
   private static Comm world = null;
   /** Name of the machine FFX is running on. */
   private static String hostName = null;
   /** Force Field X process ID. */
   private static int procID = -1;
   /** Print version and exit */
   private static boolean printVersionAndExit = false;
   /** This Runnable is used to init the GUI using SwingUtilities. */
   private final Runnable initGUI =
       new Runnable() {
         // Create the MainPanel and MainMenu, then add them to the JFrame.
         public void run() {
 
           Toolkit.getDefaultToolkit().setDynamicLayout(true);
 
           // Set the Title and Icon
           Main.this.setTitle("Force Field X");
           URL iconURL = getClass().getClassLoader().getResource("ffx/ui/icons/icon64.png");
           if (iconURL != null) {
             ImageIcon icon = new ImageIcon(iconURL);
             Main.this.setIconImage(icon.getImage());
           }
           Main.this.addWindowListener(
               new WindowAdapter() {
                 @Override
                 public void windowClosing(WindowEvent e) {
                   if (mainPanel != null) {
                     mainPanel.exit();
                   }
                   System.exit(0);
                 }
               });
 
           mainPanel = new MainPanel(Main.this);
           mainPanel.setVisible(false);
           logHandler.setMainPanel(mainPanel);
           Main.this.add(mainPanel);
           mainPanel.initialize();
           Main.this.setVisible(true);
           mainPanel.setVisible(true);
           Main.this.setJMenuBar(mainPanel.getMainMenu());
 
           // This is a hack to get GraphicsCanvis to initialize on some platform/Java3D
           // combinations.
           // mainPanel.setPanel(MainPanel.KEYWORDS);
           // mainPanel.setVisible(true);
           // mainPanel.setPanel(MainPanel.GRAPHICS);
 
           // Mac OS X specific features that help Force Field X look native
           // on Macs. This needs to be done after the MainPanel is created.
           if (SystemUtils.IS_OS_MAC_OSX) {
             OSXAdapter osxAdapter = new OSXAdapter(mainPanel);
           }
 
           SwingUtilities.updateComponentTreeUI(SwingUtilities.getRoot(Main.this));
         }
       };
 
   /**
    * Main does some window initializations.
    *
    * @param commandLineFile a {@link java.io.File} object.
    * @param argList a {@link java.util.List} object.
    */
   public Main(@Nullable File commandLineFile, List<String> argList) {
     super("Force Field X");
     // Start the clock.
     stopWatch.start();
 
     // Init the GUI.
     try {
       SwingUtilities.invokeAndWait(initGUI);
     } catch (Exception e) {
       logger.warning(" Exception initializing the GUI.\n" + e);
     }
 
     // Run the supplied command or file.
     if (commandLineFile != null) {
       runScript(mainPanel.getModelingShell(), commandLineFile, argList);
     }
 
     if (logger.isLoggable(Level.FINE)) {
       StringBuilder sb = new StringBuilder();
       sb.append(format("\n Start-up Time (msec): %s.", stopWatch.getTime()));
       Runtime runtime = Runtime.getRuntime();
       runtime.gc();
       long occupiedMemory = runtime.totalMemory() - runtime.freeMemory();
       long KB = 1024;
       sb.append(format("\n In-Use Memory   (Kb): %d", occupiedMemory / KB));
       sb.append(format("\n Free Memory     (Kb): %d", runtime.freeMemory() / KB));
       sb.append(format("\n Total Memory    (Kb): %d", runtime.totalMemory() / KB));
       logger.fine(sb.toString());
     }
   }
 
   /**
    * Create an instance of Force Field X
    *
    * @param args an array of {@link java.lang.String} objects.
    */
   public static void main(String[] args) {
 
     List<String> argList = new ArrayList<>();
     File commandLineFile = initMain(args, argList);
 
     try {
       // Start up the GUI or CLI version of Force Field X.
       if (!GraphicsEnvironment.isHeadless()) {
         startGraphicalUserInterface(commandLineFile, argList);
       } else {
         if (commandLineFile == null) {
           logger.info(" No command line file supplied.");
         } else {
           startCommandLineInterface(commandLineFile, argList);
         }
       }
     } catch (Throwable t) {
       int statusCode = 1;
       logger.info(" Uncaught exception: exiting with status code " + statusCode);
       logger.info(Utilities.stackTraceToString(t));
       System.exit(statusCode);
     }
   }
 
   /**
    * Process the input arguments into a List, start the logging, start Parallel Java and process the
    * input command.
    *
    * @param args an array of {@link java.lang.String} objects.
    * @param argList List is filled with processed arguments.
    * @return A file for FFX to operate on.
    */
   public static File initMain(String[] args, List<String> argList) {
     // Process any "-D" command line flags.
     args = processProperties(args);
 
     // Start up the Parallel Java communication layer.
     startParallelJava(args);
 
     // Configure our logging.
     startLogging();
 
     // Determine host name and process ID.
     environment();
 
     // Print the header.
     header(args);
 
     // Print out help for the command line interface.
     if (GraphicsEnvironment.isHeadless() && args.length < 2) {
       if (args.length == 1 && args[0].toUpperCase().contains("TEST")) {
         commandLineInterfaceHelp(true);
       }
       commandLineInterfaceHelp(false);
     }
 
     // Parse the specified command or structure file.
     File commandLineFile = null;
     int nArgs = args.length;
     if (nArgs > 0) {
       commandLineFile = new File(args[0]);
       // Resolve a relavtive path
       if (commandLineFile.exists()) {
         commandLineFile = new File(FilenameUtils.normalize(commandLineFile.getAbsolutePath()));
       }
     }
 
     // Convert the args to a List<String>.
     if (nArgs > 1) {
       argList.addAll(Arrays.asList(args).subList(1, nArgs));
     }
 
     return commandLineFile;
   }
 
   /**
    * A main entry point that runs a script and return a refernce to the result.
    *
    * @param args an array of {@link java.lang.String} objects.
    * @return A Groovy Script instance.
    */
   public static Script ffxScript(String[] args) {
     List<String> argList = new ArrayList<>();
     File commandLineFile = initMain(args, argList);
     try {
       if (commandLineFile == null) {
         logger.info(" No command line file supplied.");
       } else {
         return startCommandLineInterface(commandLineFile, argList);
       }
     } catch (Throwable t) {
       int statusCode = 1;
       logger.info(" Uncaught exception: exiting with status code " + statusCode);
       logger.info(Utilities.stackTraceToString(t));
       System.exit(statusCode);
     }
     return null;
   }
 
   /** Print out help for the command line version of Force Field X. */
   private static void commandLineInterfaceHelp(boolean listTestScripts) {
     logger.info(" usage: ffxc [-D<property=value>] <command> [-options] <PDB|XYZ>");
     if (listTestScripts) {
       FFXScript.listGroovyScripts(false, true);
       logger.info("\n For help on an experimental or test command use:  ffxc <command> -h\n");
     } else {
       FFXScript.listGroovyScripts(true, false);
       logger.info("\n To list experimental & test scripts: ffxc --test");
       logger.info(" For help on a specific command use:  ffxc <command> -h\n");
     }
     System.exit(0);
   }
 
   /** Determine the host name, process ID, and FFX base directory. */
   private static void environment() {
     try {
       InetAddress addr = InetAddress.getLocalHost();
       hostName = addr.getHostName();
     } catch (UnknownHostException e) {
       // Do nothing.
     }
 
     String procString = System.getProperty("app.pid");
     if (procString != null) {
       procID = Integer.parseInt(procString);
     } else {
       procID = 0;
     }
 
     String dirString = System.getProperty("basedir");
     File ffxDirectory;
     if (dirString != null) {
       ffxDirectory = new File(dirString);
     } else {
       ffxDirectory = new File(".");
     }
 
     try {
       logger.fine(format(" Force Field X directory is %s", ffxDirectory.getCanonicalPath()));
     } catch (Exception e) {
       // Do Nothing.
     }
   }
 
   /** Print out a promo. */
   private static void header(String[] args) {
     StringBuilder sb = new StringBuilder();
     sb.append(MainPanel.border).append("\n");
     sb.append(MainPanel.title).append("\n");
     sb.append(MainPanel.aboutString).append("\n");
     sb.append(MainPanel.border);
 
     // Print the FFX version and exit.
     if (printVersionAndExit && GraphicsEnvironment.isHeadless()) {
       logger.info(sb.toString());
       System.exit(0);
     }
 
     sb.append("\n ").append(new Date());
     sb.append(format("\n Process ID %d on %s.", procID, hostName));
 
     // Print out command line arguments if the array is not null.
     if (args != null && args.length > 0) {
       sb.append("\n\n Command line arguments:\n ");
       sb.append(Arrays.toString(args));
       sb.append("\n");
     }
 
     if (schedulerPort > 1) {
       sb.append(format("\n Parallel Java:\n %s", world.toString()));
       if (configuration != null) {
         sb.append(
             format(
                 "\n Scheduler %s on port %d.\n", configuration.getSchedulerHost(), schedulerPort));
       }
     }
 
     logger.info(sb.toString());
   }
 
   /** Process any "-D" command line flags. */
   private static String[] processProperties(String[] args) {
     List<String> newArgs = new ArrayList<>();
     for (String arg : args) {
       arg = arg.trim();
 
       if (arg.equals("-V") || arg.equals("--version")) {
         printVersionAndExit = true;
       }
 
       if (arg.startsWith("-D")) {
         // Remove -D from the front of String.
         arg = arg.substring(2);
         // Split at the first equals if it exists.
         if (arg.contains("=")) {
           int equalsPosition = arg.indexOf("=");
           String key = arg.substring(0, equalsPosition);
           String value = arg.substring(equalsPosition + 1);
           // Set the system property.
           System.setProperty(key, value);
         } else {
           if (!arg.isEmpty()) {
             System.setProperty(arg, "");
           }
         }
       } else {
         // Collect non "-D" arguments.
         newArgs.add(arg);
       }
     }
     // Return the remaining arguments.
     args = new String[newArgs.size()];
     newArgs.toArray(args);
     return args;
   }
 
   /**
    * Start the Force Field X command line interface.
    *
    * @param commandLineFile The command line file.
    * @param argList The command line argument list.
    */
   private static Script startCommandLineInterface(File commandLineFile, List<String> argList) {
     if (configuration == null) {
       logger.info(" Starting up the command line interface.\n");
     }
     HeadlessMain m = new HeadlessMain(logHandler);
     mainPanel = m.mainPanel;
     return runScript(mainPanel.getModelingShell(), commandLineFile, argList);
   }
 
   /**
    * Start the Force Field X graphical user interface.
    *
    * @param commandLineFile The command line file.
    * @param argList The command line argument list.
    */
   private static void startGraphicalUserInterface(File commandLineFile, List<String> argList) {
     logger.info(" Starting up the graphical user interface.");
 
     // Set some Swing Constants.
     UIManager.put("swing.boldMetal", Boolean.FALSE);
     setDefaultLookAndFeelDecorated(false);
 
     /*
      Some Mac OS X specific features that help FFX look native. These need
      to be set before the MainPanel is created.
     */
     if (SystemUtils.IS_OS_MAC_OSX) {
       OSXAdapter.setOSXProperties();
       try {
         UIManager.setLookAndFeel(UIManager.getSystemLookAndFeelClassName());
       } catch (Exception e) {
         //
       }
     }
 
     // Initialize the Main frame and Force Field X MainPanel.
     new Main(commandLineFile, argList);
   }
 
   /** Replace the default console handler with our custom FFX handler. */
   private static void startLogging() {
     // Remove all log handlers from the default logger.
     try {
       Logger defaultLogger = LogManager.getLogManager().getLogger("");
       Handler[] defaultHandlers = defaultLogger.getHandlers();
       for (Handler h : defaultHandlers) {
         defaultLogger.removeHandler(h);
       }
     } catch (Exception e) {
       String error = e.toString();
       System.err.println(error);
     }
 
     // Turn off log4j
     Properties properties = new Properties();
     properties.setProperty("log4j.threshold", "OFF");
     properties.setProperty("log4j2.level", "OFF");
     properties.setProperty("org.apache.logging.log4j.level", "OFF");
     PropertyConfigurator.configure(properties);
 
     Logger ffxLogger = Logger.getLogger("ffx");
     // Remove any existing handlers.
     for (Handler handler : ffxLogger.getHandlers()) {
       ffxLogger.removeHandler(handler);
     }
     
     // Retrieve the log level from the ffx.log system property.
     String logLevel = System.getProperty("ffx.log", "info");
     Level tempLevel;
     try {
       tempLevel = Level.parse(logLevel.toUpperCase());
     } catch (Exception e) {
       tempLevel = Level.INFO;
     }
 
     Level level = tempLevel;
     logHandler = new LogHandler();
     logHandler.setLevel(level);
     ffxLogger.addHandler(logHandler);
     ffxLogger.setLevel(level);
   }
 
   /** Start up the Parallel Java communication layer. */
   private static void startParallelJava(String[] args) {
 
     // Try to read the PJ configuration file.
     try {
       configuration = new Configuration("cluster.txt");
     } catch (IOException e) {
       // No cluster configuration file was found.
     }
 
     // Attempt to read the "pj.nn" System property.
     String numNodes = System.getProperty("pj.nn");
     int requestedNodes = -1;
     if (numNodes != null) {
       try {
         requestedNodes = Integer.parseInt(numNodes);
       } catch (Exception e) {
         // Error parsing the the pj.nn property.
       }
     }
 
     // Attempt to read the "pj.port" System property.
     String portString = System.getProperty("pj.port");
     schedulerPort = -1;
     if (portString != null) {
       try {
         schedulerPort = Integer.parseInt(portString);
       } catch (Exception e) {
         // Error parsing the the pj.nn property.
       }
     }
 
     // Configure the scheduler port if it wasn't set on the command line.
     if (schedulerPort <= 0) {
       if (requestedNodes <= 0) {
         // If the "pj.nn" property is not set, configure the port to 1 (no scheduler).
         schedulerPort = 1;
       } else if (configuration != null) {
         // Configure the port using the Configuration.
         schedulerPort = configuration.getSchedulerPort();
       } else {
         // Set the port to the PJ default.
         schedulerPort = 20617; // Must sync this value with the Scheduler.groovy script.
       }
       System.setProperty("pj.port", Integer.toString(schedulerPort));
     }
 
     try {
       Comm.init(args);
       world = Comm.world();
     } catch (Exception e) {
       // Log the exception to System.err because the logging subsystem has not be initialized yet.
       String message = " Exception starting up the Parallel Java communication layer.";
       System.err.println(message);
       String error = e.toString();
       System.err.println(error);
     }
   }
 
   public static Script runScript(ModelingShell shell, File commandLineFile, List<String> argList) {
     // Attempt to run a supplied script.
     if (commandLineFile.exists()) {
       return shell.runFFXScript(commandLineFile, argList);
     } else {
       // See if the commandLineFile is an embedded script.
       String name = commandLineFile.getName();
       Class<? extends Script> ffxScript = FFXScript.getScript(name);
       if (ffxScript != null) {
         return shell.runFFXScript(ffxScript, argList);
       }
     }
     return null;
   }
 
   /**
    * {@inheritDoc}
    *
    * <p>Commons.Lang Style toString.
    */
   @Override
   public String toString() {
     ToStringBuilder toStringBuilder =
         new ToStringBuilder(this)
             .append("Up Time: " + stopWatch)
             .append("Logger: " + logger.getName());
     return toStringBuilder.toString();
   }
 }