您最多选择25个主题
主题必须以字母或数字开头，可以包含连字符 (-)，并且长度不得超过35个字符
118 行
3.1 KiB
118 行
3.1 KiB
/*
|
|
cmake_minimum_required(VERSION 3.28)
|
|
project(codiceps)
|
|
|
|
set(CMAKE_CXX_STANDARD 23)
|
|
|
|
include(FetchContent)
|
|
|
|
FetchContent_Declare(
|
|
raylib
|
|
GIT_REPOSITORY https://github.com/raysan5/raylib.git
|
|
GIT_TAG 5.0
|
|
)
|
|
|
|
FetchContent_MakeAvailable(raylib)
|
|
|
|
add_executable(codiceps codiceps.cpp)
|
|
target_link_libraries(codiceps PUBLIC raylib_static)
|
|
*/
|
|
|
|
#include <ranges>
|
|
#include <iostream>
|
|
#include <fstream>
|
|
#include <sstream>
|
|
#include <vector>
|
|
#include <cassert>
|
|
#include "raylib.h"
|
|
|
|
std::string slurp(const char* filename) {
|
|
std::ifstream file{filename};
|
|
std::stringstream data;
|
|
file >> data.rdbuf();
|
|
return data.str();
|
|
}
|
|
|
|
std::vector<bool> to_bits(std::string str) {
|
|
std::vector<bool> d;
|
|
for(char c : str) {
|
|
d.push_back(c & 0b10000000);
|
|
d.push_back(c & 0b1000000);
|
|
d.push_back(c & 0b100000);
|
|
d.push_back(c & 0b10000);
|
|
d.push_back(c & 0b1000);
|
|
d.push_back(c & 0b100);
|
|
d.push_back(c & 0b10);
|
|
d.push_back(c & 0b1);
|
|
}
|
|
return d;
|
|
}
|
|
|
|
std::string from_bits(std::vector<bool> data) {
|
|
std::string d;
|
|
for(auto composite : data | std::views::chunk(8)) {
|
|
char c = 0;
|
|
for(auto b : composite) {
|
|
c *= 2;
|
|
c += b;
|
|
}
|
|
d.push_back(c);
|
|
}
|
|
return d;
|
|
}
|
|
|
|
void encode(const char* pic, const char* data_source) {
|
|
auto image = LoadImage(pic);
|
|
auto raw_data = slurp(data_source);
|
|
std::vector<bool> raw_bits = to_bits(raw_data);
|
|
assert(raw_bits.size() % 8 == 0);
|
|
assert(raw_bits.size() / 8 == raw_data.size());
|
|
size_t idx = 0;
|
|
for(auto j : std::ranges::iota_view(0, image.height)) {
|
|
for(auto i : std::ranges::iota_view(0, image.width)) {
|
|
auto c = GetImageColor(image, i, j);
|
|
c.r ^= c.r & 1;
|
|
c.g ^= c.g & 1;
|
|
c.b ^= c.b & 1;
|
|
c.r += idx < raw_bits.size() ? raw_bits[idx] : 0; idx++;
|
|
c.g += idx < raw_bits.size() ? raw_bits[idx] : 0; idx++;
|
|
c.b += idx < raw_bits.size() ? raw_bits[idx] : 0; idx++;
|
|
ImageDrawPixel(&image, i, j, c);
|
|
}
|
|
}
|
|
ExportImage(image, pic);
|
|
UnloadImage(image);
|
|
if(raw_bits.size() > idx) {
|
|
std::cout << "\x1b[1;31mWritten bits: " << raw_bits.size() << "/" << idx << std::endl;
|
|
} else {
|
|
std::cout << "\x1b[1;32mWritten bits: " << raw_bits.size() << "/" << idx << std::endl;
|
|
}
|
|
}
|
|
|
|
void decode(const char* pic) {
|
|
auto image = LoadImage(pic);
|
|
std::vector<bool> raw_bits;
|
|
size_t idx = 0;
|
|
for(auto j : std::ranges::iota_view(0, image.height)) {
|
|
for(auto i : std::ranges::iota_view(0, image.width)) {
|
|
auto c = GetImageColor(image, i, j);
|
|
raw_bits.push_back(c.r & 1);
|
|
raw_bits.push_back(c.g & 1);
|
|
raw_bits.push_back(c.b & 1);
|
|
}
|
|
}
|
|
UnloadImage(image);
|
|
std::cout << from_bits(raw_bits).c_str() << std::endl;
|
|
}
|
|
|
|
int main(int argc, char** argv) {
|
|
if(argc < 1) {
|
|
perror("arguments missing, expected: picture then datafile for coding, picture for decoding");
|
|
}
|
|
if(argc == 3) {
|
|
encode(argv[1], argv[2]);
|
|
} else if(argc == 2) {
|
|
decode(argv[1]);
|
|
}
|
|
return 0;
|
|
}
|