#include "message.h"

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

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

    if (!C[30]) {
        A.push_back(30);
    }

    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, C;
    std::vector<int> A;
    int cur = 0;

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

        for (int j = 0; i < 30 && j < A.size(); i ++, j ++) {
            if (!R[cur][A[j]]) {
                A.push_back(i);
            }
        }
    }

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

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

    for (int i = cur + 1; 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;
}