mirror of
https://github.com/SatDump/SatDump
synced 2026-08-13 17:47:30 -04:00
85 lines
2.7 KiB
C++
85 lines
2.7 KiB
C++
|
#include "reverse_bits.h"
|
||
|
|
#include "common/codings/differential/nrzm.h"
|
||
|
|
#include "core/style.h"
|
||
|
|
#include "imgui/imgui.h"
|
||
|
|
#include "imgui/imgui_stdlib.h"
|
||
|
|
#include "logger.h"
|
||
|
|
#include <cstdint>
|
||
|
|
#include <fstream>
|
||
|
|
|
||
|
|
namespace satdump
|
||
|
|
{
|
||
|
|
std::string ReverseBitsTool::getName() { return "Reverse Bits"; }
|
||
|
|
|
||
|
|
void ReverseBitsTool::renderMenu(std::shared_ptr<BitContainer> &container, bool is_busy)
|
||
|
|
{
|
||
|
|
if (is_busy)
|
||
|
|
style::beginDisabled();
|
||
|
|
|
||
|
|
ImGui::InputInt("Chunk Size", &chunk_size);
|
||
|
|
|
||
|
|
if (ImGui::Button("Reverse"))
|
||
|
|
should_process = true;
|
||
|
|
|
||
|
|
if (is_busy)
|
||
|
|
style::endDisabled();
|
||
|
|
}
|
||
|
|
|
||
|
|
bool ReverseBitsTool::needToProcess() { return should_process; }
|
||
|
|
|
||
|
|
void ReverseBitsTool::setProcessed() { should_process = false; }
|
||
|
|
|
||
|
|
void ReverseBitsTool::process(std::shared_ptr<BitContainer> &container, float &process_progress)
|
||
|
|
{
|
||
|
|
uint8_t *ptr = container->get_ptr();
|
||
|
|
size_t size = container->get_ptr_size();
|
||
|
|
|
||
|
|
char name[1000];
|
||
|
|
tmpnam(name);
|
||
|
|
std::string tmpfile = name;
|
||
|
|
std::ofstream file_out(tmpfile, std::ios::binary);
|
||
|
|
|
||
|
|
{
|
||
|
|
uint64_t buffer = 0;
|
||
|
|
uint8_t shifter = 0;
|
||
|
|
int in_shifter = 0;
|
||
|
|
for (size_t i = 0; i < size * 8;)
|
||
|
|
{
|
||
|
|
buffer = 0;
|
||
|
|
|
||
|
|
// Read x bits
|
||
|
|
for (size_t ii = 0; (ii < chunk_size) && (i < size * 8); i++, ii++)
|
||
|
|
buffer = buffer << 1 | ((ptr[i / 8] >> (7 - (i % 8))) & 1);
|
||
|
|
|
||
|
|
// Write back inverted
|
||
|
|
for (int ii = 0; ii < chunk_size; ii++)
|
||
|
|
{
|
||
|
|
shifter = shifter << 1 | ((buffer >> ii) & 1);
|
||
|
|
in_shifter++;
|
||
|
|
if (in_shifter == 8)
|
||
|
|
{
|
||
|
|
file_out.write((char *)&shifter, 1);
|
||
|
|
in_shifter = 0;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
process_progress = (float)i / float(size * 8);
|
||
|
|
}
|
||
|
|
|
||
|
|
if (in_shifter > 0)
|
||
|
|
file_out.write((char *)&shifter, 1);
|
||
|
|
}
|
||
|
|
|
||
|
|
file_out.close();
|
||
|
|
std::shared_ptr<satdump::BitContainer> newbitc = std::make_shared<satdump::BitContainer>("Reversed " + std::to_string(chunk_size) + " Bits", tmpfile, container->frames);
|
||
|
|
newbitc->d_display_bits = container->d_display_bits;
|
||
|
|
newbitc->d_bitperiod = container->d_bitperiod;
|
||
|
|
newbitc->init_display();
|
||
|
|
newbitc->d_is_temporary = true;
|
||
|
|
|
||
|
|
if (container->bitview != nullptr)
|
||
|
|
((BitViewHandler *)container->bitview)->addSubHandler(std::make_shared<BitViewHandler>(newbitc));
|
||
|
|
else
|
||
|
|
logger->error("Can't add container!");
|
||
|
|
}
|
||
|
|
} // namespace satdump
|