1 /* 2 * Copyright (C) 2018 The Android Open Source Project 3 * 4 * Licensed under the Apache License, Version 2.0 (the "License"); 5 * you may not use this file except in compliance with the License. 6 * You may obtain a copy of the License at 7 * 8 * http://www.apache.org/licenses/LICENSE-2.0 9 * 10 * Unless required by applicable law or agreed to in writing, software 11 * distributed under the License is distributed on an "AS IS" BASIS, 12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13 * See the License for the specific language governing permissions and 14 * limitations under the License. 15 */ 16 17 #ifndef ANDROID_SERVERS_CAMERA_CAMERA3_DEPTH_COMPOSITE_STREAM_H 18 #define ANDROID_SERVERS_CAMERA_CAMERA3_DEPTH_COMPOSITE_STREAM_H 19 20 #include "common/DepthPhotoProcessor.h" 21 #include <dynamic_depth/imaging_model.h> 22 #include <dynamic_depth/depth_map.h> 23 24 #include <gui/CpuConsumer.h> 25 26 #include "CompositeStream.h" 27 28 using dynamic_depth::DepthMap; 29 using dynamic_depth::Item; 30 using dynamic_depth::ImagingModel; 31 32 namespace android { 33 34 class CameraDeviceClient; 35 class CameraMetadata; 36 class Surface; 37 38 namespace camera3 { 39 40 class DepthCompositeStream : public CompositeStream, public Thread, 41 public CpuConsumer::FrameAvailableListener { 42 43 public: 44 DepthCompositeStream(sp<CameraDeviceBase> device, 45 wp<hardware::camera2::ICameraDeviceCallbacks> cb); 46 ~DepthCompositeStream() override; 47 48 static bool isDepthCompositeStream(const sp<Surface> &surface); 49 50 // CompositeStream overrides 51 status_t createInternalStreams(const std::vector<sp<Surface>>& consumers, 52 bool hasDeferredConsumer, uint32_t width, uint32_t height, int format, 53 camera_stream_rotation_t rotation, int *id, const String8& physicalCameraId, 54 const std::unordered_set<int32_t> &sensorPixelModesUsed, 55 std::vector<int> *surfaceIds, 56 int streamSetId, bool isShared, int32_t colorSpace, 57 int64_t dynamicProfile, int64_t streamUseCase, bool useReadoutTimestamp) override; 58 status_t deleteInternalStreams() override; 59 status_t configureStream() override; 60 status_t insertGbp(SurfaceMap* /*out*/outSurfaceMap, Vector<int32_t>* /*out*/outputStreamIds, 61 int32_t* /*out*/currentStreamId) override; 62 status_t insertCompositeStreamIds(std::vector<int32_t>* compositeStreamIds /*out*/) override; getStreamId()63 int getStreamId() override { return mBlobStreamId; } 64 65 // CpuConsumer listener implementation 66 void onFrameAvailable(const BufferItem& item) override; 67 68 // Return stream information about the internal camera streams 69 static status_t getCompositeStreamInfo(const OutputStreamInfo &streamInfo, 70 const CameraMetadata& ch, std::vector<OutputStreamInfo>* compositeOutput /*out*/); 71 72 // Get composite stream stats getStreamStats(hardware::CameraStreamStats *)73 void getStreamStats(hardware::CameraStreamStats*) override {}; 74 75 protected: 76 77 bool threadLoop() override; 78 bool onStreamBufferError(const CaptureResultExtras& resultExtras) override; 79 void onResultError(const CaptureResultExtras& resultExtras) override; 80 81 private: 82 struct InputFrame { 83 CpuConsumer::LockedBuffer depthBuffer; 84 CpuConsumer::LockedBuffer jpegBuffer; 85 CameraMetadata result; 86 bool error; 87 bool errorNotified; 88 int64_t frameNumber; 89 int32_t requestId; 90 InputFrameInputFrame91 InputFrame() : error(false), errorNotified(false), frameNumber(-1), requestId(-1) { } 92 }; 93 94 // Helper methods 95 static void getSupportedDepthSizes(const CameraMetadata& ch, bool maxResolution, 96 std::vector<std::tuple<size_t, size_t>>* depthSizes /*out*/); 97 static status_t getMatchingDepthSize(size_t width, size_t height, 98 const std::vector<std::tuple<size_t, size_t>>& supporedDepthSizes, 99 size_t *depthWidth /*out*/, size_t *depthHeight /*out*/); 100 static status_t checkAndGetMatchingDepthSize(size_t width, size_t height, 101 const std::vector<std::tuple<size_t, size_t>> &depthSizes, 102 const std::vector<std::tuple<size_t, size_t>> &depthSizesMaximumResolution, 103 const std::unordered_set<int32_t> &sensorPixelModesUsed, 104 size_t *depthWidth /*out*/, size_t *depthHeight /*out*/); 105 106 107 // Dynamic depth processing 108 status_t encodeGrayscaleJpeg(size_t width, size_t height, uint8_t *in, void *out, 109 const size_t maxOutSize, uint8_t jpegQuality, size_t &actualSize); 110 std::unique_ptr<DepthMap> processDepthMapFrame(const CpuConsumer::LockedBuffer &depthMapBuffer, 111 size_t maxJpegSize, uint8_t jpegQuality, 112 std::vector<std::unique_ptr<Item>>* items /*out*/); 113 std::unique_ptr<ImagingModel> getImagingModel(); 114 status_t processInputFrame(nsecs_t ts, const InputFrame &inputFrame); 115 116 // Buffer/Results handling 117 void compilePendingInputLocked(); 118 void releaseInputFrameLocked(InputFrame *inputFrame /*out*/); 119 void releaseInputFramesLocked(int64_t currentTs); 120 121 // Find first complete and valid frame with smallest timestamp 122 bool getNextReadyInputLocked(int64_t *currentTs /*inout*/); 123 124 // Find next failing frame number with smallest timestamp and return respective frame number 125 int64_t getNextFailingInputLocked(int64_t *currentTs /*inout*/); 126 127 static const nsecs_t kWaitDuration = 10000000; // 10 ms 128 static const auto kDepthMapPixelFormat = HAL_PIXEL_FORMAT_Y16; 129 static const auto kDepthMapDataSpace = HAL_DATASPACE_DEPTH; 130 static const auto kJpegDataSpace = HAL_DATASPACE_V0_JFIF; 131 132 int mBlobStreamId, mBlobSurfaceId, mDepthStreamId, mDepthSurfaceId; 133 size_t mBlobWidth, mBlobHeight; 134 sp<CpuConsumer> mBlobConsumer, mDepthConsumer; 135 bool mDepthBufferAcquired, mBlobBufferAcquired; 136 sp<Surface> mDepthSurface, mBlobSurface, mOutputSurface; 137 sp<ProducerListener> mProducerListener; 138 139 ssize_t mMaxJpegBufferSize; 140 ssize_t mUHRMaxJpegBufferSize; 141 142 camera3::Size mDefaultMaxJpegSize; 143 camera3::Size mUHRMaxJpegSize; 144 145 std::vector<std::tuple<size_t, size_t>> mSupportedDepthSizes; 146 std::vector<std::tuple<size_t, size_t>> mSupportedDepthSizesMaximumResolution; 147 std::vector<float> mIntrinsicCalibration, mLensDistortion; 148 bool mIsLogicalCamera; 149 150 // Keep all incoming Depth buffer timestamps pending further processing. 151 std::vector<int64_t> mInputDepthBuffers; 152 153 // Keep all incoming Jpeg/Blob buffer timestamps pending further processing. 154 std::vector<int64_t> mInputJpegBuffers; 155 156 // Map of all input frames pending further processing. 157 std::unordered_map<int64_t, InputFrame> mPendingInputFrames; 158 }; 159 160 }; //namespace camera3 161 }; //namespace android 162 163 #endif 164