mip_02_extract_matrices.py

import subprocess
import sys
import os

try:
  os.mkdir('mip_matrices')
except:
  pass

CMR_BUILD = './build'
CMR_EXTRACT_GUROBI = CMR_BUILD + '/cmr-extract-gurobi'
CMR_K_ARY = CMR_BUILD + '/cmr-k-ary'
CMR_MATRIX = CMR_BUILD + '/cmr-matrix'

for mip in sys.argv[1:]:
  if mip[-7:] == '.mps.gz':
    base = mip[:-7]
  else:
    base = mip
  base = os.path.basename(base)
  subprocess.call(f'{CMR_EXTRACT_GUROBI} {mip} -o sparse > {base}.sparse', shell=True)
  subprocess.call(f'head -n 1 {base}.sparse > mip_matrices/{base}.original.header', shell=True)
  subprocess.call(f'{CMR_K_ARY} {base}.sparse -R {base}.submat -i sparse -t', shell=True)
  print(f'Extracting actual submatrix of {base}.sparse according to {base}.submat.', flush=True)
  subprocess.call(f'{CMR_MATRIX} {base}.sparse -S {base}.submat -i sparse -d - | gzip > mip_matrices/{base}.ternary.sparse.gz', shell=True)
  subprocess.call(f'{CMR_K_ARY} {base}.sparse -R {base}.submat -i sparse -b', shell=True)
  print(f'Extracting actual submatrix of {base}.sparse according to {base}.submat.', flush=True)
  subprocess.call(f'{CMR_MATRIX} {base}.sparse -S {base}.submat -i sparse -d - | gzip > mip_matrices/{base}.binary.sparse.gz', shell=True)
  os.unlink(f'{base}.sparse')
  os.unlink(f'{base}.submat')