Introduce soloFeature to binauraliser

Author: chris-hld

examples/include/binauraliser.h

@@ -214,6 +214,20 @@ void binauraliser_setRPYflag(void* const hBin, int newState);
 /** NOT IMPLEMENTED YET */
 void binauraliser_setInterpMode(void* const hBin, int newMode);
 
+/**
+ * Sets gain factor for an input source.
+ */
+void binauraliser_setSourceGain(void* const hBin, int srcIdx, float newGain);
+
+/**
+ * Set a source to solo.
+ */
+void binauraliser_setSourceSolo(void* const hBin, int srcIdx);
+
+/**
+ * Unsolo / unmute all sources.
+ */
+void binauraliser_setUnSolo(void* const hBin);
 
 
 /* ========================================================================== */

examples/src/binauraliser/binauraliser.c

@@ -87,8 +87,10 @@ void binauraliser_create
     pData->codecStatus = CODEC_STATUS_NOT_INITIALISED;
     pData->procStatus = PROC_STATUS_NOT_ONGOING;
     pData->reInitHRTFsAndGainTables = 1;
-    for(ch=0; ch<MAX_NUM_INPUTS; ch++)
+    for(ch=0; ch<MAX_NUM_INPUTS; ch++) {
         pData->recalc_hrtf_interpFLAG[ch] = 1;
+        pData->src_gains[ch] = 1.f;
+    }
     pData->recalc_M_rotFLAG = 1; 
 }
 
@@ -214,6 +216,12 @@ void binauraliser_process
         for(; i<nSources; i++)
             memset(pData->inputFrameTD[i], 0, BINAURALISER_FRAME_SIZE * sizeof(float));
 
+        /* Apply source gains */
+        for (ch = 0; ch < nSources; ch++) {
+            if(fabsf(pData->src_gains[ch] - 1.f) > 1e-6f)
+                utility_svsmul(pData->inputFrameTD[ch], &(pData->src_gains[ch]), BINAURALISER_FRAME_SIZE, NULL);
+        }
+
         /* Apply time-frequency transform (TFT) */
         afSTFT_forward_knownDimensions(pData->hSTFT, pData->inputFrameTD, BINAURALISER_FRAME_SIZE, MAX_NUM_INPUTS, TIME_SLOTS, pData->inputframeTF);
 
@@ -435,6 +443,30 @@ void binauraliser_setInterpMode(void* const hBin, int newMode)
         pData->recalc_hrtf_interpFLAG[ch] = 1;
 }
 
+void binauraliser_setSourceGain(void* const hAmbi, int srcIdx, float newGain)
+{
+    binauraliser_data *pData = (binauraliser_data*)(hAmbi);
+    pData->src_gains[srcIdx] = newGain;
+}
+
+void binauraliser_setSourceSolo(void* const hAmbi, int srcIdx)
+{
+    binauraliser_data *pData = (binauraliser_data*)(hAmbi);
+    int i;
+    for(i=0; i<pData->nSources; i++){
+        if(i==srcIdx)
+            pData->src_gains[i] = 1.f;
+        else
+            pData->src_gains[i] = 0.f;
+    }
+}
+
+void binauraliser_setUnSolo(void* const hAmbi)
+{
+    binauraliser_data *pData = (binauraliser_data*)(hAmbi);
+    for(int i=0; i<pData->nSources; i++)
+         pData->src_gains[i] = 1.f;
+}
 
 
 /* Get Functions */

examples/src/binauraliser/binauraliser_internal.h

@@ -132,7 +132,8 @@ typedef struct _binauraliser
     int bFlipPitch;                          /**< flag to flip the sign of the pitch rotation angle */
     int bFlipRoll;                           /**< flag to flip the sign of the roll rotation angle */
     int useRollPitchYawFlag;                 /**< rotation order flag, 1: r-p-y, 0: y-p-r */
-    
+    float src_gains[MAX_NUM_INPUTS];         /**< Gains applied per source */
+
 } binauraliser_data;