// This method is required for all derived classes of EffectBase, and returns a // modified openshot::Frame object tr1::shared_ptr<Frame> Deinterlace::GetFrame(tr1::shared_ptr<Frame> frame, int frame_number) { // Get original size of frame's image int original_width = frame->GetImage()->width(); int original_height = frame->GetImage()->height(); // Get the frame's image tr1::shared_ptr<QImage> image = frame->GetImage(); const unsigned char* pixels = image->bits(); // Create a smaller, new image QImage deinterlaced_image(image->width(), image->height() / 2, QImage::Format_RGBA8888); const unsigned char* deinterlaced_pixels = deinterlaced_image.bits(); // Loop through the scanlines of the image (even or odd) int start = 0; if (isOdd) start = 1; for (int row = start; row < image->height(); row += 2) { memcpy((unsigned char*)deinterlaced_pixels, pixels + (row * image->bytesPerLine()), image->bytesPerLine()); deinterlaced_pixels += image->bytesPerLine(); } // Resize deinterlaced image back to original size, and update frame's image image = tr1::shared_ptr<QImage>(new QImage(deinterlaced_image.scaled(original_width, original_height, Qt::IgnoreAspectRatio, Qt::FastTransformation))); // Update image on frame frame->AddImage(image); // return the modified frame return frame; }
// This method is required for all derived classes of EffectBase, and returns a // modified openshot::Frame object tr1::shared_ptr<Frame> ChromaKey::GetFrame(tr1::shared_ptr<Frame> frame, int frame_number) { // Determine the current HSL (Hue, Saturation, Lightness) for the Chrome int threshold = fuzz.GetInt(frame_number); long mask_R = color.red.GetInt(frame_number); long mask_G = color.green.GetInt(frame_number); long mask_B = color.blue.GetInt(frame_number); // Get source image's pixels tr1::shared_ptr<QImage> image = frame->GetImage(); unsigned char *pixels = (unsigned char *) image->bits(); // Loop through pixels for (int pixel = 0, byte_index=0; pixel < image->width() * image->height(); pixel++, byte_index+=4) { // Get the RGB values from the pixel unsigned char R = pixels[byte_index]; unsigned char G = pixels[byte_index + 1]; unsigned char B = pixels[byte_index + 2]; unsigned char A = pixels[byte_index + 3]; // Get distance between mask color and pixel color long distance = Color::GetDistance((long)R, (long)G, (long)B, mask_R, mask_G, mask_B); // Alpha out the pixel (if color similar) if (distance <= threshold) // MATCHED - Make pixel transparent pixels[byte_index + 3] = 0; } // return the modified frame return frame; }
// Process a new layer of video or audio void Timeline::add_layer(tr1::shared_ptr<Frame> new_frame, Clip* source_clip, long int clip_frame_number, long int timeline_frame_number, bool is_top_clip) { // Get the clip's frame & image tr1::shared_ptr<Frame> source_frame = GetOrCreateFrame(source_clip, clip_frame_number); // No frame found... so bail if (!source_frame) return; // Debug output AppendDebugMethod("Timeline::add_layer", "new_frame->number", new_frame->number, "clip_frame_number", clip_frame_number, "timeline_frame_number", timeline_frame_number, "", -1, "", -1, "", -1); /* REPLACE IMAGE WITH WAVEFORM IMAGE (IF NEEDED) */ if (source_clip->Waveform()) { // Debug output AppendDebugMethod("Timeline::add_layer (Generate Waveform Image)", "source_frame->number", source_frame->number, "source_clip->Waveform()", source_clip->Waveform(), "clip_frame_number", clip_frame_number, "", -1, "", -1, "", -1); // Get the color of the waveform int red = source_clip->wave_color.red.GetInt(clip_frame_number); int green = source_clip->wave_color.green.GetInt(clip_frame_number); int blue = source_clip->wave_color.blue.GetInt(clip_frame_number); int alpha = source_clip->wave_color.alpha.GetInt(clip_frame_number); // Generate Waveform Dynamically (the size of the timeline) tr1::shared_ptr<QImage> source_image = source_frame->GetWaveform(info.width, info.height, red, green, blue, alpha); source_frame->AddImage(tr1::shared_ptr<QImage>(source_image)); } /* Apply effects to the source frame (if any). If multiple clips are overlapping, only process the * effects on the top clip. */ if (is_top_clip) source_frame = apply_effects(source_frame, timeline_frame_number, source_clip->Layer()); // Declare an image to hold the source frame's image tr1::shared_ptr<QImage> source_image; /* COPY AUDIO - with correct volume */ if (source_clip->Reader()->info.has_audio) { // Debug output AppendDebugMethod("Timeline::add_layer (Copy Audio)", "source_clip->Reader()->info.has_audio", source_clip->Reader()->info.has_audio, "source_frame->GetAudioChannelsCount()", source_frame->GetAudioChannelsCount(), "info.channels", info.channels, "clip_frame_number", clip_frame_number, "timeline_frame_number", timeline_frame_number, "", -1); if (source_frame->GetAudioChannelsCount() == info.channels) for (int channel = 0; channel < source_frame->GetAudioChannelsCount(); channel++) { float initial_volume = 1.0f; float previous_volume = source_clip->volume.GetValue(clip_frame_number - 1); // previous frame's percentage of volume (0 to 1) float volume = source_clip->volume.GetValue(clip_frame_number); // percentage of volume (0 to 1) // If no ramp needed, set initial volume = clip's volume if (isEqual(previous_volume, volume)) initial_volume = volume; // Apply ramp to source frame (if needed) if (!isEqual(previous_volume, volume)) source_frame->ApplyGainRamp(channel, 0, source_frame->GetAudioSamplesCount(), previous_volume, volume); // TODO: Improve FrameMapper (or Timeline) to always get the correct number of samples per frame. // Currently, the ResampleContext sometimes leaves behind a few samples for the next call, and the // number of samples returned is variable... and does not match the number expected. // This is a crude solution at best. =) if (new_frame->GetAudioSamplesCount() != source_frame->GetAudioSamplesCount()) // Force timeline frame to match the source frame new_frame->ResizeAudio(info.channels, source_frame->GetAudioSamplesCount(), info.sample_rate, info.channel_layout); // Copy audio samples (and set initial volume). Mix samples with existing audio samples. The gains are added together, to // be sure to set the gain's correctly, so the sum does not exceed 1.0 (of audio distortion will happen). new_frame->AddAudio(false, channel, 0, source_frame->GetAudioSamples(channel), source_frame->GetAudioSamplesCount(), initial_volume); } else // Debug output AppendDebugMethod("Timeline::add_layer (No Audio Copied - Wrong # of Channels)", "source_clip->Reader()->info.has_audio", source_clip->Reader()->info.has_audio, "source_frame->GetAudioChannelsCount()", source_frame->GetAudioChannelsCount(), "info.channels", info.channels, "clip_frame_number", clip_frame_number, "timeline_frame_number", timeline_frame_number, "", -1); } // Skip out if only an audio frame if (!source_clip->Waveform() && !source_clip->Reader()->info.has_video) // Skip the rest of the image processing for performance reasons return; // Debug output AppendDebugMethod("Timeline::add_layer (Get Source Image)", "source_frame->number", source_frame->number, "source_clip->Waveform()", source_clip->Waveform(), "clip_frame_number", clip_frame_number, "", -1, "", -1, "", -1); // Get actual frame image data source_image = source_frame->GetImage(); // Get some basic image properties int source_width = source_image->width(); int source_height = source_image->height(); /* ALPHA & OPACITY */ if (source_clip->alpha.GetValue(clip_frame_number) != 1.0) { float alpha = source_clip->alpha.GetValue(clip_frame_number); // Get source image's pixels unsigned char *pixels = (unsigned char *) source_image->bits(); // Loop through pixels for (int pixel = 0, byte_index=0; pixel < source_image->width() * source_image->height(); pixel++, byte_index+=4) { // Get the alpha values from the pixel int A = pixels[byte_index + 3]; // Apply alpha to pixel pixels[byte_index + 3] *= alpha; } // Debug output AppendDebugMethod("Timeline::add_layer (Set Alpha & Opacity)", "alpha", alpha, "source_frame->number", source_frame->number, "clip_frame_number", clip_frame_number, "", -1, "", -1, "", -1); } /* RESIZE SOURCE IMAGE - based on scale type */ switch (source_clip->scale) { case (SCALE_FIT): // keep aspect ratio source_image = tr1::shared_ptr<QImage>(new QImage(source_image->scaled(info.width, info.height, Qt::KeepAspectRatio, Qt::SmoothTransformation))); source_width = source_image->width(); source_height = source_image->height(); // Debug output AppendDebugMethod("Timeline::add_layer (Scale: SCALE_FIT)", "source_frame->number", source_frame->number, "source_width", source_width, "source_height", source_height, "", -1, "", -1, "", -1); break; case (SCALE_STRETCH): // ignore aspect ratio source_image = tr1::shared_ptr<QImage>(new QImage(source_image->scaled(info.width, info.height, Qt::IgnoreAspectRatio, Qt::SmoothTransformation))); source_width = source_image->width(); source_height = source_image->height(); // Debug output AppendDebugMethod("Timeline::add_layer (Scale: SCALE_STRETCH)", "source_frame->number", source_frame->number, "source_width", source_width, "source_height", source_height, "", -1, "", -1, "", -1); break; case (SCALE_CROP): QSize width_size(info.width, round(info.width / (float(source_width) / float(source_height)))); QSize height_size(round(info.height / (float(source_height) / float(source_width))), info.height); // respect aspect ratio if (width_size.width() >= info.width && width_size.height() >= info.height) source_image = tr1::shared_ptr<QImage>(new QImage(source_image->scaled(width_size.width(), width_size.height(), Qt::KeepAspectRatio, Qt::SmoothTransformation))); else source_image = tr1::shared_ptr<QImage>(new QImage(source_image->scaled(height_size.width(), height_size.height(), Qt::KeepAspectRatio, Qt::SmoothTransformation))); // height is larger, so resize to it source_width = source_image->width(); source_height = source_image->height(); // Debug output AppendDebugMethod("Timeline::add_layer (Scale: SCALE_CROP)", "source_frame->number", source_frame->number, "source_width", source_width, "source_height", source_height, "", -1, "", -1, "", -1); break; } /* GRAVITY LOCATION - Initialize X & Y to the correct values (before applying location curves) */ float x = 0.0; // left float y = 0.0; // top // Adjust size for scale x and scale y float sx = source_clip->scale_x.GetValue(clip_frame_number); // percentage X scale float sy = source_clip->scale_y.GetValue(clip_frame_number); // percentage Y scale float scaled_source_width = source_width * sx; float scaled_source_height = source_height * sy; switch (source_clip->gravity) { case (GRAVITY_TOP): x = (info.width - scaled_source_width) / 2.0; // center break; case (GRAVITY_TOP_RIGHT): x = info.width - scaled_source_width; // right break; case (GRAVITY_LEFT): y = (info.height - scaled_source_height) / 2.0; // center break; case (GRAVITY_CENTER): x = (info.width - scaled_source_width) / 2.0; // center y = (info.height - scaled_source_height) / 2.0; // center break; case (GRAVITY_RIGHT): x = info.width - scaled_source_width; // right y = (info.height - scaled_source_height) / 2.0; // center break; case (GRAVITY_BOTTOM_LEFT): y = (info.height - scaled_source_height); // bottom break; case (GRAVITY_BOTTOM): x = (info.width - scaled_source_width) / 2.0; // center y = (info.height - scaled_source_height); // bottom break; case (GRAVITY_BOTTOM_RIGHT): x = info.width - scaled_source_width; // right y = (info.height - scaled_source_height); // bottom break; } // Debug output AppendDebugMethod("Timeline::add_layer (Gravity)", "source_frame->number", source_frame->number, "source_clip->gravity", source_clip->gravity, "info.width", info.width, "source_width", source_width, "info.height", info.height, "source_height", source_height); /* LOCATION, ROTATION, AND SCALE */ float r = source_clip->rotation.GetValue(clip_frame_number); // rotate in degrees x += (info.width * source_clip->location_x.GetValue(clip_frame_number)); // move in percentage of final width y += (info.height * source_clip->location_y.GetValue(clip_frame_number)); // move in percentage of final height bool is_x_animated = source_clip->location_x.Points.size() > 1; bool is_y_animated = source_clip->location_y.Points.size() > 1; int offset_x = -1; int offset_y = -1; bool transformed = false; QTransform transform; if ((!isEqual(x, 0) || !isEqual(y, 0)) && (isEqual(r, 0) && isEqual(sx, 1) && isEqual(sy, 1) && !is_x_animated && !is_y_animated)) { // SIMPLE OFFSET AppendDebugMethod("Timeline::add_layer (Transform: SIMPLE)", "source_frame->number", source_frame->number, "x", x, "y", y, "r", r, "sx", sx, "sy", sy); // If only X and Y are different, and no animation is being used (just set the offset for speed) transformed = true; // Set QTransform transform.translate(x, y); } else if (!isEqual(r, 0) || !isEqual(x, 0) || !isEqual(y, 0) || !isEqual(sx, 1) || !isEqual(sy, 1)) { // COMPLEX DISTORTION AppendDebugMethod("Timeline::add_layer (Transform: COMPLEX)", "source_frame->number", source_frame->number, "x", x, "y", y, "r", r, "sx", sx, "sy", sy); // Use the QTransform object, which can be very CPU intensive transformed = true; // Set QTransform if (!isEqual(r, 0)) { // ROTATE CLIP float origin_x = x + (source_width / 2.0); float origin_y = y + (source_height / 2.0); transform.translate(origin_x, origin_y); transform.rotate(r); transform.translate(-origin_x,-origin_y); } // Set QTransform if (!isEqual(x, 0) || !isEqual(y, 0)) { // TRANSLATE/MOVE CLIP transform.translate(x, y); } if (!isEqual(sx, 0) || !isEqual(sy, 0)) { // TRANSLATE/MOVE CLIP transform.scale(sx, sy); } // Debug output AppendDebugMethod("Timeline::add_layer (Transform: COMPLEX: Completed ScaleRotateTranslateDistortion)", "source_frame->number", source_frame->number, "x", x, "y", y, "r", r, "sx", sx, "sy", sy); } // Debug output AppendDebugMethod("Timeline::add_layer (Transform: Composite Image Layer: Prepare)", "source_frame->number", source_frame->number, "offset_x", offset_x, "offset_y", offset_y, "new_frame->GetImage()->width()", new_frame->GetImage()->width(), "transformed", transformed, "", -1); /* COMPOSITE SOURCE IMAGE (LAYER) ONTO FINAL IMAGE */ tr1::shared_ptr<QImage> new_image = new_frame->GetImage(); // Load timeline's new frame image into a QPainter QPainter painter(new_image.get()); painter.setRenderHints(QPainter::Antialiasing | QPainter::SmoothPixmapTransform | QPainter::TextAntialiasing, true); // Apply transform (translate, rotate, scale)... if any if (transformed) painter.setTransform(transform); // Composite a new layer onto the image painter.setCompositionMode(QPainter::CompositionMode_SourceOver); painter.drawImage(0, 0, *source_image); painter.end(); // Debug output AppendDebugMethod("Timeline::add_layer (Transform: Composite Image Layer: Completed)", "source_frame->number", source_frame->number, "offset_x", offset_x, "offset_y", offset_y, "new_frame->GetImage()->width()", new_frame->GetImage()->width(), "transformed", transformed, "", -1); }