// Copyright (c) 2009 The Foundry Visionmongers Ltd.  All Rights Reserved.

static const char* const CLASS = "AddChannels";

static const char* const HELP =
  "Adds channels to the image. If they are not in the input they are "
  "created and filled with black or the defined color.";

#include "DDImage/NoIop.h"
#include "DDImage/Knobs.h"
#include "DDImage/Row.h"

using namespace DD::Image;

class AddChannels : public NoIop
{
  ChannelSet channels;
  ChannelSet channels2;
  ChannelSet channels3;
  ChannelSet channels4;
  float color[4];
  bool _entireFormat;   // When true, fill the entire format's channels, not just the
                        // region that's been requested.
public:
  void _validate(bool) override;
  void engine(int y, int x, int r, ChannelMask channels, Row& row) override;
  void _request(int x, int y, int r, int t, ChannelMask channels, int count) override;
  AddChannels(Node* node) : NoIop(node)
  {
    channels = Mask_None;
    channels2 = channels3 = channels4 = Mask_None;
    color[0] = color[1] = color[2] = color[3] = 0;
    _entireFormat = false;
  }
  void knobs(Knob_Callback) override;
  const char* Class() const override { return CLASS; }
  const char* node_help() const override { return HELP; }
  static Iop::Description d;
};

void AddChannels::_validate(bool for_real)
{
  copy_info();
  if (_entireFormat)  {
    info_.x(info_.format().x());
    info_.y(info_.format().y());
    info_.r(info_.format().r());
    info_.t(info_.format().t());
  }
  ChannelSet newchan = channels;
  newchan += (channels2);
  newchan += (channels3);
  newchan += (channels4);
  set_out_channels(newchan);
  info_.turn_on(newchan);
}

void AddChannels::_request(int x, int y, int r, int t, ChannelMask channels, int count)
{
  input0().request(x, y, r, t, channels, count);
}

void AddChannels::knobs(Knob_Callback f)
{
  ChannelMask_knob(f, &channels, "channels");
  ChannelMask_knob(f, &channels2, "channels2", "and");
  ChannelMask_knob(f, &channels3, "channels3", "and");
  ChannelMask_knob(f, &channels4, "channels4", "and");
  AColor_knob(f, color, "color");
  Tooltip(f, "Color to fill in any new channels. Existing channels are "
             "not changed."
             "\n@i;Colors do not "
             "work correctly in the current version except for rgba! Use "
             "only gray shades for other channels.");

  Newline(f);
  Bool_knob(f, &_entireFormat, "format_size", "add channels to format size");
  Tooltip(f, "When checked, add the channels to every pixel in the format, not just the requested region");
}

void AddChannels::engine(int y, int x, int r, ChannelMask channels, Row& row)
{
  input0().get(y, x, r, channels, row);
  ChannelSet m(channels);
  m -= (input0().channels());
  foreach (z, m) {
    int i = z - 1;
    if (i > 3)
      i = 3;
    if (!color[i])
      continue;
    float* TO = row.writable(z) + x;
    float* END = TO + r - x;
    while (TO < END)
      *TO++ = color[i];
  }
}

static Iop* build(Node* node) { return new AddChannels(node); }
Iop::Description AddChannels::d(CLASS, nullptr, build);