ParamOptimizer.h

Go to the documentation of this file.

#ifndef PARAM_OPTIMIZER_H
#define PARAM_OPTIMIZER_H
 
#include <vector>
#include <unordered_map>
#include <unordered_set>
#include <variant>
#include <memory>
#include <stdexcept>
#include <string>
#include <algorithm>
 
#include "BlockAccessor.h"
#include "Block.h"
#include "Parameter.h"
 
struct BAKeyHash {
    std::size_t operator()(const std::pair<std::string,std::string>& k) const noexcept {
        std::size_t h1 = std::hash<std::string>()(k.first);
        std::size_t h2 = std::hash<std::string>()(k.second);
        return h1 ^ (h2 + 0x9e3779b97f4a7c15ULL + (h1<<6) + (h1>>2));
    }
};
 
class ParamOptimizer {
public:
    explicit ParamOptimizer(std::shared_ptr<BlockAccessor> scope);
 
    explicit ParamOptimizer(std::vector<std::shared_ptr<BlockAccessor>> scopes);
 
    void set_value(const BlockName& block, const LhaID& id, scalar_t v);
 
    void set_param(const BlockName& block, const LhaID& id, std::shared_ptr<Parameter> p);
 
    void remove(const BlockName& block, const LhaID& id);
 
    void commit(bool coalesce = true);
 
    void clear();
 
private:
    struct OpSetValue { BlockName block; LhaID id; scalar_t value; };
 
    struct OpSetParam { BlockName block; LhaID id; std::shared_ptr<Parameter> param; };
 
    struct OpRemove   { BlockName block; LhaID id; };
 
    using Op = std::variant<OpSetValue, OpSetParam, OpRemove>;
 
    std::vector<Op> coalesce_ops_() const;
 
    std::shared_ptr<Block> find_block_(const BlockName& name) const;
 
    void freeze_all_();
 
    void unfreeze_all_();
 
    std::vector<std::shared_ptr<BlockAccessor>> scopes_;
 
    std::vector<Op> ops_;
};
 
#endif