codec2: add color formats for image component

Bug: 136731683
Test: atest CtsMediaTestCases:MediaCodecTest#testPrependHeadersToSyncFrames
Merged-In: I8233ec928e7c5700abc1bdd7f32e097892947949
Change-Id: I8233ec928e7c5700abc1bdd7f32e097892947949
(cherry picked from commit 1622326c3c5ee468cf67b769b13b3dbb3fe9b560)
diff --git a/media/codec2/sfplugin/Codec2InfoBuilder.cpp b/media/codec2/sfplugin/Codec2InfoBuilder.cpp
index c54c601..6b75eba 100644
--- a/media/codec2/sfplugin/Codec2InfoBuilder.cpp
+++ b/media/codec2/sfplugin/Codec2InfoBuilder.cpp
@@ -194,7 +194,8 @@
     // TODO: get this from intf() as well, but how do we map them to
     // MediaCodec color formats?
     bool encoder = trait.kind == C2Component::KIND_ENCODER;
-    if (mediaType.find("video") != std::string::npos) {
+    if (mediaType.find("video") != std::string::npos
+            || mediaType.find("image") != std::string::npos) {
         // vendor video codecs prefer opaque format
         if (trait.name.find("android") == std::string::npos) {
             caps->addColorFormat(COLOR_FormatSurface);