Beatmup
Public Types | Public Member Functions | Private Attributes | List of all members
Beatmup::GLES20X2UpsamplingNetwork::Layer Class Reference

Public Types

typedef InternalBitmap * Storage
 

Public Member Functions

 Layer (GL::RecycleBin &recycleBin, GraphicPipeline &gpu, Storage &outputStorage, const char *sourceCode)
 
void process (Context &ctx, GraphicPipeline &gpu, GL::TextureHandler &input)
 
void process (Context &ctx, GraphicPipeline &gpu, Layer **inputs, int inputsCount)
 
InternalBitmap & getOutput ()
 

Private Attributes

ImageShader shader
 
Storage & output
 

Detailed Description

Definition at line 31 of file cnn.h.

Member Typedef Documentation

◆ Storage

typedef InternalBitmap* Beatmup::GLES20X2UpsamplingNetwork::Layer::Storage

Definition at line 33 of file cnn.h.

Constructor & Destructor Documentation

◆ Layer()

GLES20X2UpsamplingNetwork::Layer::Layer ( GL::RecycleBin &  recycleBin,
GraphicPipeline &  gpu,
Storage &  outputStorage,
const char *  sourceCode 
)

Definition at line 29 of file cnn.cpp.

29  :
30  shader(recycleBin), output(outputStorage)
31 {
32  shader.setSourceCode(sourceCode);
33 }
Beatmup::ImageShader::setSourceCode
void setSourceCode(const std::string &sourceCode)
Passes new source code to the fragment shader.
Definition: image_shader.cpp:73

Member Function Documentation

◆ process() [1/2]

void GLES20X2UpsamplingNetwork::Layer::process ( Context &  ctx,
GraphicPipeline &  gpu,
GL::TextureHandler &  input 
)

Definition at line 36 of file cnn.cpp.

36  {
37  // reshape or instantiate output if needed
38  bool sizeChange = false;
39  if (output && (output->getWidth() != input.getWidth() || output->getHeight() != input.getHeight())) {
40  output->reshape(input.getWidth(), input.getHeight());
41  sizeChange = true;
42  }
43  else if (!output) {
44  output = new InternalBitmap(ctx, PixelFormat::QuadByte, input.getWidth(), input.getHeight(), false);
45  sizeChange = true;
46  }
47 
48  // reset size if needed
49  if (sizeChange) {
50  shader.setFloat("d1", 1.0f / input.getWidth(), 1.0f / input.getHeight());
51  shader.setFloat("d2", 2.0f / input.getWidth(), 2.0f / input.getHeight());
52  }
53 
54  // process
55  {
56  AbstractBitmap::WriteLock<ProcessingTarget::GPU> outputLock(*output);
57  shader.prepare(gpu, &input, TextureParam::INTERP_NEAREST, output, AffineMapping::IDENTITY);
58  shader.process(gpu);
59  }
60 }
Beatmup::AbstractBitmap::WriteLock
Makes a bitmap writable for a specific target device.
Definition: abstract_bitmap.h:252
Beatmup::AffineMapping::IDENTITY
static const AffineMapping IDENTITY
Definition: geometry.h:717
Beatmup::GL::TextureHandler::getHeight
virtual const int getHeight() const =0
Height of the texture in pixels.
Beatmup::GL::TextureHandler::getWidth
virtual const int getWidth() const =0
Width of the texture in pixels.
Beatmup::GL::VariablesBundle::setFloat
void setFloat(std::string name, float value)
Sets a scalar float uniform value.
Definition: variables_bundle.cpp:93
Beatmup::ImageShader::process
void process(GraphicPipeline &gpu)
Apply the shader to produce an image.
Definition: image_shader.cpp:195
Beatmup::ImageShader::prepare
void prepare(GraphicPipeline &gpu, GL::TextureHandler *input, const TextureParam texParam, AbstractBitmap *output, const AffineMapping &mapping)
Conducts required preparations for blending.
Definition: image_shader.cpp:86
Beatmup::InternalBitmap
Bitmap whose memory is managed by the Beatmup engine.
Definition: internal_bitmap.h:35
Beatmup::InternalBitmap::getHeight
const int getHeight() const
Height of the texture in pixels.
Definition: internal_bitmap.cpp:103
Beatmup::InternalBitmap::getWidth
const int getWidth() const
Width of the texture in pixels.
Definition: internal_bitmap.cpp:98
Beatmup::InternalBitmap::reshape
void reshape(int width, int height)
Changes bitmap size.
Definition: internal_bitmap.cpp:74
Beatmup::QuadByte
@ QuadByte
4 channels of 8 bits per pixel (like RGBA), unsigned integer values
Definition: abstract_bitmap.h:73
Beatmup::INTERP_NEAREST
@ INTERP_NEAREST
nearest neighbor pixel interpolation
Definition: texture_handler.h:31
ctx
Beatmup::Context * ctx
Definition: wrapper_core.cpp:80

◆ process() [2/2]

void GLES20X2UpsamplingNetwork::Layer::process ( Context &  ctx,
GraphicPipeline &  gpu,
Layer **  inputs,
int  inputsCount 
)

Definition at line 63 of file cnn.cpp.

63  {
64  const InternalBitmap& input = inputs[0]->getOutput();
65  if (output && (output->getWidth() != input.getWidth() || output->getHeight() != input.getHeight()))
66  output->reshape(input.getWidth(), input.getHeight());
67  else if (!output)
68  output = new InternalBitmap(ctx, PixelFormat::QuadByte, input.getWidth(), input.getHeight(), false);
69 
70  // prepare and process
71  {
72  AbstractBitmap::WriteLock<ProcessingTarget::GPU> outputLock(*output);
73  shader.setFloat("d1", 1.0f / input.getWidth(), 1.0f / input.getHeight());
74  shader.prepare(gpu, nullptr, output);
75 
76  // bind images
77  for (int i = 0; i < inputsCount; ++i)
78  gpu.bind(inputs[i]->getOutput(), i, TextureParam::INTERP_NEAREST);
79  shader.bindSamplerArray("images", 0, inputsCount);
80 
81  // process
82  shader.process(gpu);
83  }
84 }
Beatmup::GraphicPipeline::bind
void bind(GL::TextureHandler &texture, size_t texUnit, const TextureParam param)
Definition: pipeline.cpp:881
Beatmup::ImageShader::bindSamplerArray
void bindSamplerArray(const char *uniformId, int startingUnit, int numUnits)
Binds a bunch of texture units to a uniform sampler array variable.
Definition: image_shader.cpp:190

◆ getOutput()

InternalBitmap& Beatmup::GLES20X2UpsamplingNetwork::Layer::getOutput ( )
inline

Definition at line 43 of file cnn.h.

43  {
44  return *output;
45  }

Member Data Documentation

◆ shader

ImageShader Beatmup::GLES20X2UpsamplingNetwork::Layer::shader
private

Definition at line 35 of file cnn.h.

◆ output

Storage& Beatmup::GLES20X2UpsamplingNetwork::Layer::output
private

Definition at line 36 of file cnn.h.

The documentation for this class was generated from the following files: