A method of a user equipment (UE) in a wireless communication system is provided. The method comprises identifying E-field data of each antenna of the UE to be used for transmitting and receiving data, generating, based on the E-field data, a set of codewords including a first and second upper bounds, the set of codewords corresponding to candidate beams of each antenna, selecting at least one codeword from the set of codewords based on a performance criteria, configuring a codebook to be used for each antenna by adding the at least one codeword into the codebook, determining whether the codebook including the at least one codeword satisfies a condition to stop adding another codeword to the codebook, and applying the configured codebook for use in transmitting or receiving the data at each antenna based on whether the condition is satisfied.