#pragma once

#include "pipeline/modules/base/filestream_to_filestream.h"
#include "xrit/processor/xrit_channel_processor.h"
#include "xrit/xrit_file.h"

namespace fy4
{
    namespace lrit
    {
        class FY4LRITDataDecoderModule : public satdump::pipeline::base::FileStreamToFileStreamModule
        {
        protected:
            std::string directory;

            enum CustomFileParams
            {
                IS_ENCRYPTED,
                KEY_INDEX,
            };

            std::map<std::string, std::shared_ptr<satdump::xrit::XRITChannelProcessor>> all_processors;
            std::mutex all_processors_mtx;

#if 0
            std::map<int, uint64_t> decryption_keys;
#endif

            void processLRITFile(satdump::xrit::XRITFile &file);

        public:
            FY4LRITDataDecoderModule(std::string input_file, std::string output_file_hint, nlohmann::json parameters);
            ~FY4LRITDataDecoderModule();
            void process();
            void drawUI(bool window);

        public:
            static std::string getID();
            virtual std::string getIDM() { return getID(); };
            static nlohmann::json getParams() { return {}; } // TODOREWORK
            static std::shared_ptr<ProcessingModule> getInstance(std::string input_file, std::string output_file_hint, nlohmann::json parameters);
        };
    } // namespace lrit
} // namespace fy4