xref: /ohos5.0/foundation/graphic/graphic_2d/rosen/modules/render_service_base/src/animation/rs_render_particle_system.cpp

/*
 * Copyright (c) 2023 Huawei Device Co., Ltd.
 * 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.
 */

#include "animation/rs_render_particle_system.h"

#include <cstddef>
namespace OHOS {
namespace Rosen {
RSRenderParticleSystem::RSRenderParticleSystem(
    const std::vector<std::shared_ptr<ParticleRenderParams>>& particlesRenderParams)
    : particlesRenderParams_(particlesRenderParams)
{
    CreateEmitter();
}

void RSRenderParticleSystem::CreateEmitter()
{
    size_t index = 0;
    for (size_t iter = 0; iter < particlesRenderParams_.size(); iter++) {
        auto& particleRenderParams = particlesRenderParams_[iter];
        if (particleRenderParams != nullptr) {
            if (particleRenderParams->GetParticleType() == ParticleType::IMAGES) {
                particleRenderParams->SetImageIndex(index++);
                auto& image = particleRenderParams->GetParticleImage();
                imageVector_.push_back(image);
            }
            emitters_.push_back(std::make_shared<RSRenderParticleEmitter>(particleRenderParams));
        }
    }
}

void RSRenderParticleSystem::ClearEmitter()
{
    emitters_.clear();
}

void RSRenderParticleSystem::Emit(int64_t deltaTime, std::vector<std::shared_ptr<RSRenderParticle>>& activeParticles,
    std::vector<std::shared_ptr<RSImage>>& imageVector)
{
    for (size_t iter = 0; iter < emitters_.size(); iter++) {
        if (emitters_[iter] != nullptr) {
            emitters_[iter]->EmitParticle(deltaTime);
            auto& particles = emitters_[iter]->GetParticles();
            activeParticles.insert(activeParticles.end(), particles.begin(), particles.end());
        }
    }
    imageVector = imageVector_;
}

void RSRenderParticleSystem::UpdateParticle(
    int64_t deltaTime, std::vector<std::shared_ptr<RSRenderParticle>>& activeParticles)
{
    if (activeParticles.empty()) {
        return;
    }
    for (auto it = activeParticles.begin(); it != activeParticles.end();) {
        // std::shared_ptr<RSRenderParticle> particle = *it;
        if ((*it) == nullptr || !(*it)->IsAlive()) {
            it = activeParticles.erase(it);
        } else {
            Update((*it), particleNoiseFields_, deltaTime);
            ++it;
        }
    }
}

bool RSRenderParticleSystem::IsFinish(const std::vector<std::shared_ptr<RSRenderParticle>>& activeParticles)
{
    bool finish = true;
    if (!activeParticles.empty()) {
        return false;
    }
    for (size_t iter = 0; iter < emitters_.size(); iter++) {
        if (emitters_[iter] != nullptr) {
            finish = finish && emitters_[iter]->IsEmitterFinish();
        }
    }
    return finish;
}

void RSRenderParticleSystem::UpdateEmitter(
    const std::vector<std::shared_ptr<ParticleRenderParams>>& particlesRenderParams)
{
    particlesRenderParams_ = particlesRenderParams;
}

void RSRenderParticleSystem::UpdateNoiseField(const std::shared_ptr<ParticleNoiseFields>& particleNoiseFields)
{
    particleNoiseFields_ = particleNoiseFields;
}
} // namespace Rosen
} // namespace OHOS