From 58501b67c7f8e48c5e83e7e9bd886f6904f6596b Mon Sep 17 00:00:00 2001
From: "Mun, Gwan-gyeong" <kk.moon@samsung.com>
Date: Mon, 22 Feb 2016 20:23:59 +0900
Subject: [PATCH] layers: GH PR17, Add WSI argument passing to vk-generate.py

---
 vk-generate.py | 15 ++++++++++++---
 1 file changed, 12 insertions(+), 3 deletions(-)

diff --git a/vk-generate.py b/vk-generate.py
index b255d7f..6f478a6 100755
--- a/vk-generate.py
+++ b/vk-generate.py
@@ -27,6 +27,7 @@
 # Author: Chia-I Wu <olv@lunarg.com>
 # Author: Courtney Goeltzenleuchter <courtney@LunarG.com>
 # Author: Jon Ashburn <jon@lunarg.com>
+# Author: Gwan-gyeong Mun <kk.moon@samsung.com>
 
 import sys
 
@@ -241,18 +242,26 @@ class WinDefFileSubcommand(Subcommand):
         return "\n".join(body)
 
 def main():
+    wsi = {
+            "Win32",
+            "Android",
+            "Xcb",
+            "Xlib",
+            "Wayland",
+            "Mir"
+    }
     subcommands = {
             "dispatch-table-ops": DispatchTableOpsSubcommand,
             "win-def-file": WinDefFileSubcommand,
     }
 
-    if len(sys.argv) < 2 or sys.argv[1] not in subcommands:
-        print("Usage: %s <subcommand> [options]" % sys.argv[0])
+    if len(sys.argv) < 3 or sys.argv[1] not in wsi or sys.argv[2] not in subcommands:
+        print("Usage: %s <wsi> <subcommand> [options]" % sys.argv[0])
         print
         print("Available sucommands are: %s" % " ".join(subcommands))
         exit(1)
 
-    subcmd = subcommands[sys.argv[1]](sys.argv[2:])
+    subcmd = subcommands[sys.argv[2]](sys.argv[3:])
     subcmd.run()
 
 if __name__ == "__main__":
-- 
2.7.4