/*
 * Copyright (c) 2021-2022 Huawei Device Co., Ltd.
 * 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 UPDATER_UI_DRM_DRIVER_H
#define UPDATER_UI_DRM_DRIVER_H

#include <cerrno>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <drm_fourcc.h>
#include <fcntl.h>
#include <poll.h>
#include <sys/mman.h>
#include <xf86drm.h>
#include <xf86drmMode.h>
#include "macros_updater.h"
#include "graphic_drv.h"

namespace Updater {
struct BufferObject {
    uint32_t width {};
    uint32_t height {};
    uint32_t pitch {};
    uint32_t handle {};
    uint32_t size {};
    uint8_t *vaddr {};
    uint32_t fbId {};
};

class DrmDriver : public GraphicDrv {
    DISALLOW_COPY_MOVE(DrmDriver);
public:
    DrmDriver() : conn_(nullptr), res_(nullptr), crtc_(nullptr) {}
    ~DrmDriver() override;
    bool Init() override;
    void Flip(const uint8_t *buf) override;
    void Blank(bool blank) override;
    void Exit(void) override;
    void GetGrSurface(GrSurface &surface) override;
private:
    int ModesetCreateFb(struct BufferObject *bo);
    void ModesetDestroyFb(struct BufferObject *bo);
    int DrmInit();
    drmModeCrtc *GetCrtc(const drmModeRes &res, const int fd, const drmModeConnector &conn) const;
    drmModeConnector *GetFirstConnector(const drmModeRes &res, const int fd) const;
    drmModeConnector *GetConnectorByType(const drmModeRes &res, const int fd, const uint32_t type) const;
    drmModeConnector *GetConnector(const drmModeRes &res, const int fd, uint32_t &modeId) const;
    drmModeRes *GetResources(int &fd) const;
    drmModeRes *GetOneResources(const int devIndex, int &fd) const;
    drmModeConnector *conn_;
    drmModeRes *res_;
    drmModeCrtc *crtc_;
    struct BufferObject buff_ {};
};
} // namespace Updater
#endif