#!/usr/bin/env python3

from sys import argv, exit
from glob import glob

program = 'jobdisks'
version = '1.2'
verdate = '20091114'
author  = 'Walter Brisken'


fill = '   --   '

def usage():
	print('\n%s ver. %s   %s %s' % (program, version, verdate, author))
	print('\nA program to list all the modules used in a project')
	print('\nUsage: %s [options] [<file 1> [<file 2> ... ] ] ]' % (argv[0]))
	print('\n<file X> is either a .fx correlator job script file or a .input DiFX input')
	print('          file.  Any number of files can be listed.  If no files are listed,')
	print('          all .input files in the current directory are used.  If still no')
	print('          files are found, all the .fx files in the directory are used.')
	print('\nOptions can include:\n')
	print('    -h or --help     Print this help info\n')
	print('    -c or --changes  Print module changes only\n')
	print('Example:  jobdisks\n')
	print('Example:  jobdisks job1420.*.input\n')
	print('Example:  jobdisks *.fx\n')
	exit(0)

def getmodules_fx(filename, antlist):
	mods = {}
	data = open(filename).readlines()
	for d in data:
		if d[:7] == 'tape_id':
			s = split(d)
			ant = s[-2]
			mod = s[2]
			if not ant in antlist:
				antlist.append(ant)
			
			mods[ant] = mod
	return mods
	

def getmodules_input(filename, antlist):
	mods = {}
	data = open(filename).readlines()
	ants = []
	for d in data:
		if d[:14] == 'TELESCOPE NAME':
			ant = d[20:].strip()
			ants.append(ant)
			if ant in antlist:
				pass
				#ant = 'XXX'
			else:
				antlist.append(ant)
		elif d[:5] == 'FILE ':
			if len(ants) > 0:
				ant = ants[0]
				ants.remove(ant)
				mods[ant] = d[20:].strip().split()[0]
	return mods

def getmodules(filename, antlist):
	if filename[-3:] == '.fx':
		return getmodules_fx(filename, antlist)
	elif filename[-6:] == '.input':
		return getmodules_input(filename, antlist)
	else:
		print('Unknown filetype : %s. Ignoring.' % filename)
		return []

# use '-' to indicate no module for that station
def fillmodlist(antlist, modlist):
	for m in modlist:
		for a in antlist:
			if not a in m:
				m[a] = fill

def printmodlist(files, antlist, modlist):
	nf = len(files)
	na = len(antlist)
	antsused = []
	lastused = {}
	nspace = len(files[0])-6-4+2+3
	if files[0][-3:] == '.fx':
		nspace += 3
	out = 'FILE' + (' '*nspace)
	for a in range(na):
		out += ('%-10s' % antlist[a])
		lastused[antlist[a]] = 'X'
	out = out.strip()
	print(out)
	for f in range(nf):
		if files[f][-6:] == '.input':
			out = files[f][:-6]
		elif files[f][-3:] == '.fx':
			out = files[f][:-3]
		else:
			out = files[f]
		for a in range(na):
			ant = antlist[a]
			change = False
			if modlist[f][ant] == fill:
				change = False
			elif a not in antsused:
				antsused.append(a)
				change = True
				lastused[ant] = modlist[f][ant]
			elif modlist[f][ant] != lastused[ant]:
				change = True
				lastused[ant] = modlist[f][ant]
			if change:
				out += ' *'
			else:
				out += '  '
			out += modlist[f][ant]
		out = out.strip()
		print(out)

def printchangelist(files, antlist, modlist):
	nf = len(files)
	na = len(antlist)
	lastused = {}
	for ant in antlist:
		lastused[ant] = 'X'
	for f in range(nf):
		jobchange = False
		for a in range(na):
			ant = antlist[a]
			change = False
			if modlist[f][ant] == fill:
				change = False
			elif f == 0:
				change = True
				lastused[ant] = modlist[f][ant]
			elif modlist[f][ant] != lastused[ant]:
				change = True
				lastused[ant] = modlist[f][ant]
			if change:
				if not jobchange:
					jobchange = True
					if files[f][-6:] == '.input':
						out = files[f][:-6]
					elif files[f][-3:] == '.fx':
						out = files[f][:-3]
					else:
						out = files[f]
					print(out)
				print('  %-2s %s' % (ant, modlist[f][ant]))
	
def checkfiles(files):
	types = {}
	quit = False

	for f in files:
		if len(glob(f)) < 1:
			print('File not found : %s' % f)
			quit = True
		else:
			if f[-3:] == '.fx':
				types['.fx'] = True
			elif f[-6:] == '.input':
				types['.input'] = True
			else:
				types[f] = True
			
	if len(types) > 1:
		print('Mixture of types encountered.')
		quit = True
	if quit:
		exit(0)

def getmoduleset(files):
	antlist = []
	modlist = []
	for filename in files:
		modlist.append(getmodules(filename, antlist))

	antlist.sort()

	fillmodlist(antlist, modlist)

	return antlist, modlist


mode = 'Modules'

files = []
masterant = ''
for arg in argv[1:]:
	if arg == '-h' or arg == '--help':
		usage()
	elif arg == '-c' or arg == '--changes':
		mode = 'Changes'
	else:
		if arg[-6:] == '.input' or arg[-3:] == '.fx':
			files.append(arg)
		else:
			print('unknown file type : %s' % arg)
			exit(0)

if len(files) < 1:
	files = glob('*.input')
	if len(files) < 1:
		files = glob('*.fx')
		if len(files) < 1:
			print('No files found.  Quitting.')
			exit(0)

files.sort()

checkfiles(files)

antlist, modlist = getmoduleset(files)

if mode == 'Modules':
	printmodlist(files, antlist, modlist)
elif mode == 'Changes':
	printchangelist(files, antlist, modlist)
else:
	print('Unknown mode of operation : %s' % mode)