LogonModule.hxx

Go to the documentation of this file.

 
#ifndef _DCC_LOGONMODULE_HXX_
#define _DCC_LOGONMODULE_HXX_
 
#include <map>
#include <vector>
 
#include "dcc/Defs.hxx"
#include "dcc/Logon.hxx"
 
namespace dcc
{
 
template<class Base>
class ParameterizedLogonModule : public LogonHandlerModule
{
public:
    struct LocoInfo : public Base::Storage
    {
        uint8_t flags_ {0};
 
        uint16_t assignedAddress_ {Defs::ADR_INVALID};
 
        uint64_t decoderId_;
    };
 
    std::vector<LocoInfo> locos_;
    std::map<uint64_t, uint16_t> ids_;
 
    unsigned num_locos()
    {
        return locos_.size();
    }
 
    bool is_valid_loco_id(unsigned loco_id)
    {
        return loco_id < num_locos();
    }
 
    uint8_t &loco_flags(unsigned loco_id)
    {
        return locos_[loco_id].flags_;
    }
 
    uint64_t loco_did(unsigned loco_id)
    {
        return locos_[loco_id].decoderId_;
    }
 
    unsigned create_or_lookup_loco(uint64_t decoder_id)
    {
        auto it = ids_.find(decoder_id);
        if (it == ids_.end())
        {
            // create new.
            uint16_t lid = locos_.size();
            locos_.emplace_back();
            locos_[lid].decoderId_ = decoder_id;
            ids_[decoder_id] = lid;
            return lid;
        }
        else
        {
            return it->second;
        }
    }
 
    void run_address_policy(unsigned loco_id, uint16_t desired_address)
    {
 
        // Note: we ignore the desired address and start assigning addresses
        // from 10000 and up.
        locos_[loco_id].assignedAddress_ = nextAddress_++;
    }
 
    uint16_t assigned_address(unsigned loco_id)
    {
        return locos_[loco_id].assignedAddress_;
    }
 
    void assign_complete(unsigned loco_id)
    {
        loco_flags(loco_id) |= LogonHandlerModule::FLAG_COMPLETE;
    }
 
    uint16_t nextAddress_ {(Defs::ADR_MOBILE_LONG << 8) + 10000};
 
}; // class ParameterizedLogonModule
 
class DefaultBase {
public:
    struct Storage {};
};
 
using DefaultLogonModule = ParameterizedLogonModule<DefaultBase>;
 
} // namespace dcc
 
#endif //  _DCC_LOGONMODULE_HXX_