static bool GenerateFile(string sourceRealFileName, bool generateIrradiance, int destSize, string destRealFileName, List <Vector3> outIrradianceValues, out string error)
{
var tempDirectory = GetTemporaryDirectory();
string arguments;
if (generateIrradiance)
{
arguments = $"--format=hdr --size={destSize} --type=cubemap --ibl-irradiance=\"{tempDirectory}\" \"{sourceRealFileName}\"";
}
else
{
arguments = $"--format=hdr --size={destSize} --type=cubemap -x \"{tempDirectory}\" \"{sourceRealFileName}\"";
}
var process = new Process();
process.StartInfo.FileName = Path.Combine(VirtualFileSystem.Directories.EngineInternal, @"Tools\Filament\cmgen.exe");
process.StartInfo.Arguments = arguments;
process.Start();
process.WaitForExit();
var exitCode = process.ExitCode;
if (exitCode != 0)
{
error = $"cmgen.exe exit code = {exitCode}.";
return(false);
}
var folder = Path.Combine(tempDirectory, Path.GetFileNameWithoutExtension(sourceRealFileName));
int size;
bool need16bit = false;
{
var file = Path.Combine(folder, generateIrradiance ? "i_nx.hdr" : "m0_nx.hdr");
var bytes = File.ReadAllBytes(file);
if (!ImageUtility.LoadFromBuffer(bytes, "hdr", out var data, out var size2, out _, out var format, out _, out _, out error))
{
return(false);
}
size = size2.X;
if (format != PixelFormat.Float32RGB)
{
error = "format != PixelFormat.Float32RGB";
return(false);
}
var image2D = new ImageUtility.Image2D(format, size2, data);
for (int y = 0; y < image2D.Size.Y; y++)
{
for (int x = 0; x < image2D.Size.X; x++)
{
var v = image2D.GetPixel(new Vector2I(x, y));
if (v.X > 1.001f || v.Y >= 1.001f || v.Z >= 1.001f)
{
need16bit = true;
goto end16bit;
}
}
}
end16bit :;
}
var surfaces = new List <DDSImage.Surface>();
for (int face = 0; face < 6; face++)
{
int counter = 0;
int currentSize = size;
while (currentSize > 0)
{
var postfixes = new string[] { "px", "nx", "py", "ny", "pz", "nz" };
string file;
if (generateIrradiance)
{
file = Path.Combine(folder, $"i_{postfixes[ face ]}.hdr");
}
else
{
file = Path.Combine(folder, $"m{counter}_{postfixes[ face ]}.hdr");
}
var bytes = File.ReadAllBytes(file);
if (!ImageUtility.LoadFromBuffer(bytes, "hdr", out var data, out var size2, out _, out var format, out _, out _, out error))
{
return(false);
}
if (format != PixelFormat.Float32RGB)
{
error = "format != PixelFormat.Float32RGB";
return(false);
}
if (size2.X != currentSize)
{
error = "size2.X != currentSize";
return(false);
}
if (need16bit)
{
byte[] newData = new byte[currentSize * currentSize * 4 * 2];
unsafe
{
fixed(byte *pData = data)
{
float *pData2 = (float *)pData;
fixed(byte *pNewData = newData)
{
Half *pNewData2 = (Half *)pNewData;
for (int n = 0; n < currentSize * currentSize; n++)
{
pNewData2[n * 4 + 0] = new Half(pData2[n * 3 + 0]);
pNewData2[n * 4 + 1] = new Half(pData2[n * 3 + 1]);
pNewData2[n * 4 + 2] = new Half(pData2[n * 3 + 2]);
pNewData2[n * 4 + 3] = new Half(1.0f);
}
}
}
}
surfaces.Add(new DDSImage.Surface(new Vector2I(currentSize, currentSize), newData));
}
else
{
byte[] newData = new byte[currentSize * currentSize * 4];
unsafe
{
fixed(byte *pData = data)
{
float *pData2 = (float *)pData;
for (int n = 0; n < currentSize * currentSize; n++)
{
newData[n * 4 + 3] = 255;
newData[n * 4 + 2] = (byte)(MathEx.Saturate(pData2[n * 3 + 0]) * 255.0);
newData[n * 4 + 1] = (byte)(MathEx.Saturate(pData2[n * 3 + 1]) * 255.0);
newData[n * 4 + 0] = (byte)(MathEx.Saturate(pData2[n * 3 + 2]) * 255.0);
}
}
}
surfaces.Add(new DDSImage.Surface(new Vector2I(currentSize, currentSize), newData));
}
counter++;
currentSize /= 2;
if (generateIrradiance)
{
break;
}
}
}
var image = new DDSImage(need16bit ? DDSImage.FormatEnum.R16G16B16A16 : DDSImage.FormatEnum.X8R8G8B8, surfaces.ToArray(), true);
if (!WriteToFile(destRealFileName, image, out error))
{
return(false);
}
if (outIrradianceValues != null)
{
var shFile = Path.Combine(folder, "sh.txt");
var lines = File.ReadAllLines(shFile);
foreach (var line in lines)
{
var index1 = line.IndexOf('(');
var index2 = line.IndexOf(')');
if (index1 != -1 && index2 != -1)
{
var str = line.Substring(index1 + 1, index2 - index1 - 1).Trim();
var strs = str.Split(new char[] { ',' }, StringSplitOptions.RemoveEmptyEntries);
if (strs.Length != 3)
{
error = "Unable to parse \"sh.txt\".";
return(false);
}
var x = double.Parse(strs[0].Trim());
var y = double.Parse(strs[1].Trim());
var z = double.Parse(strs[2].Trim());
outIrradianceValues.Add(new Vector3(x, y, z));
}
}
}
DeleteDirectory(tempDirectory);
error = "";
return(true);
}
public void UpdateCaptureCubemap()
{
if (Mode.Value != ModeEnum.Capture)
{
return;
}
Component_Image texture = null;
Component_Image textureRead = null;
try
{
//create
var resolution = Resolution.Value;
var hdr = HDR.Value;
var size = int.Parse(resolution.ToString().Replace("_", ""));
//!!!!16 бит достаточно, тогда нужно поддержку для Image2D
PixelFormat format = hdr ? PixelFormat.Float32RGBA : PixelFormat.A8R8G8B8;
//PixelFormat format = hdr ? PixelFormat.Float16RGBA : PixelFormat.A8R8G8B8;
texture = ComponentUtility.CreateComponent <Component_Image>(null, true, false);
texture.CreateType = Component_Image.TypeEnum._2D;
texture.CreateSize = new Vector2I(size, size);
texture.CreateMipmaps = false;
texture.CreateFormat = format;
texture.CreateUsage = Component_Image.Usages.RenderTarget;
texture.CreateFSAA = 0;
texture.Enabled = true;
var renderTexture = texture.Result.GetRenderTarget(0, 0);
//!!!!
var viewport = renderTexture.AddViewport(true, false);
//var viewport = renderTexture.AddViewport( false, false );
viewport.Mode = Viewport.ModeEnum.ReflectionProbeCubemap;
viewport.AttachedScene = ParentScene;
textureRead = ComponentUtility.CreateComponent <Component_Image>(null, true, false);
textureRead.CreateType = Component_Image.TypeEnum._2D;
textureRead.CreateSize = new Vector2I(size, size);
textureRead.CreateMipmaps = false;
textureRead.CreateFormat = format;
textureRead.CreateUsage = Component_Image.Usages.ReadBack | Component_Image.Usages.BlitDestination;
textureRead.CreateFSAA = 0;
textureRead.Enabled = true;
//!!!!
textureRead.Result.PrepareNativeObject();
//render
var image2D = new ImageUtility.Image2D(PixelFormat.Float32RGB, new Vector2I(size * 4, size * 3));
var position = Transform.Value.Position;
for (int face = 0; face < 6; face++)
{
Vector3 dir = Vector3.Zero;
Vector3 up = Vector3.Zero;
//flipped
switch (face)
{
case 0: dir = -Vector3.YAxis; up = Vector3.ZAxis; break;
case 1: dir = Vector3.YAxis; up = Vector3.ZAxis; break;
case 2: dir = Vector3.ZAxis; up = -Vector3.XAxis; break;
case 3: dir = -Vector3.ZAxis; up = Vector3.XAxis; break;
case 4: dir = Vector3.XAxis; up = Vector3.ZAxis; break;
case 5: dir = -Vector3.XAxis; up = Vector3.ZAxis; break;
}
//!!!!renderingPipelineOverride
var scene = ParentScene;
var cameraEditor = scene.Mode.Value == Component_Scene.ModeEnum._2D ? scene.CameraEditor2D.Value : scene.CameraEditor.Value;
if (cameraEditor == null)
{
cameraEditor = new Component_Camera();
}
var cameraSettings = new Viewport.CameraSettingsClass(viewport, 1, 90, NearClipPlane.Value, FarClipPlane.Value, position, dir, up, ProjectionType.Perspective, 1, cameraEditor.Exposure, cameraEditor.EmissiveFactor, renderSky: RenderSky);
viewport.Update(true, cameraSettings);
//clear temp data
viewport.RenderingContext.MultiRenderTarget_DestroyAll();
viewport.RenderingContext.DynamicTexture_DestroyAll();
texture.Result.GetRealObject(true).BlitTo(viewport.RenderingContext.CurrentViewNumber, textureRead.Result.GetRealObject(true), 0, 0);
//get data
var totalBytes = PixelFormatUtility.GetNumElemBytes(format) * size * size;
var data = new byte[totalBytes];
unsafe
{
fixed(byte *pBytes = data)
{
var demandedFrame = textureRead.Result.GetRealObject(true).Read((IntPtr)pBytes, 0);
while (RenderingSystem.CallBgfxFrame() < demandedFrame)
{
}
}
}
Vector2I index = Vector2I.Zero;
switch (face)
{
case 1: index = new Vector2I(2, 1); break;
case 0: index = new Vector2I(0, 1); break;
case 2: index = new Vector2I(1, 0); break;
case 3: index = new Vector2I(1, 2); break;
case 4: index = new Vector2I(1, 1); break;
case 5: index = new Vector2I(3, 1); break;
}
//switch( face )
//{
//case 0: index = new Vector2I( 2, 1 ); break;
//case 1: index = new Vector2I( 0, 1 ); break;
//case 2: index = new Vector2I( 1, 0 ); break;
//case 3: index = new Vector2I( 1, 2 ); break;
//case 4: index = new Vector2I( 1, 1 ); break;
//case 5: index = new Vector2I( 3, 1 ); break;
//}
var faceImage = new ImageUtility.Image2D(format, new Vector2I(size, size), data);
//flip by X
var faceImageFlip = new ImageUtility.Image2D(format, new Vector2I(size, size));
for (int y = 0; y < size; y++)
{
for (int x = 0; x < size; x++)
{
var pixel = faceImage.GetPixel(new Vector2I(x, y));
faceImageFlip.SetPixel(new Vector2I(size - 1 - x, y), pixel);
}
}
image2D.Blit(index * size, faceImageFlip);
}
//reset alpha channel
for (int y = 0; y < image2D.Size.Y; y++)
{
for (int x = 0; x < image2D.Size.X; x++)
{
var pixel = image2D.GetPixel(new Vector2I(x, y));
pixel.W = 1.0f;
image2D.SetPixel(new Vector2I(x, y), pixel);
}
}
var destRealFileName = VirtualPathUtility.GetRealPathByVirtual(GetDestVirtualFileName());
if (!Directory.Exists(Path.GetDirectoryName(destRealFileName)))
{
Directory.CreateDirectory(Path.GetDirectoryName(destRealFileName));
}
if (!ImageUtility.Save(destRealFileName, image2D.Data, image2D.Size, 1, image2D.Format, 1, 0, out var error))
{
throw new Exception(error);
}
//delete Gen files
var names = new string[] { "_Gen.info", "_GenEnv.dds", "_GenIrr.dds" };
foreach (var name in names)
{
var fileName2 = VirtualPathUtility.GetRealPathByVirtual(destRealFileName + name);
if (File.Exists(fileName2))
{
File.Delete(fileName2);
}
}
}
catch (Exception e)
{
Log.Error(e.Message);
}
finally
{
texture?.Dispose();
textureRead?.Dispose();
}
processedCubemapNeedUpdate = true;
}
