]> git.sesse.net Git - ccbs/blobdiff - bigscreen/ccbs_bigscreen.cpp
Fixed a misleading comment.
[ccbs] / bigscreen / ccbs_bigscreen.cpp
index 6afc0a082b2ea90b2a8f916af02961a4312be582..f6d8da537f22d7064e3d31efc0c65cb0e5eba29a 100644 (file)
 #include <cstdio>
+#include <cstring>
+#include <iconv.h>
 #include <unistd.h>
 #include <pqxx/pqxx>
-#include "glwindow.h"
+#include <ft2build.h>
+#include FT_FREETYPE_H
+#include <tinyptc.h>
+#include <endian.h>
+
+iconv_t ucs4_iconv;
+
+// UCS-4 string with support for getting from UTF-8
+class widestring : public std::wstring
+{
+public:
+       void operator= (const char *from)
+       {
+               unsigned bytes = std::strlen(from);
+               char *from_buf = strdup(from);
+               wchar_t *to_buf = new wchar_t[bytes + 1];
+
+               char *inptr = from_buf, *outptr = reinterpret_cast<char *> (to_buf);
+
+               size_t in_left = bytes;
+               size_t out_left = bytes * sizeof(wchar_t);
+
+               size_t ret = iconv(ucs4_iconv, NULL, NULL, &outptr, &out_left);
+               if (ret == (size_t)(-1)) {
+                       throw std::runtime_error("Error in iconv during initialization");
+               }
+
+               ret = iconv(ucs4_iconv, &inptr, &in_left, &outptr, &out_left);
+               if (ret == (size_t)(-1)) {
+                       perror("iconv");
+                       throw std::runtime_error("Error in iconv during conversion");
+               }
+
+               erase(begin(), end());
+               std::copy(to_buf, reinterpret_cast<wchar_t *> (outptr), std::back_inserter(*this));
+
+               free(from_buf);
+               delete[] to_buf;
+       }
+};
+
+template<>
+void pqxx::from_string<widestring>(const char *from, widestring &to)
+{
+       to = from;
+}
+
+int my_draw_text(const widestring &str, unsigned char *buf, int xpos, int ypos, bool real_render, int r, int g, int b, std::vector<FT_Face> &fontlist);
+
+class Tournament {
+public:
+       int id;
+       widestring name;
+};
+
+Tournament active_tournament;
+std::vector<FT_Face> fonts;
+
+/* A trigger that sets a flag whenever it's trigged. */
+class FlagTrigger : pqxx::trigger {
+private:
+       bool flag;
+       
+public:
+       FlagTrigger(pqxx::connection_base &conn, const PGSTD::string &name)
+               : pqxx::trigger(conn, name), flag(false) {}
+       virtual ~FlagTrigger() throw () {}
+       
+       virtual void operator() (int pid)
+       {
+               flag = true;
+               std::fprintf(stderr, "Received a flag trigger from pid %u\n", pid);
+       }
+
+       bool get_flag() const
+       {
+               return flag;
+       }
+
+       void reset_flag()
+       {
+               flag = false;
+       }
+};
+
+/* A transactor that fetches the current tournament and some information about it. */
+class FetchCurrentTournament : public pqxx::transactor<> {
+private:
+       Tournament *tourn;
+
+public:
+       FetchCurrentTournament(Tournament *tourn) : tourn(tourn) {}
+       void operator() (pqxx::transaction<> &t)
+       {
+               pqxx::result res( t.exec("SELECT * FROM bigscreen.active_tournament NATURAL JOIN tournaments") );
+               try {
+                       pqxx::result::tuple tournament = res.at(0);
+
+                       tourn->id = tournament["tournament"].as(tourn->id);
+                       tourn->name = tournament["tournamentname"].as(tourn->name);
+               } catch (PGSTD::out_of_range &e) {
+                       tourn->id = -1;
+                       tourn->name = "";
+               }
+       }
+};
+
+void init(pqxx::connection &conn)
+{
+       conn.perform(FetchCurrentTournament(&active_tournament));
+
+       if (active_tournament.id == -1) {
+               std::fprintf(stderr, "No active tournament\n");
+       } else {
+               std::fprintf(stderr, "Current tournament is %d (name: '%s')\n",
+                       active_tournament.id, active_tournament.name.c_str());
+       }
+}
+
+unsigned char framebuf[800 * 600 * 4];
+
+void main_loop(pqxx::connection &conn)
+{
+       if (active_tournament.id == -1) {
+               // No active tournament, sleep a second or so and exit
+               sleep(1);
+               return;
+       }
+
+       memset(framebuf, 0, 800*600*4);
+       
+       pqxx::work t(conn, "trx");
+
+       // fetch all songs
+       pqxx::result res( t.exec("SELECT * FROM songs WHERE title LIKE 'M%'") );
+       unsigned y = 0;
+       for (pqxx::result::const_iterator i = res.begin(); i != res.end(); ++i) {
+               my_draw_text(i["title"].as(widestring()), framebuf, 0, y, 1, 255, 255, 255, fonts);
+               y += 20;
+//             std::fprintf(stderr, "%s\n", i["title"].c_str());
+       }
+       t.commit();
+
+       ptc_update(framebuf);
+       sleep(1);
+}
+
+void init_freetype()
+{
+       FT_Library library;
+       FT_Face face;
+       if (FT_Init_FreeType(&library))
+               throw std::runtime_error("FreeType init failed.");
+
+       // Georgia
+       if (FT_New_Face(library, "/usr/share/fonts/truetype/msttcorefonts/Georgia.ttf", 0, &face))
+               throw std::runtime_error("Face opening failed.");
+       if (FT_Set_Char_Size(face, 0, 12 * 64, 96, 96))
+               throw std::runtime_error("Size set failed.");
+       fonts.push_back(face);
+
+       // FreeSerif
+       if (FT_New_Face(library, "/usr/share/fonts/truetype/freefont/FreeSerif.ttf", 0, &face)) {
+               std::fprintf(stderr, "Warning: Couldn't open FreeSerif, some glyphs might not be available\n");
+       } else {
+               if (FT_Set_Char_Size(face, 0, 12 * 64, 96, 96))
+                       throw std::runtime_error("Size set failed.");
+               fonts.push_back(face);
+       }
+       
+       // Arial Unicode MS
+       if (FT_New_Face(library, "arialuni.ttf", 0, &face)) {
+               std::fprintf(stderr, "Warning: Couldn't open Arial Unicode MS, some glyphs might not be available\n");
+       } else {
+               if (FT_Set_Char_Size(face, 0, 12 * 64, 96, 96))
+                       throw std::runtime_error("Size set failed.");
+               fonts.push_back(face);
+       }
+}
+
+int my_draw_text(const widestring &str, unsigned char *buf, int xpos, int ypos, bool real_render, int r, int g, int b, std::vector<FT_Face> &fontlist)
+{
+       FT_GlyphSlot slot;
+       int x = 0;
+
+       for (widestring::const_iterator i = str.begin(); i != str.end(); ++i) {
+               int glyph_index;
+               for (std::vector<FT_Face>::const_iterator j = fontlist.begin(); j != fontlist.end(); ++j) {
+                       glyph_index = FT_Get_Char_Index(*j, *i);
+                       if (glyph_index == 0)
+                               continue;
+
+                       if (FT_Load_Glyph(*j, glyph_index, FT_LOAD_RENDER))
+                               throw std::runtime_error("Couldn't load glyph");
+                       slot = (*j)->glyph;
+                       break;
+               }
+               if (glyph_index == 0) {
+                       std::fprintf(stderr, "Warning: Could not find a glyph in any font for U+%x, ignoring\n", *i);
+                       continue;
+               }
+
+               if (real_render) {
+                       int y;
+                       FT_Bitmap *bm = &(slot->bitmap);
+                       for (y = 0; y < bm->rows; y++) {
+                               int xx;
+                               int dsty = ypos - slot->bitmap_top + y;
+                               if (dsty < 0 || dsty > 599) continue;
+
+                               unsigned char *dst = buf + dsty * 800*4 + (x + xpos + slot->bitmap_left)*4;
+                               unsigned char *src = bm->buffer + y * bm->width;
+                               for (xx = 0; xx < bm->width; xx++) {
+                                       *dst = (*dst * (256-*src) + r * *src) >> 8;
+                                       *dst++;
+                                       *dst = (*dst * (256-*src) + g * *src) >> 8;
+                                       *dst++;
+                                       *dst = (*dst * (256-*src) + b * *src) >> 8;
+                                       *dst++;
+                                       *dst++ = 0;
+                                       src++;
+                               }
+                       }
+               }
+
+               x += slot->advance.x >> 6;
+       }
+
+       return x;
+}
+
 
 int main(int argc, char **argv)
 {
-       GLWindow glw("CCBS bigscreen", 800, 600, 32, false, 16, -1);
+#if __BYTE_ORDER == __LITTLE_ENDIAN
+       ucs4_iconv = iconv_open("ucs-4le", "utf-8");
+#else  
+       ucs4_iconv = iconv_open("ucs-4be", "utf-8");
+#endif
+       
+       ptc_open("CCBS bigscreen", 800, 600);
+       
        try {
-               pqxx::connection conn("dbname=ccbs host=sql.samfundet.no user=ccbs password=GeT|>>B_");
-               pqxx::work t(conn, "trx");
-
-               // fetch all songs
-               pqxx::result res( t.exec("SELECT * FROM songs") );
-               for (pqxx::result::const_iterator i = res.begin(); i != res.end(); ++i) {
-                       std::fprintf(stderr, "%s\n", i["title"].c_str());
+               init_freetype();
+               pqxx::connection conn("dbname=ccbs host=altersex.samfundet.no user=ccbs password=GeT|>>B_");
+               FlagTrigger tournament_changed(conn, "active_tournament");
+               
+               // when active_tournament is changed, we destroy everything and start from scratch
+               for ( ;; ) {
+                       tournament_changed.reset_flag();
+                       init(conn);
+                       do {
+                               main_loop(conn);
+                               conn.get_notifs();
+                       } while (!tournament_changed.get_flag());
+                       std::fprintf(stderr, "active_tournament changed, resetting...\n");
                }
-               t.commit();
        } catch (const std::exception &e) {
                std::fprintf(stderr, "Exception: %s\n", e.what());
                exit(1);