//色ごとに画面をクリアするシェーダの例
RWTexture2D<float4> screen :register(u0);
[numthreads(1,1,1)]
void clearRed(uint2 DTid : SV_DispatchThreadID) {
  screen[DTid] = float4(1, 0, 0, 1);//赤色でクリアする
}
[numthreads(1,1,1)]
void clearGreen(uint2 DTid : SV_DispatchThreadID) {
  screen[DTid] = float4(0, 1, 0, 1);//緑色でクリアする
}
//以下、必要な色分エントリポイントを用意する

//c++の引数みたいにエントリポイントに値を渡したいが、シェーダではこの書き方はできない
RWTexture2D<float4> screen :register(u0);
[numthreads(1,1,1)]
void clearRed(uint2 DTid : SV_DispatchThreadID, float4 clearColor/*<- コンパイルエラー*/) {
  screen[DTid] = clearColor;//任意の色でクリアする
}

//定数バッファの定義
cbuffer Param : register(b0) {
  float4 clearColor;
  uint2 screenSize;
};

RWTexture2D<float4> screen : register(u0);
[numthreads(1, 1, 1)]
void main( uint2 DTid : SV_DispatchThreadID ) {
  [branch]
  if(DTid.x < screenSize.x && DTid.y < screenSize.y) {
    screen[DTid] = clearColor;//定数バッファの変数はグローバル変数のように使える
  }
}

std::array<ID3D11Buffer*, 1> ppCBs = { {
  this->mpCB.Get(),
} };
this->mpImmediateContext->CSSetConstantBuffers(0, static_cast<UINT>(ppCBs.size()), ppCBs.data());

Param param;	//ID3D11Bufferに設定するデータ
param.clearColor = DirectX::SimpleMath::Vector4(1, 0.7f, 0.7f, 1.f);
param.screenSize = DirectX::SimpleMath::Vector2(static_cast<float>(this->width()), static_cast<float>(this->height()));

//作成するID3D11Bufferの情報
D3D11_BUFFER_DESC desc = {};
desc.ByteWidth = sizeof(param) + (sizeof(param) % 16 == 0 ? 0 : 16 - sizeof(param) % 16);//サイズは16の倍数でないといけない
desc.BindFlags = D3D11_BIND_CONSTANT_BUFFER;//ID3D11Bufferを定数バッファとして使うよう宣言している
desc.Usage = D3D11_USAGE_DEFAULT;//GPU上からしかID3D11Bufferの内容にアクセスできないよう宣言している
desc.CPUAccessFlags = 0;//CPUからのアクセスフラグの指定。今回はアクセスしないので何も設定しない

//CPUとGPU間のデータ転送の時に使う構造体
//ここではID3D11Bufferの初期データを設定するために使っている
D3D11_SUBRESOURCE_DATA initData = {};
initData.pSysMem = &param;
initData.SysMemPitch = sizeof(param);//設定するデータのサイズ
auto hr = this->mpDevice->CreateBuffer(&desc, &initData, this->mpCB.GetAddressOf());
if (FAILED(hr)) {
  throw std::runtime_error("定数バッファの作成に失敗");
}

//Scene::onUpdate関数一部
//ID3D11DeviceCOntext::UpdateSubresource関数を使った定数バッファの更新
Param param;
param.clearColor = DirectX::SimpleMath::Vector4(abs(sin(t)), (cos(t)), (sin(t)), 1.f);
param.screenSize.x = static_cast<float>(this->width());
param.screenSize.y = static_cast<float>(this->height());
//定数バッファの内容更新
this->mpImmediateContext->UpdateSubresource(this->mpCB.Get(), 0, nullptr, &param, 0, 0);

//Sccene::onUpdate関数の一部
//Map関数を使うときはD3D11_BUFFER_DESCのUsageにD3D11_USAGE_DYNAMICを
// CPUAccessFlagsにD3D11_CPU_ACCESS_WRITEを指定してください。
UINT subresourceIndex = 0;
D3D11_MAPPED_SUBRESOURCE mapped;
auto hr = this->mpImmediateContext->Map(this->mpCBMappable.Get(), subresourceIndex, D3D11_MAP_WRITE_DISCARD, 0, &mapped);
if (SUCCEEDED(hr)) {
  Param* p = static_cast<Param*>(mapped.pData);
  p->clearColor = DirectX::SimpleMath::Vector4(abs(sin(t*2.f)), (cos(t)), (sin(t)), 1.f);
  p->screenSize.x = static_cast<float>(this->width());
  p->screenSize.y = static_cast<float>(this->height());
  this->mpImmediateContext->Unmap(this->mpCBMappable.Get(), subresourceIndex);
}