BootImageGenerator.h

/*
 * File:    BootImageGenerator.h
 *
 * Copyright (c) Freescale Semiconductor, Inc. All rights reserved.
 * See included license file for license details.
 */
#if !defined(_BootImageGenerator_h_)
#define _BootImageGenerator_h_

#include "OutputSection.h"
#include "BootImage.h"
#include "OptionContext.h"

namespace elftosb
{
class BootImageGenerator
{
public:
    BootImageGenerator() {}
    virtual ~BootImageGenerator() {}
    void addOutputSection(OutputSection *section) { m_sections.push_back(section); }
    void setOptionContext(OptionContext *context) { m_options = context; }
    virtual BootImage *generate() = 0;

protected:
    typedef std::vector<OutputSection *> section_vector_t;

    section_vector_t m_sections; 
    OptionContext *m_options;    

    void processVersionOptions(BootImage *image);

    void processDriveTagOption(BootImage *image);
};

}; // namespace elftosb

#endif // _BootImageGenerator_h_