Skip to content

feat: add code samples and tests for overlay creation #5464

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
busunkim96 merged 12 commits into from
Mar 30, 2021
Merged

feat: add code samples and tests for overlay creation #5464

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
12 commits
Select commit Hold shift + click to select a range
4642845
feat: add code samples and tests for overlay creation
Mar 5, 2021
52b2c00
Fix lixence header
Mar 5, 2021
3478fbb
Merge branch 'master' into master
irataxy Mar 5, 2021
83c0fe5
Merge branch 'master' into master
irataxy Mar 8, 2021
db90d69
Merge branch 'master' into master
engelke Mar 11, 2021
46ace6f
Merge branch 'master' into master
gcf-merge-on-green[bot] Mar 11, 2021
595b2cc
Merge branch 'master' into master
irataxy Mar 16, 2021
5763c98
Merge branch 'master' into master
irataxy Mar 29, 2021
b8b8507
Update media/transcoder/create_job_with_animated_overlay.py
irataxy Mar 30, 2021
96e7fbc
Fix lint
Mar 30, 2021
d800481
Merge branch 'master' into master
irataxy Mar 30, 2021
a04dc91
Merge branch 'master' into master
irataxy Mar 30, 2021
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Binary file modified media/testdata/ChromeCast.mp4
View file
Open in desktop
Binary file not shown.
Binary file added media/testdata/overlay.jpg
View file
Open in desktop
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
160 changes: 160 additions & 0 deletions media/transcoder/create_job_with_animated_overlay.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1,160 @@
#!/usr/bin/env python

# Copyright 2021 Google LLC
#
# 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.

"""Google Cloud Transcoder sample for creating a job based on a supplied job config that includes an animated overlay.

Example usage:
python create_job_with_animated_overlay.py --project-id <project-id> --location <location> --input-uri <uri> --overlay-image-uri <uri> --output-uri <uri>
"""

import argparse

from google.cloud.video import transcoder_v1beta1
from google.cloud.video.transcoder_v1beta1.services.transcoder_service import (
TranscoderServiceClient,
)
from google.protobuf import duration_pb2 as duration

# [START transcoder_create_job_with_animated_overlay]


def create_job_with_animated_overlay(
project_id, location, input_uri, overlay_image_uri, output_uri
):
"""Creates a job based on an ad-hoc job configuration that includes an animated image overlay.

Args:
project_id: The GCP project ID.
location: The location to start the job in.
input_uri: Uri of the video in the Cloud Storage bucket.
overlay_image_uri: Uri of the JPEG image for the overlay in the Cloud Storage bucket. Must be a JPEG.
output_uri: Uri of the video output folder in the Cloud Storage bucket."""

client = TranscoderServiceClient()
Comment thread
irataxy marked this conversation as resolved.

parent = f"projects/{project_id}/locations/{location}"
job = transcoder_v1beta1.types.Job()
job.input_uri = input_uri
job.output_uri = output_uri
job.config = transcoder_v1beta1.types.JobConfig(
elementary_streams=[
transcoder_v1beta1.types.ElementaryStream(
key="video-stream0",
video_stream=transcoder_v1beta1.types.VideoStream(
codec="h264",
height_pixels=360,
width_pixels=640,
bitrate_bps=550000,
frame_rate=60,
),
),
Comment thread
irataxy marked this conversation as resolved.
transcoder_v1beta1.types.ElementaryStream(
key="audio-stream0",
audio_stream=transcoder_v1beta1.types.AudioStream(
codec="aac", bitrate_bps=64000
),
),
],
mux_streams=[
transcoder_v1beta1.types.MuxStream(
key="sd",
container="mp4",
elementary_streams=["video-stream0", "audio-stream0"],
),
],
overlays=[
transcoder_v1beta1.types.Overlay(
image=transcoder_v1beta1.types.Overlay.Image(
uri=overlay_image_uri,
resolution=transcoder_v1beta1.types.Overlay.NormalizedCoordinate(
x=0,
y=0,
),
alpha=1,
),
animations=[
transcoder_v1beta1.types.Overlay.Animation(
animation_fade=transcoder_v1beta1.types.Overlay.AnimationFade(
fade_type=transcoder_v1beta1.types.Overlay.FadeType.FADE_IN,
xy=transcoder_v1beta1.types.Overlay.NormalizedCoordinate(
x=0.5,
y=0.5,
),
start_time_offset=duration.Duration(
seconds=5,
),
end_time_offset=duration.Duration(
seconds=10,
),
),
),
transcoder_v1beta1.types.Overlay.Animation(
animation_fade=transcoder_v1beta1.types.Overlay.AnimationFade(
fade_type=transcoder_v1beta1.types.Overlay.FadeType.FADE_OUT,
xy=transcoder_v1beta1.types.Overlay.NormalizedCoordinate(
x=0.5,
y=0.5,
),
start_time_offset=duration.Duration(
seconds=12,
),
end_time_offset=duration.Duration(
seconds=15,
),
),
),
],
),
],
)
response = client.create_job(parent=parent, job=job)
print(f"Job: {response.name}")
return response


# [END transcoder_create_job_with_animated_overlay]

if __name__ == "__main__":
parser = argparse.ArgumentParser()
parser.add_argument("--project-id", help="Your Cloud project ID.", required=True)
parser.add_argument(
"--location",
help="The location to start this job in.",
default="us-central1",
)
parser.add_argument(
"--input-uri",
help="Uri of the video in the Cloud Storage bucket.",
required=True,
)
parser.add_argument(
"--overlay-image-uri",
help="Uri of the overlay JPEG image in the Cloud Storage bucket. Must be a JPEG.",
required=True,
)
parser.add_argument(
"--output-uri",
help="Uri of the video output folder in the Cloud Storage bucket. Must end in '/'.",
required=True,
)
args = parser.parse_args()
create_job_with_animated_overlay(
args.project_id,
args.location,
args.input_uri,
args.overlay_image_uri,
args.output_uri,
)
148 changes: 148 additions & 0 deletions media/transcoder/create_job_with_static_overlay.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1,148 @@
#!/usr/bin/env python

# Copyright 2021 Google LLC
#
# 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.

"""Google Cloud Transcoder sample for creating a job based on a supplied job config that includes a static overlay.

Example usage:
python create_job_with_static_overlay.py --project-id <project-id> --location <location> --input-uri <uri> --overlay-image-uri <uri> --output-uri <uri>
"""

import argparse

from google.cloud.video import transcoder_v1beta1
from google.cloud.video.transcoder_v1beta1.services.transcoder_service import (
TranscoderServiceClient,
)
from google.protobuf import duration_pb2 as duration

# [START transcoder_create_job_with_static_overlay]


def create_job_with_static_overlay(
project_id, location, input_uri, overlay_image_uri, output_uri
):
"""Creates a job based on an ad-hoc job configuration that includes a static image overlay.

Args:
project_id: The GCP project ID.
location: The location to start the job in.
input_uri: Uri of the video in the Cloud Storage bucket.
overlay_image_uri: Uri of the JPEG image for the overlay in the Cloud Storage bucket. Must be a JPEG.
output_uri: Uri of the video output folder in the Cloud Storage bucket."""

client = TranscoderServiceClient()

parent = f"projects/{project_id}/locations/{location}"
job = transcoder_v1beta1.types.Job()
job.input_uri = input_uri
job.output_uri = output_uri
job.config = transcoder_v1beta1.types.JobConfig(
elementary_streams=[
transcoder_v1beta1.types.ElementaryStream(
key="video-stream0",
video_stream=transcoder_v1beta1.types.VideoStream(
codec="h264",
height_pixels=360,
width_pixels=640,
bitrate_bps=550000,
frame_rate=60,
),
),
transcoder_v1beta1.types.ElementaryStream(
key="audio-stream0",
audio_stream=transcoder_v1beta1.types.AudioStream(
codec="aac", bitrate_bps=64000
),
),
],
mux_streams=[
transcoder_v1beta1.types.MuxStream(
key="sd",
container="mp4",
elementary_streams=["video-stream0", "audio-stream0"],
),
],
overlays=[
transcoder_v1beta1.types.Overlay(
image=transcoder_v1beta1.types.Overlay.Image(
uri=overlay_image_uri,
resolution=transcoder_v1beta1.types.Overlay.NormalizedCoordinate(
x=1,
y=0.5,
),
alpha=1,
),
animations=[
transcoder_v1beta1.types.Overlay.Animation(
animation_static=transcoder_v1beta1.types.Overlay.AnimationStatic(
xy=transcoder_v1beta1.types.Overlay.NormalizedCoordinate(
x=0,
y=0,
),
start_time_offset=duration.Duration(
seconds=0,
),
),
),
transcoder_v1beta1.types.Overlay.Animation(
animation_end=transcoder_v1beta1.types.Overlay.AnimationEnd(
start_time_offset=duration.Duration(
seconds=10,
),
),
),
],
),
],
)
response = client.create_job(parent=parent, job=job)
print(f"Job: {response.name}")
return response


# [END transcoder_create_job_with_static_overlay]

if __name__ == "__main__":
parser = argparse.ArgumentParser()
parser.add_argument("--project-id", help="Your Cloud project ID.", required=True)
parser.add_argument(
"--location",
help="The location to start this job in.",
default="us-central1",
)
parser.add_argument(
"--input-uri",
help="Uri of the video in the Cloud Storage bucket.",
required=True,
)
parser.add_argument(
"--overlay-image-uri",
help="Uri of the overlay JPEG image in the Cloud Storage bucket. Must be a JPEG.",
required=True,
)
parser.add_argument(
"--output-uri",
help="Uri of the video output folder in the Cloud Storage bucket. Must end in '/'.",
required=True,
)
args = parser.parse_args()
create_job_with_static_overlay(
args.project_id,
args.location,
args.input_uri,
args.overlay_image_uri,
args.output_uri,
)
Loading