# Bojan Nikolic  <b.nikolic@mrao.cam.ac.uk>, <bojan@bnikolic.co.uk>
# October 2007,
# Revised 2008
#

import numpy
from itertools import ifilter

def readantenna(fnamein):

    def nc_p(l):
        """Defines the antenna files comments"""
        return l[0] != '#'

    stnx, stny, stnz, stnd, =[], [] , [] ,[]

    for line in ifilter(nc_p,
                        open(fnamein)):
            ld=line.split()
            if(len(ld) >3):
                stnx.append(float(ld[0]))
                stny.append(float(ld[1]))
                stnz.append(float(ld[2]))
                stnd.append(float(ld[3]))
                
    return ( numpy.array(stnx),
             numpy.array(stny),
             numpy.array(stnz),
             numpy.array(stnd) )



def AntennaFileToLists(fnamein):

    x , y , z, d = readantenna(fnamein)

    x = x-x.mean() ;
    y = y-y.mean() ;
    z = z-z.mean() ;

    names = [ 'A%02d'% i for i in range(len(x)) ]
    return x, y, z, d , names

def Conf(n):

    "Return the file containing configuration n"

    if n == None:
        raise "Must specify configuration"

    # For now assume n is an integer
    return "../configs/almaconfig.out%02i" % n