public CppFileLocation GetPreferredRootFile(CppFileLocation currentFile)
{
using (ReadLockCookie.Create())
{
var sourceFile = currentFile.GetRandomSourceFile(mySolution);
if (myShaderContext.TryGetFromCache(sourceFile.GetLocation(), out var result))
{
var path = myManager.TryGetProjectFile(result.Document)?.Location;
if (path != null)
{
var location = new CppFileLocation(new FileSystemPathWithRange(path, result.Range));
if (!myLocationTracker.Exists(location))
{
myLogger.Trace(
$"Reset context for {sourceFile.GetPersistentIdForLogging()}, because inject is not registered");
myShaderContext.RemoveFromCache(sourceFile.GetLocation());
return(CppFileLocation.EMPTY);
}
return(location);
}
}
}
return(CppFileLocation.EMPTY);
}
public (IEnumerable <CppFileLocation> includeLocations, Dictionary <string, string> defines) GetProgramInfo(CppFileLocation cppFileLocation)
{
// PSI is not commited here
// TODO: cpp global cache should calculate cache only when PSI for file with cpp injects is committed.
var sourceFile = cppFileLocation.GetRandomSourceFile(mySolution);
var range = cppFileLocation.RootRange;
Assertion.Assert(range.IsValid, "range.IsValid");
var buffer = sourceFile.Document.Buffer;
var type = GetShaderProgramType(buffer, range.StartOffset);
var defines = GetDefinedMacroses(ProjectedBuffer.Create(buffer, range), out var isSurface);
var includes = GetIncludes(sourceFile, type, isSurface);
return(includes, defines);
}
public IEnumerable <CppFileLocation> GetIncludes(CppFileLocation cppFileLocation)
{
var sourceFile = cppFileLocation.GetRandomSourceFile(mySolution);
var cgInclude = GetCgIncludeLocation(sourceFile);
if (!cgInclude.Equals(CppFileLocation.EMPTY))
{
yield return(cgInclude);
}
var cgIncludeFolder = CgIncludeDirectoryTracker.GetCgIncludeFolderPath(myUnityVersion);
if (!cgIncludeFolder.ExistsDirectory)
{
yield break;
}
var range = cppFileLocation.RootRange;
Assertion.Assert(range.IsValid, "range.IsValid");
var program = sourceFile.GetDominantPsiFile <ShaderLabLanguage>()?.GetContainingNodeAt <ICgContent>(new TreeOffset(range.StartOffset));
Assertion.Assert(program != null, "program != null");
if (program?.Parent?.FirstChild is ICgProgramBlock)
{
var hlslSupport = cgIncludeFolder.Combine("HLSLSupport.cginc");
if (hlslSupport.ExistsFile)
{
yield return(new CppFileLocation(hlslSupport));
}
var variables = cgIncludeFolder.Combine("UnityShaderVariables.cginc");
if (variables.ExistsFile)
{
yield return(new CppFileLocation(variables));
}
}
}
private ShaderContextData GetContextDataFor(CppFileLocation root)
{
var range = myShaderContextDataPresentationCache.GetRangeForShaderProgram(root.GetRandomSourceFile(mySolution), root.RootRange);
if (!range.HasValue)
{
return(null);
}
var folderFullPath = root.Location.Parent;
var folderPath = folderFullPath.TryMakeRelativeTo(mySolution.SolutionDirectory);
var folderHint = folderPath.FullPath.ShortenTextWithEllipsis(40);
return(new ShaderContextData(root.Location.FullPath, root.Location.Name, folderHint, root.RootRange.StartOffset, root.RootRange.EndOffset, range.Value.startLine + 1));
}
public IEnumerable <CppFileLocation> GetIncludes(CppFileLocation cppFileLocation)
{
// PSI is not commited here
// TODO: cpp global cache should calculate cache only when PSI for file with cpp injects is committed.
var sourceFile = cppFileLocation.GetRandomSourceFile(mySolution);
var range = cppFileLocation.RootRange;
Assertion.Assert(range.IsValid, "range.IsValid");
var buffer = sourceFile.Document.Buffer;
var type = GetShaderProgramType(buffer, range.StartOffset);
var includeType = GetIncludeProgramType(type);
yield return(new CppFileLocation(myCppExternalModule, mySolution.SolutionDirectory.Combine(Utils.ShaderConfigFile)));
if (includeType != InjectedHlslProgramType.Uknown)
{
var includes = GetIncludesLocation(sourceFile, includeType);
foreach (var include in includes)
{
yield return(include);
}
}
var cgIncludeFolder = CgIncludeDirectoryTracker.GetCgIncludeFolderPath(myUnityVersion);
if (!cgIncludeFolder.ExistsDirectory)
{
yield break;
}
if (type == InjectedHlslProgramType.CGProgram)
{
var hlslSupport = cgIncludeFolder.Combine("HLSLSupport.cginc");
if (hlslSupport.ExistsFile)
{
yield return(new CppFileLocation(myCppExternalModule, hlslSupport));
}
var variables = cgIncludeFolder.Combine("UnityShaderVariables.cginc");
if (variables.ExistsFile)
{
yield return(new CppFileLocation(myCppExternalModule, variables));
}
}
var lexer = CppLexer.Create(ProjectedBuffer.Create(buffer, range));
lexer.Start();
while (lexer.TokenType != null)
{
var tokenType = lexer.TokenType;
if (tokenType is CppDirectiveTokenNodeType)
{
lexer.Advance();
var context = lexer.GetTokenText().TrimStart();
if (context.StartsWith("surface"))
{
var unityCG = cgIncludeFolder.Combine("UnityCG.cginc");
if (unityCG.ExistsFile)
{
yield return(new CppFileLocation(myCppExternalModule, unityCG));
}
var lighting = cgIncludeFolder.Combine("Lighting.cginc");
if (lighting.ExistsFile)
{
yield return(new CppFileLocation(myCppExternalModule, lighting));
}
var unityPbsLighting = cgIncludeFolder.Combine("UnityPBSLighting.cginc");
if (unityPbsLighting.ExistsFile)
{
yield return(new CppFileLocation(myCppExternalModule, unityPbsLighting));
}
var autoLight = cgIncludeFolder.Combine("AutoLight.cginc");
if (autoLight.ExistsFile)
{
yield return(new CppFileLocation(myCppExternalModule, autoLight));
}
break;
}
// TODO: optimization if we found vertex/fragment/geometry/hull/domain?
}
lexer.Advance();
}
}
public bool Exists(CppFileLocation cppFileLocation)
{
return(Exists(cppFileLocation.GetRandomSourceFile(mySolution), cppFileLocation));
}
