@aguacerowx/react-native 0.0.17 → 0.0.19

package/ios/InspectorModule.swift

@@ -1,64 +1,64 @@
-import Foundation
-import React
-
-@objc(InspectorModule)
-class InspectorModule: NSObject {
-    
-@objc(getValueAtGridIndex:j:resolver:rejecter:)
-    func getValueAtGridIndex(
-        i: NSNumber,
-        j: NSNumber,
-        resolve: @escaping RCTPromiseResolveBlock,
-        reject: @escaping RCTPromiseRejectBlock
-    ) {
-        DispatchQueue.global(qos: .userInitiated).async {
-            let cache = InspectorDataCache.shared
-            
-            guard let data = cache.lastDecompressedData else {
-                print("🔍 [Inspector] ERROR: No data in cache")
-                resolve(nil)
-                return
-            }
-            
-            let nx = cache.nx
-            let scale = cache.scale
-            let offset = cache.offset
-            let missing = cache.missing
-            let scaleType = cache.scaleType
-            
-            let index = j.intValue * nx + i.intValue
-            
-            guard index >= 0 && index < data.count else {
-                print("🔍 [Inspector] ERROR: Index out of bounds")
-                resolve(nil)
-                return
-            }
-            
-            // Read as signed value
-            let signedQuantizedValue = Int(Int8(bitPattern: data[index]))
-            let missingQuantized = Int(round(missing))
-            
-            // Check for missing value with tolerance
-            if abs(signedQuantizedValue - missingQuantized) < 1 {
-                resolve(nil)
-                return
-            }
-            
-            // Step 1: Linear dequantization
-            let intermediateValue = Double(signedQuantizedValue) * Double(scale) + Double(offset)
-            var finalValue = intermediateValue
-            
-            // Step 2: Apply sqrt inverse transformation if needed
-            if scaleType == 1 {
-                finalValue = intermediateValue * abs(intermediateValue)
-            } 
-            
-            resolve(finalValue)
-        }
-    }
-    
-    @objc
-    static func requiresMainQueueSetup() -> Bool {
-        return false
-    }
+import Foundation
+import React
+
+@objc(InspectorModule)
+class InspectorModule: NSObject {
+    
+@objc(getValueAtGridIndex:j:resolver:rejecter:)
+    func getValueAtGridIndex(
+        i: NSNumber,
+        j: NSNumber,
+        resolve: @escaping RCTPromiseResolveBlock,
+        reject: @escaping RCTPromiseRejectBlock
+    ) {
+        DispatchQueue.global(qos: .userInitiated).async {
+            let cache = InspectorDataCache.shared
+            
+            guard let data = cache.lastDecompressedData else {
+                print("🔍 [Inspector] ERROR: No data in cache")
+                resolve(nil)
+                return
+            }
+            
+            let nx = cache.nx
+            let scale = cache.scale
+            let offset = cache.offset
+            let missing = cache.missing
+            let scaleType = cache.scaleType
+            
+            let index = j.intValue * nx + i.intValue
+            
+            guard index >= 0 && index < data.count else {
+                print("🔍 [Inspector] ERROR: Index out of bounds")
+                resolve(nil)
+                return
+            }
+            
+            // Read as signed value
+            let signedQuantizedValue = Int(Int8(bitPattern: data[index]))
+            let missingQuantized = Int(round(missing))
+            
+            // Check for missing value with tolerance
+            if abs(signedQuantizedValue - missingQuantized) < 1 {
+                resolve(nil)
+                return
+            }
+            
+            // Step 1: Linear dequantization
+            let intermediateValue = Double(signedQuantizedValue) * Double(scale) + Double(offset)
+            var finalValue = intermediateValue
+            
+            // Step 2: Apply sqrt inverse transformation if needed
+            if scaleType == 1 {
+                finalValue = intermediateValue * abs(intermediateValue)
+            } 
+            
+            resolve(finalValue)
+        }
+    }
+    
+    @objc
+    static func requiresMainQueueSetup() -> Bool {
+        return false
+    }
 }

package/ios/Shaders.metal

@@ -1,131 +1,179 @@
-#include <metal_stdlib>
-using namespace metal;
-
-// Struct to match the vertex buffer layout in Swift
-struct Vertex {
-    float2 position [[attribute(0)]];
-    float2 texCoord [[attribute(1)]];
-};
-
-// Struct for passing data from vertex to fragment shader
-struct RasterizerData {
-    float4 position [[position]];
-    float2 texCoord;
-};
-
-// Uniforms passed from the CPU to the fragment shader
-struct FragmentUniforms {
-    float opacity;
-    float2 dataRange;
-    float scale;
-    float offset;
-    float missingQuantized;
-    float2 textureSize;
-    int smoothing;
-    int scaleType; // 0 for linear, 1 for sqrt
-};
-
-// --- Vertex Shader (Unchanged) ---
-vertex RasterizerData vertex_main(const Vertex in [[stage_in]],
-                                  constant float4x4 &mvp [[buffer(1)]]) {
-    RasterizerData out;
-    out.position = mvp * float4(in.position, 0.0, 1.0);
-    out.texCoord = in.texCoord;
-    return out;
-}
-
-float sampleDataTextureSmooth(
-    texture2d<float, access::sample> dataTexture,
-    sampler dataSampler,
-    float2 uv,
-    constant FragmentUniforms &uniforms)
-{
-    float missing_value = uniforms.missingQuantized + 128.0;
-
-    // Calculate the fractional part for interpolation and the size of one texel.
-    float2 texel_size = 1.0 / uniforms.textureSize;
-    float2 pixel_coord = uv * uniforms.textureSize;
-    float2 pixel_floor = floor(pixel_coord - 0.5);
-    float2 f = fract(pixel_coord - 0.5);
-
-    // Calculate the precise UV coordinates for the CENTER of the four surrounding texels.
-    float2 uv00 = (pixel_floor + float2(0.5, 0.5)) * texel_size;
-    float2 uv10 = (pixel_floor + float2(1.5, 0.5)) * texel_size;
-    float2 uv01 = (pixel_floor + float2(0.5, 1.5)) * texel_size;
-    float2 uv11 = (pixel_floor + float2(1.5, 1.5)) * texel_size;
-
-    // Sample the four corner texels.
-    float s00 = dataTexture.sample(dataSampler, uv00).r * 255.0;
-    float s10 = dataTexture.sample(dataSampler, uv10).r * 255.0;
-    float s01 = dataTexture.sample(dataSampler, uv01).r * 255.0;
-    float s11 = dataTexture.sample(dataSampler, uv11).r * 255.0;
-
-    // THE CRITICAL CHECK: If any surrounding pixel is a "missing" value,
-    // treat this entire fragment as missing to prevent interpolation.
-    if (abs(s00 - missing_value) < 0.5 ||
-        abs(s10 - missing_value) < 0.5 ||
-        abs(s01 - missing_value) < 0.5 ||
-        abs(s11 - missing_value) < 0.5)
-    {
-        return missing_value;
-    }
-
-    // All neighbors are valid data, so we can safely perform manual bilinear interpolation.
-    float s0 = mix(s00, s10, f.x); // Interpolate the top row
-    float s1 = mix(s01, s11, f.x); // Interpolate the bottom row
-    
-    // --- THE FIX IS HERE ---
-    // Remove the incorrect "1.0 - f.y" inversion. The vertical mix should use f.y directly.
-    return mix(s0, s1, f.y);
-}
-
-// --- Fragment Shader (NEW & IMPROVED) ---
-fragment float4 fragment_main(RasterizerData in [[stage_in]],
-                                constant FragmentUniforms &uniforms [[buffer(0)]],
-                                texture2d<float, access::sample> dataTexture [[texture(0)]],
-                                texture2d<float, access::sample> colormapTexture [[texture(1)]],
-                                sampler dataSampler [[sampler(0)]],
-                                sampler colormapSampler [[sampler(1)]])
-{
-    // Choose the sampling method based on the smoothing uniform.
-    float value_in_texture;
-    if (uniforms.smoothing == 1) {
-        // Use our custom function that prevents edge artifacts.
-        value_in_texture = sampleDataTextureSmooth(dataTexture, dataSampler, in.texCoord, uniforms);
-    } else {
-        // Smoothing is off. The sampler is already set to 'nearest' by the Swift code,
-        // so a standard sample call produces the correct pixelated result.
-        value_in_texture = dataTexture.sample(dataSampler, in.texCoord).r * 255.0;
-    }
-
-    float missing_value = uniforms.missingQuantized + 128.0;
-
-    // Discard if the pixel is a predefined missing value (either originally or from our function).
-    if (abs(value_in_texture - missing_value) < 0.5) {
-        discard_fragment();
-    }
-
-    // Dequantize the value to its real-world representation.
-    float quantized_value = value_in_texture - 128.0;
-    float intermediate_value = quantized_value * uniforms.scale + uniforms.offset;
-    float raw_value = intermediate_value;
-    
-    // Apply non-linear inverse scaling if specified.
-    if (uniforms.scaleType == 1) { // 'sqrt'
-        raw_value = intermediate_value * abs(intermediate_value);
-    }
-    
-    // Discard if the data is below the specified visible range.
-    if (raw_value < uniforms.dataRange.x) {
-        discard_fragment();
-    }
-    
-    // Normalize the value to a 0-1 coordinate for colormap lookup.
-    float colormap_coord = (raw_value - uniforms.dataRange.x) / (uniforms.dataRange.y - uniforms.dataRange.x);
-    colormap_coord = clamp(colormap_coord, 0.0, 1.0);
-    
-    // Sample the colormap and apply final opacity.
-    float4 color = colormapTexture.sample(colormapSampler, float2(colormap_coord, 0.5));
-    
-    return float4(color.rgb, color.a * uniforms.opacity);
+#include <metal_stdlib>
+using namespace metal;
+
+// Struct to match the vertex buffer layout in Swift
+struct Vertex {
+    float2 position [[attribute(0)]];
+    float2 texCoord [[attribute(1)]];
+};
+
+// Struct for passing data from vertex to fragment shader
+struct RasterizerData {
+    float4 position [[position]];
+    float2 texCoord;
+};
+
+// Uniforms passed from the CPU to the fragment shader
+struct FragmentUniforms {
+    float opacity;
+    float2 dataRange;
+    float scale;
+    float offset;
+    float missingQuantized;
+    float2 textureSize;
+    int smoothing;
+    int scaleType; // 0 for linear, 1 for sqrt
+    int isPtype;   // ADD THIS
+};
+
+// --- Vertex Shader (Unchanged) ---
+vertex RasterizerData vertex_main(const Vertex in [[stage_in]],
+                                  constant float4x4 &mvp [[buffer(1)]]) {
+    RasterizerData out;
+    out.position = mvp * float4(in.position, 0.0, 1.0);
+    out.texCoord = in.texCoord;
+    return out;
+}
+
+
+// --- HELPER FUNCTIONS FOR FRAGMENT SHADER ---
+
+// Dequantizes a signed value (-127 to 127) into a physical value
+float dequantize_val(float q_val, constant FragmentUniforms &uniforms) {
+    if (abs(q_val - uniforms.missingQuantized) < 0.5) {
+        return -9999.0;
+    }
+    float intermediate = q_val * uniforms.scale + uniforms.offset;
+    if (uniforms.scaleType == 1) { // 1 = sqrt
+        return intermediate * abs(intermediate);
+    }
+    return intermediate;
+}
+
+// Samples texture, converts 0-255 to signed -128-127 range.
+// Returns a quantized value, or a large positive sentinel for missing data.
+float get_value(texture2d<float, access::sample> dataTexture, sampler dataSampler, float2 coord, constant FragmentUniforms &uniforms) {
+    // Note: The sampler should be set to 'nearest' when calling this for ptype logic.
+    float value_0_to_255 = dataTexture.sample(dataSampler, coord).r * 255.0;
+    float val = value_0_to_255 - 128.0;
+    float missing_in_texture_range_0_255 = uniforms.missingQuantized + 128.0;
+    if (abs(value_0_to_255 - missing_in_texture_range_0_255) < 0.5) {
+        return 99999.0; // Sentinel for "missing"
+    }
+    return val;
+}
+
+// Maps a physical value to a precipitation type category
+float get_ptype_from_physical(float val) {
+    if (val < 0.0) return -1.0;   // Missing/Invalid
+    if (val < 100.0) return 0.0; // Rain
+    if (val < 200.0) return 1.0; // Mix
+    if (val < 300.0) return 2.0; // Snow
+    return 3.0;                  // Pellets
+}
+
+
+// --- Main Fragment Shader ---
+fragment float4 fragment_main(RasterizerData in [[stage_in]],
+                                constant FragmentUniforms &uniforms [[buffer(0)]],
+                                texture2d<float, access::sample> dataTexture [[texture(0)]],
+                                texture2d<float, access::sample> colormapTexture [[texture(1)]],
+                                sampler dataSampler [[sampler(0)]],
+                                sampler colormapSampler [[sampler(1)]])
+{
+    float raw_value;
+    float alpha_multiplier = 1.0; // Used for alpha feathering at data edges
+
+    if (uniforms.smoothing == 0) { // Nearest neighbor
+        float quantized_value = get_value(dataTexture, dataSampler, in.texCoord, uniforms);
+        if (quantized_value >= 99999.0) {
+             discard_fragment();
+        }
+        raw_value = dequantize_val(quantized_value, uniforms);
+
+    } else { // Bilinear interpolation
+        float2 texel_size = 1.0 / uniforms.textureSize;
+        float2 tex_coord_in_texels = in.texCoord * uniforms.textureSize;
+        float2 bl_texel_index = floor(tex_coord_in_texels - 0.5);
+        float2 f = fract(tex_coord_in_texels - 0.5); // Interpolation factors
+
+        // Coordinates of the 4 neighboring texels
+        float2 v00_coord = (bl_texel_index + float2(0.5, 0.5)) * texel_size;
+        float2 v10_coord = (bl_texel_index + float2(1.5, 0.5)) * texel_size;
+        float2 v01_coord = (bl_texel_index + float2(0.5, 1.5)) * texel_size;
+        float2 v11_coord = (bl_texel_index + float2(1.5, 1.5)) * texel_size;
+
+        if (uniforms.isPtype == 1) {
+            // --- PTYPE-SPECIFIC INTERPOLATION (CATEGORICAL) ---
+            float v00 = dequantize_val(get_value(dataTexture, dataSampler, v00_coord, uniforms), uniforms);
+            float v10 = dequantize_val(get_value(dataTexture, dataSampler, v10_coord, uniforms), uniforms);
+            float v01 = dequantize_val(get_value(dataTexture, dataSampler, v01_coord, uniforms), uniforms);
+            float v11 = dequantize_val(get_value(dataTexture, dataSampler, v11_coord, uniforms), uniforms);
+
+            float p00 = get_ptype_from_physical(v00);
+            float p10 = get_ptype_from_physical(v10);
+            float p01 = get_ptype_from_physical(v01);
+            float p11 = get_ptype_from_physical(v11);
+
+            // Determine the dominant precipitation type for this pixel
+            float dominant_ptype = -1.0;
+            if (p00 != -1.0) dominant_ptype = p00;
+            else if (p10 != -1.0) dominant_ptype = p10;
+            else if (p01 != -1.0) dominant_ptype = p01;
+            else if (p11 != -1.0) dominant_ptype = p11;
+            
+            if (dominant_ptype == -1.0) discard_fragment();
+
+            // Interpolate ONLY the values that match the dominant ptype
+            float total_value = 0.0;
+            float total_weight = 0.0;
+            if (p00 == dominant_ptype) { float w = (1.0-f.x)*(1.0-f.y); total_value+=v00*w; total_weight+=w; }
+            if (p10 == dominant_ptype) { float w = f.x*(1.0-f.y);       total_value+=v10*w; total_weight+=w; }
+            if (p01 == dominant_ptype) { float w = (1.0-f.x)*f.y;       total_value+=v01*w; total_weight+=w; }
+            if (p11 == dominant_ptype) { float w = f.x*f.y;             total_value+=v11*w; total_weight+=w; }
+
+            if (total_weight <= 0.0) discard_fragment();
+            raw_value = total_value / total_weight;
+            
+            // Force full alpha to prevent dark borders between ptype regions
+            alpha_multiplier = 1.0;
+
+        } else {
+            // --- STANDARD INTERPOLATION (CONTINUOUS) ---
+            float v00_q = get_value(dataTexture, dataSampler, v00_coord, uniforms);
+            float v10_q = get_value(dataTexture, dataSampler, v10_coord, uniforms);
+            float v01_q = get_value(dataTexture, dataSampler, v01_coord, uniforms);
+            float v11_q = get_value(dataTexture, dataSampler, v11_coord, uniforms);
+
+            float total_value_q = 0.0;
+            float total_weight = 0.0;
+            if (v00_q < 99999.0) { float w = (1.0-f.x)*(1.0-f.y); total_value_q+=v00_q*w; total_weight+=w; }
+            if (v10_q < 99999.0) { float w = f.x*(1.0-f.y);       total_value_q+=v10_q*w; total_weight+=w; }
+            if (v01_q < 99999.0) { float w = (1.0-f.x)*f.y;       total_value_q+=v01_q*w; total_weight+=w; }
+            if (v11_q < 99999.0) { float w = f.x*f.y;             total_value_q+=v11_q*w; total_weight+=w; }
+
+            if (total_weight <= 0.0) discard_fragment();
+            raw_value = dequantize_val(total_value_q / total_weight, uniforms);
+            alpha_multiplier = total_weight; // Feather the edges of the data
+        }
+    }
+
+    // --- COMMON FINAL STEPS ---
+
+    if (raw_value < uniforms.dataRange.x) {
+        discard_fragment();
+    }
+
+    float colormap_coord = (raw_value - uniforms.dataRange.x) / (uniforms.dataRange.y - uniforms.dataRange.x);
+    colormap_coord = clamp(colormap_coord, 0.0, 1.0);
+    
+    float4 color = colormapTexture.sample(colormapSampler, float2(colormap_coord, 0.5));
+    
+    // Discard if the colormap specifies transparency
+    if (color.a < 0.1) {
+        discard_fragment();
+    }
+
+    float final_alpha = color.a * uniforms.opacity * alpha_multiplier;
+    return float4(color.rgb * final_alpha, final_alpha);
 }

package/ios/WeatherFrameProcessorModule.m

@@ -1,16 +1,16 @@
-#import <React/RCTBridgeModule.h>
-
-@interface RCT_EXTERN_MODULE(WeatherFrameProcessorModule, NSObject)
-
-RCT_EXTERN_METHOD(processFrame:(NSDictionary *)options
-                  resolver:(RCTPromiseResolveBlock)resolve
-                  rejecter:(RCTPromiseRejectBlock)reject)
-
-RCT_EXTERN_METHOD(cancelAllFrames)
-
-+ (BOOL)requiresMainQueueSetup
-{
-    return NO;
-}
-
+#import <React/RCTBridgeModule.h>
+
+@interface RCT_EXTERN_MODULE(WeatherFrameProcessorModule, NSObject)
+
+RCT_EXTERN_METHOD(processFrame:(NSDictionary *)options
+                  resolver:(RCTPromiseResolveBlock)resolve
+                  rejecter:(RCTPromiseRejectBlock)reject)
+
+RCT_EXTERN_METHOD(cancelAllFrames)
+
++ (BOOL)requiresMainQueueSetup
+{
+    return NO;
+}
+
 @end