#!amber

# Copyright 2020 Valve Corporation.
# Copyright 2020 The Khronos Group Inc.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.


SHADER compute comp SPIRV-ASM
;
; The shader below is based on the following GLSL shader:
;
;     #version 450
;
;     struct Pair {
;         int first;
;         int second;
;     };
;
;     const Pair constant_pair = { 100, 200 };
;
;     layout(set=0, binding=0, std430) buffer InputBlock {
;         int array[10];
;     } inputValues;
;
;     layout(set=0, binding=1, std430) buffer OutputBlock {
;         int array[10];
;     } outputValues;
;
;     int add_second (int value, Pair pair) {
;         return value + pair.second;
;     }
;
;     void main() {
;         uint idx = gl_GlobalInvocationID.x;
;         outputValues.array[idx] = add_second(inputValues.array[idx], constant_pair);
;     }
;
; However, the first element of constant_pair has been modified to be undefined.
;
                            OpCapability Shader
                  %std450 = OpExtInstImport "GLSL.std.450"
                            OpMemoryModel Logical GLSL450
                            OpEntryPoint GLCompute %main "main" %gl_GlobalInvocationID
                            OpExecutionMode %main LocalSize 1 1 1
                            OpDecorate %gl_GlobalInvocationID BuiltIn GlobalInvocationId
                            OpDecorate %_arr_int_uint_10 ArrayStride 4
                            OpMemberDecorate %OutputBlock 0 Offset 0
                            OpDecorate %OutputBlock BufferBlock
                            OpDecorate %outputValues DescriptorSet 0
                            OpDecorate %outputValues Binding 1
                            OpMemberDecorate %InputBlock 0 Offset 0
                            OpDecorate %InputBlock BufferBlock
                            OpDecorate %inputValues DescriptorSet 0
                            OpDecorate %inputValues Binding 0
                    %void = OpTypeVoid
               %void_func = OpTypeFunction %void
                     %int = OpTypeInt 32 1
                    %uint = OpTypeInt 32 0
                  %v3uint = OpTypeVector %uint 3
                   %int_0 = OpConstant %int 0
                   %int_1 = OpConstant %int 1
                 %int_200 = OpConstant %int 200
                  %uint_0 = OpConstant %uint 0
                 %uint_10 = OpConstant %uint 10
       %_ptr_Function_int = OpTypePointer Function %int
                    %Pair = OpTypeStruct %int %int
      %_ptr_Function_Pair = OpTypePointer Function %Pair
    %add_second_func_type = OpTypeFunction %int %_ptr_Function_int %_ptr_Function_Pair
      %_ptr_Function_uint = OpTypePointer Function %uint
       %_ptr_Input_v3uint = OpTypePointer Input %v3uint
         %_ptr_Input_uint = OpTypePointer Input %uint
        %_arr_int_uint_10 = OpTypeArray %int %uint_10
             %OutputBlock = OpTypeStruct %_arr_int_uint_10
%_ptr_Uniform_OutputBlock = OpTypePointer Uniform %OutputBlock
            %outputValues = OpVariable %_ptr_Uniform_OutputBlock Uniform
              %InputBlock = OpTypeStruct %_arr_int_uint_10
 %_ptr_Uniform_InputBlock = OpTypePointer Uniform %InputBlock
             %inputValues = OpVariable %_ptr_Uniform_InputBlock Uniform
                            ; Replaced %int_100 with an undefined int.
               %undef_int = OpUndef %int
                            ; Composed a constant Pair with the undefined int in the first member.
              %const_Pair = OpConstantComposite %Pair %undef_int %int_200
        %_ptr_Uniform_int = OpTypePointer Uniform %int
   %gl_GlobalInvocationID = OpVariable %_ptr_Input_v3uint Input
                    %main = OpFunction %void None %void_func
              %main_label = OpLabel
                 %param_1 = OpVariable %_ptr_Function_int Function
                 %param_2 = OpVariable %_ptr_Function_Pair Function
                %gidx_ptr = OpAccessChain %_ptr_Input_uint %gl_GlobalInvocationID %uint_0
                    %gidx = OpLoad %uint %gidx_ptr
         %input_value_ptr = OpAccessChain %_ptr_Uniform_int %inputValues %int_0 %gidx
             %input_value = OpLoad %int %input_value_ptr
                            OpStore %param_1 %input_value
                            OpStore %param_2 %const_Pair
                  %retval = OpFunctionCall %int %add_second %param_1 %param_2
        %output_value_ptr = OpAccessChain %_ptr_Uniform_int %outputValues %int_0 %gidx
                            OpStore %output_value_ptr %retval
                            OpReturn
                            OpFunctionEnd
              %add_second = OpFunction %int None %add_second_func_type
               %value_ptr = OpFunctionParameter %_ptr_Function_int
                    %pair = OpFunctionParameter %_ptr_Function_Pair
        %add_second_label = OpLabel
                   %value = OpLoad %int %value_ptr
                            ; Access the second struct member, which is defined.
         %pair_second_ptr = OpAccessChain %_ptr_Function_int %pair %int_1
             %pair_second = OpLoad %int %pair_second_ptr
              %add_result = OpIAdd %int %value %pair_second
                            OpReturnValue %add_result
                            OpFunctionEnd
END

BUFFER input_buffer DATA_TYPE int32 SIZE 10 SERIES_FROM 1000 INC_BY 1
BUFFER output_buffer DATA_TYPE int32 SIZE 10 FILL 0
BUFFER reference_buffer DATA_TYPE int32 SIZE 10 SERIES_FROM 1200 INC_BY 1

PIPELINE compute compute_pipeline
	ATTACH comp
	BIND BUFFER input_buffer AS storage DESCRIPTOR_SET 0 BINDING 0
	BIND BUFFER output_buffer AS storage DESCRIPTOR_SET 0 BINDING 1
END

RUN compute_pipeline 10 1 1
EXPECT output_buffer EQ_BUFFER reference_buffer