Working polartest

openair1/PHY/CODING/TESTBENCH/polartest.c

@@ -10,8 +10,8 @@
 #include "PHY/CODING/coding_defs.h"
 #include "SIMULATION/TOOLS/sim.h"
 
-//#define DEBUG_POLAR_PARAMS
 //#define DEBUG_DCI_POLAR_PARAMS
+//#define DEBUG_POLAR_TIMING
 
 int main(int argc, char *argv[]) {
 
@@ -24,7 +24,6 @@ int main(int argc, char *argv[]) {
 
 	randominit(0);
 	crcTableInit();
-	uint32_t crc;
 	//Default simulation values (Aim for iterations = 1000000.)
 	int itr, iterations = 1000, arguments, polarMessageType = 0; //0=PBCH, 1=DCI, -1=UCI
 	double SNRstart = -20.0, SNRstop = 0.0, SNRinc= 0.5; //dB
@@ -34,14 +33,13 @@ int main(int argc, char *argv[]) {
 	int8_t decoderState=0, blockErrorState=0; //0 = Success, -1 = Decoding failed, 1 = Block Error.
 	uint16_t testLength = 0, coderLength = 0, blockErrorCumulative=0, bitErrorCumulative=0;
 	double timeEncoderCumulative = 0, timeDecoderCumulative = 0;
-	uint8_t aggregation_level, decoderListSize, pathMetricAppr;
+	uint8_t aggregation_level = 8, decoderListSize = 8, pathMetricAppr = 0;
 
 	while ((arguments = getopt (argc, argv, "s:d:f:m:i:l:a:")) != -1)
 	switch (arguments)
 	{
 		case 's':
 			SNRstart = atof(optarg);
-			printf("SNRstart = %f\n", SNRstart);
 			break;
 
 		case 'd':
@@ -93,7 +91,12 @@ int main(int argc, char *argv[]) {
 
 	folderName=getenv("HOME");
 	strcat(folderName,"/Desktop/polartestResults");
+
+	#ifdef DEBUG_POLAR_TIMING
+	sprintf(fileName,"%s/TIMING_ListSize_%d_pmAppr_%d_Payload_%d_Itr_%d",folderName,decoderListSize,pathMetricAppr,testLength,iterations);
+	#else
 	sprintf(fileName,"%s/_ListSize_%d_pmAppr_%d_Payload_%d_Itr_%d",folderName,decoderListSize,pathMetricAppr,testLength,iterations);
+	#endif
 	strftime(currentTimeInfo, 25, "_%Y-%m-%d-%H-%M-%S.csv", localtime(&currentTime));
 	strcat(fileName,currentTimeInfo);
 
@@ -107,65 +110,45 @@ int main(int argc, char *argv[]) {
         fprintf(stderr,"[polartest.c] Problem creating file %s with fopen\n",fileName);
         exit(-1);
       }
+
+#ifdef DEBUG_POLAR_TIMING
+    fprintf(logFile,",timeEncoderCRCByte[us],timeEncoderCRCBit[us],timeEncoderInterleaver[us],timeEncoderBitInsertion[us],timeEncoder1[us],timeEncoder2[us],timeEncoderRateMatching[us],timeEncoderByte2Bit[us]\n");
+#else
     fprintf(logFile,",SNR,nBitError,blockErrorState,t_encoder[us],t_decoder[us]\n");
+#endif
 
-	//uint8_t *testInput = malloc(sizeof(uint8_t) * testLength); //generate randomly
-	//uint8_t *encoderOutput = malloc(sizeof(uint8_t) * coderLength);
-	uint32_t testInput[4], encoderOutput[4];
-	memset(testInput,0,sizeof(testInput));
-	memset(encoderOutput,0,sizeof(encoderOutput));
+    uint8_t testArrayLength = ceil(testLength / 32.0);
+    uint8_t coderArrayLength = ceil(coderLength / 32.0);
 
-	double *modulatedInput = malloc (sizeof(double) * coderLength); //channel input
+	uint32_t *testInput = malloc(sizeof(uint32_t) * testArrayLength); //generate randomly
+	uint32_t *encoderOutput = malloc(sizeof(uint32_t) * coderArrayLength);
+	uint32_t *estimatedOutput = malloc(sizeof(uint32_t) * testArrayLength); //decoder output
+	memset(testInput,0,sizeof(uint32_t) * testArrayLength);
+	memset(encoderOutput,0,sizeof(uint32_t) * coderArrayLength);
+	memset(estimatedOutput,0,sizeof(uint32_t) * testArrayLength);
 
+	uint8_t *encoderOutputByte = malloc(sizeof(uint8_t) * coderLength);
+	double *modulatedInput = malloc (sizeof(double) * coderLength); //channel input
 	double *channelOutput  = malloc (sizeof(double) * coderLength); //add noise
-	uint32_t *estimatedOutput = malloc(sizeof(uint8_t) * testLength); //decoder output
 
 	t_nrPolar_paramsPtr nrPolar_params = NULL, currentPtr = NULL;
 	nr_polar_init(&nrPolar_params, polarMessageType, testLength, aggregation_level);
+	currentPtr = nr_polar_params(nrPolar_params, polarMessageType, testLength, aggregation_level);
+
 #ifdef DEBUG_DCI_POLAR_PARAMS
+	uint32_t crc;
 	unsigned int poly24c = 0xb2b11700;
+	testInput[0]=0x01189400;
 	printf("testInput: [0]->0x%08x \t [1]->0x%08x \t [2]->0x%08x \t [3]->0x%08x\n",
 			testInput[0], testInput[1], testInput[2], testInput[3]);
-	printf("encOutput: [0]->0x%08x \t [1]->0x%08x \t [2]->0x%08x \t [3]->0x%08x\n",
-			  encoderOutput[0], encoderOutput[1], encoderOutput[2], encoderOutput[3]);
-	testInput[0]=0x01189400;
 	uint8_t testInput2[8];
 	nr_crc_bit2bit_uint32_8_t(testInput, 32, testInput2);
-	printf("testInput2: [0]->%x \t [1]->%x \t [2]->%x \t [3]->%x\n            [4]->%x \t [5]->%x \t [6]->%x \t [7]->%x \t\n",
+	printf("testInput2: [0]->%x \t [1]->%x \t [2]->%x \t [3]->%x\n"
+		   "            [4]->%x \t [5]->%x \t [6]->%x \t [7]->%x\n",
 				testInput2[0], testInput2[1], testInput2[2], testInput2[3],
 				testInput2[4], testInput2[5], testInput2[6], testInput2[7]);
 	printf("crc32: [0]->0x%08x\n",crc24c(testInput2, 32));
 	printf("crc56: [0]->0x%08x\n",crc24c(testInput2, 56));
-	return 0;
-	uint8_t testInput8[4];
-	/*testInput8[0]=0x00;
-	testInput8[1]=0x49;
-	testInput8[2]=0x81;
-	testInput8[3]=0x10;
-	testInput8[4]=0x00;*/
-	testInput8[0]=0xff;
-	testInput8[1]=0xd0;
-	testInput8[2]=0xff;
-	testInput8[3]=0x82;
-	crc = crc24c(testInput8, 31);
-	for (int i=0;i<24;i++) printf("[i]=%d\n",(crc>>i)&1);
-	printf("crc: [0]->0x%08x\n",crc);
-	printf("crcbit: %x\n",crcbit(testInput8, 3, poly24c));
-	return 0;
-	unsigned char test[] = "Thebigredfox";
-
-	for (int i=0;i<8;i++) printf("[i]=%d\n",(test[0]>>i)&1);
-	printf("test[0]=%x\n",test[0]);
-	printf("%s -- sizeof=%d\n",test,sizeof(test));
-	printf("%x\n", crcbit(test, sizeof(test) - 1, poly24c));
-	printf("%x\n", crc24c(test, (sizeof(test) - 1)*8));
-	polarMessageType = 1;
-	testLength = 41;
-	aggregation_level=1;
-	coderLength = 108;
-	nr_polar_init(&nrPolar_params, polarMessageType, testLength, aggregation_level);
-	nr_polar_print_polarParams(nrPolar_params);
-
     crc = crc24c(testInput, testLength)>>8;
     for (int i=0;i<24;i++) printf("[i]=%d\n",(crc>>i)&1);
     printf("crc: [0]->0x%08x\n",crc);
@@ -174,18 +157,30 @@ int main(int argc, char *argv[]) {
     testInput[2+(testLength>>3)] = ((uint8_t*)&crc)[0];
     printf("testInput: [0]->0x%08x \t [1]->0x%08x \t [2]->0x%08x \t [3]->0x%08x\n",
     			testInput[0], testInput[1], testInput[2], testInput[3]);
-	return (0);
-	currentPtr = nr_polar_params(nrPolar_params, polarMessageType, testLength, aggregation_level);
-	polar_encoder(testInput, encoderOutput, currentPtr);
-	printf("AFTER POLAR ENCODING\n");
-	printf("testInput: [0]->0x%08x \t [1]->0x%08x \t [2]->0x%08x \t [3]->0x%08x\n",
-			testInput[0], testInput[1], testInput[2], testInput[3]);
-	printf("encOutput: [0]->0x%08x \t [1]->0x%08x \t [2]->0x%08x \t [3]->0x%08x\n",
-			encoderOutput[0], encoderOutput[1], encoderOutput[2], encoderOutput[3]);
-	return (0);
 #endif
 
-	currentPtr = nr_polar_params(nrPolar_params, polarMessageType, testLength, aggregation_level);
+#ifdef DEBUG_POLAR_TIMING
+	for (SNR = SNRstart; SNR <= SNRstop; SNR += SNRinc) {
+		SNR_lin = pow(10, SNR / 10);
+		for (itr = 1; itr <= iterations; itr++) {
+			for (int j=0; j<ceil(testLength / 32.0); j++) {
+				for(int i=0; i<32; i++) {
+					testInput[j] |= ( ((uint32_t) (rand()%2)) &1);
+					testInput[j]<<=1;
+				}
+			}
+			printf("testInput: [0]->0x%08x \n", testInput[0]);
+			polar_encoder_timing(testInput, encoderOutput, currentPtr, cpu_freq_GHz, logFile);
+		}
+	}
+	fclose(logFile);
+	free(testInput);
+	free(encoderOutput);
+	free(modulatedInput);
+	free(channelOutput);
+	free(estimatedOutput);
+	return (0);
+#endif
 
 	// We assume no a priori knowledge available about the payload.
 	double aPrioriArray[currentPtr->payloadBits];
@@ -195,25 +190,38 @@ int main(int argc, char *argv[]) {
 		SNR_lin = pow(10, SNR/10);
 		for (itr = 1; itr <= iterations; itr++) {
 
-		for(int i=0; i<testLength; i++) testInput[i]=(uint8_t) (rand() % 2);
+			for (int i = 0; i < testArrayLength; i++) {
+				for (int j = 0; j < (sizeof(testInput[0])*8)-1; j++) {
+					testInput[i] |= ( ((uint32_t) (rand()%2)) &1);
+					testInput[i]<<=1;
+				}
+				testInput[i] |= ( ((uint32_t) (rand()%2)) &1);
+			}
+
+			/*printf("testInput: [0]->0x%08x\n", testInput[0]);
+			for (int i=0; i<32; i++)
+				printf("%d\n",(testInput[0]>>i)&1);*/
 
 			start_meas(&timeEncoder);
 			polar_encoder(testInput, encoderOutput, currentPtr);
 			stop_meas(&timeEncoder);
 
+			/*printf("encoderOutput: [0]->0x%08x\n", encoderOutput[0]);
+			printf("encoderOutput: [1]->0x%08x\n", encoderOutput[1]);
+*/
+			//Bit-to-byte:
+			nr_bit2byte_uint32_8_t(encoderOutput, coderLength, encoderOutputByte);
+
 			//BPSK modulation
 			for(int i=0; i<coderLength; i++) {
-			if (encoderOutput[i] == 0)
+				if (encoderOutputByte[i] == 0)
 					modulatedInput[i]=1/sqrt(2);
 				else
 					modulatedInput[i]=(-1)/sqrt(2);
 
 				channelOutput[i] = modulatedInput[i] + (gaussdouble(0.0,1.0) * (1/sqrt(2*SNR_lin)));
-			//printf("%f\n",channelOutput[i]);
 			}
 
-
-
 			start_meas(&timeDecoder);
 			/*decoderState = polar_decoder(channelOutput,
 									 	 estimatedOutput,
@@ -228,15 +236,21 @@ int main(int argc, char *argv[]) {
 												 NR_POLAR_DECODER_PATH_METRIC_APPROXIMATION,
 												 aPrioriArray);
 			stop_meas(&timeDecoder);
+			/*printf("testInput: [0]->0x%08x\n", testInput[0]);
+			printf("estimatedOutput: [0]->0x%08x\n", estimatedOutput[0]);
+*/
 
 			//calculate errors
 			if (decoderState==-1) {
 				blockErrorState=-1;
 				nBitError=-1;
 			} else {
-			for(int i=0; i<testLength; i++){
-				if (estimatedOutput[i]!=testInput[i]) nBitError++;
+				for (int i = 0; i < testArrayLength; i++) {
+					for (int j = 0; j < (sizeof(testInput[0])*8); j++) {
+						if (((estimatedOutput[i]>>j) & 1) != ((testInput[i]>>j) & 1)) nBitError++;
+					}
 				}
+
 				if (nBitError>0) blockErrorState=1;
 			}
 
@@ -273,11 +287,14 @@ int main(int argc, char *argv[]) {
 	print_meas(&timeDecoder,"polar_decoder",NULL,NULL);
 
 	fclose(logFile);
-	//free(testInput);
-	//free(encoderOutput);
+	//Bit
+	free(testInput);
+	free(encoderOutput);
+	free(estimatedOutput);
+	//Byte
+	free(encoderOutputByte);
 	free(modulatedInput);
 	free(channelOutput);
-	free(estimatedOutput);
 
 	return (0);
 }

openair1/PHY/CODING/nrPolar_tools/nr_polar_defs.h

@@ -42,6 +42,8 @@
 #include "PHY/CODING/nrPolar_tools/nr_polar_dci_defs.h"
 #include "PHY/CODING/nrPolar_tools/nr_polar_uci_defs.h"
 #include "PHY/CODING/nrPolar_tools/nr_polar_pbch_defs.h"
+#include "PHY/CODING/coding_defs.h"
+#include "SIMULATION/TOOLS/sim.h"
 
 #define NR_POLAR_DECODER_LISTSIZE 8 //uint8_t
 #define NR_POLAR_DECODER_PATH_METRIC_APPROXIMATION 0 //uint8_t; 0 --> eq. (8a) and (11b), 1 --> eq. (9) and (12)
@@ -111,6 +113,12 @@ void polar_encoder_dci(uint32_t *in,
 					   t_nrPolar_paramsPtr polarParams,
 					   uint16_t n_RNTI);
 
+void polar_encoder_timing(uint32_t *in,
+						  uint32_t *out,
+						  t_nrPolar_paramsPtr polarParams,
+						  double cpuFreqGHz,
+						  FILE* logFile);
+
 int8_t polar_decoder(double *input,
 		 	 	 	 uint8_t *output,
 					 t_nrPolar_paramsPtr polarParams,
@@ -124,6 +132,15 @@ int8_t polar_decoder_aPriori(double *input,
 							 uint8_t pathMetricAppr,
 							 double *aPrioriPayload);
 
+int8_t polar_decoder_aPriori_timing(double *input,
+									uint32_t *output,
+									t_nrPolar_paramsPtr polarParams,
+									uint8_t listSize,
+									uint8_t pathMetricAppr,
+									double *aPrioriPayload,
+									double cpuFreqGHz,
+									FILE* logFile);
+
 void nr_polar_init(t_nrPolar_paramsPtr *polarParams,
 				   int8_t messageType,
 				   uint16_t messageLength,

openair1/PHY/CODING/nrPolar_tools/nr_polar_encoder.c

@@ -30,10 +30,15 @@
  * \warning
 */
 
+//#define DEBUG_POLAR_ENCODER
 //#define DEBUG_POLAR_ENCODER_DCI
+//#define DEBUG_POLAR_ENCODER_TIMING
 
 #include "PHY/CODING/nrPolar_tools/nr_polar_defs.h"
 
+//input  [a_31 a_30 ... a_0]
+//output [f_31 f_30 ... f_0] [f_63 f_62 ... f_32] ...
+
 void polar_encoder(uint32_t *in,
 				   uint32_t *out,
 				   t_nrPolar_paramsPtr polarParams)
@@ -95,6 +100,10 @@ void polar_encoder(uint32_t *in,
 	/*
 	 * Return bits.
 	 */
+#ifdef DEBUG_POLAR_ENCODER
+	for (int i=0; i< polarParams->encoderLength;i++) printf("f[%d]=%d\n", i, polarParams->nr_polar_E[i]);
+#endif
+
 	nr_byte2bit_uint8_32_t(polarParams->nr_polar_E, polarParams->encoderLength, out);
 }
 
@@ -187,3 +196,76 @@ void polar_encoder_dci(uint32_t *in,
 	}
 #endif
 }
+
+void polar_encoder_timing(uint32_t *in,
+						  uint32_t *out,
+						  t_nrPolar_paramsPtr polarParams,
+						  double cpuFreqGHz,
+						  FILE* logFile)
+{
+	//Initiate timing.
+	time_stats_t timeEncoderCRCByte, timeEncoderCRCBit, timeEncoderInterleaver, timeEncoderBitInsertion, timeEncoder1, timeEncoder2, timeEncoderRateMatching, timeEncoderByte2Bit;
+	reset_meas(&timeEncoderCRCByte); reset_meas(&timeEncoderCRCBit); reset_meas(&timeEncoderInterleaver); reset_meas(&timeEncoderBitInsertion); reset_meas(&timeEncoder1); reset_meas(&timeEncoder2); reset_meas(&timeEncoderRateMatching); reset_meas(&timeEncoderByte2Bit);
+	uint16_t n_RNTI=0x0000;
+
+	start_meas(&timeEncoderCRCByte);
+	nr_crc_bit2bit_uint32_8_t(in, polarParams->payloadBits, polarParams->nr_polar_aPrime); //(a to a')
+	polarParams->crcBit = crc24c(polarParams->nr_polar_aPrime, (polarParams->payloadBits+polarParams->crcParityBits)); //Parity bits computation (p)
+	uint8_t arrayInd = ceil(polarParams->payloadBits / 8.0); //(a to b)
+	for (int i=0; i<arrayInd-1; i++)
+		for (int j=0; j<8; j++)
+			polarParams->nr_polar_B[j+(i*8)] = ((polarParams->nr_polar_aPrime[3+i]>>(7-j)) & 1);
+	for (int i=0; i<((polarParams->payloadBits)%8); i++) polarParams->nr_polar_B[i+(arrayInd-1)*8] = ((polarParams->nr_polar_aPrime[3+(arrayInd-1)]>>(7-i)) & 1);
+	for (int i=0; i<8; i++) polarParams->nr_polar_B[polarParams->payloadBits+i] = ((polarParams->crcBit)>>(31-i))&1;
+	for (int i=0; i<16; i++) polarParams->nr_polar_B[polarParams->payloadBits+8+i] =	( (((polarParams->crcBit)>>(23-i))&1) + ((n_RNTI>>(15-i))&1) ) % 2; //Scrambling (b to c)
+	stop_meas(&timeEncoderCRCByte);
+
+
+	start_meas(&timeEncoderCRCBit);
+	nr_bit2byte_uint32_8_t(in, polarParams->payloadBits, polarParams->nr_polar_A);
+	nr_matrix_multiplication_uint8_t_1D_uint8_t_2D(polarParams->nr_polar_A, polarParams->crc_generator_matrix, polarParams->nr_polar_crc, polarParams->payloadBits, polarParams->crcParityBits); //Calculate CRC.
+	for (uint8_t i = 0; i < polarParams->crcParityBits; i++) polarParams->nr_polar_crc[i] = (polarParams->nr_polar_crc[i] % 2);
+	for (uint16_t i = 0; i < polarParams->payloadBits; i++) polarParams->nr_polar_B[i] = polarParams->nr_polar_A[i]; //Attach CRC to the Transport Block. (a to b)
+	for (uint16_t i = polarParams->payloadBits; i < polarParams->K; i++) polarParams->nr_polar_B[i]= polarParams->nr_polar_crc[i-(polarParams->payloadBits)];
+	stop_meas(&timeEncoderCRCBit);
+
+
+	start_meas(&timeEncoderInterleaver); //Interleaving (c to c')
+	nr_polar_interleaver(polarParams->nr_polar_B, polarParams->nr_polar_CPrime, polarParams->interleaving_pattern, polarParams->K);
+	stop_meas(&timeEncoderInterleaver);
+
+
+	start_meas(&timeEncoderBitInsertion); //Bit insertion (c' to u)
+	nr_polar_bit_insertion(polarParams->nr_polar_CPrime, polarParams->nr_polar_U, polarParams->N, polarParams->K, polarParams->Q_I_N, polarParams->Q_PC_N, polarParams->n_pc);
+	stop_meas(&timeEncoderBitInsertion);
+
+
+	start_meas(&timeEncoder1); //Encoding (u to d)
+	nr_matrix_multiplication_uint8_t_1D_uint8_t_2D(polarParams->nr_polar_U, polarParams->G_N, polarParams->nr_polar_D, polarParams->N, polarParams->N);
+	stop_meas(&timeEncoder1);
+
+
+	start_meas(&timeEncoder2);
+	for (uint16_t i = 0; i < polarParams->N; i++) polarParams->nr_polar_D[i] = (polarParams->nr_polar_D[i] % 2);
+	stop_meas(&timeEncoder2);
+
+
+	start_meas(&timeEncoderRateMatching);//Rate matching //Sub-block interleaving (d to y) and Bit selection (y to e)
+	nr_polar_interleaver(polarParams->nr_polar_D, polarParams->nr_polar_E, polarParams->rate_matching_pattern, polarParams->encoderLength);
+	stop_meas(&timeEncoderRateMatching);
+
+
+	start_meas(&timeEncoderByte2Bit); //Return bits.
+	nr_byte2bit_uint8_32_t(polarParams->nr_polar_E, polarParams->encoderLength, out);
+	stop_meas(&timeEncoderByte2Bit);
+
+	fprintf(logFile,",%f,%f,%f,%f,%f,%f,%f,%f\n",
+			(timeEncoderCRCByte.diff_now/(cpuFreqGHz*1000.0)),
+			(timeEncoderCRCBit.diff_now/(cpuFreqGHz*1000.0)),
+			(timeEncoderInterleaver.diff_now/(cpuFreqGHz*1000.0)),
+			(timeEncoderBitInsertion.diff_now/(cpuFreqGHz*1000.0)),
+			(timeEncoder1.diff_now/(cpuFreqGHz*1000.0)),
+			(timeEncoder2.diff_now/(cpuFreqGHz*1000.0)),
+			(timeEncoderRateMatching.diff_now/(cpuFreqGHz*1000.0)),
+			(timeEncoderByte2Bit.diff_now/(cpuFreqGHz*1000.0)));
+}