lib/codegen/InstanceInfo.h

Go to the documentation of this file.

//===-- codegen/InstanceInfo.h -------------------------------- -*- C++ -*-===//
//
// This file is distributed under the MIT license. See LICENSE.txt for details.
//
// Copyright (C) 2009, Stephen Wilson
//
//===----------------------------------------------------------------------===//

//===----------------------------------------------------------------------===//
//===----------------------------------------------------------------------===//

#ifndef COMMA_CODEGEN_INSTANCEINFO_HDR_GUARD
#define COMMA_CODEGEN_INSTANCEINFO_HDR_GUARD

#include "comma/ast/Decl.h"

#include "llvm/ADT/DenseMap.h"

namespace comma {

class CodeGen;
class SRInfo;

class InstanceInfo {

public:

    const Domoid *getDefinition() const { return instance->getDefinition(); }
    Domoid *getDefinition() { return instance->getDefinition(); }


    const DomainInstanceDecl *getInstanceDecl() const { return instance; }
    DomainInstanceDecl *getInstanceDecl() { return instance; }

    llvm::StringRef getLinkName() const { return linkName; }

    SRInfo *lookupSRInfo(SubroutineDecl *srDecl) {
        return srInfoTable.lookup(getKeySRDecl(srDecl));
    }

    SRInfo *getSRInfo(SubroutineDecl *srDecl) {
        SRInfo *info = lookupSRInfo(srDecl);
        assert(info && "SRInfo lookup failed!");
        return info;
    }

    void markAsCompiled() { compiledFlag = true; }

    bool isCompiled() const { return compiledFlag; }

private:
    InstanceInfo(CodeGen &CG, DomainInstanceDecl *instance);
    friend class CodeGen;

    DomainInstanceDecl *instance;

    typedef llvm::DenseMap<SubroutineDecl*, SRInfo*> SRInfoMap;
    SRInfoMap srInfoTable;

    std::string linkName;

    bool compiledFlag;            

    static SubroutineDecl *getKeySRDecl(SubroutineDecl *srDecl);
};

} // end comma namespace.

#endif

---