src/core/raster/qgscontrastenhancement.h

Go to the documentation of this file.

/* **************************************************************************
                qgscontrastenhancement.h -  description
                       -------------------
begin                : Mon Oct 22 2007
copyright            : (C) 2007 by Peter J. Ersts
email                : ersts@amnh.org

This class contains code that was originally part of the larger QgsRasterLayer
class originally created circa 2004 by T.Sutton, Gary E.Sherman, Steve Halasz
****************************************************************************/

/* **************************************************************************
 *                                                                         *
 *   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.                                   *
 *                                                                         *
 ***************************************************************************/

#ifndef QGSCONTRASTENHANCEMENT_H
#define QGSCONTRASTENHANCEMENT_H

#include <limits>

#include "qgis.h"

class QgsContrastEnhancementFunction;
class QDomDocument;
class QDomElement;
class QString;

class CORE_EXPORT QgsContrastEnhancement
{

  public:

    enum ContrastEnhancementAlgorithm
    {
      NoEnhancement,                  //this should be the default color scaling algorithm
      StretchToMinimumMaximum,        //linear histogram enhanceContrast
      StretchAndClipToMinimumMaximum,
      ClipToMinimumMaximum,
      UserDefinedEnhancement
    };

    QgsContrastEnhancement( QGis::DataType theDatatype = QGis::Byte );
    QgsContrastEnhancement( const QgsContrastEnhancement& ce );
    ~QgsContrastEnhancement();

    /*
     *
     * Static methods
     *
     */
    static double maximumValuePossible( QGis::DataType );

    static double minimumValuePossible( QGis::DataType );

    /*
     *
     * Non-Static Inline methods
     *
     */
    double maximumValue() const { return mMaximumValue; }

    double minimumValue() const { return mMinimumValue; }

    ContrastEnhancementAlgorithm contrastEnhancementAlgorithm() const { return mContrastEnhancementAlgorithm; }

    static QString contrastEnhancementAlgorithmString( ContrastEnhancementAlgorithm algorithm );

    static ContrastEnhancementAlgorithm contrastEnhancementAlgorithmFromString( const QString& contrastEnhancementString );

    /*
     *
     * Non-Static methods
     *
     */
    int enhanceContrast( double );

    bool isValueInDisplayableRange( double );

    void setContrastEnhancementAlgorithm( ContrastEnhancementAlgorithm, bool generateTable = true );

    void setContrastEnhancementFunction( QgsContrastEnhancementFunction* );

    void setMaximumValue( double, bool generateTable = true );

    void setMinimumValue( double, bool generateTable = true );

    void writeXML( QDomDocument& doc, QDomElement& parentElem ) const;

    void readXML( const QDomElement& elem );

  private:
    ContrastEnhancementAlgorithm mContrastEnhancementAlgorithm;

    QgsContrastEnhancementFunction* mContrastEnhancementFunction;

    bool mEnhancementDirty;

    double mLookupTableOffset;

    int *mLookupTable;

    double mMinimumValue;

    double mMaximumValue;

    QGis::DataType mRasterDataType;

    double mRasterDataTypeRange;



    bool generateLookupTable();

    int calculateContrastEnhancementValue( double );
};

#endif