128 lines
3.4 KiB
C++
Executable File
128 lines
3.4 KiB
C++
Executable File
/*
|
|
* Copyright 2021 Rockchip Electronics 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.
|
|
*
|
|
* author: zj@rock-chips.com
|
|
*/
|
|
#ifndef __MPPENCODESERVER_H__
|
|
#define __MPPENCODESERVER_H__
|
|
|
|
#include <media/stagefright/foundation/ABase.h>
|
|
#include <media/stagefright/foundation/ADebug.h>
|
|
#include <media/stagefright/foundation/AHandler.h>
|
|
#include <media/stagefright/foundation/ALooper.h>
|
|
#include <media/stagefright/foundation/AMessage.h>
|
|
#include <media/stagefright/foundation/AString.h>
|
|
#include <media/stagefright/foundation/Mutexed.h>
|
|
|
|
#include <thread>
|
|
|
|
#include "OutFrameThread.h"
|
|
#include "RKMppEncApi.h"
|
|
#include "rk_mpi.h"
|
|
using namespace android;
|
|
|
|
/**
|
|
* Called when an input buffer becomes available.
|
|
* The specified index is the index of the available input buffer.
|
|
*/
|
|
typedef void (*OnInputAvailable)(int32_t fd);
|
|
|
|
void OnInputAvailableCB(int32_t fd);
|
|
|
|
typedef struct NotifyCallback {
|
|
OnInputAvailable onInputAvailable;
|
|
} NotifyCallback;
|
|
|
|
class MppEncodeServer : public std::enable_shared_from_this<MppEncodeServer>,
|
|
public Runnable {
|
|
public:
|
|
MppEncodeServer();
|
|
~MppEncodeServer();
|
|
|
|
typedef struct {
|
|
char dev_name[64]; // v4l2 device name
|
|
int width; // v4l2 vfmt width
|
|
int height; // v4l2 vfmt height
|
|
int fps;
|
|
char stream_name[64]; // rtsp url stream name
|
|
int port_num; // rtsp port number
|
|
} MetaInfo;
|
|
|
|
bool init(MetaInfo* meta);
|
|
bool setNotifyCallback(NotifyCallback callback, void* userdata);
|
|
bool start();
|
|
bool stop();
|
|
bool reset();
|
|
bool release();
|
|
|
|
// for handler
|
|
bool processQueue();
|
|
// to implement Runnable
|
|
void run();
|
|
|
|
RKMppEncApi* mEncoder;
|
|
NotifyCallback mNotifyCallback;
|
|
FILE* mInputFile = nullptr;
|
|
FILE* mOutputFile = nullptr;
|
|
// This is used by one thread to tell another thread to exit. So it must be
|
|
// atomic.
|
|
std::atomic<bool> mThreadEnabled{false};
|
|
std::atomic<bool> mThreadExited{false};
|
|
// encode getoutPacket thread
|
|
OutFrameThread mOutFrameThread;
|
|
|
|
private:
|
|
class WorkHandler : public AHandler {
|
|
public:
|
|
enum {
|
|
kWhatProcess,
|
|
kWhatInit,
|
|
kWhatStart,
|
|
kWhatStop,
|
|
kWhatReset,
|
|
kWhatRelease,
|
|
};
|
|
|
|
WorkHandler();
|
|
~WorkHandler() override = default;
|
|
|
|
void setComponent(MppEncodeServer* thiz);
|
|
|
|
protected:
|
|
void onMessageReceived(const sp<AMessage>& msg) override;
|
|
|
|
private:
|
|
MppEncodeServer* mThiz;
|
|
};
|
|
|
|
enum {
|
|
UNINITIALIZED,
|
|
STOPPED,
|
|
RUNNING,
|
|
};
|
|
|
|
struct ExecState {
|
|
ExecState() : mState(UNINITIALIZED) {}
|
|
int mState;
|
|
};
|
|
Mutexed<ExecState> mExecState;
|
|
sp<ALooper> mLooper;
|
|
sp<WorkHandler> mHandler;
|
|
|
|
bool initOther(MetaInfo* meta);
|
|
};
|
|
|
|
#endif // __MPPENCODESERVER_H__
|