mirror of
https://github.com/SatDump/SatDump
synced 2026-08-13 17:47:30 -04:00
182 lines
6.3 KiB
C++
182 lines
6.3 KiB
C++
/**********************************************************************
|
|
* This file is used for testing random stuff without running the
|
|
* whole of SatDump, which comes in handy for debugging individual
|
|
* elements before putting them all together in modules...
|
|
*
|
|
* If you are an user, ignore this file which will not be built by
|
|
* default, and if you're a developper in need of doing stuff here...
|
|
* Go ahead!
|
|
*
|
|
* Don't judge the code you might see in there! :)
|
|
**********************************************************************/
|
|
|
|
#include "logger.h"
|
|
#include <fstream>
|
|
#include <vector>
|
|
#include "common/image/image.h"
|
|
#include "common/repack.h"
|
|
|
|
int main(int argc, char *argv[])
|
|
{
|
|
initLogger();
|
|
|
|
std::string msg_native_file = argv[1]; //"/home/alan/Downloads/MSG4-SEVI-MSG15-0100-NA-20220701121243.901000000Z-NA/MSG4-SEVI-MSG15-0100-NA-20220701121243.901000000Z-NA.nat";
|
|
|
|
// We will in the future want to decode from memory, so load it all up in RAM
|
|
std::vector<uint8_t> msg_file;
|
|
{
|
|
std::ifstream input_file(msg_native_file);
|
|
uint8_t byte;
|
|
while (!input_file.eof())
|
|
{
|
|
input_file.read((char *)&byte, 1);
|
|
msg_file.push_back(byte);
|
|
}
|
|
}
|
|
|
|
uint8_t *buf = msg_file.data();
|
|
|
|
// Parse header
|
|
std::string mh_strs[48];
|
|
mh_strs[0] = (char *)buf;
|
|
mh_strs[1] = (char *)buf + 80;
|
|
mh_strs[2] = (char *)buf + 160;
|
|
mh_strs[3] = (char *)buf + 240;
|
|
mh_strs[4] = (char *)buf + 320;
|
|
mh_strs[5] = (char *)buf + 400;
|
|
mh_strs[6] = (char *)buf + 480;
|
|
mh_strs[6] += (char *)buf + 526;
|
|
mh_strs[7] = (char *)buf + 542;
|
|
mh_strs[7] += (char *)buf + 588;
|
|
mh_strs[8] = (char *)buf + 604;
|
|
mh_strs[8] += (char *)buf + 650;
|
|
mh_strs[9] = (char *)buf + 666;
|
|
mh_strs[9] += (char *)buf + 712;
|
|
mh_strs[10] = (char *)buf + 728;
|
|
mh_strs[10] += (char *)buf + 774;
|
|
mh_strs[11] = (char *)buf + 2154;
|
|
mh_strs[12] = (char *)buf + 2234;
|
|
mh_strs[13] = (char *)buf + 2314;
|
|
mh_strs[14] = (char *)buf + 2394;
|
|
mh_strs[15] = (char *)buf + 2474;
|
|
mh_strs[16] = (char *)buf + 2554;
|
|
mh_strs[17] = (char *)buf + 2634;
|
|
mh_strs[18] = (char *)buf + 2714;
|
|
mh_strs[19] = (char *)buf + 2794;
|
|
mh_strs[20] = (char *)buf + 2874;
|
|
mh_strs[21] = (char *)buf + 2954;
|
|
mh_strs[22] = (char *)buf + 3034;
|
|
mh_strs[23] = (char *)buf + 3114;
|
|
mh_strs[24] = (char *)buf + 3194;
|
|
mh_strs[25] = (char *)buf + 3274;
|
|
mh_strs[26] = (char *)buf + 3354;
|
|
mh_strs[27] = (char *)buf + 3434;
|
|
mh_strs[28] = (char *)buf + 3514;
|
|
mh_strs[29] = (char *)buf + 3594;
|
|
mh_strs[30] = (char *)buf + 3674;
|
|
mh_strs[31] = (char *)buf + 3754;
|
|
mh_strs[32] = (char *)buf + 3834;
|
|
mh_strs[33] = (char *)buf + 3914;
|
|
mh_strs[34] = (char *)buf + 3994;
|
|
mh_strs[35] = (char *)buf + 4074;
|
|
mh_strs[36] = (char *)buf + 4154;
|
|
mh_strs[37] = (char *)buf + 4234;
|
|
mh_strs[38] = (char *)buf + 4314;
|
|
mh_strs[39] = (char *)buf + 4394;
|
|
mh_strs[40] = (char *)buf + 4474;
|
|
mh_strs[41] = (char *)buf + 4554;
|
|
mh_strs[42] = (char *)buf + 4634;
|
|
mh_strs[43] = (char *)buf + 4714;
|
|
mh_strs[44] = (char *)buf + 4794;
|
|
mh_strs[45] = (char *)buf + 4874;
|
|
mh_strs[46] = (char *)buf + 4954;
|
|
mh_strs[47] = (char *)buf + 5034;
|
|
|
|
// for (int i = 0; i < 48; i++)
|
|
// logger->debug("L{:d} {:s}", i, mh_strs[i].substr(0, mh_strs[i].size() - 1));
|
|
|
|
// IMG Size
|
|
int vis_ir_x_size, vis_ir_y_size;
|
|
int hrv_x_size, hrv_y_size;
|
|
sscanf(mh_strs[44].c_str(), "NumberLinesVISIR : %d", &vis_ir_y_size);
|
|
sscanf(mh_strs[45].c_str(), "NumberColumnsVISIR : %d", &vis_ir_x_size);
|
|
sscanf(mh_strs[46].c_str(), "NumberLinesHRV : %d", &hrv_y_size);
|
|
sscanf(mh_strs[47].c_str(), "NumberColumnsHRV : %d", &hrv_x_size);
|
|
|
|
logger->warn("VIS/IR Size : {:d}x{:d}", vis_ir_x_size, vis_ir_y_size);
|
|
logger->warn("HRV Size : {:d}x{:d}", hrv_x_size, hrv_y_size);
|
|
|
|
// Other data
|
|
long int headerpos, datapos, trailerpos;
|
|
sscanf(mh_strs[8].c_str(), "%*s : %*d %ld\n", &headerpos);
|
|
sscanf(mh_strs[9].c_str(), "%*s : %*d %ld\n", &datapos);
|
|
sscanf(mh_strs[10].c_str(), "%*s : %*d %ld\n", &trailerpos);
|
|
|
|
char bandsel[16];
|
|
sscanf(mh_strs[39].c_str(), "%*s : %s", bandsel);
|
|
|
|
image::Image<uint16_t> vis_ir_imgs[12];
|
|
for (int channel = 0; channel < 12; channel++)
|
|
{
|
|
if (bandsel[channel] != 'X')
|
|
continue;
|
|
|
|
vis_ir_imgs[channel].init((channel == 11 ? hrv_x_size : vis_ir_x_size), (channel == 11 ? hrv_y_size : vis_ir_y_size), 1);
|
|
}
|
|
|
|
uint16_t repacked_line[81920];
|
|
|
|
// Read line data
|
|
// uint8_t pkt_hdr[48];
|
|
uint8_t *data_ptr = buf + datapos;
|
|
for (int line = 0; line < vis_ir_y_size; line++)
|
|
{
|
|
for (int channel = 0; channel < 12; channel++)
|
|
{
|
|
if (bandsel[channel] != 'X')
|
|
continue;
|
|
|
|
for (int i = 0; i < (channel == 11 ? 3 : 1); i++)
|
|
{
|
|
uint16_t seq_cnt = (data_ptr + 16)[0] << 8 | (data_ptr + 16)[1];
|
|
uint32_t pkt_len = (data_ptr + 18)[0] << 24 | (data_ptr + 18)[1] << 16 | (data_ptr + 18)[2] << 8 | (data_ptr + 18)[3];
|
|
|
|
// logger->info("PKT {:d} CHANNEL {:d} SEQ {:d} LEN {:d}", line, channel, seq_cnt, pkt_len);
|
|
|
|
int datasize = pkt_len - 15 - 27;
|
|
|
|
repackBytesTo10bits(data_ptr + 38 + 27, datasize, repacked_line);
|
|
|
|
for (int x = 0; x < (channel == 11 ? hrv_x_size : vis_ir_x_size); x++)
|
|
vis_ir_imgs[channel][(channel == 11 ? (line * 3 + i) : line) * (channel == 11 ? hrv_x_size : vis_ir_x_size) + x] = repacked_line[x] << 6;
|
|
|
|
data_ptr += 38 + 27 + datasize;
|
|
}
|
|
}
|
|
}
|
|
|
|
// Saving
|
|
for (int channel = 0; channel < 12; channel++)
|
|
{
|
|
if (bandsel[channel] != 'X')
|
|
continue;
|
|
|
|
std::string path = "SEVIRI_" + std::to_string(channel + 1) + ".png";
|
|
|
|
vis_ir_imgs[channel].mirror(true, true);
|
|
|
|
logger->info("Saving " + path);
|
|
vis_ir_imgs[channel].save_png(path);
|
|
}
|
|
|
|
std::string path = "SEVIRI_321.png";
|
|
|
|
image::Image<uint16_t> compo_321(vis_ir_x_size, vis_ir_y_size, 3);
|
|
|
|
compo_321.draw_image(0, vis_ir_imgs[2]);
|
|
compo_321.draw_image(1, vis_ir_imgs[1]);
|
|
compo_321.draw_image(2, vis_ir_imgs[0]);
|
|
|
|
logger->info("Saving " + path);
|
|
compo_321.save_png(path);
|
|
}
|