# copyright 2000 greg landrum
""" Generic file manipulation stuff

"""


import re

import numpy


class ReFile:
  """convenience class for dealing with files with comments

  blank (all whitespace) lines, and lines beginning with comment
    characters are skipped.

  anything following a comment character on a line is stripped off
  """

  def readline(self):
    """ read the next line and return it.

    return '' on EOF

    """
    result = ''
    while result == '':
      inLine = self.inFile.readline()
      if inLine == '':
        return ''
      result = self.regExp.split(inLine)[0].strip()
    return result

  def readlines(self):
    """ return a list of all the lines left in the file

    return [] if there are none

    """
    res = []
    inLines = self.inFile.readlines()
    for line in inLines:
      result = self.regExp.split(line)[0].strip()
      if result != '':
        res.append(result)

    return res

  def rewind(self):
    """ rewinds the file (seeks to the beginning)

    """
    self.inFile.seek(0)

  def __init__(self, fileName, mode='r', comment=r'#', trailer=r'\n'):
    if trailer is not None and trailer != '':
      comment = comment + r'|' + trailer
    self.regExp = re.compile(comment)
    self.inFile = open(fileName, mode)


def ReadDataFile(fileName, comment=r'#', depVarCol=0, dataType=numpy.float):
  """ read in the data file and return a tuple of two Numeric arrays:
  (independent variables, dependant variables).

  **ARGUMENTS:**

  - fileName: the fileName

  - comment: the comment character for the file

  - depVarcol: the column number containing the dependant variable

  - dataType: the Numeric short-hand for the data type

  RETURNS:

   a tuple of two Numeric arrays:

    (independent variables, dependant variables).

  """
  inFile = ReFile(fileName)
  dataLines = inFile.readlines()
  nPts = len(dataLines)

  if dataType in [numpy.float, numpy.float32, numpy.float64]:
    _convfunc = float
  else:
    _convfunc = int

  nIndVars = len(dataLines[0].split()) - 1
  indVarMat = numpy.zeros((nPts, nIndVars), dataType)
  depVarVect = numpy.zeros(nPts, dataType)
  for i in range(nPts):
    splitLine = dataLines[i].split()
    depVarVect[i] = _convfunc(splitLine[depVarCol])
    del splitLine[depVarCol]
    indVarMat[i, :] = map(_convfunc, splitLine)

  return indVarMat, depVarVect


if __name__ == '__main__':
  import sys

  fileN = sys.argv[1]
  iV, dV = ReadDataFile(fileN)
  print('iV:', iV)
  print('dV:', dV)