OptimizationArray.hpp

Go to the documentation of this file.

#ifndef OPTIMIZATIONARRAY_HPP_
#define OPTIMIZATIONARRAY_HPP_

#include <string>
#include <map>
#include <stdio.h>


// Forward declaration of class boost::serialization::access
namespace boost {
namespace serialization {
class access;
}//: serialization
}//: access

namespace mic {
namespace neural_nets {
namespace optimization {


template<typename T>
class OptimizationArray {
public:

    OptimizationArray() = default;

    template <typename opT>
    void add( std::string name_, size_t rows_, size_t cols_){
        keys_map[name_] = functions.size();
        functions.push_back ( std::make_shared< opT > ( opT  ( rows_, cols_ ) ) );

    }

    template <typename opT>
    void add( std::string name_, std::shared_ptr<opT> opt_){
        keys_map[name_] = functions.size();
        functions.push_back ( opt_ );
    }

    void clear(){
        keys_map.clear();
        functions.clear();
    }


    std::shared_ptr<mic::neural_nets::optimization::OptimizationFunction<T> > & operator[] ( size_t number_ ) {
        // TODO: throw exception when out of the scope?

        return functions[number_];

    }

    std::shared_ptr<mic::neural_nets::optimization::OptimizationFunction<T> > & operator[] ( char key_ ) {
        // TODO: throw exception when out of the scope?

        return ( *this ) [std::string ( 1, key_ )];

    }

    std::shared_ptr<mic::neural_nets::optimization::OptimizationFunction<T> > & operator[] ( std::string key ) {

        if ( keys_map.find ( key ) == keys_map.end() )
            std::cout << "Warning !!! " <<
                      "::[] - key not found:" << key << std::endl;

        // TODO: throw exception when out of the scope?

        return functions[keys_map[key]];

    }

    friend std::ostream& operator<<(std::ostream& os_, const mic::neural_nets::optimization::OptimizationArray<T> & obj_) {
        // Display name
        for (auto& i: obj_.keys_map) {
            // Display elements.
            os_ << "[" << i.first << "] \n";
//          os_ << (*obj_.functions[i.second]) << std::endl;
        }
        return os_;
    }

    std::map<std::string, size_t> keys() {
        return keys_map;
    }

    std::string size() {
        return functions.size();
    }

protected:
    std::vector<std::shared_ptr< mic::neural_nets::optimization::OptimizationFunction<T> > > functions;


    std::map<std::string, size_t> keys_map;

private:
    // Friend class - required for using boost serialization.
    friend class boost::serialization::access;

};


} /* namespace optimzation */
} /* namespace neural_nets */
} /* namespace mic */


#endif /* OPTIMIZATIONARRAY_HPP_ */