Files
swift-mirror/lib/Serialization/ModuleFile.h
Jordan Rose c8958a232e Replace PointerIntUnion with a PointerUnion-compatible Fixnum type.
Per discussion on swift-dev, we'd prefer to have a "pointer-like" integer
with limited bit width, to reuse all the existing infrastructure for
pointer-like types, rather than invent another new data structure.

Swift SVN r5529
2013-06-08 00:18:19 +00:00

146 lines
4.5 KiB
C++

//===--- ModuleFile.h - Info about a loaded serialized module ---*- C++ -*-===//
//
// This source file is part of the Swift.org open source project
//
// Copyright (c) 2014 - 2015 Apple Inc. and the Swift project authors
// Licensed under Apache License v2.0 with Runtime Library Exception
//
// See http://swift.org/LICENSE.txt for license information
// See http://swift.org/CONTRIBUTORS.txt for the list of Swift project authors
//
//===----------------------------------------------------------------------===//
#ifndef SWIFT_SERIALIZATION_MODULEFILE_H
#define SWIFT_SERIALIZATION_MODULEFILE_H
#include "ModuleFormat.h"
#include "swift/AST/Identifier.h"
#include "swift/AST/Type.h"
#include "swift/Basic/Fixnum.h"
#include "swift/Basic/LLVM.h"
#include "llvm/ADT/ArrayRef.h"
#include "llvm/ADT/DenseMap.h"
#include "llvm/ADT/OwningPtr.h"
#include "llvm/Bitcode/BitstreamReader.h"
namespace llvm {
class BitstreamCursor;
class BitstreamReader;
class MemoryBuffer;
}
namespace swift {
class Decl;
class Module;
class ValueDecl;
/// Describes whether a loaded module can be used.
enum class ModuleStatus {
/// The module is valid.
Valid,
/// The module contains pointers to source files, which should be loaded
/// instead.
///
/// This is a bring-up hack and will eventually go away.
FallBackToTranslationUnit,
/// The module file format is too new to be used by this version of the
/// compiler.
FormatTooNew,
/// The module file is malformed in some way.
Malformed
};
/// A serialized module, along with the tools to access it.
class ModuleFile {
/// A reference back to the AST representation of the module.
Module *ModuleContext;
/// The module file data.
llvm::OwningPtr<llvm::MemoryBuffer> InputFile;
/// The reader attached to InputFile.
llvm::BitstreamReader InputReader;
/// The cursor used to lazily load things from the file.
llvm::BitstreamCursor DeclTypeCursor;
/// Paths to the source files used to build this module.
SmallVector<StringRef, 4> SourcePaths;
/// Decls referenced by this module.
std::vector<PointerUnion<Decl*, serialization::BitOffset>> Decls;
/// Types referenced by this module.
std::vector<PointerUnion<Type, serialization::BitOffset>> Types;
/// All top-level decls in this module.
// FIXME: A single identifier may refer to multiple decls.
llvm::DenseMap<Identifier, serialization::DeclID> TopLevelIDs;
/// FIXME: HACK: an array of the top-level decl IDs.
std::vector<serialization::DeclID> RawTopLevelIDs;
/// Whether this module file can be used.
ModuleStatus Status;
/// Constructs an new module and validates it.
ModuleFile(llvm::OwningPtr<llvm::MemoryBuffer> &&input);
/// Convenience function for module loading.
void error(ModuleStatus issue = ModuleStatus::Malformed) {
assert(issue != ModuleStatus::Valid &&
issue != ModuleStatus::FallBackToTranslationUnit);
Status = issue;
}
/// Returns the decl with the given ID, deserializing it if needed.
Decl *getDecl(serialization::DeclID DID);
/// Returns the type with the current ID, deserializing it if needed.
Type getType(serialization::TypeID TID);
/// Populates TopLevelIDs for name lookup.
void buildTopLevelDeclMap();
public:
/// Loads a module from the given memory buffer.
///
/// \param input A memory buffer containing the serialized module data.
/// The created module takes ownership of the buffer, even if
/// there's an error in loading.
/// \param[out] module The loaded module.
/// \returns Whether the module was successfully loaded, or what went wrong
/// if it was not.
static ModuleStatus load(llvm::OwningPtr<llvm::MemoryBuffer> &&input,
llvm::OwningPtr<ModuleFile> &module) {
module.reset(new ModuleFile(std::move(input)));
return module->getStatus();
}
/// Associates this module file with an AST module.
void associateWithModule(Module *module) {
assert(!ModuleContext && "already associated with an AST module");
ModuleContext = module;
}
/// Checks whether this module can be used.
ModuleStatus getStatus() const { return Status; }
/// Returns paths to the source files that were used to build this module.
ArrayRef<StringRef> getInputSourcePaths() const {
assert(getStatus() == ModuleStatus::Valid ||
getStatus() == ModuleStatus::FallBackToTranslationUnit);
return SourcePaths;
}
/// Searches the module's top-level decls for the given identifier.
void lookupValue(Identifier name, SmallVectorImpl<ValueDecl*> &results);
};
} // end namespace swift
#endif