Support for Bitwise OR operator in translator codegen frontend

crosstl/translator/codegen/directx_codegen.py

@@ -258,6 +258,7 @@ def map_operator(self, op):
             "MULTIPLY": "*",
             "DIVIDE": "/",
             "BITWISE_XOR": "^",
+            "BITWISE_OR": "|",
             "BITWISE_AND": "&",
             "LESS_THAN": "<",
             "GREATER_THAN": ">",

crosstl/translator/codegen/metal_codegen.py

@@ -343,6 +343,7 @@ def map_operator(self, op):
             "MULTIPLY": "*",
             "DIVIDE": "/",
             "BITWISE_XOR": "^",
+            "BITWISE_OR": "|",
             "BITWISE_AND": "&",
             "LESS_THAN": "<",
             "GREATER_THAN": ">",

tests/test_translator/test_codegen/test_directx_codegen.py

@@ -683,5 +683,42 @@ def test_shift_operators():
         pytest.fail("Shift right assignment operator codegen not implemented.")
 
 
+def test_bitwise_or_operator():
+    code = """
+    shader main {
+    struct VSInput {
+        vec2 texCoord @ TEXCOORD0;
+    };
+    struct VSOutput {
+        vec4 color @ COLOR;
+    };
+    sampler2D iChannel0;
+    vertex {
+        VSOutput main(VSInput input) {
+            VSOutput output;
+            // Use bitwise OR on texture coordinates (for testing purposes)
+            output.color = vec4(float(int(input.texCoord.x * 100.0) | 15), 
+                                float(int(input.texCoord.y * 100.0) | 15), 
+                                0.0, 1.0);
+            return output;
+        }
+    }
+    fragment {
+        vec4 main(VSOutput input) @ gl_FragColor {
+            // Simple fragment shader to display the result of the AND operation
+            return vec4(input.color.rgb, 1.0);
+        }
+    }
+}
+    """
+    try:
+        tokens = tokenize_code(code)
+        ast = parse_code(tokens)
+        generated_code = generate_code(ast)
+        print(generated_code)
+    except SyntaxError:
+        pytest.fail("Bitwise OR codegen not implemented")
+
+
 if __name__ == "__main__":
     pytest.main()

tests/test_translator/test_codegen/test_metal_codegen.py

@@ -366,5 +366,42 @@ def test_bitwise_operators():
         pytest.fail("Bitwise Shift codegen not implemented.")
 
 
+def test_bitwise_or_operator():
+    code = """
+    shader main {
+    struct VSInput {
+        vec2 texCoord @ TEXCOORD0;
+    };
+    struct VSOutput {
+        vec4 color @ COLOR;
+    };
+    sampler2D iChannel0;
+    vertex {
+        VSOutput main(VSInput input) {
+            VSOutput output;
+            // Use bitwise AND on texture coordinates (for testing purposes)
+            output.color = vec4(float(int(input.texCoord.x * 100.0) | 15), 
+                                float(int(input.texCoord.y * 100.0) | 15), 
+                                0.0, 1.0);
+            return output;
+        }
+    }
+    fragment {
+        vec4 main(VSOutput input) @ gl_FragColor {
+            // Simple fragment shader to display the result of the AND operation
+            return vec4(input.color.rgb, 1.0);
+        }
+    }
+}
+    """
+    try:
+        tokens = tokenize_code(code)
+        ast = parse_code(tokens)
+        generated_code = generate_code(ast)
+        print(generated_code)
+    except SyntaxError:
+        pytest.fail("Bitwise OR codegen not implemented")
+
+
 if __name__ == "__main__":
     pytest.main()