/****************************************************************************
 *
 *    Copyright (c) 2023 by Rockchip Corp.  All rights reserved.
 *
 *    The material in this file is confidential and contains trade secrets
 *    of Rockchip Corporation. This is proprietary information owned by
 *    Rockchip Corporation. No part of this work may be disclosed,
 *    reproduced, copied, transmitted, or used in any way for any purpose,
 *    without the express written permission of Rockchip Corporation.
 *
 *****************************************************************************/

#ifndef MEMC_AUTO_FD_H_
#define MEMC_AUTO_FD_H_

#include <unistd.h>

namespace android {

class MemcUniqueFd
{
public:
    MemcUniqueFd() = default;
    MemcUniqueFd(int fd) : fd_(fd) {}
    MemcUniqueFd(const MemcUniqueFd &rhs) { fd_ = rhs.Dup(); }

    MemcUniqueFd(MemcUniqueFd &&rhs)
    {
        fd_ = rhs.fd_;
        rhs.fd_ = -1;
    }

    MemcUniqueFd &operator=(MemcUniqueFd &&rhs)
    {
        Set(rhs.Release());
        return *this;
    }

    ~MemcUniqueFd()
    {
        if (fd_ > 0) close(fd_);
    }

    int Release()
    {
        int old_fd = fd_;
        fd_ = -1;
        return old_fd;
    }

    int Set(int fd)
    {
        if (fd_ > 0) close(fd_);
        fd_ = fd;
        return fd_;
    }

    void Close()
    {
        if (fd_ > 0) close(fd_);
        fd_ = -1;
    }

    int get() const { return fd_; }

    int Dup() const { return dup(fd_); }

    int *get_ptr() { return &fd_; }

private:
    int fd_ = -1;
};

struct MemcOutputFd
{
    MemcOutputFd() = default;
    MemcOutputFd(int *fd) : fd_(fd) {}
    MemcOutputFd(MemcOutputFd &&rhs)
    {
        fd_ = rhs.fd_;
        rhs.fd_ = NULL;
    }

    MemcOutputFd &operator=(MemcOutputFd &&rhs)
    {
        fd_ = rhs.fd_;
        rhs.fd_ = NULL;
        return *this;
    }

    int Set(int fd)
    {
        if (*fd_ >= 0) close(*fd_);
        *fd_ = fd;
        return fd;
    }

    int get() { return *fd_; }

    operator bool() const { return fd_ != NULL; }

private:
    int *fd_ = NULL;
};

}  // namespace android

#endif