/*
 * Copyright (C) 2016 The Android Open Source Project
 *
 * 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 _ACAMERA_CAPTURE_SESSION_H
#define _ACAMERA_CAPTURE_SESSION_H

#include <set>
#include <string>
#include <hardware/camera3.h>
#include <camera/NdkCameraDevice.h>

#ifdef __ANDROID_VNDK__
#include "ndk_vendor/impl/ACameraDevice.h"
#else
#include "ACameraDevice.h"
#endif

using namespace android;

struct ACaptureSessionOutput {
    explicit ACaptureSessionOutput(ANativeWindow* window, bool isShared = false,
            const char* physicalCameraId = "") :
            mWindow(window), mIsShared(isShared), mPhysicalCameraId(physicalCameraId) {};

    bool operator == (const ACaptureSessionOutput& other) const {
        return mWindow == other.mWindow;
    }
    bool operator != (const ACaptureSessionOutput& other) const {
        return mWindow != other.mWindow;
    }
    bool operator < (const ACaptureSessionOutput& other) const {
        return mWindow < other.mWindow;
    }
    bool operator > (const ACaptureSessionOutput& other) const {
        return mWindow > other.mWindow;
    }

    inline bool isWindowEqual(ANativeWindow* window) const {
        return mWindow == window;
    }

    // returns true if the window was successfully added, false otherwise.
    inline bool addSharedWindow(ANativeWindow* window) {
        auto ret = mSharedWindows.insert(window);
        return ret.second;
    }

    // returns the number of elements removed.
    inline size_t removeSharedWindow(ANativeWindow* window) {
        return mSharedWindows.erase(window);
    }

    ANativeWindow* mWindow;
    std::set<ANativeWindow*> mSharedWindows;
    bool           mIsShared;
    int            mRotation = CAMERA3_STREAM_ROTATION_0;
    std::string mPhysicalCameraId;
};

struct ACaptureSessionOutputContainer {
    std::set<ACaptureSessionOutput> mOutputs;
};

/**
 * Capture session state callbacks used in {@link ACameraDevice_setPrepareCallbacks}
 */
typedef struct ACameraCaptureSession_prepareCallbacks {
    /// optional application context. This will be passed in the context
    /// parameter of the {@link onWindowPrepared} callback.
    void*                               context;

    ACameraCaptureSession_prepareCallback onWindowPrepared;
} ACameraCaptureSession_prepareCallbacks;

/**
 * ACameraCaptureSession opaque struct definition
 * Leave outside of android namespace because it's NDK struct
 */
struct ACameraCaptureSession : public RefBase {
  public:
#ifdef __ANDROID_VNDK__
    ACameraCaptureSession(
            int id,
            const ACaptureSessionOutputContainer* outputs,
            const ACameraCaptureSession_stateCallbacks* cb,
            std::weak_ptr<android::acam::CameraDevice> device) :
            mId(id), mOutput(*outputs), mUserSessionCallback(*cb),
            mDevice(std::move(device)) {}
#else
    ACameraCaptureSession(
            int id,
            const ACaptureSessionOutputContainer* outputs,
            const ACameraCaptureSession_stateCallbacks* cb,
            android::acam::CameraDevice* device) :
            mId(id), mOutput(*outputs), mUserSessionCallback(*cb),
            mDevice(device) {}
#endif

    // This can be called in app calling close() or after some app callback is finished
    // Make sure the caller does not hold device or session lock!
    ~ACameraCaptureSession();

    // No API except Session_Close will work if device is closed
    // A session will enter closed state when one of the following happens:
    //     1. Explicitly closed by app
    //     2. Replaced by a newer session
    //     3. Device is closed
    bool isClosed() { Mutex::Autolock _l(mSessionLock); return mIsClosed; }

    // Close the session and mark app no longer need this session.
    void closeByApp();

    camera_status_t stopRepeating();

    camera_status_t abortCaptures();

    template<class T>
    camera_status_t setRepeatingRequest(
            /*optional*/T* cbs,
            int numRequests, ACaptureRequest** requests,
            /*optional*/int* captureSequenceId);

    template<class T>
    camera_status_t capture(
            /*optional*/T* cbs,
            int numRequests, ACaptureRequest** requests,
            /*optional*/int* captureSequenceId);

    camera_status_t updateOutputConfiguration(ACaptureSessionOutput *output);

    void setWindowPreparedCallback(void *context,
            ACameraCaptureSession_prepareCallback cb) {
        Mutex::Autolock _l(mSessionLock);
        mPreparedCb.context = context;
        mPreparedCb.onWindowPrepared = cb;
    }
    camera_status_t prepare(ANativeWindow *window);

    ACameraDevice* getDevice();

  private:
    friend class android::acam::CameraDevice;

    // Close session because app close camera device, camera device got ERROR_DISCONNECTED,
    // or a new session is replacing this session.
    void closeByDevice();

#ifdef __ANDROID_VNDK__
    std::shared_ptr<android::acam::CameraDevice> getDevicePtr();
#else
    sp<android::acam::CameraDevice> getDeviceSp();
#endif

    const int mId;
    const ACaptureSessionOutputContainer mOutput;
    const ACameraCaptureSession_stateCallbacks mUserSessionCallback;
#ifdef __ANDROID_VNDK__
    const std::weak_ptr<android::acam::CameraDevice> mDevice;
#else
    const wp<android::acam::CameraDevice> mDevice;
#endif

    bool  mIsClosed = false;
    bool  mClosedByApp = false;
    ACameraCaptureSession_prepareCallbacks mPreparedCb;
    Mutex mSessionLock;
};

#endif // _ACAMERA_CAPTURE_SESSION_H