iLab Neuromorphic Robotics Toolkit  0.1
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Groups Pages
ColorMappingImpl.H
Go to the documentation of this file.
1 /*! @file
2  @author Randolph Voorhies
3  @copyright GNU Public License (GPL v3)
4  @section License
5  @verbatim
6  // ////////////////////////////////////////////////////////////////////////
7  // The iLab Neuromorphic Robotics Toolkit (NRT) //
8  // Copyright 2010-2012 by the University of Southern California (USC) //
9  // and the iLab at USC. //
10  // //
11  // iLab - University of Southern California //
12  // Hedco Neurociences Building, Room HNB-10 //
13  // Los Angeles, Ca 90089-2520 - USA //
14  // //
15  // See http://ilab.usc.edu for information about this project. //
16  // ////////////////////////////////////////////////////////////////////////
17  // This file is part of The iLab Neuromorphic Robotics Toolkit. //
18  // //
19  // The iLab Neuromorphic Robotics Toolkit is free software: you can //
20  // redistribute it and/or modify it under the terms of the GNU General //
21  // Public License as published by the Free Software Foundation, either //
22  // version 3 of the License, or (at your option) any later version. //
23  // //
24  // The iLab Neuromorphic Robotics Toolkit is distributed in the hope //
25  // that it will be useful, but WITHOUT ANY WARRANTY; without even the //
26  // implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR //
27  // PURPOSE. See the GNU General Public License for more details. //
28  // //
29  // You should have received a copy of the GNU General Public License //
30  // along with The iLab Neuromorphic Robotics Toolkit. If not, see //
31  // <http://www.gnu.org/licenses/>. //
32  // ////////////////////////////////////////////////////////////////////////
33  @endverbatim */
34 
35 #ifndef INCLUDE_NRT_IMAGEPROC_DRAWING_DETAILS_COLORMAPPINGIMPL_H
36 #define INCLUDE_NRT_IMAGEPROC_DRAWING_DETAILS_COLORMAPPINGIMPL_H
37 
38 // ######################################################################
39 template<class T> inline
40 nrt::PixRGB<nrt::byte> nrt::jetPixel(T v, T vmin, T vmax)
41 {
42  nrt::PixRGB<T> c = {1.0, 1.0, 1.0}; // white
43  T dv;
44 
45  if (v < vmin)
46  v = vmin;
47  if (v > vmax)
48  v = vmax;
49  dv = vmax - vmin;
50 
51  if (v < (vmin + 0.25 * dv)) {
52  c.channels[0] = 0;
53  c.channels[1] = 4 * (v - vmin) / dv;
54  } else if (v < (vmin + 0.5 * dv)) {
55  c.channels[0] = 0;
56  c.channels[2] = 1 + 4 * (vmin + 0.25 * dv - v) / dv;
57  } else if (v < (vmin + 0.75 * dv)) {
58  c.channels[0] = 4 * (v - vmin - 0.5 * dv) / dv;
59  c.channels[2] = 0;
60  } else {
61  c.channels[1] = 1 + 4 * (vmin + 0.75 * dv - v) / dv;
62  c.channels[2] = 0;
63  }
64 
65  for(T & chan : c.channels) chan *= 255.0;
66 
67  return nrt::PixRGB<byte>(c);
68 }
69 
70 // ######################################################################
71 template<class T> inline
72 nrt::Image<nrt::PixRGB<nrt::byte>> nrt::jet(Image<PixGray<T>> const & input, T min, T max)
73 {
74  Image<PixRGB<byte>> result(input.dims(), nrt::ImageInitPolicy::None);
75  std::transform(input.pod_begin(), input.pod_end(), result.begin(),
76  [min, max](T val) { return jetPixel(val, min, max); });
77  return result;
78 }
79 
80 #endif // INCLUDE_NRT_IMAGEPROC_DRAWING_DETAILS_COLORMAPPINGIMPL_H