AVR Z-LINK®

createhex.cpp

Go to the documentation of this file.

// This file has been prepared for Doxygen automatic documentation generation.
#include "hexfile.h"
#include <sstream>
#include <iomanip>
#include <string>
#include <stdexcept>
#include <vector>
#include <fstream>
#include <algorithm>



class ParameterParser
{
private:
        std::string m_filename; 
        std::vector< byte > m_MACAddress; 

        std::vector< byte > ParseHexString( const std::string & hexString ) const;
        std::string CreateHexString( const std::vector< byte > & byteVector ) const;

public:
        ParameterParser( size_t argc, const char * * argv );
        ~ParameterParser() {}

        //
        // Query methods.
        // 
        const std::string & GetFilename() const { return m_filename; }
        const std::vector< byte > & GetMACAddress() const { return m_MACAddress; }

        std::string GetMACAddressString() const { return CreateHexString( GetMACAddress() ); }
};



std::vector< byte > ParameterParser::ParseHexString( const std::string & hexString ) const
{
        // String must be even number of hex digits.
        if( hexString.size() % 2 != 0 ) {
                throw std::runtime_error( "Hex string does not have even character count" );
        }

        //
        // Parse hex digit pairs and insert into byte vector.
        // 
        std::vector< byte > bytes;
        for( size_t idx = 0; idx < hexString.size(); idx += 2 ) {
                std::istringstream iss( hexString.substr( idx, 2 ) );
                byte data;
                iss >> std::hex >> data;
                if( !iss ) {
                        throw std::runtime_error( "Error parsing hex string \"" + hexString + "\"" );
                }
                bytes.push_back( data );
        }
        return bytes;
}



std::string ParameterParser::CreateHexString( const std::vector< byte > & byteVector ) const
{
        std::ostringstream oss;
        oss << std::setfill('0') << std::hex;

        // Add all byte last byte with spaces in between.
        for( size_t idx = 0; idx < byteVector.size() - 1; ++idx ) {
                oss << std::setw(2) << byteVector.at( idx ) << ' ';
        }

        // Add last byte.
        if( byteVector.size() > 0 ) {
                oss << std::setw(2) << byteVector.at( byteVector.size()-1 );
        }

        return oss.str();
}



ParameterParser::ParameterParser( size_t argc, const char * * argv )
{
        // Check that argument count is valid.
        if( argc != 3 ) {
                std::cout << "Parameters:\n"
                        "  filename     - Filename for resulting Intel HEX file\n"
                        "  macaddress   - Hex representation of the MAC address (32 hex digits)\n"
                        << std::endl;
                throw std::runtime_error( "Not exactly 2 parameters" );
        }

        //
        // Create strings or string streams from argument character arrays.
        // 
        std::string arg1( argv[1] ); // Filename
        std::string arg2( argv[2] ); // MAC Address

        //
        // Copy string arguments and parse non-string arguments.
        // 
        m_filename = arg1;
        
        m_MACAddress = ParseHexString( arg2 );
        
        //
        // Check argument validity.
        // 
        if( m_MACAddress.size() != 16 ) {
                throw std::runtime_error( "MAC address has invalid length" );
        }
}



void CreateHEXFile( const ParameterParser & parser )
{
        HEXFile hexData;
        std::ofstream outFile( parser.GetFilename().c_str() );

        // Prepare counter value bytes.
        HEXRecord MACAddress;
        std::vector< byte > MACAddressReversed = parser.GetMACAddress();
        std::reverse( MACAddressReversed.begin(), MACAddressReversed.end() );
        MACAddress << MACAddressReversed;

        // Add records to HEX file.
        hexData << MACAddress;

        // Write to disk.
        outFile << hexData;
}



int main( size_t argc, const char * * argv )
{
        try {
                const std::string endian( " (Byte order reversed in HEX file)" );
                ParameterParser parser( argc, argv );
                std::cout << "Filename      : " << parser.GetFilename() << std::endl
                        << "MAC address   : " << parser.GetMACAddressString() << endian << std::endl;

                CreateHEXFile( parser );
                std::cout << std::endl << parser.GetFilename() << " created!" << std::endl;
        
        } catch( const std::exception & e ) {
                std::cout << "Error: " << e.what() << std::endl;
        } catch(...) {
                std::cout << "Unknown error" << std::endl;
        }
}