Aegisub/aegisub/src/audio_colorscheme.h

// Copyright (c) 2009-2010, Niels Martin Hansen
// All rights reserved.
//
// Redistribution and use in source and binary forms, with or without
// modification, are permitted provided that the following conditions are met:
//
//   * Redistributions of source code must retain the above copyright notice,
//     this list of conditions and the following disclaimer.
//   * Redistributions in binary form must reproduce the above copyright notice,
//     this list of conditions and the following disclaimer in the documentation
//     and/or other materials provided with the distribution.
//   * Neither the name of the Aegisub Group nor the names of its contributors
//     may be used to endorse or promote products derived from this software
//     without specific prior written permission.
//
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
// AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
// IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
// ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
// LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
// CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
// SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
// INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
// CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
// ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
// POSSIBILITY OF SUCH DAMAGE.
//
// Aegisub Project http://www.aegisub.org/
//
// $Id$

/// @file audio_colorscheme.h
/// @see audio_colorscheme.cpp
/// @ingroup audio_ui
///
/// Manage colour schemes for the audio display


#ifndef AGI_PRE
#include <wx/colour.h>
#endif


/// @class AudioSpectrumColorMap
/// @brief Provides colour maps for audio display rendering
///
/// Maps values from floats in range 0..1 into RGB colour values.
///
/// First create an instance of this class, then call an initialisation function
/// in it to fill the palette with a colour map.
///
/// @todo Let consumers of this class specify their own palette generation function.
class AudioColorScheme {
	/// The palette data for the map
	unsigned char *palette;

	/// Factor to multiply 0..1 values by to map them into the palette range
	size_t factor;

public:
	/// @brief Constructor
	/// @param prec Bit precision to create the colour map with
	///
	/// Allocates the palette array to 2^prec entries
	AudioColorScheme(int prec)
		: palette(new unsigned char[(4<<prec) + 4])
		, factor(1<<prec)
	{
	}

	/// @brief Destructor
	///
	/// De-allocates the palette array
	~AudioColorScheme()
	{
		delete[] palette;
	}

	/// @brief Initialise the palette to the Aegisub 2.1 "Icy Blue" scheme (unselected)
	void InitIcyBlue_Normal();
	/// @brief Initialise the palette to the Aegisub 2.1 "Icy Blue" scheme (selected)
	void InitIcyBlue_Selected();

	/// @brief Map a floating point value to RGB
	/// @param val   [in] The value to map from
	/// @param pixel [out] First byte of the pixel to write
	///
	/// Writes into the XRGB pixel (assumed 32 bit without alpha) passed.
	/// The pixel format is assumed to be the same as that in the palette.
	inline void map(float val, unsigned char *pixel)
	{
		if (val < 0.0) val = 0.0;
		if (val > 1.0) val = 1.0;
		// Find the colour in the palette
		unsigned char *color = palette + ((int)(val*factor) * 4);
		// Copy to the destination.
		// Has to be done one byte at a time since we're writing RGB and not RGBX or RGBA
		// data, and we otherwise write past the end of the pixel we're writing, possibly
		// hitting adjacent memory blocks or just overwriting the start of the following
		// scanline in the image.
		// As the image is 24 bpp, 3 of every 4 uint32_t writes would  be unaligned anyway.
		pixel[0] = color[0];
		pixel[1] = color[1];
		pixel[2] = color[2];
	}

	/// @brief Get a floating point value's colour as a wxColour
	/// @param val The value to map from
	/// @return The corresponding wxColour
	wxColour get(float val)
	{
		if (val < 0.0) val = 0.0;
		if (val > 1.0) val = 1.0;
		unsigned char *color = palette + ((int)(val*factor) * 4);
		return wxColour(color[0], color[1], color[2]);
	}
};