mirror of
https://github.com/SatDump/SatDump
synced 2026-08-13 17:47:30 -04:00
124 lines
No EOL
4.5 KiB
C++
124 lines
No EOL
4.5 KiB
C++
#include "module_cluster_instruments.h"
|
|
#include <fstream>
|
|
#include "common/ccsds/ccsds_standard/vcdu.h"
|
|
#include "logger.h"
|
|
#include <filesystem>
|
|
#include "imgui/imgui.h"
|
|
#include "common/utils.h"
|
|
#include "common/ccsds/ccsds_standard/demuxer.h"
|
|
#include "products/products.h"
|
|
#include "products/dataset.h"
|
|
#include "common/simple_deframer.h"
|
|
|
|
namespace cluster
|
|
{
|
|
namespace instruments
|
|
{
|
|
CLUSTERInstrumentsDecoderModule::CLUSTERInstrumentsDecoderModule(std::string input_file, std::string output_file_hint, nlohmann::json parameters)
|
|
: ProcessingModule(input_file, output_file_hint, parameters)
|
|
{
|
|
}
|
|
|
|
void CLUSTERInstrumentsDecoderModule::process()
|
|
{
|
|
filesize = getFilesize(d_input_file);
|
|
std::ifstream data_in(d_input_file, std::ios::binary);
|
|
|
|
logger->info("Using input frames " + d_input_file);
|
|
|
|
time_t lastTime = 0;
|
|
uint8_t cadu[1279];
|
|
|
|
// Demuxers
|
|
ccsds::ccsds_standard::Demuxer demuxer_vcid1(1101, false);
|
|
|
|
def::SimpleDeframer wbddeframer(0xFAF334, 24, 8768, 0);
|
|
|
|
|
|
std::ofstream output("file.ccsds");
|
|
|
|
while (!data_in.eof())
|
|
{
|
|
// Read buffer
|
|
data_in.read((char *)&cadu, 1279);
|
|
|
|
// Parse this transport frame
|
|
ccsds::ccsds_standard::VCDU vcdu = ccsds::ccsds_standard::parseVCDU(cadu);
|
|
|
|
// logger->info(pkt.header.apid);
|
|
// logger->info(vcdu.vcid);
|
|
|
|
if (vcdu.vcid == 5) // Parse VCID
|
|
{
|
|
std::vector<ccsds::CCSDSPacket> ccsdsFrames = demuxer_vcid1.work(cadu);
|
|
for (ccsds::CCSDSPacket &pkt : ccsdsFrames)
|
|
{
|
|
std::vector<std::vector<uint8_t>> minorframes = wbddeframer.work(pkt.payload.data(), pkt.payload.size());
|
|
for(auto &frm : minorframes){
|
|
output.write((char *)frm.data(),frm.size());
|
|
}
|
|
}
|
|
}
|
|
|
|
progress = data_in.tellg();
|
|
if (time(NULL) % 10 == 0 && lastTime != time(NULL))
|
|
{
|
|
lastTime = time(NULL);
|
|
logger->info("Progress " + std::to_string(round(((double)progress / (double)filesize) * 1000.0) / 10.0) + "%%");
|
|
}
|
|
}
|
|
|
|
data_in.close();
|
|
|
|
// dataset.save(d_output_file_hint.substr(0, d_output_file_hint.rfind('/')));
|
|
}
|
|
|
|
void CLUSTERInstrumentsDecoderModule::drawUI(bool window)
|
|
{
|
|
ImGui::Begin("Cluster Instruments Decoder", NULL, window ? 0 : NOWINDOW_FLAGS);
|
|
// /
|
|
// /if (ImGui::BeginTable("##clusterinstrumentstable", 3, ImGuiTableFlags_Borders | ImGuiTableFlags_RowBg))
|
|
// /{
|
|
// / ImGui::TableNextRow();
|
|
// / ImGui::TableSetColumnIndex(0);
|
|
// / ImGui::Text("Instrument");
|
|
// / ImGui::TableSetColumnIndex(1);
|
|
// / ImGui::Text("Images / Frames");
|
|
// / ImGui::TableSetColumnIndex(2);
|
|
// / ImGui::Text("Status");
|
|
// /
|
|
// / for (int i = 0; i < 4; i++)
|
|
// / {
|
|
// / ImGui::TableNextRow();
|
|
// / ImGui::TableSetColumnIndex(0);
|
|
// / ImGui::Text("CIPS %d", i + 1);
|
|
// / ImGui::TableSetColumnIndex(1);
|
|
// / ImGui::TextColored(ImColor(0, 255, 0), "%d", (int)cips_readers[i].images.size());
|
|
// / ImGui::TableSetColumnIndex(2);
|
|
// / drawStatus(cips_status[i]);
|
|
// / }
|
|
// /
|
|
// / ImGui::EndTable();
|
|
// /}
|
|
// /
|
|
ImGui::ProgressBar((double)progress / (double)filesize, ImVec2(ImGui::GetWindowWidth() - 10, 20 * ui_scale));
|
|
|
|
ImGui::End();
|
|
}
|
|
|
|
std::string CLUSTERInstrumentsDecoderModule::getID()
|
|
{
|
|
return "cluster_instruments";
|
|
}
|
|
|
|
std::vector<std::string> CLUSTERInstrumentsDecoderModule::getParameters()
|
|
{
|
|
return {};
|
|
}
|
|
|
|
std::shared_ptr<ProcessingModule> CLUSTERInstrumentsDecoderModule::getInstance(std::string input_file, std::string output_file_hint, nlohmann::json parameters)
|
|
{
|
|
return std::make_shared<CLUSTERInstrumentsDecoderModule>(input_file, output_file_hint, parameters);
|
|
}
|
|
} // namespace amsu
|
|
} // namespace metop
|