/*
 * Copyright (C) 2017 Mentor Graphics Development (Deutschland) GmbH
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#ifndef WM_WAYLAND_HPP
#define WM_WAYLAND_HPP

#include "controller_hooks.hpp"
#include "ivi-controller-client-protocol.h"
#include "util.hpp"

#include <functional>
#include <memory>
#include <unordered_map>
#include <vector>

//                      _                 _
// __      ____ _ _   _| | __ _ _ __   __| |     _ __  _ __ _____  ___   _
// \ \ /\ / / _` | | | | |/ _` | '_ \ / _` |    | '_ \| '__/ _ \ \/ / | | |
//  \ V  V / (_| | |_| | | (_| | | | | (_| |    | |_) | | | (_) >  <| |_| |
//   \_/\_/ \__,_|\__, |_|\__,_|_| |_|\__,_|____| .__/|_|  \___/_/\_\\__, |
//                |___/                   |_____|_|                  |___/
template <typename ProxyT>
struct wayland_proxy {
   std::unique_ptr<ProxyT, std::function<void(ProxyT *)>> proxy;
   wayland_proxy(wayland_proxy const &) = delete;
   wayland_proxy &operator=(wayland_proxy const &) = delete;
   wayland_proxy(void *p)
      : wayland_proxy(p,
                      reinterpret_cast<void (*)(ProxyT *)>(wl_proxy_destroy)) {}
   wayland_proxy(void *p, std::function<void(ProxyT *)> &&p_del)
      : proxy(std::unique_ptr<ProxyT, std::function<void(ProxyT *)>>(
           static_cast<ProxyT *>(p), p_del)) {}
};

//                                                                  _
//  _ __   __ _ _ __ ___   ___  ___ _ __   __ _  ___ ___  __      _| |
// | '_ \ / _` | '_ ` _ \ / _ \/ __| '_ \ / _` |/ __/ _ \ \ \ /\ / / |
// | | | | (_| | | | | | |  __/\__ \ |_) | (_| | (_|  __/  \ V  V /| |
// |_| |_|\__,_|_| |_| |_|\___||___/ .__/ \__,_|\___\___|   \_/\_/ |_|
//                                 |_|
namespace wl {
//                 _     _
//  _ __ ___  __ _(_)___| |_ _ __ _   _
// | '__/ _ \/ _` | / __| __| '__| | | |
// | | |  __/ (_| | \__ \ |_| |  | |_| |
// |_|  \___|\__, |_|___/\__|_|   \__, |
//           |___/                |___/
struct registry : public wayland_proxy<struct wl_registry> {
   typedef std::function<void(struct wl_registry *, uint32_t, uint32_t)> binder;
   std::unordered_map<std::string, binder> bindings;

   registry(registry const &) = delete;
   registry &operator=(registry const &) = delete;
   registry(struct wl_display *d);

   void add_global_handler(char const *iface, binder bind);

   // Events
   void global(uint32_t name, char const *iface, uint32_t v);
   void global_remove(uint32_t name);
};

//      _ _           _
//   __| (_)___ _ __ | | __ _ _   _
//  / _` | / __| '_ \| |/ _` | | | |
// | (_| | \__ \ |_) | | (_| | |_| |
//  \__,_|_|___/ .__/|_|\__,_|\__, |
//             |_|            |___/
struct display {
   std::unique_ptr<struct wl_display, void (*)(struct wl_display *)> d;
   struct registry r;

   display(display const &) = delete;
   display &operator=(display const &) = delete;
   display();
   bool ok() const;
   void roundtrip();
   int dispatch();
   int dispatch_pending();
   int read_events();
   void flush();
   int get_fd() const;
   int get_error();

   // Lets just proxy this for the registry
   inline void add_global_handler(char const *iface, registry::binder bind) {
      this->r.add_global_handler(iface, bind);
   }
};

//              _               _
//   ___  _   _| |_ _ __  _   _| |_
//  / _ \| | | | __| '_ \| | | | __|
// | (_) | |_| | |_| |_) | |_| | |_
//  \___/ \__,_|\__| .__/ \__,_|\__|
//                 |_|
struct output : wayland_proxy<struct wl_output> {
   int width{};
   int height{};
   int refresh{};
   int transform{};

   output(output const &) = delete;
   output &operator=(output const &) = delete;
   output(struct wl_registry *r, uint32_t name, uint32_t v);

   // Events
   void geometry(int32_t x, int32_t y, int32_t pw, int32_t ph, int32_t subpel,
                 char const *make, char const *model, int32_t tx);
   void mode(uint32_t flags, int32_t w, int32_t h, int32_t r);
   void done();
   void scale(int32_t factor);
};
}  // namespace wl

//  _ __   __ _ _ __ ___   ___  ___ _ __   __ _  ___ ___
// | '_ \ / _` | '_ ` _ \ / _ \/ __| '_ \ / _` |/ __/ _ \
// | | | | (_| | | | | | |  __/\__ \ |_) | (_| | (_|  __/
// |_| |_|\__,_|_| |_| |_|\___||___/ .__/ \__,_|\___\___|
//                                 |_|
//                   _       _
//   __ _  ___ _ __ (_)_   _(_)
//  / _` |/ _ \ '_ \| \ \ / / |
// | (_| |  __/ | | | |\ V /| |
//  \__, |\___|_| |_|_| \_/ |_|
//  |___/
namespace genivi {

struct size {
   uint32_t w, h;
};

struct rect {
   int32_t w, h;
   int32_t x, y;
};

static const constexpr rect full_rect = rect{-1, -1, 0, 0};

inline bool operator==(struct rect a, struct rect b) {
   return a.w == b.w && a.h == b.h && a.x == b.x && a.y == b.y;
}

struct controller;

struct controller_child {
   struct controller *parent;
   uint32_t id;

   controller_child(controller_child const &) = delete;
   controller_child &operator=(controller_child const &) = delete;
   controller_child(struct controller *c, uint32_t i) : parent(c), id(i) {}
   virtual ~controller_child() {}
};

struct surface_properties {
   uint32_t id;  // let's just save an ID here too
   struct rect dst_rect;
   struct rect src_rect;
   struct size size;
   int32_t orientation;
   int32_t visibility;
   float opacity;
};

//                  __
//  ___ _   _ _ __ / _| __ _  ___ ___
// / __| | | | '__| |_ / _` |/ __/ _ \
// \__ \ |_| | |  |  _| (_| | (_|  __/
// |___/\__,_|_|  |_|  \__,_|\___\___|
//
struct surface : public wayland_proxy<struct ivi_controller_surface>,
                 controller_child {
   surface(surface const &) = delete;
   surface &operator=(surface const &) = delete;
   surface(uint32_t i, struct controller *c);

   // Requests
   void set_visibility(uint32_t visibility);
   void set_opacity(wl_fixed_t opacity);
   void set_source_rectangle(int32_t x, int32_t y, int32_t width,
                             int32_t height);
   void set_destination_rectangle(int32_t x, int32_t y, int32_t width,
                                  int32_t height);
   void set_configuration(int32_t width, int32_t height);
   void set_orientation(int32_t orientation);
   void screenshot(const char *filename);
   void send_stats();
   void destroy(int32_t destroy_scene_object);
};

//  _
// | | __ _ _   _  ___ _ __
// | |/ _` | | | |/ _ \ '__|
// | | (_| | |_| |  __/ |
// |_|\__,_|\__, |\___|_|
//          |___/
struct layer : public wayland_proxy<struct ivi_controller_layer>,
               controller_child {
   layer(layer const &) = delete;
   layer &operator=(layer const &) = delete;
   layer(uint32_t i, struct controller *c);
   layer(uint32_t i, int32_t w, int32_t h, struct controller *c);

   // Requests
   void set_visibility(uint32_t visibility);
   void set_opacity(wl_fixed_t opacity);
   void set_source_rectangle(int32_t x, int32_t y, int32_t width,
                             int32_t height);
   void set_destination_rectangle(int32_t x, int32_t y, int32_t width,
                                  int32_t height);
   void set_configuration(int32_t width, int32_t height);
   void set_orientation(int32_t orientation);
   void screenshot(const char *filename);
   void clear_surfaces();
   void add_surface(struct surface *surface);
   void remove_surface(struct surface *surface);
   void set_render_order(std::vector<uint32_t> const &ro);
};

//
//  ___  ___ _ __ ___  ___ _ __
// / __|/ __| '__/ _ \/ _ \ '_ \
// \__ \ (__| | |  __/  __/ | | |
// |___/\___|_|  \___|\___|_| |_|
//
struct screen : public wayland_proxy<struct ivi_controller_screen>,
                controller_child {
   screen(screen const &) = delete;
   screen &operator=(screen const &) = delete;
   screen(uint32_t i, struct controller *c, struct ivi_controller_screen *p);
   void clear();
   void add_layer(layer *l);
   void set_render_order(std::vector<uint32_t> const &ro);
};

//                  _             _ _
//   ___ ___  _ __ | |_ _ __ ___ | | | ___ _ __
//  / __/ _ \| '_ \| __| '__/ _ \| | |/ _ \ '__|
// | (_| (_) | | | | |_| | | (_) | | |  __/ |
//  \___\___/|_| |_|\__|_|  \___/|_|_|\___|_|
//
struct controller : public wayland_proxy<struct ivi_controller> {
   // This controller is still missing ivi-input

   typedef std::unordered_map<uintptr_t, uint32_t> proxy_to_id_map_type;
   typedef std::unordered_map<uint32_t, std::unique_ptr<struct surface>>
      surface_map_type;
   typedef std::unordered_map<uint32_t, std::unique_ptr<struct layer>>
      layer_map_type;
   typedef std::unordered_map<uint32_t, std::unique_ptr<struct screen>>
      screen_map_type;

   typedef std::unordered_map<uint32_t, struct surface_properties> props_map;

   // HACK:
   // The order of these member is mandatory, as when objects are destroyed
   // they will call their parent (that's us right here!) and remove their
   // proxy-to-id mapping. I.e. the *_proxy_to_id members need to be valid
   // when the surfaces/layers/screens maps are destroyed. This sucks, but
   // I cannot see a better solution w/o globals or some other horrible
   // call-our-parent construct.
   proxy_to_id_map_type surface_proxy_to_id;
   proxy_to_id_map_type layer_proxy_to_id;
   proxy_to_id_map_type screen_proxy_to_id;

   props_map sprops;
   props_map lprops;

   surface_map_type surfaces;
   layer_map_type layers;
   screen_map_type screens;

   size output_size;

   wm::controller_hooks *chooks;

   void add_proxy_to_id_mapping(struct ivi_controller_surface *p, uint32_t id);
   void remove_proxy_to_id_mapping(struct ivi_controller_surface *p);
   void add_proxy_to_id_mapping(struct ivi_controller_layer *p, uint32_t id);
   void remove_proxy_to_id_mapping(struct ivi_controller_layer *p);
   void add_proxy_to_id_mapping(struct wl_output *p, uint32_t id);
   void remove_proxy_to_id_mapping(struct wl_output *p);

   bool surface_exists(uint32_t id) const {
      return this->surfaces.find(id) != this->surfaces.end();
   }

   bool layer_exists(uint32_t id) const {
      return this->layers.find(id) != this->layers.end();
   }

   controller(struct wl_registry *r, uint32_t name, uint32_t version);

   // Requests
   void commit_changes() const {
      ivi_controller_commit_changes(this->proxy.get());
   }
   void layer_create(uint32_t id, int32_t w, int32_t h);
   void surface_create(uint32_t id);

   // Events
   // controller
   void controller_screen(uint32_t id, struct ivi_controller_screen *screen);
   void controller_layer(uint32_t id);
   void controller_surface(uint32_t id);
   void controller_error(int32_t object_id, int32_t object_type,
                         int32_t error_code, char const *error_text);

   // surface
   void surface_visibility(struct surface *s, int32_t visibility);
   void surface_opacity(struct surface *s, float opacity);
   void surface_source_rectangle(struct surface *s, int32_t x, int32_t y,
                                 int32_t width, int32_t height);
   void surface_destination_rectangle(struct surface *s, int32_t x, int32_t y,
                                      int32_t width, int32_t height);
   void surface_configuration(struct surface *s, int32_t width, int32_t height);
   void surface_orientation(struct surface *s, int32_t orientation);
   void surface_pixelformat(struct surface *s, int32_t pixelformat);
   void surface_layer(struct surface *s, struct ivi_controller_layer *layer);
   void surface_stats(struct surface *s, uint32_t redraw_count,
                      uint32_t frame_count, uint32_t update_count, uint32_t pid,
                      const char *process_name);
   void surface_destroyed(struct surface *s);
   void surface_content(struct surface *s, int32_t content_state);

   // layer
   void layer_visibility(struct layer *l, int32_t visibility);
   void layer_opacity(struct layer *l, float opacity);
   void layer_source_rectangle(struct layer *l, int32_t x, int32_t y,
                               int32_t width, int32_t height);
   void layer_destination_rectangle(struct layer *l, int32_t x, int32_t y,
                                    int32_t width, int32_t height);
   void layer_configuration(struct layer *l, int32_t width, int32_t height);
   void layer_orientation(struct layer *l, int32_t orientation);
   void layer_screen(struct layer *l, struct wl_output *screen);
   void layer_destroyed(struct layer *l);
};
}  // namespace genivi

#endif  // !WM_WAYLAND_HPP