#!/usr/bin/env python

# usage: makedep.py $(DEP_FILE) $(OBJ_DIR) $(SRC_DIR) [$(SRC_DIR2)]

# Generate $DEP_FILE in $OBJ_DIR (arguments 1 and 2, respectively)
# Read in every file in $SRC_DIR and $SRC_DIR2 (arguments 3 and 4)
# Only depend on modules located in $SRC_DIR or $SRC_DIR2

from __future__ import print_function, unicode_literals

import os
import sys
import re

try:
  dep_file = sys.argv[1]
except:
  dep_file = "depends.d"

try:
  obj_dir = sys.argv[2]
except:
  obj_dir = '.'

try:
  src_dir = sys.argv[3]
except:
  src_dir = '.'

try:
  src_dir2 = sys.argv[4]
except:
  src_dir2 = src_dir

try:
  inc_dir = sys.argv[5]
  files_in_inc_dir = os.listdir(inc_dir)
except:
  inc_dir = 'NONE'

fout = open(dep_file, 'w')
files_in_src_dir =  os.listdir(src_dir)
if src_dir != src_dir2:
  files_in_src_dir.extend(os.listdir(src_dir2))

for src_file in files_in_src_dir:
  file_name, file_ext = os.path.splitext(src_file)
  if file_ext == '.F90':
    try:
      fin = open(src_dir+'/'+src_file,"r")
    except:
      fin = open(src_dir2+'/'+src_file,"r")
    for line in fin:
      if re.match('^ *[Uu][Ss][Ee]',line):
        line_array = line.split()
        # statements are usually "use module, only : subroutine"
        # so we need to strip away the , to get the module name
        file_used = line_array[1].split(',')[0]
        if file_used+'.F90' in files_in_src_dir:
          print(file_name+'.o depends on '+file_used+'.o')
          fout.write(obj_dir+'/'+file_name+'.o: '+obj_dir+'/'+file_used+'.o\n')
        else:
          if inc_dir != 'NONE':
            if file_used+'.mod' in files_in_inc_dir:
              print(file_name+'.o depends on '+file_used+'.mod')
              fout.write(obj_dir+'/'+file_name+'.o: '+inc_dir+'/'+file_used+'.mod\n')
            else:
              # Check for upper case
              file_used = file_used.upper()
              if file_used+'.mod' in files_in_inc_dir:
                print(file_name+'.o depends on '+file_used+'.mod')
                fout.write(obj_dir+'/'+file_name+'.o: '+inc_dir+'/'+file_used+'.mod\n')
    fin.close