Skip to content
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

SCD-4 Append bg photo name to output photo #11

Merged
merged 1 commit into from
Jun 17, 2024
+23 −10
Merged

SCD-4 Append bg photo name to output photo #11

Changes from all commits
Commits
Show all changes
1 commit
Select commit
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
33 changes: 23 additions & 10 deletions StudioClient/DownloadHelper.cs
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -7,12 +7,12 @@ namespace SkylabStudio
{
public class PhotoOptions
{
public List<Image>? Bgs { get; set; }
public List<BgImageResult>? Bgs { get; set; }
public bool? ReturnOnError { get; set; }

public PhotoOptions()
{
Bgs = new List<Image>();
Bgs = new List<BgImageResult>();
ReturnOnError = false;
}
}
Expand All @@ -29,23 +29,36 @@ public DownloadAllPhotosResult()
ErroredPhotos = new List<string>();
}
}

public class BgImageResult
{
public string BgName { get; set; }
public Image BgImage { get; set; }

public BgImageResult(string bgName, Image bgImage)
{
BgName = bgName;
BgImage = bgImage;
}
}

public partial class StudioClient
{
/// <summary>
/// Downloads background images based on the provided profile.
/// </summary>
/// <param name="profile">The profile associated to the job.</param>
/// <returns>List of downloaded background images.</returns>
private async Task<List<Image>?> DownloadBgImages(dynamic profile)
private async Task<List<BgImageResult>?> DownloadBgImages(dynamic profile)
{
List<Image> tempBgs = new List<Image>();
List<BgImageResult> tempBgs = new List<BgImageResult>();
List<JToken> bgPhotos = ((JArray) profile!.photos).Where(photo => photo["jobId"] != null).ToList();

foreach (dynamic bg in bgPhotos)
{
byte[] bgBuffer = await DownloadImageAsync(bg.originalUrl.Value);
Image bgImage = Image.NewFromBuffer(bgBuffer);
tempBgs.Add(bgImage);
tempBgs.Add(new BgImageResult(bg.name.Value, bgImage));
}

return tempBgs;
Expand Down Expand Up @@ -92,7 +105,7 @@ public partial class StudioClient
/// <param name="profile">The profile associated to the job.</param>
/// <param name="bgs">List of background images.</param>
/// <returns>True if the operation is successful; otherwise, false.</returns>
private async Task<bool> DownloadReplacedBackgroundImage(string fileName, Image inputImage, string outputPath, dynamic? profile = null, List<Image>? bgs = null)
private async Task<bool> DownloadReplacedBackgroundImage(string fileName, Image inputImage, string outputPath, dynamic? profile = null, List<BgImageResult>? bgs = null)
{
try
{
Expand All @@ -108,8 +121,8 @@ private async Task<bool> DownloadReplacedBackgroundImage(string fileName, Image

if (bgs != null && bgs.Count > 0){
for (int i = 0; i < bgs.Count; i++) {
string newFileName = i == 0 ? $"{Path.GetFileNameWithoutExtension(fileName)}.{outputFileType}": $"{Path.GetFileNameWithoutExtension(fileName)} ({i+1}).{outputFileType}";
Image resizedBgImage = bgs[i].ThumbnailImage(inputImage.Width, inputImage.Height, crop: Enums.Interesting.Centre);
string newFileName = $"{Path.GetFileNameWithoutExtension(fileName)} ({bgs[i].BgName}).{outputFileType}";
Image resizedBgImage = bgs[i].BgImage.ThumbnailImage(inputImage.Width, inputImage.Height, crop: Enums.Interesting.Centre);
Image resultImage = resizedBgImage.Composite2(rgbCutout, Enums.BlendMode.Over);
resultImage.WriteToFile(Path.Combine(outputPath, newFileName));
}
Expand Down Expand Up @@ -142,7 +155,7 @@ public async Task<DownloadAllPhotosResult> DownloadAllPhotos(JArray photosList,

List<string> successPhotos = new List<string>();
List<string> erroredPhotos = new List<string>();
List<Image> bgs = new List<Image>();
List<BgImageResult> bgs = new List<BgImageResult>();

try {
profile = await GetProfile(profile.id.Value);
Expand Down Expand Up @@ -234,7 +247,7 @@ public async Task<Tuple<string,bool>> DownloadPhoto(long photoId, string output
bool replaceBackground = Convert.ToBoolean(profile.replaceBackground.Value);
bool isDualFileOutput = Convert.ToBoolean(profile.dualFileOutput.Value);
bool enableStripPngMetadata = Convert.ToBoolean(profile.enableStripPngMetadata.Value);
List<Image>? bgs = options?.Bgs;
List<BgImageResult>? bgs = options?.Bgs;

// Load output image
byte[] imageBuffer = await DownloadImageAsync(photo.retouchedUrl.Value);
Expand Down
Loading