Changeset 3079

Timestamp:

Oct 11, 2010 11:54:31 AM (11 years ago)

Author:

jwatson

Message:

Improvements to GUROBI plug-in to allow execution on windows and linux, and eliminated Coopr module import dependencies on claird home diretory existence and accessibility :)

File:

• coopr.plugins/trunk/coopr/plugins/mip/GUROBI.py (modified) (9 diffs)

coopr.plugins/trunk/coopr/plugins/mip/GUROBI.py

@@ -10 +10 @@
 
 import os
+import sys
 import re
 from coopr.opt.base import *
@@ -27 +28 @@
 import time
 
-# TODO: CHECK IF IT IS ILMLicensedSystemCallSolver
 class GUROBI(ILMLicensedSystemCallSolver):
     """The GUROBI LP/MIP solver
@@ -63 +63 @@
         self._capabilities = pyutilib.misc.Options()
         self._capabilities.linear = True
+        self._capabilities.quadratic = True        
         self._capabilities.integer = True
+        self._capabilities.sos1 = True
+        self._capabilities.sos2 = True        
 
     #
@@ -69 +72 @@
     #
     def warm_start_capable(self):
+
        return False
 
     def executable(self):
-        executable = pyutilib.services.registered_executable("gurobi.sh")
+        
+        if sys.platform == "win32":
+           executable = pyutilib.services.registered_executable("gurobi.bat")
+        else:
+           executable = pyutilib.services.registered_executable("gurobi.sh")            
         if executable is None:
             pyutilib.component.core.PluginGlobals.env().log.error("Could not locate the 'gurobi' executable, which is required for solver %s" % self.name)
@@ -80 +88 @@
 
     def create_command_line(self,executable,problem_files):
+
         #
         # Define log file
@@ -96 +105 @@
         # Write the GUROBI execution script
         #
+
+        # NOTE: the gurobi shell is independent of Coopr python virtualized environment, so any
+        #       imports - specifically that required to get GUROBI_RUN - must be handled explicitly.
         self.gurobi_script_file_name = pyutilib.services.TempfileManager.create_tempfile(suffix = '.gurobi.script')
         gurobi_script_file = open(self.gurobi_script_file_name,'w')
         gurobi_script_file.write("import sys\n")
         gurobi_script_file.write("from gurobipy import *\n")
-        gurobi_script_file.write("sys.path.append(\'/home/claird/Research/CooprWork/coopr/src/coopr.plugins/coopr/plugins/mip\')\n")
+        # NOTE: The gurobi plugin and GUROBI_RUN.py live in the same directory.
+        gurobi_script_file.write("sys.path.append("+"\""+os.path.dirname(__file__)+"\""+")\n")
         gurobi_script_file.write("from GUROBI_RUN import *\n")
         gurobi_script_file.write("gurobi_run(\""+self._problem_files[0]+"\",\""+self.soln_file+"\")\n")
@@ -109 +122 @@
         # user if we're keeping files around.
         if self.keepFiles:
-           print "Solver script file=" + self.gurobi_script_file_name
+           print "Solver script file: '%s'" % self.gurobi_script_file_name
 
         #
@@ -118 +131 @@
 
     def process_logfile(self):
+        
         return ILMLicensedSystemCallSolver.process_logfile(self)
 
     def process_soln_file(self,results):
+        
         # check for existence of the solution file
         # not sure why we just return - would think that we 
@@ -173 +188 @@
         results.solution.insert(soln)
 
-        # todo:
-        # is this necessary????
-#        return ILMLicensedSystemCallSolver.process_soln_file(self,results)
- 
-pyutilib.services.register_executable(name="gurobi.sh")
+if sys.platform == "win32":
+   pyutilib.services.register_executable(name="gurobi.bat")
+else:
+   pyutilib.services.register_executable(name="gurobi.sh")