talksik/mediapipe
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Project import generated by Copybara.

Browse Source 
GitOrigin-RevId: d073f8e21be2fcc0e503cb97c6695078b6b75310
This commit is contained in:
MediaPipe Team
2021-02-27 03:30:05 -05:00
committed by chuoling
parent 39309bedba
commit 350fbb2100
755 changed files with 16391 additions and 11075 deletions
Download Patch File Download Diff File Expand all files Collapse all files
mediapipe/objc/MPPGraphTests.mm
+22 -28
View File
@@ -28,15 +28,14 @@ namespace mediapipe {
class GrayscaleCalculator : public Calculator {
public:
static ::mediapipe::Status FillExpectations(const CalculatorOptions& options,
PacketTypeSet* inputs, PacketTypeSet* outputs,
PacketTypeSet* input_side_packets) {
static absl::Status FillExpectations(const CalculatorOptions& options, PacketTypeSet* inputs,
PacketTypeSet* outputs, PacketTypeSet* input_side_packets) {
inputs->Index(0).Set<ImageFrame>();
outputs->Index(0).Set<ImageFrame>();
return ::util::OkStatus();
return absl::OkStatus();
}
::mediapipe::Status Process() final {
absl::Status Process() final {
const auto& input = Input()->Get<ImageFrame>();
int w = input.Width();
int h = input.Height();
@@ -49,7 +48,7 @@ class GrayscaleCalculator : public Calculator {
NSCAssert(vErr == kvImageNoError, @"vImageRGBAToGray failed: %zd", vErr);
Output()->Add(output.release(), InputTimestamp());
return ::util::OkStatus();
return absl::OkStatus();
}
};
REGISTER_CALCULATOR(GrayscaleCalculator);
@@ -58,41 +57,37 @@ REGISTER_CALCULATOR(GrayscaleCalculator);
// if the video header is not present in the input stream.
class VideoHeaderCalculator : public Calculator {
public:
static ::mediapipe::Status FillExpectations(const CalculatorOptions& options,
PacketTypeSet* inputs, PacketTypeSet* outputs,
PacketTypeSet* input_side_packets) {
static absl::Status FillExpectations(const CalculatorOptions& options, PacketTypeSet* inputs,
PacketTypeSet* outputs, PacketTypeSet* input_side_packets) {
inputs->Index(0).Set<ImageFrame>();
outputs->Index(0).Set<ImageFrame>();
return ::util::OkStatus();
return absl::OkStatus();
}
::mediapipe::Status Open() final {
absl::Status Open() final {
if (Input()->Header().IsEmpty()) {
return ::util::UnknownError("No video header present.");
return absl::UnknownError("No video header present.");
}
return ::util::OkStatus();
return absl::OkStatus();
}
::mediapipe::Status Process() final {
absl::Status Process() final {
Output()->AddPacket(Input()->Value());
return ::util::OkStatus();
return absl::OkStatus();
}
};
REGISTER_CALCULATOR(VideoHeaderCalculator);
class ErrorCalculator : public Calculator {
public:
static ::mediapipe::Status FillExpectations(const CalculatorOptions& options,
PacketTypeSet* inputs, PacketTypeSet* outputs,
PacketTypeSet* input_side_packets) {
static absl::Status FillExpectations(const CalculatorOptions& options, PacketTypeSet* inputs,
PacketTypeSet* outputs, PacketTypeSet* input_side_packets) {
inputs->Index(0).SetAny();
outputs->Index(0).SetSameAs(&inputs->Index(0));
return ::util::OkStatus();
return absl::OkStatus();
}
::mediapipe::Status Process() final {
return ::mediapipe::Status(absl::StatusCode::kUnknown, kExpectedError);
}
absl::Status Process() final { return absl::Status(absl::StatusCode::kUnknown, kExpectedError); }
};
REGISTER_CALCULATOR(ErrorCalculator);
@@ -127,7 +122,7 @@ REGISTER_CALCULATOR(ErrorCalculator);
_graph = [[MPPGraph alloc] initWithGraphConfig:config];
[_graph addFrameOutputStream:"output_frames" outputPacketType:MPPPacketTypePixelBuffer];
CFHolder<CVPixelBufferRef> inputBuffer;
::mediapipe::Status status = CreateCVPixelBufferFromCGImage(_sourceImage.CGImage, &inputBuffer);
absl::Status status = CreateCVPixelBufferFromCGImage(_sourceImage.CGImage, &inputBuffer);
XCTAssert(status.ok());
CVPixelBufferRef outputBuffer = [self runGraph:_graph
withPixelBuffer:*inputBuffer
@@ -166,7 +161,7 @@ REGISTER_CALCULATOR(ErrorCalculator);
[_graph addFrameOutputStream:"gray_frames" outputPacketType:MPPPacketTypeImageFrame];
CFHolder<CVPixelBufferRef> inputBuffer;
::mediapipe::Status status = CreateCVPixelBufferFromCGImage(_sourceImage.CGImage, &inputBuffer);
absl::Status status = CreateCVPixelBufferFromCGImage(_sourceImage.CGImage, &inputBuffer);
XCTAssert(status.ok());
WEAKIFY(self);
@@ -204,7 +199,7 @@ REGISTER_CALCULATOR(ErrorCalculator);
_graph = [[MPPGraph alloc] initWithGraphConfig:config];
[_graph addFrameOutputStream:"output_frames" outputPacketType:MPPPacketTypeImageFrame];
CFHolder<CVPixelBufferRef> inputBuffer;
::mediapipe::Status status = CreateCVPixelBufferFromCGImage(grayImage.CGImage, &inputBuffer);
absl::Status status = CreateCVPixelBufferFromCGImage(grayImage.CGImage, &inputBuffer);
XCTAssert(status.ok());
CVPixelBufferRef outputBuffer = [self runGraph:_graph
withPixelBuffer:*inputBuffer
@@ -222,8 +217,7 @@ REGISTER_CALCULATOR(ErrorCalculator);
node->add_input_stream("input_frames");
node->add_output_stream("output_frames");
CFHolder<CVPixelBufferRef> srcPixelBuffer;
::mediapipe::Status status =
CreateCVPixelBufferFromCGImage(_sourceImage.CGImage, &srcPixelBuffer);
absl::Status status = CreateCVPixelBufferFromCGImage(_sourceImage.CGImage, &srcPixelBuffer);
XCTAssert(status.ok());
_graph = [[MPPGraph alloc] initWithGraphConfig:config];
[_graph addFrameOutputStream:"output_frames" outputPacketType:MPPPacketTypeImageFrame];
@@ -290,7 +284,7 @@ REGISTER_CALCULATOR(ErrorCalculator);
_graph = [[MPPGraph alloc] initWithGraphConfig:config];
[_graph addFrameOutputStream:"output_frames" outputPacketType:MPPPacketTypePixelBuffer];
CFHolder<CVPixelBufferRef> inputBuffer;
::mediapipe::Status status = CreateCVPixelBufferFromCGImage(_sourceImage.CGImage, &inputBuffer);
absl::Status status = CreateCVPixelBufferFromCGImage(_sourceImage.CGImage, &inputBuffer);
XCTAssert(status.ok());
CVPixelBufferRef outputBuffer = [self runGraph:_graph
withPixelBuffer:*inputBuffer