mvc_env.pxd


from libcpp.vector cimport vector
from libcpp.set cimport set
from libcpp.memory cimport shared_ptr
from libcpp cimport bool
from graph cimport Graph

cdef extern from "./src/lib/mvc_env.h":
    cdef cppclass MvcEnv:
        MvcEnv(double _norm)
        void s0(shared_ptr[Graph] _g)except+
        double step(int a)except+
        void stepWithoutReward(int a)except+
        int randomAction()except+
        int betweenAction()except+
        bool isTerminal()except+
        # double getReward(double oldCcNum)except+
        double getReward(int a)except+
        double getMaxConnectedNodesNum()except+
        double norm
        double CcNum
        shared_ptr[Graph] graph
        vector[vector[int]]  state_seq
        vector[int] act_seq
        vector[int] action_list
        vector[double] reward_seq
        vector[double] sum_rewards
        int numCoveredEdges
        set[int] covered_set
        vector[int] avail_list