src.acoustools.Utilities.Boards

 1import torch
 2from torch import Tensor
 3from acoustools.Utilities.Setup import device
 4
 5
 6def create_board(N:int, z:float) -> Tensor: 
 7    '''
 8    Create a single transducer array \n
 9    :param N: Number of transducers + 1 per side eg for 16 transducers `N=17`
10    :param z: z-coordinate of board
11    :return: tensor of transducer positions
12    '''
13    pitch=0.0105
14    grid_vec=pitch*(torch.arange(-N/2+1, N/2, 1)).to(device)
15    x, y = torch.meshgrid(grid_vec,grid_vec,indexing="ij")
16    x = x.to(device)
17    y= y.to(device)
18    trans_x=torch.reshape(x,(torch.numel(x),1))
19    trans_y=torch.reshape(y,(torch.numel(y),1))
20    trans_z=z*torch.ones((torch.numel(x),1)).to(device)
21    trans_pos=torch.cat((trans_x, trans_y, trans_z), axis=1)
22    return trans_pos
23
24# BOARD_POSITIONS = .234/2
25BOARD_POSITIONS:float = 0.2365/2
26'''
27Static variable for the z-position of the boards, positive for top board, negative for bottom board
28'''
29  
30def transducers(N=16,z=BOARD_POSITIONS) -> Tensor:
31  '''
32  :return: the 'standard' transducer arrays with 2 16x16 boards at `z = +-234/2 `
33  '''
34  return torch.cat((create_board(N+1,z),create_board(N+1,-1*z)),axis=0).to(device)
35
36
37
38TRANSDUCERS:Tensor = transducers()
39'''
40Static variable for `transducers()` result
41'''
42TOP_BOARD:Tensor = create_board(17,BOARD_POSITIONS)
43'''
44Static variable for a 16x16 array at `z=.234/2` - top board of a 2 array setup
45'''
46BOTTOM_BOARD:Tensor = create_board(17,-1*BOARD_POSITIONS)
47'''
48Static variable for a 16x16 array at `z=-.234/2` - bottom board of a 2 array setup
49'''