You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
58 lines
1.7 KiB
58 lines
1.7 KiB
4 years ago
|
using System.Collections.Generic;
|
||
|
|
using UnityEngine;
|
||
|
|
using UnityEngine.Rendering;
|
||
|
|
|
||
|
|
|
||
|
|
namespace UTJ.FrameCapturer
|
||
|
|
{
|
||
|
|
public class ExrEncoder : MovieEncoder
|
||
|
|
{
|
||
|
|
static readonly string[] s_channelNames = { "R", "G", "B", "A" };
|
||
|
|
fcAPI.fcExrContext m_ctx;
|
||
|
|
fcAPI.fcExrConfig m_config;
|
||
|
|
string m_outPath;
|
||
|
|
int m_frame;
|
||
|
|
|
||
|
|
public override void Release() { m_ctx.Release(); }
|
||
|
|
public override bool IsValid() { return m_ctx; }
|
||
|
|
public override Type type { get { return Type.Exr; } }
|
||
|
|
|
||
|
|
public override void Initialize(object config, string outPath)
|
||
|
|
{
|
||
|
|
if (!fcAPI.fcExrIsSupported())
|
||
|
|
{
|
||
|
|
Debug.LogError("Exr encoder is not available on this platform.");
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
|
||
|
|
m_config = (fcAPI.fcExrConfig)config;
|
||
|
|
m_ctx = fcAPI.fcExrCreateContext(ref m_config);
|
||
|
|
m_outPath = outPath;
|
||
|
|
m_frame = 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
public override void AddVideoFrame(byte[] frame, fcAPI.fcPixelFormat format, double timestamp = -1.0)
|
||
|
|
{
|
||
|
|
if (m_ctx)
|
||
|
|
{
|
||
|
|
string path = m_outPath + "_" + m_frame.ToString("0000") + ".exr";
|
||
|
|
int channels = System.Math.Min(m_config.channels, (int)format & 7);
|
||
|
|
|
||
|
|
fcAPI.fcExrBeginImage(m_ctx, path, m_config.width, m_config.height);
|
||
|
|
for (int i = 0; i < channels; ++i)
|
||
|
|
{
|
||
|
|
fcAPI.fcExrAddLayerPixels(m_ctx, frame, format, i, s_channelNames[i]);
|
||
|
|
}
|
||
|
|
fcAPI.fcExrEndImage(m_ctx);
|
||
|
|
}
|
||
|
|
++m_frame;
|
||
|
|
}
|
||
|
|
|
||
|
|
public override void AddAudioSamples(float[] samples)
|
||
|
|
{
|
||
|
|
// not supported
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|
||
|
|
}
|