QemuVideo: prepare to support more hardware

[mirror_edk2.git] / OvmfPkg / QemuVideoDxe / Gop.c

diff --git a/OvmfPkg/QemuVideoDxe/Gop.c b/OvmfPkg/QemuVideoDxe/Gop.c
index d364630c47609407a1062d84a35557bea4570474..bd49ee91391fb6e995465b8868a5eb258e9c5ff8 100644 (file)
--- a/OvmfPkg/QemuVideoDxe/Gop.c
+++ b/OvmfPkg/QemuVideoDxe/Gop.c
@@ -181,7 +181,17 @@ Routine Description:
     return EFI_OUT_OF_RESOURCES;\r
   }\r
 \r
-  InitializeGraphicsMode (Private, &QemuVideoVideoModes[ModeData->ModeNumber]);\r
+  switch (Private->Variant) {\r
+  case QEMU_VIDEO_CIRRUS_5430:\r
+  case QEMU_VIDEO_CIRRUS_5446:\r
+    InitializeCirrusGraphicsMode (Private, &QemuVideoCirrusModes[ModeData->ModeNumber]);\r
+    break;\r
+  default:\r
+    ASSERT (FALSE);\r
+    gBS->FreePool (Private->LineBuffer);\r
+    Private->LineBuffer = NULL;\r
+    return EFI_DEVICE_ERROR;\r
+  }\r
 \r
   This->Mode->Mode = ModeNumber;\r
   This->Mode->Info->HorizontalResolution = ModeData->HorizontalResolution;\r