reactics/symrs.hh

/*
    Copyright (c) 2012-2014
    Artur Meski <meski@ipipan.waw.pl>

    Reuse of the code or its part for any purpose
    without the author's permission is strictly prohibited.
*/

#ifndef RS_SYMRSSTATE_HH
#define RS_SYMRSSTATE_HH

#include <vector>
#include <algorithm>
#include <map>
#include <cassert>
#include "cudd.hh"
#include "types.hh"
#include "macro.hh"
#include "bdd_macro.hh"
#include "rs.hh"
#include "options.hh"
#include "memtime.hh"

using std::cout;
using std::cerr;
using std::endl;
using std::vector;
using std::map;

class SymRS
{
    friend class ModelChecker;
    friend class FormRSCTL;

    RctSys *rs;
    Cudd *cuddMgr;
    Options *opts;

	// Mapping: entity ID -> action/context entity ID
    StateEntityToAction stateToAct;

    BDD *initStates;
	
    vector<BDD> *pv;
    vector<BDD> *pv_succ;
    BDD *pv_E;
    BDD *pv_succ_E;

    vector<BDD> *pv_rs;
    vector<BDD> *pv_rs_succ;
    BDD *pv_rs_E;
    BDD *pv_rs_succ_E;

    vector<BDD> *partTrans;

    BDD *monoTrans;

	// Context automaton
	vector<BDD> *pv_ca;
	vector<BDD> *pv_ca_succ;
	BDD *pv_ca_E;
	BDD *pv_ca_succ_E;
	BDD *tr_ca;

    vector<BDD> *pv_act;
    BDD *pv_act_E;

    unsigned int totalStateVars;
    unsigned int totalReactions;
    unsigned int totalRctSysStateVars;
    unsigned int totalActions;
	unsigned int totalCtxAutStateVars;
	
    BDD encEntity_raw(Entity entity, bool succ) const;
    BDD encEntity(Entity entity) const {
        return encEntity_raw(entity, false);
    }
    BDD encActEntity(Entity entity) const {
		assert(entity < pv_act->size());
        return (*pv_act)[entity];
    }
    BDD encEntitySucc(Entity entity) const {
        return encEntity_raw(entity, true); 
    }
    BDD encEntitiesConj_raw(const Entities &entities, bool succ);
    BDD encEntitiesConj(const Entities &entities) {
        return encEntitiesConj_raw(entities, false);
    }
    BDD encEntitiesConjSucc(const Entities &entities) {
        return encEntitiesConj_raw(entities, true);
    }
    BDD encEntitiesDisj_raw(const Entities &entities, bool succ);
    BDD encEntitiesDisj(const Entities &entities) {
        return encEntitiesDisj_raw(entities, false);
    }
    BDD encEntitiesDisjSucc(const Entities &entities) {
        return encEntitiesDisj_raw(entities, true);
    }
    BDD encStateActEntitiesConj(const Entities &entities);
    BDD encStateActEntitiesDisj(const Entities &entities);

	BDD encActEntitiesConj(const Entities &entities);

    /**
     * @brief Complements an encoding of a given state by negating all the variables that are not set to true
     *
     * @return Returns the encoded state
     */
    BDD compState(const BDD &state) const;

    BDD compContext(const BDD &context) const;

    std::string decodedRctSysStateToStr(const BDD &state);
    void printDecodedRctSysStates(const BDD &states);

	BDD encNoContext(void);

    void initBDDvars(void);
    void encodeTransitions(void);
    void encodeTransitions_old(void);
    void encodeInitStates(void);
	void encodeInitStatesForCtxAut(void);
	void encodeInitStatesNoCtxAut(void);
    void mapStateToAct(void);
    void encode(void);

    int getMappedStateToActID(int stateID) const 
	{ 
		assert(stateID < static_cast<int>(totalRctSysStateVars)); 
		return stateToAct[stateID]; 
	}

	size_t getCtxAutStateEncodingSize(void);
	
public:
    SymRS(RctSys *rs, Options *opts);
	
    vector<BDD> *getEncPV(void) { return pv; }
    vector<BDD> *getEncPVsucc(void) { return pv_succ; }
    BDD *getEncPV_E(void) { return pv_E; }
    BDD *getEncPVsucc_E(void) { return pv_succ_E; }
    BDD *getEncPVact_E(void) { return pv_act_E; }
    vector<BDD> *getEncPartTrans(void) { return partTrans; }
    BDD *getEncMonoTrans(void) { return monoTrans; }
    BDD getEncState(const Entities &entities);
    BDD *getEncInitStates(void) { return initStates; }
    Cudd *getCuddMgr(void) { return cuddMgr; }
    unsigned int getTotalStateVars(void) { return totalStateVars; }
    unsigned int getTotalRctSysStateVars(void) { return totalRctSysStateVars; }
    BDD encEntity(std::string name) const {
        return encEntity(rs->getEntityID(name));
    }
	
    BDD encActStrEntity(std::string name) const;
    BDD getBDDtrue(void) const { return BDD_TRUE; }
    BDD getBDDfalse(void) const { return BDD_FALSE; }
	
    /**
     * @brief Checks if context automaton is used
     *
     * @return True if CA is used
     */
	bool usingContextAutomaton(void) { return rs->ctx_aut != nullptr; }
	
    /**
     * @brief Encodes a context automaton's state
     *
     * @return Returns the encoded state
     */
	BDD encCtxAutState_raw(State state_id, bool succ) const;

	/**
     * @brief Encodes a context automaton's state (as predecessor/non-primed)
     *
     * @return Returns the encoded state
     */
	BDD encCtxAutState(State state_id) const { return encCtxAutState_raw(state_id, false); }
	
    /**
     * @brief Encodes a context automaton's state (as successor/primed)
     *
     * @return Returns the encoded state
     */
	BDD encCtxAutStateSucc(State state_id) const { return encCtxAutState_raw(state_id, true); }
	
    /**
     * @brief Encodes the initial state of context automaton
     *
     * @return Returns the encoded state(s)
     */
	BDD getEncCtxAutInitState(void);
	
    /**
     * @brief Getter for context automaton's state variables (predecessor/non-primed)
     *
     * @return Returns a vector of BDDs
     */
	vector<BDD> *getEncCtxAutPV(void) { return pv_ca; }

    /**
     * @brief Getter for context automaton's successor (primed) state variables
     *
     * @return Returns a vector of BDDs
     */
	vector<BDD> *getEncCtxAutPVsucc(void) { return pv_ca_succ; }

    /**
     * @brief Getter for context automaton's quantification BDD (for non-primed vars)
     *
     * @return Returns a BDD for quantification 
     */
	BDD *getEncCtxAutPV_E(void) { return pv_ca_E; }

    /**
     * @brief Getter for context automaton's quantification BDD (for primed vars)
     *
     * @return Returns a BDD for quantification 
     */
	BDD *getEncCtxAutPVsucc_E(void) { return pv_ca_succ_E; }

    /**
     * @brief Encodes the monolithic transition relation
     */	
	void encodeCtxAutTrans(void);
	
    /**
     * @brief Getter for context automaton's transition relation
     *
     * @return Returns a BDD encoding the transition relation
     */	
	BDD *getEncCtxAutTrans(void) { return tr_ca; }	
};

#endif

/** EOF **/