react-native-mask-segment-canvas 0.1.0

package/src/utils/maskSemanticPalette.ts

@@ -0,0 +1,187 @@
+import {
+  getMaskRuntimeRevision,
+  getMaskSegmentRuntimeConfig,
+} from './maskSegmentRuntime';
+
+export type MaskSemanticColor = {
+  name: string;
+  hex: string;
+  /** 参考色（BGR，与掩码 buffer 通道一致） */
+  bgr: { b: number; g: number; r: number };
+};
+
+/** 掩码语义色表（与后端分区颜色参考一致） */
+export const MASK_SEMANTIC_COLORS: MaskSemanticColor[] = [
+  { name: 'door', hex: '#E6194B', bgr: { b: 75, g: 25, r: 230 } },
+  { name: 'ceiling', hex: '#3CB44B', bgr: { b: 75, g: 180, r: 60 } },
+  { name: 'cabinet', hex: '#FFE119', bgr: { b: 25, g: 225, r: 255 } },
+  { name: 'wall', hex: '#4363D8', bgr: { b: 216, g: 99, r: 67 } },
+  { name: 'baseboard', hex: '#F58231', bgr: { b: 49, g: 130, r: 245 } },
+  { name: 'windowFrame', hex: '#911EB4', bgr: { b: 180, g: 30, r: 145 } },
+  { name: 'garageDoor', hex: '#46F0F0', bgr: { b: 240, g: 240, r: 70 } },
+  { name: 'roof', hex: '#F032E6', bgr: { b: 230, g: 50, r: 240 } },
+  { name: 'eave', hex: '#BCF60C', bgr: { b: 12, g: 246, r: 188 } },
+];
+
+export const BASEBOARD_SEMANTIC_NAME = 'baseboard';
+
+type SemanticRgbEntry = {
+  name: string;
+  rgb: { r: number; g: number; b: number };
+};
+
+type SemanticContext = {
+  baseboardMaxColorDistSq: number;
+  semanticRgb: SemanticRgbEntry[];
+  baseboardRgb: SemanticRgbEntry;
+  cabinetRgb: SemanticRgbEntry;
+  wallRgb: SemanticRgbEntry;
+};
+
+let contextRevision = -1;
+let cachedContext: SemanticContext | null = null;
+
+function buildSemanticRgb(colors: MaskSemanticColor[]): SemanticRgbEntry[] {
+  return colors.map(entry => ({
+    name: entry.name,
+    rgb: {
+      r: entry.bgr.r,
+      g: entry.bgr.g,
+      b: entry.bgr.b,
+    },
+  }));
+}
+
+function getSemanticContext(): SemanticContext {
+  const revision = getMaskRuntimeRevision();
+  if (contextRevision === revision && cachedContext) {
+    return cachedContext;
+  }
+
+  const mask = getMaskSegmentRuntimeConfig().mask;
+  const semanticRgb = buildSemanticRgb(mask.semanticColors);
+  const baseboardRgb = semanticRgb.find(
+    entry => entry.name === BASEBOARD_SEMANTIC_NAME,
+  )!;
+  const cabinetRgb = semanticRgb.find(entry => entry.name === 'cabinet')!;
+  const wallRgb = semanticRgb.find(entry => entry.name === 'wall')!;
+  const maxDist = mask.baseboardMaxColorDist;
+
+  cachedContext = {
+    baseboardMaxColorDistSq: maxDist * maxDist,
+    semanticRgb,
+    baseboardRgb,
+    cabinetRgb,
+    wallRgb,
+  };
+  contextRevision = revision;
+  return cachedContext;
+}
+
+function colorDistanceSq(
+  a: { r: number; g: number; b: number },
+  b: { r: number; g: number; b: number },
+): number {
+  const dr = a.r - b.r;
+  const dg = a.g - b.g;
+  const db = a.b - b.b;
+  return dr * dr + dg * dg + db * db;
+}
+
+/** 将掩码像素归类到最近的语义色（baseboard 仅严格橙色命中） */
+export function classifyBgrPixelToSemantic(
+  b: number,
+  g: number,
+  r: number,
+): string {
+  const ctx = getSemanticContext();
+  const pixel = { r, g, b };
+  const distToBaseboard = colorDistanceSq(pixel, ctx.baseboardRgb.rgb);
+  const distToCabinet = colorDistanceSq(pixel, ctx.cabinetRgb.rgb);
+  const distToWall = colorDistanceSq(pixel, ctx.wallRgb.rgb);
+
+  if (
+    distToBaseboard <= ctx.baseboardMaxColorDistSq &&
+    distToBaseboard < distToCabinet &&
+    distToBaseboard < distToWall
+  ) {
+    return BASEBOARD_SEMANTIC_NAME;
+  }
+
+  let best = ctx.semanticRgb[0];
+  let bestDist = Number.POSITIVE_INFINITY;
+
+  for (const entry of ctx.semanticRgb) {
+    if (entry.name === BASEBOARD_SEMANTIC_NAME) {
+      continue;
+    }
+    const dist = colorDistanceSq(pixel, entry.rgb);
+    if (dist < bestDist) {
+      bestDist = dist;
+      best = entry;
+    }
+  }
+
+  return best.name;
+}
+
+export function getSemanticColorByName(name: string): MaskSemanticColor | undefined {
+  const colors = getMaskSegmentRuntimeConfig().mask.semanticColors;
+  return colors.find(entry => entry.name === name);
+}
+
+/**
+ * 踢脚线须更接近 #F58231 且明显优于黄柜 / 蓝墙，避免整块黄区被误判。
+ */
+export function isStrictBaseboardPixel(
+  b: number,
+  g: number,
+  r: number,
+): boolean {
+  const ctx = getSemanticContext();
+  const pixel = { r, g, b };
+  const distToBaseboard = colorDistanceSq(pixel, ctx.baseboardRgb.rgb);
+  const distToCabinet = colorDistanceSq(pixel, ctx.cabinetRgb.rgb);
+  const distToWall = colorDistanceSq(pixel, ctx.wallRgb.rgb);
+
+  return (
+    distToBaseboard <= ctx.baseboardMaxColorDistSq &&
+    distToBaseboard < distToCabinet &&
+    distToBaseboard < distToWall
+  );
+}
+
+export function isBaseboardPixel(b: number, g: number, r: number): boolean {
+  return isStrictBaseboardPixel(b, g, r);
+}
+
+/** 掩码上墙/柜交界细条的量化色 */
+export const BASEBOARD_STRIP_QUANT_KEYS = new Set(['0,255,255', '64,255,255']);
+
+/** 掩码上墙面量化色 */
+export const WALL_QUANT_KEYS = new Set([
+  '192,128,64',
+  '192,64,64',
+  '128,64,64',
+  '192,192,128',
+  '128,128,64',
+]);
+
+/** 掩码上柜/地面量化色 */
+export const CABINET_QUANT_KEYS = new Set([
+  '0,192,255',
+  '64,192,255',
+  '128,192,255',
+]);
+
+export function getBaseboardStripQuantKeys(): Set<string> {
+  return getMaskSegmentRuntimeConfig().mask.baseboardStripQuantKeys;
+}
+
+export function getWallQuantKeys(): Set<string> {
+  return getMaskSegmentRuntimeConfig().mask.wallQuantKeys;
+}
+
+export function getCabinetQuantKeys(): Set<string> {
+  return getMaskSegmentRuntimeConfig().mask.cabinetQuantKeys;
+}