CCSDSSecondaryHeaderFactory.h

Go to the documentation of this file.

#ifndef CCSDS_SECONDARY_HEADER_FACTORY_H
#define CCSDS_SECONDARY_HEADER_FACTORY_H
 
#include <memory>
#include <unordered_map>
#include <functional>
#include <string>
#include "CCSDSSecondaryHeaderAbstract.h"  // Base class header
 
namespace CCSDS {
 
class SecondaryHeaderFactory {
public:
  SecondaryHeaderFactory() = default;
 
  using CreatorFunc = std::function<std::shared_ptr<SecondaryHeaderAbstract>()>;
 
  ResultBool registerType(std::shared_ptr<SecondaryHeaderAbstract> header) {
    RET_IF_ERR_MSG(!header, INVALID_HEADER_DATA, "Cannot register, invalid Header provided.");
    m_creators[header->getType()] = std::move(header);
    return true;
  }
 
  std::shared_ptr<SecondaryHeaderAbstract> create(const std::string& type) {
    if (const auto it = m_creators.find(type); it != m_creators.end()) {
      return it->second; // Call the stored creation function
    }
    return nullptr; // Return nullptr if type not found
  }
 
  bool typeIsRegistered(const std::string& type) {
    if (const auto it = m_creators.find(type); it != m_creators.end()) {
      return true; // return true if found
    }
    return false; // Return false if type not found
  }
 
private:
  std::unordered_map<std::string, std::shared_ptr<SecondaryHeaderAbstract> > m_creators;
};
 
} // namespace CCSDS
 
#endif // CCSDS_SECONDARY_HEADER_FACTORY_H