Merged hamming code from testing to main branch.

This commit is contained in:
K 2024-10-03 01:12:32 +05:30
parent a27584fc48
commit 197fc7f126
Signed by: notkshitij
GPG Key ID: C5B8BC7530F8F43F

View File

@ -0,0 +1,103 @@
#include <iostream>
#include <cmath>
#include <vector>
using namespace std;
// Function to calculate the number of parity bits needed
int calculateParityBits(int dataBits) {
int parityBits = 0;
while (pow(2, parityBits) < dataBits + parityBits + 1) {
parityBits++;
}
return parityBits;
}
// Function to encode the data using Hamming code
vector<int> encodeData(vector<int> data) {
int dataBits = data.size();
int parityBits = calculateParityBits(dataBits);
vector<int> encoded(dataBits + parityBits, 0);
// Set the data bits
int j = 0;
for (int i = 0; i < encoded.size(); i++) {
if (i + 1 == pow(2, j)) {
j++;
} else {
encoded[i] = data[i - j];
}
}
// Calculate and set the parity bits
for (int i = 0; i < parityBits; i++) {
int parityBit = pow(2, i);
int sum = 0;
for (int j = parityBit - 1; j < encoded.size(); j += 2 * parityBit) {
for (int k = 0; k < parityBit; k++) {
if (j + k < encoded.size()) {
sum += encoded[j + k];
}
}
}
encoded[parityBit - 1] = sum % 2;
}
return encoded;
}
// Function to check for errors in the encoded data
int checkForErrors(vector<int> encoded) {
int parityBits = calculateParityBits(encoded.size() - calculateParityBits(encoded.size()));
int errorPosition = 0;
for (int i = 0; i < parityBits; i++) {
int parityBit = pow(2, i);
int sum = 0;
for (int j = parityBit - 1; j < encoded.size(); j += 2 * parityBit) {
for (int k = 0; k < parityBit; k++) {
if (j + k < encoded.size()) {
sum += encoded[j + k];
}
}
}
errorPosition += (sum % 2) * parityBit;
}
return errorPosition;
}
int main() {
int dataBits;
cout<<"Enter the number of data bits:\t";
cin >> dataBits;
vector<int> data(dataBits);
cout<<endl<<"NOTE: Make sure the bits are entered in binary format, separated by spaces.\nEg. 1 0 0 1 (for 4 data bits).";
cout<<endl<<"Enter the data bits:\t";
for (int i = 0; i < dataBits; i++) {
cin >> data[i];
}
vector<int> encoded = encodeData(data);
cout<<endl<<"--------------------"<<endl;
cout<<"Encoded bits are:\t";
for (int bit : encoded) {
cout << bit << " ";
}
cout<<endl<<"--------------------"<<endl;
cout<<endl<<"Enter the encoded bits:\t";
vector<int> receivedEncoded(encoded.size());
for (int i = 0; i < encoded.size(); i++) {
cin >> receivedEncoded[i];
}
int errorPosition = checkForErrors(receivedEncoded);
if (errorPosition == 0) {
cout<<"No errors detected."<<endl;
} else {
cout<<"Error detected at position: "<<errorPosition<<endl;
}
return 0;
}