Program Listing for File onlineencoder.hpp

↰ Return to documentation for file (processors/onlineencoder/onlineencoder.hpp)

// ---------------------------------------------------------------------
// This file is part of falcon-core.
//
// Copyright (C) 2021-now Neuro-Electronics Research Flanders
//
// Falcon-server is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// Falcon-server is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with falcon-core. If not, see <http://www.gnu.org/licenses/>.
// ---------------------------------------------------------------------
#pragma once

#include "iprocessor.hpp"
#include "columnsdata/columnsdata.hpp"
#include "compressed_decoder_custom/compressed_decoder_custom.hpp"


class OnlineEncoder : public IProcessor{


    // CONSTRUCTOR and OVERLOADED METHODS
   public:
    OnlineEncoder();
    void Configure(const GlobalContext &context) override;
    void CreatePorts() override;
    void CompleteStreamInfo() override;
    void Process(ProcessingContext &context) override;
    void Postprocess(ProcessingContext &context) override;

    // VARIABLES
   protected:
    unsigned int slot_;
    SlotType nslots_;
    std::string path_;

    int MAX_NCHANNELS = 385;
    // DATA PORTS
   protected:
    PortIn<ColumnsType<double>> *  data_in_port_;
    FollowerState<bool> *to_encode_;

    BroadcasterState<std::map<std::string, std::shared_ptr<PoissonLikelihood>>*> *shared_likelihoods_ ;
    BroadcasterState<std::shared_ptr<Decoder>*> *shared_decoder_ ;
    std::shared_ptr<Decoder> decoder_;
    std::map<std::string, std::shared_ptr<PoissonLikelihood>> likelihoods_;

    // OPTIONS
   protected:
    options::Double update_frequency_{1};
    options::String model_path_{};
    options::Bool save_model_{true};
    options::Bool training_{false};
};