connectivity_cons.h

#include <objscip/objscip.h>
#include "graph.h"

using namespace scip;

class ConnectivityCons : public ObjConshdlr 
{
public:
    ConnectivityCons(
        SCIP* scip, 
        Graph& g_, 
        std::vector<Graph::vertex_descriptor>& master_nodes_, 
        Graph::vertex_descriptor master_node_, 
        std::vector<SCIP_VAR*>& superpixel_vars_ 
        );

    virtual SCIP_DECL_CONSTRANS(scip_trans);
    
    SCIP_DECL_CONSSEPALP(scip_sepalp);

    SCIP_DECL_CONSSEPASOL(scip_sepasol);

    virtual SCIP_DECL_CONSENFOLP(scip_enfolp);

    virtual SCIP_DECL_CONSENFOPS(scip_enfops);

    virtual SCIP_DECL_CONSCHECK(scip_check);
 
    virtual SCIP_DECL_CONSLOCK(scip_lock);

private:
    size_t findComponents(
        SCIP* scip, 
        SCIP_SOL* sol, 
        Graph& subgraph, 
        std::vector<int>& component 
    );

    SCIP_RETCODE sepaConnectivity(
        SCIP* scip,
        SCIP_CONSHDLR* conshdlr,
        SCIP_SOL* sol,
        SCIP_RESULT* result
    );

    Graph& g;
    std::vector<Graph::vertex_descriptor>& master_nodes;
    Graph::vertex_descriptor master_node;
    std::vector<SCIP_VAR*>& superpixel_vars;
};

SCIP_RETCODE SCIPcreateConsConnectivity(
    SCIP* scip, 
    SCIP_CONS** cons, 
    const char* name, 
    SCIP_Bool initial, 
    SCIP_Bool separate, 
    SCIP_Bool enforce, 
    SCIP_Bool check, 
    SCIP_Bool propagate, 
    SCIP_Bool local, 
    SCIP_Bool modifiable, 
    SCIP_Bool dynamic, 
    SCIP_Bool removable 
);