1 files changed, 4 insertions, 1 deletions
diff --git a/drmresources.h b/drmresources.h
index 0cc2456..2058dbc 100644
--- a/drmresources.h
+++ b/drmresources.h
@@ -21,10 +21,13 @@
 #include "drmcrtc.h"
 #include "drmencoder.h"
 #include "drmeventlistener.h"
+#include "drmhwcgralloc.h"
 #include "drmplane.h"
 
 #include <stdint.h>
 
+#include <hardware/hwcomposer.h>
+
 namespace android {
 
 class DrmResources {
@@ -32,7 +35,7 @@ class DrmResources {
   DrmResources();
   ~DrmResources();
 
-  int Init();
+  int Init(const gralloc_module_t *gralloc);
 
   int fd() const {
     return fd_.get();