80: Map compute specialization constants r=kvark a=grovesNL

Shares code between graphics/compute pipelines setup to map compute specialization constants

Co-authored-by: Joshua Groves <josh@joshgroves.com>
This commit is contained in:
bors[bot] 2018-05-24 11:02:25 +00:00
commit 1da17c2050
2 changed files with 43 additions and 57 deletions

View file

@ -727,3 +727,43 @@ pub fn map_pipeline_statistics(flags: VkQueryPipelineStatisticFlags) -> query::P
// Vulkan and HAL flags are equal
unsafe { mem::transmute(flags) }
}
pub fn map_specialization_info(specialization: &VkSpecializationInfo) -> Vec<pso::Specialization> {
let data = unsafe { slice::from_raw_parts(
specialization.pData as *const u8,
specialization.dataSize as _,
)};
let entries = unsafe { slice::from_raw_parts(
specialization.pMapEntries,
specialization.mapEntryCount as _,
)};
entries
.into_iter()
.map(|entry| {
let offset = entry.offset as usize;
pso::Specialization {
id: entry.constantID,
value: match entry.size {
4 => pso::Constant::U32(
data[offset] as u32 |
(data[offset+1] as u32) << 8 |
(data[offset+2] as u32) << 16 |
(data[offset+3] as u32) << 24
),
8 => pso::Constant::U64(
data[offset] as u64 |
(data[offset+1] as u64) << 8 |
(data[offset+2] as u64) << 16 |
(data[offset+3] as u64) << 24 |
(data[offset+4] as u64) << 32 |
(data[offset+5] as u64) << 40 |
(data[offset+6] as u64) << 48 |
(data[offset+7] as u64) << 56
),
size => panic!("Unexpected specialization constant size: {:?}", size),
},
}
})
.collect::<Vec<_>>()
}

View file

@ -1404,6 +1404,7 @@ pub extern "C" fn gfxMergePipelineCaches(
) -> VkResult {
VkResult::VK_SUCCESS
}
#[inline]
pub extern "C" fn gfxCreateGraphicsPipelines(
gpu: VkDevice,
@ -1433,45 +1434,7 @@ pub extern "C" fn gfxCreateGraphicsPipelines(
let specialization = unsafe { stage
.pSpecializationInfo
.as_ref()
.map(|specialization| {
let data = slice::from_raw_parts(
specialization.pData as *const u8,
specialization.dataSize as _,
);
let entries = slice::from_raw_parts(
specialization.pMapEntries,
specialization.mapEntryCount as _,
);
entries
.into_iter()
.map(|entry| {
let offset = entry.offset as usize;
pso::Specialization {
id: entry.constantID,
value: match entry.size {
4 => pso::Constant::U32(
data[offset] as u32 |
(data[offset+1] as u32) << 8 |
(data[offset+2] as u32) << 16 |
(data[offset+3] as u32) << 24
),
8 => pso::Constant::U64(
data[offset] as u64 |
(data[offset+1] as u64) << 8 |
(data[offset+2] as u64) << 16 |
(data[offset+3] as u64) << 24 |
(data[offset+4] as u64) << 32 |
(data[offset+5] as u64) << 40 |
(data[offset+6] as u64) << 48 |
(data[offset+7] as u64) << 56
),
size => panic!("Unexpected specialization constant size: {:?}", size),
},
}
})
.collect::<Vec<pso::Specialization>>()
})
.map(conv::map_specialization_info)
.unwrap_or(vec![])
};
@ -1853,24 +1816,7 @@ pub extern "C" fn gfxCreateComputePipelines(
let specialization = unsafe { info.stage
.pSpecializationInfo
.as_ref()
.map(|specialization| {
let _data = slice::from_raw_parts(
specialization.pData,
specialization.dataSize as _,
);
let entries = slice::from_raw_parts(
specialization.pMapEntries,
specialization.mapEntryCount as _,
);
entries
.into_iter()
.map(|_entry| {
// Currently blocked due to lack of specialization type knowledge
unimplemented!()
})
.collect::<Vec<pso::Specialization>>()
})
.map(conv::map_specialization_info)
.unwrap_or(vec![])
};