@NotNull
private MutablePackageFragmentDescriptor getOrCreatePackageFragmentForFile(
@NotNull JetFile file) {
JetPackageDirective packageDirective = file.getPackageDirective();
assert packageDirective != null : "scripts are not supported";
MutablePackageFragmentDescriptor fragment =
packageFragmentProvider.getOrCreateFragment(packageDirective.getFqName());
ModuleDescriptor module = packageFragmentProvider.getModule();
DescriptorResolver.resolvePackageHeader(
packageDirective, module, TypeHierarchyResolver.this.trace);
trace.record(BindingContext.FILE_TO_PACKAGE_FRAGMENT, file, fragment);
// Register files corresponding to this package
// The trace currently does not support bi-di multimaps that would handle this task nicer
FqName fqName = fragment.getFqName();
Collection<JetFile> files = trace.get(PACKAGE_TO_FILES, fqName);
if (files == null) {
files = Sets.newIdentityHashSet();
}
files.add(file);
trace.record(BindingContext.PACKAGE_TO_FILES, fqName, files);
return fragment;
}
@Override
public void visitJetFile(@NotNull JetFile file) {
MutablePackageFragmentDescriptor packageFragment = getOrCreatePackageFragmentForFile(file);
c.getPackageFragments().put(file, packageFragment);
c.addFile(file);
PackageViewDescriptor packageView =
packageFragment.getContainingDeclaration().getPackage(packageFragment.getFqName());
ChainedScope rootPlusPackageScope =
new ChainedScope(
packageView, "Root scope for " + file, packageView.getMemberScope(), outerScope);
WriteThroughScope packageScope =
new WriteThroughScope(
rootPlusPackageScope,
packageFragment.getMemberScope(),
new TraceBasedRedeclarationHandler(trace),
"package in file " + file.getName());
packageScope.changeLockLevel(WritableScope.LockLevel.BOTH);
c.getFileScopes().put(file, packageScope);
if (file.isScript()) {
scriptHeaderResolver.processScriptHierarchy(c, file.getScript(), packageScope);
}
prepareForDeferredCall(packageScope, packageFragment, file);
}