blob: ea6ba8b3c2f9ca1d2693e264caeedde6d3ed6016 [file] [log] [blame]
// Copyright 2020 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 COMPUTEPIPE_EXAMPLE_FACE_TRACKER_H
#define COMPUTEPIPE_EXAMPLE_FACE_TRACKER_H
#include <aidl/android/automotive/computepipe/registry/BnClientInfo.h>
#include <aidl/android/automotive/computepipe/registry/IPipeQuery.h>
#include <aidl/android/automotive/computepipe/registry/IPipeRegistration.h>
#include <aidl/android/automotive/computepipe/runner/BnPipeStateCallback.h>
#include <aidl/android/automotive/computepipe/runner/BnPipeStream.h>
#include <aidl/android/automotive/computepipe/runner/PipeState.h>
#include <condition_variable>
#include <iostream>
#include <memory>
#include "FaceOutput.pb.h"
namespace android {
namespace automotive {
namespace computepipe {
using ::aidl::android::automotive::computepipe::registry::BnClientInfo;
using ::aidl::android::automotive::computepipe::registry::IPipeQuery;
using ::aidl::android::automotive::computepipe::runner::BnPipeStateCallback;
using ::aidl::android::automotive::computepipe::runner::BnPipeStream;
using ::aidl::android::automotive::computepipe::runner::IPipeRunner;
using ::aidl::android::automotive::computepipe::runner::IPipeStream;
using ::aidl::android::automotive::computepipe::runner::PacketDescriptor;
using ::aidl::android::automotive::computepipe::runner::PipeState;
using ::android::automotive::computepipe::example::BoundingBox;
class RemoteState {
public:
explicit RemoteState(std::function<void(bool, std::string)>& cb);
PipeState GetCurrentState();
void UpdateCurrentState(const PipeState& state);
private:
bool hasChanged = false;
PipeState mState = PipeState::RESET;
std::mutex mStateLock;
std::condition_variable mWait;
std::function<void(bool, std::string)> mTerminationCb;
};
class ClientInfo : public BnClientInfo {
public:
ndk::ScopedAStatus getClientId(int32_t* _aidl_return) {
if (_aidl_return) {
*_aidl_return = 0;
return ndk::ScopedAStatus::ok();
}
return ndk::ScopedAStatus::fromExceptionCode(EX_TRANSACTION_FAILED);
}
};
class StreamCallback : public BnPipeStream {
public:
StreamCallback() = default;
ndk::ScopedAStatus deliverPacket(const PacketDescriptor& in_packet) override;
private:
BoundingBox mLastBox;
};
class StateCallback : public BnPipeStateCallback {
public:
explicit StateCallback(std::shared_ptr<RemoteState> s);
ndk::ScopedAStatus handleState(PipeState state) override;
private:
std::shared_ptr<RemoteState> mStateTracker = nullptr;
};
class FaceTracker {
public:
FaceTracker() = default;
ndk::ScopedAStatus init(std::function<void(bool, std::string)>&& termination);
void start();
void stop();
private:
ndk::ScopedAStatus setupConfig();
std::shared_ptr<IPipeRunner> mPipeRunner = nullptr;
std::shared_ptr<ClientInfo> mClientInfo = nullptr;
std::shared_ptr<StreamCallback> mStreamCallback = nullptr;
std::shared_ptr<StateCallback> mStateCallback = nullptr;
std::shared_ptr<RemoteState> mRemoteState = nullptr;
};
} // namespace computepipe
} // namespace automotive
} // namespace android
#endif