Blame lib/cfapattern.cpp

rpm-build d2b433
/*
rpm-build d2b433
 * libopenraw - cfapattern.cpp
rpm-build d2b433
 *
rpm-build d2b433
 * Copyright (C) 2012 Hubert Figuière
rpm-build d2b433
 *
rpm-build d2b433
 * This library is free software: you can redistribute it and/or
rpm-build d2b433
 * modify it under the terms of the GNU Lesser General Public License
rpm-build d2b433
 * as published by the Free Software Foundation, either version 3 of
rpm-build d2b433
 * the License, or (at your option) any later version.
rpm-build d2b433
 *
rpm-build d2b433
 * This library is distributed in the hope that it will be useful,
rpm-build d2b433
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
rpm-build d2b433
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
rpm-build d2b433
 * Lesser General Public License for more details.
rpm-build d2b433
 *
rpm-build d2b433
 * You should have received a copy of the GNU Lesser General Public
rpm-build d2b433
 * License along with this library.  If not, see
rpm-build d2b433
 * <http://www.gnu.org/licenses/>.
rpm-build d2b433
 */
rpm-build d2b433
rpm-build d2b433
#include <stdint.h>
rpm-build d2b433
#include <stddef.h>
rpm-build d2b433
rpm-build d2b433
#include <array>
rpm-build d2b433
rpm-build d2b433
#include <boost/static_assert.hpp>
rpm-build d2b433
rpm-build d2b433
#include <libopenraw/consts.h>
rpm-build d2b433
rpm-build d2b433
#include "cfapattern.hpp"
rpm-build d2b433
rpm-build d2b433
namespace OpenRaw {
rpm-build d2b433
rpm-build d2b433
namespace Internals {
rpm-build d2b433
rpm-build d2b433
/** alias the colours. */
rpm-build d2b433
static const uint8_t RED = OR_PATTERN_COLOUR_RED;
rpm-build d2b433
static const uint8_t GREEN = OR_PATTERN_COLOUR_GREEN;
rpm-build d2b433
static const uint8_t BLUE = OR_PATTERN_COLOUR_BLUE;
rpm-build d2b433
rpm-build d2b433
static const uint8_t RGGB_PATTERN[] = { RED, GREEN, GREEN, BLUE };
rpm-build d2b433
static const uint8_t GBRG_PATTERN[] = { GREEN, BLUE, RED, GREEN };
rpm-build d2b433
static const uint8_t BGGR_PATTERN[] = { BLUE, GREEN, GREEN, RED };
rpm-build d2b433
static const uint8_t GRBG_PATTERN[] = { GREEN, RED, BLUE, GREEN };
rpm-build d2b433
rpm-build d2b433
class Cfa2x2RgbPattern
rpm-build d2b433
  : public CfaPattern
rpm-build d2b433
{
rpm-build d2b433
public:
rpm-build d2b433
  Cfa2x2RgbPattern(::or_cfa_pattern pattern)
rpm-build d2b433
    : CfaPattern(pattern, 2, 2)
rpm-build d2b433
    {
rpm-build d2b433
      switch(pattern) {
rpm-build d2b433
      case OR_CFA_PATTERN_RGGB:
rpm-build d2b433
        setPatternPattern(RGGB_PATTERN, 4);
rpm-build d2b433
        break;
rpm-build d2b433
      case OR_CFA_PATTERN_GBRG:
rpm-build d2b433
        setPatternPattern(GBRG_PATTERN, 4);
rpm-build d2b433
        break;
rpm-build d2b433
      case OR_CFA_PATTERN_BGGR:
rpm-build d2b433
        setPatternPattern(BGGR_PATTERN, 4);
rpm-build d2b433
        break;
rpm-build d2b433
      case OR_CFA_PATTERN_GRBG:
rpm-build d2b433
        setPatternPattern(GRBG_PATTERN, 4);
rpm-build d2b433
        break;
rpm-build d2b433
rpm-build d2b433
      default:
rpm-build d2b433
        break;
rpm-build d2b433
      }
rpm-build d2b433
    }
rpm-build d2b433
rpm-build d2b433
};
rpm-build d2b433
rpm-build d2b433
}
rpm-build d2b433
rpm-build d2b433
const CfaPattern*
rpm-build d2b433
CfaPattern::twoByTwoPattern(::or_cfa_pattern pattern)
rpm-build d2b433
{
rpm-build d2b433
  static std::array<CfaPattern*, _OR_CFA_PATTERN_INVALID> s_patterns
rpm-build d2b433
    = { { NULL, NULL, NULL, NULL, NULL, NULL } };
rpm-build d2b433
  // this should be updated if we change the enum
rpm-build d2b433
  BOOST_STATIC_ASSERT(_OR_CFA_PATTERN_INVALID == 6);
rpm-build d2b433
rpm-build d2b433
  if((pattern == OR_CFA_PATTERN_NON_RGB22) ||
rpm-build d2b433
     (pattern >= _OR_CFA_PATTERN_INVALID)) {
rpm-build d2b433
    return NULL;
rpm-build d2b433
  }
rpm-build d2b433
rpm-build d2b433
  CfaPattern* pat = s_patterns[pattern];
rpm-build d2b433
  if(!pat) {
rpm-build d2b433
    pat = new Internals::Cfa2x2RgbPattern(pattern);
rpm-build d2b433
    s_patterns[pattern] = pat;
rpm-build d2b433
  }
rpm-build d2b433
rpm-build d2b433
  return pat;
rpm-build d2b433
}
rpm-build d2b433
rpm-build d2b433
rpm-build d2b433
class CfaPattern::Private
rpm-build d2b433
{
rpm-build d2b433
public:
rpm-build d2b433
  friend class Internals::Cfa2x2RgbPattern;
rpm-build d2b433
rpm-build d2b433
  Private()
rpm-build d2b433
    : x(0), y(0), n_colours(0)
rpm-build d2b433
    , pattern_type(OR_CFA_PATTERN_NONE)
rpm-build d2b433
    , pattern(NULL)
rpm-build d2b433
    {}
rpm-build d2b433
rpm-build d2b433
  uint16_t x;
rpm-build d2b433
  uint16_t y;
rpm-build d2b433
  uint16_t n_colours;
rpm-build d2b433
  ::or_cfa_pattern pattern_type;
rpm-build d2b433
  const uint8_t* pattern;
rpm-build d2b433
};
rpm-build d2b433
rpm-build d2b433
CfaPattern::CfaPattern()
rpm-build d2b433
  : d(new CfaPattern::Private)
rpm-build d2b433
{
rpm-build d2b433
}
rpm-build d2b433
rpm-build d2b433
CfaPattern::CfaPattern(::or_cfa_pattern pattern,
rpm-build d2b433
                       uint16_t width, uint16_t height)
rpm-build d2b433
  : d(new CfaPattern::Private)
rpm-build d2b433
{
rpm-build d2b433
  setSize(width, height);
rpm-build d2b433
  setPatternType(pattern);
rpm-build d2b433
}
rpm-build d2b433
rpm-build d2b433
CfaPattern::~CfaPattern()
rpm-build d2b433
{
rpm-build d2b433
  delete d;
rpm-build d2b433
}
rpm-build d2b433
rpm-build d2b433
void CfaPattern::setSize(uint16_t x, uint16_t y)
rpm-build d2b433
{
rpm-build d2b433
  d->x = x;
rpm-build d2b433
  d->y = y;
rpm-build d2b433
  if(x != 2 || y != 2) {
rpm-build d2b433
    d->pattern_type = OR_CFA_PATTERN_NON_RGB22;
rpm-build d2b433
  }
rpm-build d2b433
  else if(!is2by2Rgb()) {
rpm-build d2b433
    d->pattern_type = OR_CFA_PATTERN_NONE;
rpm-build d2b433
  }
rpm-build d2b433
}
rpm-build d2b433
rpm-build d2b433
bool CfaPattern::is2by2Rgb() const
rpm-build d2b433
{
rpm-build d2b433
  return (d->pattern_type != OR_CFA_PATTERN_NONE)
rpm-build d2b433
    && (d->pattern_type != OR_CFA_PATTERN_NON_RGB22);
rpm-build d2b433
}
rpm-build d2b433
rpm-build d2b433
void
rpm-build d2b433
CfaPattern::setPatternPattern(const uint8_t* pattern, uint16_t count)
rpm-build d2b433
{
rpm-build d2b433
  if(count != d->x * d->y) {
rpm-build d2b433
    d->pattern = NULL;
rpm-build d2b433
    // TODO deal with the error
rpm-build d2b433
    return;
rpm-build d2b433
  }
rpm-build d2b433
  d->pattern = pattern;
rpm-build d2b433
}
rpm-build d2b433
rpm-build d2b433
const uint8_t*
rpm-build d2b433
CfaPattern::patternPattern(uint16_t& count) const
rpm-build d2b433
{
rpm-build d2b433
  if(d->pattern) {
rpm-build d2b433
    count = d->x * d->y;
rpm-build d2b433
    return d->pattern;
rpm-build d2b433
  }
rpm-build d2b433
  
rpm-build d2b433
  count = 0;
rpm-build d2b433
  return NULL;
rpm-build d2b433
}
rpm-build d2b433
rpm-build d2b433
void CfaPattern::setPatternType(::or_cfa_pattern pattern)
rpm-build d2b433
{
rpm-build d2b433
  d->pattern_type = pattern;
rpm-build d2b433
  if(is2by2Rgb()) {
rpm-build d2b433
    setSize(2, 2);
rpm-build d2b433
    d->n_colours = 3;
rpm-build d2b433
  }
rpm-build d2b433
}
rpm-build d2b433
rpm-build d2b433
::or_cfa_pattern
rpm-build d2b433
CfaPattern::patternType() const
rpm-build d2b433
{
rpm-build d2b433
  return d->pattern_type;
rpm-build d2b433
}
rpm-build d2b433
rpm-build d2b433
}
rpm-build d2b433
/*
rpm-build d2b433
  Local Variables:
rpm-build d2b433
  mode:c++
rpm-build d2b433
  c-file-style:"stroustrup"
rpm-build d2b433
  c-file-offsets:((innamespace . 0))
rpm-build d2b433
  tab-width:2
rpm-build d2b433
  c-basic-offset:2
rpm-build d2b433
  indent-tabs-mode:nil
rpm-build d2b433
  fill-column:80
rpm-build d2b433
  End:
rpm-build d2b433
*/