tests/db.cpp

   1 #include <cstdio>
   2 #include <algorithm>
   3
   4 #include <xf86drm.h>
   5 #include <xf86drmMode.h>
   6 #include <drm_fourcc.h>
   7
   8 #include "kms++.h"
   9
  10 #include "test.h"
  11
  12 using namespace std;
  13 using namespace kms;
  14
  15 static void main_loop(Card& card);
  16
  17 class OutputFlipHandler : PageFlipHandlerBase
  18 {
  19 public:
  20         OutputFlipHandler(Connector* conn, Crtc* crtc, DumbFramebuffer* fb1, DumbFramebuffer* fb2)
  21                 : m_connector(conn), m_crtc(crtc), m_fbs { fb1, fb2 }, m_front_buf(1), m_bar_xpos(0)
  22         {
  23         }
  24
  25         ~OutputFlipHandler()
  26         {
  27                 delete m_fbs[0];
  28                 delete m_fbs[1];
  29         }
  30
  31         OutputFlipHandler(const OutputFlipHandler& other) = delete;
  32         OutputFlipHandler& operator=(const OutputFlipHandler& other) = delete;
  33
  34         void set_mode()
  35         {
  36                 auto mode = m_connector->get_default_mode();
  37                 int r = m_crtc->set_mode(m_connector, *m_fbs[0], mode);
  38                 ASSERT(r == 0);
  39         }
  40
  41         void handle_page_flip(uint32_t frame, double time)
  42         {
  43                 m_front_buf = (m_front_buf + 1) % 2;
  44
  45                 const int bar_width = 20;
  46                 const int bar_speed = 8;
  47
  48                 auto crtc = m_crtc;
  49                 auto fb = m_fbs[(m_front_buf + 1) % 2];
  50
  51                 ASSERT(crtc);
  52                 ASSERT(fb);
  53
  54                 int current_xpos = m_bar_xpos;
  55                 int old_xpos = (current_xpos + (fb->width() - bar_width - bar_speed)) % (fb->width() - bar_width);
  56                 int new_xpos = (current_xpos + bar_speed) % (fb->width() - bar_width);
  57
  58                 draw_color_bar(*fb, old_xpos, new_xpos, bar_width);
  59
  60                 m_bar_xpos = new_xpos;
  61
  62                 auto& card = crtc->card();
  63
  64                 if (card.has_atomic()) {
  65                         int r;
  66
  67                         AtomicReq req(card);
  68
  69                         req.add(m_crtc, "FB_ID", fb->id());
  70
  71                         r = req.test();
  72                         ASSERT(r == 0);
  73
  74                         r = req.commit(this);
  75                         ASSERT(r == 0);
  76                 } else {
  77                         int r = crtc->page_flip(*fb, this);
  78                         ASSERT(r == 0);
  79                 }
  80         }
  81
  82 private:
  83         Connector* m_connector;
  84         Crtc* m_crtc;
  85         DumbFramebuffer* m_fbs[2];
  86
  87         int m_front_buf;
  88         int m_bar_xpos;
  89 };
  90
  91 int main()
  92 {
  93         Card card;
  94
  95         if (card.master() == false)
  96                 printf("Not DRM master, modeset may fail\n");
  97
  98         //card.print_short();
  99
 100         vector<OutputFlipHandler*> outputs;
 101
 102         for (auto pipe : card.get_connected_pipelines())
 103         {
 104                 auto conn = pipe.connector;
 105                 auto crtc = pipe.crtc;
 106
 107                 auto mode = conn->get_default_mode();
 108
 109                 auto fb1 = new DumbFramebuffer(card, mode.hdisplay, mode.vdisplay, PixelFormat::XRGB8888);
 110                 auto fb2 = new DumbFramebuffer(card, mode.hdisplay, mode.vdisplay, PixelFormat::XRGB8888);
 111
 112                 printf("conn %u, crtc %u, fb1 %u, fb2 %u\n", conn->id(), crtc->id(), fb1->id(), fb2->id());
 113
 114                 auto output = new OutputFlipHandler(conn, crtc, fb1, fb2);
 115                 outputs.push_back(output);
 116         }
 117
 118         for(auto out : outputs)
 119                 out->set_mode();
 120
 121         for(auto out : outputs)
 122                 out->handle_page_flip(0, 0);
 123
 124         main_loop(card);
 125
 126         for(auto out : outputs)
 127                 delete out;
 128 }
 129
 130 static void main_loop(Card& card)
 131 {
 132         fd_set fds;
 133
 134         FD_ZERO(&fds);
 135
 136         int fd = card.fd();
 137
 138         printf("press enter to exit\n");
 139
 140         while (true) {
 141                 int r;
 142
 143                 FD_SET(0, &fds);
 144                 FD_SET(fd, &fds);
 145
 146                 r = select(fd + 1, &fds, NULL, NULL, NULL);
 147                 if (r < 0) {
 148                         fprintf(stderr, "select() failed with %d: %m\n", errno);
 149                         break;
 150                 } else if (FD_ISSET(0, &fds)) {
 151                         fprintf(stderr, "exit due to user-input\n");
 152                         break;
 153                 } else if (FD_ISSET(fd, &fds)) {
 154                         card.call_page_flip_handlers();
 155                 }
 156         }
 157 }