#include "message.h"

void send_message(std::vector<bool> M, std::vector<bool> C) {
    std::vector<int> A;

    for (int i = 0; i < 30; i ++) {
        send_packet(std::vector<bool>(31, C[i]));
        if (!C[i]) {
            A.push_back(i);
        }
    }

    std::vector<bool> P(31);
    for (int i = 0; i < 4; i ++) {
        P[A[i]] = M.size() >> i & 1;
    }
    send_packet(P);

    for (int i = 0; i < M.size(); i += 16) {
        for (int j = i; j < i + 16 && j < M.size(); j ++) {
            P[A[j - i]] = M[j];
        }
        send_packet(P);
    }
}

std::vector<bool> receive_message(std::vector<std::vector<bool>> R) {
    std::vector<bool> M;
    std::vector<int> A;

    for (int i = 0; i < 30; i ++) {
        int cnt = 0;
        for (int x : R[i]) {
            cnt += x;
        }
        if (cnt << 1 < 31) {
            A.push_back(i);
        }
    }

    if (A.size() < 16) {
        A.push_back(30);
    }

    int res = 0;
    for (int i = 0; i < 4; i ++) {
        res |= R[30][A[i]] << i;
    }
    if (!res) {
        res = 16;
    }

    for (int i = 31; i + 1 < R.size(); i ++) {
        for (int j = 0; j < 16; j ++) {
            M.push_back(R[i][A[j]]);
        }
    }
    for (int i = 0; i < res; i ++) {
        M.push_back(R.back()[A[i]]);
    }

    return M;
}