/// <summary>
///
/// </summary>
/// <param name="desc"></param>
public void Initialize(InitDesc desc)
{
TextureBuffer.BufferDesc buffer_desc = new TextureBuffer.BufferDesc {
width = desc.width,
height = desc.height,
depth = desc.depth,
format = desc.format,
bindFlag = desc.bindFlag,
optionFlags = desc.optionFlags,
mips = desc.mips,
initData = desc.initData,
initStream = desc.initStream,
};
buffer_ = new TextureBuffer(buffer_desc);
}
/// <summary>
/// 生成
/// </summary>
/// <param name="desc"></param>
public ComputeShader(InitDesc desc)
{
var device = GraphicsCore.D3D11Device;
var inc = new ShaderIncludeHandler(System.IO.Path.GetDirectoryName(desc.file_name));
using (var bytecode = ShaderBytecode.CompileFromFile(desc.file_name, desc.main, "cs_5_0", ShaderFlags.None, EffectFlags.None, desc.macro, inc)) {
shader_ = new D3DComputeShader(device, bytecode);
using (ShaderReflection reflect = new ShaderReflection(bytecode)) {
// コンスタントバッファ解決
cbInst_ = reflect.CreateConstantBufferInstance("CB"); // 現状名前固定
if (cbInst_ != null)
{
BufferDescription buffer_desc = new BufferDescription();
buffer_desc.Usage = ResourceUsage.Default;
buffer_desc.CpuAccessFlags = CpuAccessFlags.None;
buffer_desc.BindFlags = BindFlags.ConstantBuffer;
buffer_desc.SizeInBytes = cbInst_.BufferSize;
constantBuffer_ = new D3D11Buffer(GraphicsCore.D3D11Device, buffer_desc);
}
}
}
}
public override void InitParams(Initializer initializer = null, NDArrayDict arg_params = null,
NDArrayDict aux_params = null, bool allow_missing = false, bool force_init = false,
bool allow_extra = false)
{
if (ParamsInitialized && !force_init)
{
Logger.Warning("Parameters already initialized and force_init=False. init_params call ignored.");
return;
}
if (!Binded)
{
throw new Exception("call bind before initializing the parameters");
}
void impl(InitDesc name, ref NDArray arr, NDArrayDict cache)
{
if (cache != null)
{
NDArray cache_arr = null;
if (cache.Contains(name.Name))
{
cache_arr = cache[name.Name];
if (cache_arr != arr)
{
cache_arr.CopyTo(arr);
}
}
else
{
if (!allow_missing)
{
throw new Exception($"{name.Name} is not presented");
}
if (initializer != null)
{
initializer.InitWeight(name.Name, ref arr);
}
}
}
}
var attr = Symbol.AttrDict();
foreach (var name in _arg_params.Keys)
{
var arr = _arg_params[name];
var desc = new InitDesc(name, attr.ContainsKey(name) ? attr[name] : null);
impl(desc, ref arr, arg_params);
}
foreach (var name in _aux_params.Keys)
{
var arr = _aux_params[name];
var desc = new InitDesc(name, attr.ContainsKey(name) ? attr[name] : null);
impl(desc, ref arr, aux_params);
}
ParamsInitialized = true;
_params_dirty = false;
_exec_group.SetParams(_arg_params, _aux_params, allow_extra);
}
/// <summary>
/// パラメータから初期化
/// </summary>
/// <param name="desc"></param>
public Texture(InitDesc desc)
{
Initialize(desc);
}
/// <summary>
/// 初期化
/// </summary>
/// <param name="data"></param>
public void Initialize(InitDesc desc)
{
var device = GraphicsCore.D3D11Device;
shaderId_ = desc.id;
name_ = desc.name;
if (desc.file_name != null)
{
// ファイルから生成
// バイナリキャッシング
string cacheDir = System.IO.Path.GetDirectoryName(desc.file_name) + "/cache";
if (!System.IO.Directory.Exists(cacheDir))
{
System.IO.Directory.CreateDirectory(cacheDir);
}
string cacheFile = cacheDir + "/" + desc.name + "_" + desc.vs_main + "_" + desc.ps_main;
string vsCacheFile = cacheFile + ".vsc";
string psCacheFile = cacheFile + ".psc";
// 頂点シェーダキャッシュ読み取り
ShaderBytecode vsBytecode = null;
if (System.IO.File.Exists(vsCacheFile))
{
DateTime cacheTime = System.IO.File.GetLastWriteTime(vsCacheFile);
DateTime editTime = System.IO.File.GetLastWriteTime(desc.file_name);
if (cacheTime > editTime)
{
System.IO.FileStream stream = new System.IO.FileStream(vsCacheFile, FileMode.Open);
var bytes = new byte[stream.Length];
stream.Read(bytes, 0, (int)stream.Length);
vsBytecode = new ShaderBytecode(new SlimDX.DataStream(bytes, true, false));
}
}
// キャッシュになければコンパイルして生成
ShaderIncludeHandler inc = null;
if (vsBytecode == null)
{
inc = new ShaderIncludeHandler(System.IO.Path.GetDirectoryName(desc.file_name));
vsBytecode = ShaderBytecode.CompileFromFile(desc.file_name, desc.vs_main, "vs_" + desc.profile, ShaderFlags.None, EffectFlags.None, desc.macro, inc);
// キャッシュファイル保存
System.IO.FileStream stream = new System.IO.FileStream(vsCacheFile, FileMode.Create);
vsBytecode.Data.CopyTo(stream);
stream.Close();
}
using (vsBytecode) {
vertexShader_ = new VertexShader(device, vsBytecode);
using (ShaderReflection reflect = new ShaderReflection(vsBytecode)) {
vertexAttr_ = reflect.GetVertexLayoutAttribute();
int layout_id = Shader.GetVertexLayoutID(vertexAttr_);
var signeture = ShaderSignature.GetInputSignature(vsBytecode);
inputLayout_ = new InputLayout(device, signeture, sLayoutList[layout_id]);
signeture.Dispose();
// コンスタントバッファ解決
ResolveConstantBuffer(reflect);
}
}
// ピクセルシェーダキャッシュ読み取り
if (desc.ps_main != null)
{
ShaderBytecode psBytecode = null;
if (System.IO.File.Exists(psCacheFile))
{
DateTime cacheTime = System.IO.File.GetLastWriteTime(psCacheFile);
DateTime editTime = System.IO.File.GetLastWriteTime(desc.file_name);
if (cacheTime > editTime)
{
System.IO.FileStream stream = new System.IO.FileStream(psCacheFile, FileMode.Open);
var bytes = new byte[stream.Length];
stream.Read(bytes, 0, (int)stream.Length);
psBytecode = new ShaderBytecode(new SlimDX.DataStream(bytes, true, false));
}
}
if (psBytecode == null)
{
if (inc == null)
{
inc = new ShaderIncludeHandler(System.IO.Path.GetDirectoryName(desc.file_name));
}
psBytecode = ShaderBytecode.CompileFromFile(desc.file_name, desc.ps_main, "ps_" + desc.profile, ShaderFlags.None, EffectFlags.None, desc.macro, inc);
// キャッシュファイル保存
System.IO.FileStream stream = new System.IO.FileStream(psCacheFile, FileMode.Create);
psBytecode.Data.CopyTo(stream);
stream.Close();
}
using (psBytecode) {
pixelShader_ = new PixelShader(device, psBytecode);
using (ShaderReflection reflect = new ShaderReflection(psBytecode)) {
// コンスタントバッファ解決
ResolveConstantBuffer(reflect);
}
}
}
}
else
{
// stringから生成
// 頂点シェーダ
using (var bytecode = ShaderBytecode.Compile(desc.vs_string, desc.vs_main, "vs_" + desc.profile, ShaderFlags.None, EffectFlags.None)) {
vertexShader_ = new VertexShader(device, bytecode);
using (ShaderReflection reflect = new ShaderReflection(bytecode)) {
vertexAttr_ = reflect.GetVertexLayoutAttribute();
int layout_id = Shader.GetVertexLayoutID(vertexAttr_);
var signeture = ShaderSignature.GetInputSignature(bytecode);
inputLayout_ = new InputLayout(device, signeture, sLayoutList[layout_id]);
signeture.Dispose();
// コンスタントバッファ解決
ResolveConstantBuffer(reflect);
}
}
// ピクセルシェーダ
if (desc.ps_main != null)
{
using (var bytecode = ShaderBytecode.Compile(desc.ps_string, desc.ps_main, "ps_" + desc.profile, ShaderFlags.None, EffectFlags.None)) {
pixelShader_ = new PixelShader(device, bytecode);
using (ShaderReflection reflect = new ShaderReflection(bytecode)) {
// コンスタントバッファ解決
ResolveConstantBuffer(reflect);
}
}
}
}
Initilaized = true;
}
/// <summary>
/// ファイルから初期化
/// </summary>
public Shader(InitDesc desc)
{
Initilaized = false;
Initialize(desc);
}