色名を取得する

pinfort · Jan 9, 2019 · 698cfa4 · 698cfa4
1 parent 8bcf4fa
commit 698cfa4
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 0 deletions.
diff --git a/colors.py b/colors.py
@@ -0,0 +1,11 @@
+def colors(nf):
+    if nf == 1:
+        return ["Y"]
+    elif nf == 2:
+        return ["Y", "A"]
+    elif nf == 3:
+        return ["Y", "Cb", "Cr"]
+    elif nf == 4:
+        return ["Y", "Cb", "Cr", "A"]
+    else:
+        raise ValueError("invalid value")
diff --git a/segments/SOFs/SOF2.py b/segments/SOFs/SOF2.py
@@ -1,5 +1,6 @@
 from segments.segmentsCommon import SegmentsCommon
 from markers import Markers
+from colors import colors
 
 class SOF2(SegmentsCommon):
     def __init__(self):
@@ -20,6 +21,8 @@ def analyze(self, marker, body):
         nfLength = 3
         for i in range(analyzed["nf"]):
             nfDict = {}
+            nfDict["id"] = i + 1
+            nfDict["name"] = colors(analyzed["nf"])[i]
             nf = body[nfHead:nfHead + nfLength]
             nfDict["cn"] = nf[0]
             hv = nf[1]