satdump/src-core/dsp/io/file_source.h

#pragma once

#include "common/utils.h"
#include "dsp/block.h"
#include "logger.h"
#include <fstream>

namespace satdump
{
    namespace ndsp
    {
        template <typename T>
        class FileSourceBlock : public Block
        {
        public:
            std::string p_file = "/tmp/afile";
            int p_buffer_size = 8192;

        private:
            int d_buffer_size;

            std::atomic<bool> d_eof;

            std::ifstream file_reader;

            bool work();

        public:
            // TODOREWORK
            std::atomic<uint64_t> d_filesize;
            std::atomic<uint64_t> d_progress;

        public:
            FileSourceBlock();
            ~FileSourceBlock();

            void init()
            {
                d_buffer_size = p_buffer_size;

                file_reader = std::ifstream(p_file, std::ios::binary);
                if (!file_reader.good())
                    logger->error("Invalid file " + p_file);

                d_filesize = getFilesize(p_file);
                d_progress = 0;
                d_eof = !file_reader.good();
            }

            nlohmann::ordered_json get_cfg_list()
            {
                nlohmann::ordered_json p;
                add_param_simple(p, "file", "string");
                p["file"]["disable"] = is_work_running();
                add_param_simple(p, "buffer_size", "int");
                p["buffer_size"]["disable"] = is_work_running();
                return p;
            }

            nlohmann::json get_cfg(std::string key)
            {
                if (key == "file")
                    return p_file;
                else if (key == "buffer_size")
                    return p_buffer_size;
                else
                    throw satdump_exception(key);
            }

            cfg_res_t set_cfg(std::string key, nlohmann::json v)
            {
                if (key == "file")
                    p_file = v;
                else if (key == "buffer_size")
                    p_buffer_size = v;
                else
                    throw satdump_exception(key);
                return RES_OK;
            }
        };
    } // namespace ndsp
} // namespace satdump