include/dlvhex2/AnnotatedGroundProgram.h

Go to the documentation of this file.

/* dlvhex -- Answer-Set Programming with external interfaces.
 * Copyright (C) 2005-2007 Roman Schindlauer
 * Copyright (C) 2006-2015 Thomas Krennwallner
 * Copyright (C) 2009-2016 Peter Schüller
 * Copyright (C) 2011-2016 Christoph Redl
 * Copyright (C) 2015-2016 Tobias Kaminski
 * Copyright (C) 2015-2016 Antonius Weinzierl
 *
 * This file is part of dlvhex.
 *
 * dlvhex is free software; you can redistribute it and/or modify it
 * under the terms of the GNU Lesser General Public License as
 * published by the Free Software Foundation; either version 2.1 of
 * the License, or (at your option) any later version.
 *
 * dlvhex is distributed in the hope that it will be useful, but
 * WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with dlvhex; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
 * 02110-1301 USA.
 */

#ifndef _DLVHEX_ANNOTATEDGROUNDPPROGRAM_HPP_
#define _DLVHEX_ANNOTATEDGROUNDPPROGRAM_HPP_

#include "dlvhex2/PlatformDefinitions.h"
#include "dlvhex2/ID.h"
#include "dlvhex2/Error.h"
#include "dlvhex2/Registry.h"
#include "dlvhex2/OrdinaryASPProgram.h"
#include "dlvhex2/PredicateMask.h"
#include "dlvhex2/Nogood.h"

#include <boost/graph/graph_traits.hpp>
#include <boost/graph/adjacency_list.hpp>
#include <boost/shared_ptr.hpp>

#include <boost/shared_ptr.hpp>
#include <boost/unordered_map.hpp>

#include <vector>
#include <list>

DLVHEX_NAMESPACE_BEGIN

class DLVHEX_EXPORT AnnotatedGroundProgram
{

    ProgramCtx* ctx;
    RegistryPtr reg;
    OrdinaryASPProgram groundProgram;
    std::vector<ID> dependencyIDB;
    bool haveGrounding;

    // back-mapping of (ground) external auxiliaries to their nonground external atoms
    std::vector<ID> indexedEatoms;
    std::vector<boost::shared_ptr<ExternalAtomMask> > eaMasks;
    boost::unordered_map<IDAddress, std::vector<ID> > auxToEA;

    SimpleNogoodContainerPtr supportSets;

    InterpretationPtr programMask;

    struct ProgramComponent
    {
        InterpretationConstPtr componentAtoms;
        OrdinaryASPProgram program;
        ProgramComponent(InterpretationConstPtr componentAtoms, OrdinaryASPProgram& program) : componentAtoms(componentAtoms), program(program) {}
        typedef boost::shared_ptr<ProgramComponent> Ptr;
    };
    typedef ProgramComponent::Ptr ProgramComponentPtr;
    typedef boost::adjacency_list<boost::vecS, boost::vecS, boost::bidirectionalS, IDAddress> Graph;
    typedef Graph::vertex_descriptor Node;
    boost::unordered_map<IDAddress, Node> depNodes;
    Graph depGraph;
    std::vector<std::set<IDAddress> > depSCC;
    boost::unordered_map<IDAddress, int> componentOfAtom;
    std::vector<std::pair<IDAddress, IDAddress> > externalEdges;
    std::vector<bool> headCycles;
    InterpretationPtr headCyclicRules;
    std::vector<bool> eCycles;
    std::vector<ProgramComponentPtr> programComponents;

    bool headCyclesTotal;
    bool eCyclesTotal;

    // initialization members
    void createProgramMask();
    void createEAMasks();
    void mapAuxToEAtoms();
    void initialize();
    void computeAtomDependencyGraph();
    void computeAdditionalDependencies();
    void computeStronglyConnectedComponents();
    void computeHeadCycles();
    void computeECycles();
    public:
        AnnotatedGroundProgram();
        AnnotatedGroundProgram(ProgramCtx& ctx, const OrdinaryASPProgram& groundProgram, std::vector<ID> indexedEatoms = std::vector<ID>(), std::vector<ID> dependencyIDB  = std::vector<ID>());

        AnnotatedGroundProgram(ProgramCtx& ctx, std::vector<ID> indexedEatoms);

        void addProgram(const AnnotatedGroundProgram& other);

        const AnnotatedGroundProgram& operator=(const AnnotatedGroundProgram& other);

        void setIndexEAtoms(std::vector<ID> indexedEatoms);

        bool containsHeadCycles(ID ruleID) const;
        int getComponentCount() const;
        const OrdinaryASPProgram& getProgramOfComponent(int compNr) const;
        InterpretationConstPtr getAtomsOfComponent(int compNr) const;
        bool hasHeadCycles(int compNr) const;
        bool hasECycles(int compNr) const;
        bool hasECycles(int compNr, InterpretationConstPtr intr) const;
        bool hasHeadCycles() const;
        bool hasECycles() const;
        bool hasECycles(InterpretationConstPtr intr) const;

        bool mapsAux(IDAddress ida) const;
        typedef std::pair<IDAddress, std::vector<ID> > AuxToExternalAtoms;
        const boost::unordered_map<IDAddress, std::vector<ID> >& getAuxToEA() const;
        const std::vector<ID>& getAuxToEA(IDAddress ida) const;
        const boost::shared_ptr<ExternalAtomMask> getEAMask(int eaIndex) const;

        const OrdinaryASPProgram& getGroundProgram() const;
        const std::vector<ID>& getIndexedEAtoms() const;
        ID getIndexedEAtom(int index) const;
        int getIndexOfEAtom(ID eatomID) const;
        InterpretationConstPtr getProgramMask() const;

        void setCompleteSupportSetsForVerification(SimpleNogoodContainerPtr supportSets);

        bool allowsForVerificationUsingCompleteSupportSets() const;
        SimpleNogoodContainerPtr getCompleteSupportSetsForVerification();

        bool verifyExternalAtomsUsingCompleteSupportSets(int eaIndex, InterpretationConstPtr interpretation, InterpretationConstPtr auxiliariesToVerify) const;
};

DLVHEX_NAMESPACE_END
#endif                           // _DLVHEX_ANNOTATEDGROUNDPPROGRAM_HPP_


// vim:expandtab:ts=4:sw=4:
// mode: C++
// End: