import numpy as np

def nrPBCHDMRSIndices(ncellid, style = 'matlab'):
    indices_sym1_3 = np.arange(ncellid%4, 240, 4)
    indices_sym2 = np.concatenate((np.arange(ncellid%4, 48, 4), np.arange(192 + ncellid%4, 240, 4)))
    if style == 'python':
        indices = np.concatenate((indices_sym1_3, indices_sym2, indices_sym1_3))
        symbol_idx = np.concatenate((1*np.ones(len(indices_sym1_3)), 2*np.ones(len(indices_sym2)), 3*np.ones(len(indices_sym1_3)))).astype('int')
        return (tuple(indices), tuple(symbol_idx))
    elif style == 'matlab':
        indices = np.concatenate((indices_sym1_3+240, indices_sym2+2*240, indices_sym1_3+3*240))
        return indices
    else:
        print("Error: Unknown style!")
        return