ParameterLifter.h

Go to the documentation of this file.

#pragma once
 
#include <memory>
#include <set>
#include <unordered_map>
#include <vector>
 
#include "storm-pars/modelchecker/region/monotonicity/LocalMonotonicityResult.h"
#include "storm-pars/modelchecker/region/monotonicity/MonotonicityResult.h"
#include "storm-pars/utility/parametric.h"
#include "storm/adapters/RationalFunctionAdapter.h"
#include "storm/solver/OptimizationDirection.h"
#include "storm/storage/BitVector.h"
#include "storm/storage/SparseMatrix.h"
 
namespace storm {
 
namespace analysis {
class Order;
}  // namespace analysis
 
namespace storage {
template<typename ParametricType>
class ParameterRegion;
}  // namespace storage
 
namespace transformer {
 
template<typename ParametricType, typename ConstantType>
class ParameterLifter {
   public:
    typedef typename storm::utility::parametric::VariableType<ParametricType>::type VariableType;
    typedef typename storm::utility::parametric::CoefficientType<ParametricType>::type CoefficientType;
    typedef typename storm::analysis::MonotonicityResult<VariableType>::Monotonicity Monotonicity;
 
    ParameterLifter(storm::storage::SparseMatrix<ParametricType> const& pMatrix, std::vector<ParametricType> const& pVector,
                    storm::storage::BitVector const& selectedRows, storm::storage::BitVector const& selectedColumns, bool generateRowLabels = false,
                    bool useMonotonicity = false);
 
    void specifyRegion(storm::storage::ParameterRegion<ParametricType> const& region, storm::solver::OptimizationDirection const& dirForParameters);
 
    void specifyRegion(storm::storage::ParameterRegion<ParametricType> const& region, storm::solver::OptimizationDirection const& dirForParameters,
                       storm::storage::BitVector const& selectedRows);
 
    void specifyRegion(storm::storage::ParameterRegion<ParametricType> const& region, storm::solver::OptimizationDirection const& dirForParameters,
                       std::shared_ptr<storm::analysis::Order> reachabilityOrder,
                       std::shared_ptr<storm::analysis::LocalMonotonicityResult<VariableType>> localMonotonicityResult);
 
    // Returns the resulting matrix. Should only be called AFTER specifying a region
    storm::storage::SparseMatrix<ConstantType> const& getMatrix() const;
 
    // Returns the resulting vector. Should only be called AFTER specifying a region
    std::vector<ConstantType> const& getVector() const;
 
    std::vector<std::set<VariableType>> const& getOccurringVariablesAtState() const;
 
    std::map<VariableType, std::set<uint_fast64_t>> const& getOccuringStatesAtVariable() const;
 
    uint_fast64_t getRowGroupIndex(uint_fast64_t originalState) const;
    uint_fast64_t getOriginalStateNumber(uint_fast64_t newState) const;
    uint_fast64_t getRowGroupSize(uint_fast64_t originalState) const;
    uint_fast64_t getRowGroupCount() const;
 
    /*
     * During initialization, the actual regions are not known. Hence, we consider abstract valuations,
     * where it is only known whether a parameter will be set to either the lower/upper bound of the region or whether this is unspecified
     */
    class AbstractValuation {
       public:
        AbstractValuation() = default;
        AbstractValuation(AbstractValuation const& other) = default;
        bool operator==(AbstractValuation const& other) const;
 
        void addParameterLower(VariableType const& var);
        void addParameterUpper(VariableType const& var);
        void addParameterUnspecified(VariableType const& var);
 
        std::size_t getHashValue() const;
        AbstractValuation getSubValuation(std::set<VariableType> const& pars) const;
        std::set<VariableType> const& getLowerParameters() const;
        std::set<VariableType> const& getUpperParameters() const;
        std::set<VariableType> const& getUnspecifiedParameters() const;
 
        std::vector<storm::utility::parametric::Valuation<ParametricType>> getConcreteValuations(
            storm::storage::ParameterRegion<ParametricType> const& region) const;
 
       private:
        std::set<VariableType> lowerPars, upperPars, unspecifiedPars;
    };
 
    // Returns for each row the abstract valuation for this row
    // Note: the returned vector might be empty if row label generaion was disabled initially
    std::vector<AbstractValuation> const& getRowLabels() const;
 
   private:
    /*
     * We minimize the number of function evaluations by only calling evaluate() once for each unique pair of function and valuation.
     * The result of each evaluation is then written to all positions in the matrix (and the vector) where the corresponding (function,valuation) occurred.
     */
 
    class FunctionValuationCollector {
       public:
        FunctionValuationCollector() = default;
 
        ConstantType& add(ParametricType const& function, AbstractValuation const& valuation);
 
        void evaluateCollectedFunctions(storm::storage::ParameterRegion<ParametricType> const& region,
                                        storm::solver::OptimizationDirection const& dirForUnspecifiedParameters);
 
       private:
        // Stores a function and a valuation. The valuation is stored as an index of the collectedValuations-vector.
        typedef std::pair<ParametricType, AbstractValuation> FunctionValuation;
 
        class FuncValHash {
           public:
            std::size_t operator()(FunctionValuation const& fv) const {
                std::size_t seed = 0;
                carl::hash_add(seed, fv.first);
                carl::hash_add(seed, fv.second.getHashValue());
                return seed;
            }
        };
 
        // Stores the collected functions with the valuations together with a placeholder for the result.
        std::unordered_map<FunctionValuation, ConstantType, FuncValHash> collectedFunctions;
    };
 
    FunctionValuationCollector functionValuationCollector;
 
    // Returns the 2^(variables.size()) vertices of the region
    std::vector<AbstractValuation> getVerticesOfAbstractRegion(std::set<VariableType> const& variables) const;
 
    std::vector<AbstractValuation> rowLabels;
 
    std::vector<uint_fast64_t> oldToNewColumnIndexMapping;  // Mapping from old to new columnIndex used for monotonicity
    std::vector<uint_fast64_t> rowGroupToStateNumber;       // Mapping from new to old columnIndex used for monotonicity
 
    storm::storage::SparseMatrix<ConstantType> matrix;  // The resulting matrix;
    std::vector<std::pair<typename storm::storage::SparseMatrix<ConstantType>::iterator, ConstantType&>>
        matrixAssignment;  // Connection of matrix entries with placeholders
 
    std::vector<ConstantType> vector;                                                                      // The resulting vector
    std::vector<std::pair<typename std::vector<ConstantType>::iterator, ConstantType&>> vectorAssignment;  // Connection of vector entries with placeholders
 
    // Used for monotonicity in sparsedtmcparameterlifter
    std::vector<std::set<VariableType>> occurringVariablesAtState;
    std::map<VariableType, std::set<uint_fast64_t>> occuringStatesAtVariable;
};
 
}  // namespace transformer
}  // namespace storm