clear 
ueConfig=struct('NCellID',25,'RNTI',11,'NULRB',6,'NSubframe',1,'CyclicPrefixUL','Normal','NTxAnts',1,'Hopping','Off');
pucchConfig=struct('NLayers',1,'OrthCover','Off','Shortened',0,'ResourceSize',0);

addpath('../../build/srslte/lib/phch/test')

format_str={'1','1a','1b','2','2a','2b'};

k=1;
for f=0:5
    for n=0:7:130
        for d=1:3
            for ncs=0:d:7
                pucchConfig.ResourceIdx= n;
                pucchConfig.DeltaShift = d;
                pucchConfig.CyclicShifts = ncs;
                if (f >= 3) 
                    nb=20;
                    nb_ack=f-3;
                else 
                    nb=f;
                    nb_ack=0;
                end
                bits=randi(2,nb,1)-1;
                bits_ack=randi(2,nb_ack,1)-1;
                fprintf('Testing PUCCH Format: %s, n_pucch=%d, DeltaShift=%d, CyclicShift=%d\n',format_str{f+1},n,d,ncs);
                if (f >= 3)
                    [sym_mat, info]=ltePUCCH2(ueConfig,pucchConfig,bits);
                    idx=ltePUCCH2Indices(ueConfig,pucchConfig);
                    [dmrs_mat, info_dmrs]=ltePUCCH2DRS(ueConfig,pucchConfig,bits_ack);
                    idx_dmrs=ltePUCCH2DRSIndices(ueConfig,pucchConfig);                                        
                else
                    [sym_mat, info]=ltePUCCH1(ueConfig,pucchConfig,bits);
                    idx=ltePUCCH1Indices(ueConfig,pucchConfig);
                    [dmrs_mat, info_dmrs]=ltePUCCH1DRS(ueConfig,pucchConfig);
                    idx_dmrs=ltePUCCH1DRSIndices(ueConfig,pucchConfig);                                        
                end
                subframe_mat = lteULResourceGrid(ueConfig);
                subframe_mat(idx)=sym_mat;
                subframe_mat(idx_dmrs)=dmrs_mat;

                [sym, dmrs, subframe]=srslte_pucch_encode(ueConfig,pucchConfig,[bits; bits_ack]);
                
                error_sym=max(abs(sym-sym_mat));
                error_dmrs=max(abs(dmrs-dmrs_mat));
                error_sf=max(abs(subframe_mat(:)-subframe));
                k=k+1;

                if (error_sym > 1e-5)
                    disp(info)
                    plot(1:length(sym),sym,1:length(sym_mat),sym_mat)
                    legend('srsLTE','Matlab')
                    error('Error in symbols');                
                end
                if (error_dmrs > 1e-5)
                    disp(info_dmrs)
                    plot(angle(dmrs)-angle(dmrs_mat))
                    error('Error in DMRS');                
                end
                if (error_sf > 1e-5)
                    disp(info)
                    plot(abs(subframe-subframe_mat(:)))
                    error('Error in subframe');
                end
            end
        end
    end
end

if (k == 2) 
    disp(info)
    disp(error_sym)
    disp(error_sf)
    n=1:length(sym);
    plot(n,real(sym(n)),n,real(sym_mat(n)))
end