90 lines
2.9 KiB
C++
90 lines
2.9 KiB
C++
|
/*
|
||
|
|
* This program is free software; you can redistribute it and/or
|
||
|
|
* modify it under the terms of the GNU General Public License
|
||
|
|
* as published by the Free Software Foundation; either version 2
|
||
|
|
* of the License, or (at your option) any later version.
|
||
|
|
*
|
||
|
|
* This program is distributed in the hope that it will be useful,
|
||
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||
|
|
* GNU General Public License for more details.
|
||
|
|
*
|
||
|
|
* You should have received a copy of the GNU General Public License
|
||
|
|
* along with this program; if not, write to the Free Software Foundation,
|
||
|
|
* Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
|
||
|
|
*/
|
||
|
|
|
||
|
|
#ifndef __SIM_PARTICLE_FUNCTION_HH__
|
||
|
|
#define __SIM_PARTICLE_FUNCTION_HH__
|
||
|
|
|
||
|
|
#include "FN_attributes_ref.hh"
|
||
|
|
#include "FN_multi_function.hh"
|
||
|
|
|
||
|
|
#include "BLI_resource_collector.hh"
|
||
|
|
|
||
|
|
namespace blender::sim {
|
||
|
|
|
||
|
|
class ParticleFunctionInput {
|
||
|
|
public:
|
||
|
|
virtual ~ParticleFunctionInput() = default;
|
||
|
|
virtual void add_input(fn::AttributesRef attributes,
|
||
|
|
fn::MFParamsBuilder ¶ms,
|
||
|
|
ResourceCollector &resources) const = 0;
|
||
|
|
};
|
||
|
|
|
||
|
|
class ParticleFunction {
|
||
|
|
private:
|
||
|
|
const fn::MultiFunction *global_fn_;
|
||
|
|
const fn::MultiFunction *per_particle_fn_;
|
||
|
|
Array<const ParticleFunctionInput *> global_inputs_;
|
||
|
|
Array<const ParticleFunctionInput *> per_particle_inputs_;
|
||
|
|
Array<bool> output_is_global_;
|
||
|
|
Vector<uint> global_output_indices_;
|
||
|
|
Vector<uint> per_particle_output_indices_;
|
||
|
|
Vector<fn::MFDataType> output_types_;
|
||
|
|
Vector<StringRefNull> output_names_;
|
||
|
|
|
||
|
|
friend class ParticleFunctionEvaluator;
|
||
|
|
|
||
|
|
public:
|
||
|
|
ParticleFunction(const fn::MultiFunction *global_fn,
|
||
|
|
const fn::MultiFunction *per_particle_fn,
|
||
|
|
Span<const ParticleFunctionInput *> global_inputs,
|
||
|
|
Span<const ParticleFunctionInput *> per_particle_inputs,
|
||
|
|
Span<bool> output_is_global);
|
||
|
|
};
|
||
|
|
|
||
|
|
class ParticleFunctionEvaluator {
|
||
|
|
private:
|
||
|
|
ResourceCollector resources_;
|
||
|
|
const ParticleFunction &particle_fn_;
|
||
|
|
IndexMask mask_;
|
||
|
|
fn::MFContextBuilder global_context_;
|
||
|
|
fn::MFContextBuilder per_particle_context_;
|
||
|
|
fn::AttributesRef particle_attributes_;
|
||
|
|
Vector<void *> outputs_;
|
||
|
|
bool is_computed_ = false;
|
||
|
|
|
||
|
|
public:
|
||
|
|
ParticleFunctionEvaluator(const ParticleFunction &particle_fn,
|
||
|
|
IndexMask mask,
|
||
|
|
fn::AttributesRef particle_attributes);
|
||
|
|
~ParticleFunctionEvaluator();
|
||
|
|
|
||
|
|
void compute();
|
||
|
|
fn::GVSpan get(uint output_index, StringRef expected_name) const;
|
||
|
|
|
||
|
|
template<typename T> fn::VSpan<T> get(uint output_index, StringRef expected_name) const
|
||
|
|
{
|
||
|
|
return this->get(output_index, expected_name).typed<T>();
|
||
|
|
}
|
||
|
|
|
||
|
|
private:
|
||
|
|
void compute_globals();
|
||
|
|
void compute_per_particle();
|
||
|
|
};
|
||
|
|
|
||
|
|
} // namespace blender::sim
|
||
|
|
|
||
|
|
#endif /* __SIM_PARTICLE_FUNCTION_HH__ */
|