Added ue location calculation for a particular L

lib/include/srslte/phy/phch/pdcch.h

@@ -167,7 +167,15 @@ SRSLTE_API uint32_t srslte_pdcch_ue_locations(srslte_pdcch_t *q,
 SRSLTE_API uint32_t srslte_pdcch_ue_locations_ncce(uint32_t nof_cce, 
                                                    srslte_dci_location_t *c, 
                                                    uint32_t max_candidates, 
-                                                   uint32_t nsubframe, uint16_t rnti); 
+                                                   uint32_t nsubframe,
+                                                   uint16_t rnti);
+
+SRSLTE_API uint32_t srslte_pdcch_ue_locations_ncce_L(uint32_t nof_cce,
+                                                     srslte_dci_location_t *c,
+                                                     uint32_t max_candidates,
+                                                     uint32_t nsubframe,
+                                                     uint16_t rnti,
+                                                     int L);
 
 /* Function for generation of common search space DCI locations */
 SRSLTE_API uint32_t srslte_pdcch_common_locations(srslte_pdcch_t *q, 

lib/src/phy/phch/pdcch.c

@@ -218,13 +218,20 @@ uint32_t srslte_pdcch_ue_locations(srslte_pdcch_t *q, srslte_dci_location_t *c,
   return srslte_pdcch_ue_locations_ncce(q->nof_cce, c, max_candidates, nsubframe, rnti);
 }
 
+
+uint32_t srslte_pdcch_ue_locations_ncce(uint32_t nof_cce, srslte_dci_location_t *c, uint32_t max_candidates,
+                                        uint32_t nsubframe, uint16_t rnti)
+{
+  return srslte_pdcch_ue_locations_ncce_L(nof_cce, c, max_candidates, nsubframe, rnti, -1);
+}
+
 /** 36.213 v9.1.1 
  * Computes up to max_candidates UE-specific candidates for DCI messages and saves them 
  * in the structure pointed by c.
  * Returns the number of candidates saved in the array c.   
  */
-uint32_t srslte_pdcch_ue_locations_ncce(uint32_t nof_cce, srslte_dci_location_t *c, uint32_t max_candidates,
-                        uint32_t nsubframe, uint16_t rnti) {
+uint32_t srslte_pdcch_ue_locations_ncce_L(uint32_t nof_cce, srslte_dci_location_t *c, uint32_t max_candidates,
+                        uint32_t nsubframe, uint16_t rnti, int Ls) {
   
   int l; // this must be int because of the for(;;--) loop
   uint32_t i, k, L, m; 
@@ -241,6 +248,7 @@ uint32_t srslte_pdcch_ue_locations_ncce(uint32_t nof_cce, srslte_dci_location_t
   // All aggregation levels from 8 to 1
   for (l = 3; l >= 0; l--) {
     L = (1 << l);
+    if (Ls<0 || Ls==L) {
       // For all candidates as given in table 9.1.1-1
       for (i = 0; i < nof_candidates[l]; i++) {
         if (nof_cce >= L) {
@@ -259,6 +267,7 @@ uint32_t srslte_pdcch_ue_locations_ncce(uint32_t nof_cce, srslte_dci_location_t
         }
       }
     }
+  }
 
   DEBUG("Initiated %d candidate(s) in the UE-specific search space for C-RNTI: 0x%x, nsubframe=%d, nof_cce=%d\n", 
          k, rnti, nsubframe, nof_cce);