TransitionToActionRewardTransformer.h

Go to the documentation of this file.

#pragma once
 
#include <memory>
#include <string>
#include <vector>
 
#include "storm/models/sparse/Model.h"
#include "storm/storage/BitVector.h"
 
namespace storm::transformer {
 
template<typename ValueType, typename RewardModelType = storm::models::sparse::StandardRewardModel<ValueType>>
struct TransitionToActionRewardTransformerReturnType {
    std::shared_ptr<storm::models::sparse::Model<ValueType, RewardModelType>> model;
    std::vector<uint64_t> originalToNewStateIndices;
};

template<typename ValueType, typename RewardModelType>
TransitionToActionRewardTransformerReturnType<ValueType, RewardModelType> transformTransitionToActionRewards(
    std::shared_ptr<storm::models::sparse::Model<ValueType, RewardModelType>> originalModel, std::vector<std::string> const& relevantRewardModelNames);
 
template<typename ValueType>
TransitionToActionRewardTransformerReturnType<ValueType> transformTransitionToActionRewards(
    std::shared_ptr<storm::models::sparse::Model<ValueType>> originalModel, std::vector<std::string> const& relevantRewardModelNames) {
    return transformTransitionToActionRewards<ValueType, storm::models::sparse::StandardRewardModel<ValueType>>(originalModel, relevantRewardModelNames);
}
 
}  // namespace storm::transformer