#include "vision.h"
using namespace std;
void construct_network(int H, int W, int K) {
vector<int> Ns;
int totalDiagnols = H + W - 1;
// printf("Total Diagonals: %d\n", totalDiagnols);
vector <vector<int>> diagonals(totalDiagnols), crossDiagonals(totalDiagnols);
for (int i = 0; i < H; i++) {
for (int j = 0; j < W; j++) {
diagonals[i + j].push_back(i * W + j);
crossDiagonals[H - 1 - i + j].push_back(i * W + j);
}
}
vector<int> diagXorIndices, crossDiagXorsIndices;
for (int d = 0; d < totalDiagnols; d++) {
// printf("Diagonal %d: ", d);
// for (int idx : diagonals[d]) {
// printf("%d ", idx);
// }
// printf("\n");
diagXorIndices.push_back(add_xor(diagonals[d]));
}
for (int d = 0; d < totalDiagnols; d++) {
// printf("Cross Diagonal %d: ", d);
// for (int idx : crossDiagonals[d]) {
// printf("%d ", idx);
// }
// printf("\n");
crossDiagXorsIndices.push_back(add_xor(crossDiagonals[d]));
}
vector<int> answers;
for (int i = 0;i < totalDiagnols - K;i++) {
answers.push_back(add_and({ diagXorIndices[i], diagXorIndices[i + K] }));
answers.push_back(add_and({ crossDiagXorsIndices[i], crossDiagXorsIndices[i + K] }));
}
add_or(answers);
}
