def hera_to_info(hexnum, nant, pols=['x'], connections=None, red_type=[], **kwargs):
'''Go from hera array to info.
connections is a txt file that has the hera ant numbers and paper antenna numbers.
red_type is a list of the baselines (in paper_ant format with (i,j), i<j) in the redundant groups you want.'''
antpos = -np.ones((nant*len(pols),3))
if connections:
paper_ants = get_paper_ants(connections)
hex_index = 0
for row in range(hexnum-1, -(hexnum), -1):
for col in range(2*hexnum-abs(row)-1):
for z, pol in enumerate(pols):
z = 2**z
x = ((-(2*hexnum-abs(row))+2)/2.0 + col)
y = row*-1*np.sqrt(3)/2
i = omni.Antpol(paper_ants[hex_index],pol,nant)
antpos[i,0],antpos[i,1],antpos[i,2] = x,y,z
hex_index+=1
reds = omni.compute_reds(nant, pols, antpos[:nant],tol=.1)
ex_ants = [omni.Antpol(i,nant).ant() for i in range(antpos.shape[0]) if antpos[i,-1] < 0] # look at z (pol) component to see if flagged antenna. Positions can be positive and negative.
#import IPython; IPython.embed()
kwargs['ex_ants'] = kwargs.get('ex_ants',[]) + ex_ants
reds = omni.filter_reds(reds, **kwargs)
info = omni.FirstCalRedundantInfo(nant)
info.init_from_reds(reds,antpos)
return info
def aa_to_info_hera(aa, pols=['x'], fcal=False, **kwargs):
nant = len(aa)
try:
antpos_ideal = aa.antpos_ideal
xs,ys,zs = antpos_ideal.T
layout = np.arange(len(xs))
#antpos = np.concatenat([antpos_ideal for i in len(pols)])
except(AttributeError):
layout = aa.ant_layout
xs,ys = np.indices(layout.shape)
antpos = -np.ones((nant*len(pols),3)) #remake antpos with pol information. -1 to flag
for ant,x,y in zip(layout.flatten(), xs.flatten(), ys.flatten()):
for z, pol in enumerate(pols):
z = 2**z
i = omni.Antpol(ant, pol, len(aa))
antpos[i,0], antpos[i,1], antpos[i,2] = x,y,z
reds = omni.compute_reds(nant, pols, antpos[:nant], tol=.1)
ex_ants = [omni.Antpol(i,nant).ant() for i in range(antpos.shape[0]) if antpos[i,0] == -1]
kwargs['ex_ants'] = kwargs.get('ex_ants',[]) + ex_ants
reds = omni.filter_reds(reds, **kwargs)
if fcal:
info = omni.FirstCalRedundantInfo(nant)
else:
info = omni.RedundantInfo(nant)
info.init_from_reds(reds,antpos)
return info
def hex_to_info(hexnum, pols=['x'], **kwargs):
'''Given a hex number generate redundancies, assuming antenna
numbering starts at 0 in the bottom left of the hex.
Returns an omnical info class'''
nant = 3*hexnum**2 - 3*hexnum + 1
antpos = -np.ones((nant*len(pols),3))
ant=0
for row in range(hexnum-1, -(hexnum), -1):
for col in range(2*hexnum-abs(row)-1):
for z, pol in enumerate(pols):
x = ((-(2*hexnum-abs(row))+2)/2.0 + col)
y = row*-1*np.sqrt(3)/2
i = omni.Antpol(ant,pol,nant)
antpos[i,0],antpos[i,1],antpos[i,2] = x,y,z
ant+=1
reds = omni.compute_reds(nant, pols, antpos[:nant],tol=.1)
ex_ants = []
kwargs['ex_ants'] = kwargs.get('ex_ants',[]) + ex_ants
reds = omni.filter_reds(reds, **kwargs)
info = omni.RedundantInfo(nant)
info.init_from_reds(reds,antpos)
return info