tvector_arithmetic.h.html - numeric - C++ library with numerical algorithms
(HTM) git clone git://src.adamsgaard.dk/numeric
(DIR) Log
(DIR) Files
(DIR) Refs
(DIR) LICENSE
---
tvector_arithmetic.h.html (10691B)
---
1 <!DOCTYPE HTML PUBLIC "-//W3C//DTD HTML 4.01//EN" "http://www.w3.org/TR/html4/strict.dtd">
2 <html>
3 <head>
4 <meta http-equiv="content-type" content="text/html; charset=UTF-8">
5 <title>~/code/numeric/exam/vector_arithmetic.h.html</title>
6 <meta name="Generator" content="Vim/7.4">
7 <meta name="plugin-version" content="vim7.4_v1">
8 <meta name="syntax" content="cpp">
9 <meta name="settings" content="number_lines,use_css,pre_wrap,no_foldcolumn,expand_tabs,line_ids,prevent_copy=">
10 <meta name="colorscheme" content="desert">
11 <style type="text/css">
12 <!--
13 pre { white-space: pre-wrap; font-family: monospace; color: #ffffff; background-color: #000000; }
14 body { font-family: monospace; color: #ffffff; background-color: #000000; }
15 * { font-size: 1em; }
16 .Type { color: #008000; }
17 .Statement { color: #804000; }
18 .LineNr { color: #804000; }
19 .Comment { color: #008080; }
20 .Constant { color: #af5f00; }
21 .PreProc { color: #c000c0; }
22 -->
23 </style>
24
25 <script type='text/javascript'>
26 <!--
27
28 /* function to open any folds containing a jumped-to line before jumping to it */
29 function JumpToLine()
30 {
31 var lineNum;
32 lineNum = window.location.hash;
33 lineNum = lineNum.substr(1); /* strip off '#' */
34
35 if (lineNum.indexOf('L') == -1) {
36 lineNum = 'L'+lineNum;
37 }
38 lineElem = document.getElementById(lineNum);
39 /* Always jump to new location even if the line was hidden inside a fold, or
40 * we corrected the raw number to a line ID.
41 */
42 if (lineElem) {
43 lineElem.scrollIntoView(true);
44 }
45 return true;
46 }
47 if ('onhashchange' in window) {
48 window.onhashchange = JumpToLine;
49 }
50
51 -->
52 </script>
53 </head>
54 <body onload='JumpToLine();'>
55 <pre id='vimCodeElement'>
56 <span id="L1" class="LineNr"> 1 </span><span class="Comment">// Make sure header is only included once</span>
57 <span id="L2" class="LineNr"> 2 </span><span class="PreProc">#ifndef VECTOR_ARITHMETIC_H_</span>
58 <span id="L3" class="LineNr"> 3 </span><span class="PreProc">#define VECTOR_ARITHMETIC_H_</span>
59 <span id="L4" class="LineNr"> 4 </span>
60 <span id="L5" class="LineNr"> 5 </span><span class="PreProc">#include </span><span class="Constant"><vector></span>
61 <span id="L6" class="LineNr"> 6 </span><span class="PreProc">#include </span><span class="Constant"><cmath></span> <span class="Comment">// for sqrt</span>
62 <span id="L7" class="LineNr"> 7 </span><span class="PreProc">#include </span><span class="Constant">"typedefs.h"</span>
63 <span id="L8" class="LineNr"> 8 </span>
64 <span id="L9" class="LineNr"> 9 </span><span class="Comment">//// Overload vector methods to allow scalar-</span>
65 <span id="L10" class="LineNr">10 </span><span class="Comment">//// and element-wise arithmetic operations</span>
66 <span id="L11" class="LineNr">11 </span>
67 <span id="L12" class="LineNr">12 </span><span class="Comment">// Scalar multiplication (same scalar for real and imaginary parts)</span>
68 <span id="L13" class="LineNr">13 </span>std::vector<std::<span class="Type">complex</span><Floattype> >
69 <span id="L14" class="LineNr">14 </span> <span class="Statement">operator</span>*(<span class="Type">const</span> std::vector<std::<span class="Type">complex</span><Floattype> > vec,
70 <span id="L15" class="LineNr">15 </span> <span class="Type">const</span> Floattype scalar)
71 <span id="L16" class="LineNr">16 </span>{
72 <span id="L17" class="LineNr">17 </span> std::vector<std::<span class="Type">complex</span><Floattype> > result(vec.size());
73 <span id="L18" class="LineNr">18 </span> <span class="Statement">for</span> (Inttype i=<span class="Constant">0</span>; i<vec.size(); ++i) {
74 <span id="L19" class="LineNr">19 </span> result[i].real() = real(vec[i])*scalar;
75 <span id="L20" class="LineNr">20 </span> result[i].imag() = imag(vec[i])*scalar;
76 <span id="L21" class="LineNr">21 </span> }
77 <span id="L22" class="LineNr">22 </span> <span class="Statement">return</span> result;
78 <span id="L23" class="LineNr">23 </span>}
79 <span id="L24" class="LineNr">24 </span>
80 <span id="L25" class="LineNr">25 </span><span class="Comment">// Scalar multiplication</span>
81 <span id="L26" class="LineNr">26 </span>std::vector<std::<span class="Type">complex</span><Floattype> >
82 <span id="L27" class="LineNr">27 </span> <span class="Statement">operator</span>*(<span class="Type">const</span> std::vector<std::<span class="Type">complex</span><Floattype> > vec,
83 <span id="L28" class="LineNr">28 </span> <span class="Type">const</span> std::<span class="Type">complex</span><Floattype> scalar)
84 <span id="L29" class="LineNr">29 </span>{
85 <span id="L30" class="LineNr">30 </span> std::vector<std::<span class="Type">complex</span><Floattype> > result(vec.size());
86 <span id="L31" class="LineNr">31 </span> <span class="Statement">for</span> (Inttype i=<span class="Constant">0</span>; i<(Inttype)vec.size(); ++i)
87 <span id="L32" class="LineNr">32 </span> result[i] = vec[i]*scalar;
88 <span id="L33" class="LineNr">33 </span> <span class="Statement">return</span> result;
89 <span id="L34" class="LineNr">34 </span>}
90 <span id="L35" class="LineNr">35 </span>
91 <span id="L36" class="LineNr">36 </span><span class="Comment">// Scalar division </span>
92 <span id="L37" class="LineNr">37 </span>std::vector<std::<span class="Type">complex</span><Floattype> >
93 <span id="L38" class="LineNr">38 </span> <span class="Statement">operator</span>/(<span class="Type">const</span> std::vector<std::<span class="Type">complex</span><Floattype> > vec,
94 <span id="L39" class="LineNr">39 </span> <span class="Type">const</span> std::<span class="Type">complex</span><Floattype> scalar)
95 <span id="L40" class="LineNr">40 </span>{
96 <span id="L41" class="LineNr">41 </span> std::vector<std::<span class="Type">complex</span><Floattype> > result(vec.size());
97 <span id="L42" class="LineNr">42 </span> <span class="Statement">for</span> (Inttype i=<span class="Constant">0</span>; i<(Inttype)vec.size(); ++i)
98 <span id="L43" class="LineNr">43 </span> result[i] = vec[i]/scalar;
99 <span id="L44" class="LineNr">44 </span> <span class="Statement">return</span> result;
100 <span id="L45" class="LineNr">45 </span>}
101 <span id="L46" class="LineNr">46 </span>
102 <span id="L47" class="LineNr">47 </span><span class="Comment">// Element-wise addition</span>
103 <span id="L48" class="LineNr">48 </span>std::vector<std::<span class="Type">complex</span><Floattype> >
104 <span id="L49" class="LineNr">49 </span> <span class="Statement">operator</span>+(<span class="Type">const</span> std::vector<std::<span class="Type">complex</span><Floattype> > vec1,
105 <span id="L50" class="LineNr">50 </span> <span class="Type">const</span> std::vector<std::<span class="Type">complex</span><Floattype> > vec2)
106 <span id="L51" class="LineNr">51 </span>{
107 <span id="L52" class="LineNr">52 </span> std::vector<std::<span class="Type">complex</span><Floattype> > result(vec1.size());
108 <span id="L53" class="LineNr">53 </span> <span class="Statement">for</span> (Inttype i=<span class="Constant">0</span>; i<(Inttype)vec1.size(); ++i)
109 <span id="L54" class="LineNr">54 </span> result[i] = vec1[i] + vec2[i];
110 <span id="L55" class="LineNr">55 </span> <span class="Statement">return</span> result;
111 <span id="L56" class="LineNr">56 </span>}
112 <span id="L57" class="LineNr">57 </span>
113 <span id="L58" class="LineNr">58 </span><span class="Comment">// Element-wise subtraction</span>
114 <span id="L59" class="LineNr">59 </span>std::vector<std::<span class="Type">complex</span><Floattype> >
115 <span id="L60" class="LineNr">60 </span> <span class="Statement">operator</span>-(<span class="Type">const</span> std::vector<std::<span class="Type">complex</span><Floattype> > vec1,
116 <span id="L61" class="LineNr">61 </span> <span class="Type">const</span> std::vector<std::<span class="Type">complex</span><Floattype> > vec2)
117 <span id="L62" class="LineNr">62 </span>{
118 <span id="L63" class="LineNr">63 </span> std::vector<std::<span class="Type">complex</span><Floattype> > result(vec1.size());
119 <span id="L64" class="LineNr">64 </span> <span class="Statement">for</span> (Inttype i=<span class="Constant">0</span>; i<(Inttype)vec1.size(); ++i)
120 <span id="L65" class="LineNr">65 </span> result[i] = vec1[i] - vec2[i];
121 <span id="L66" class="LineNr">66 </span> <span class="Statement">return</span> result;
122 <span id="L67" class="LineNr">67 </span>}
123 <span id="L68" class="LineNr">68 </span>
124 <span id="L69" class="LineNr">69 </span><span class="Comment">// Element-wise multiplication</span>
125 <span id="L70" class="LineNr">70 </span>std::vector<std::<span class="Type">complex</span><Floattype> >
126 <span id="L71" class="LineNr">71 </span> <span class="Statement">operator</span>*(<span class="Type">const</span> std::vector<std::<span class="Type">complex</span><Floattype> > vec1,
127 <span id="L72" class="LineNr">72 </span> <span class="Type">const</span> std::vector<std::<span class="Type">complex</span><Floattype> > vec2)
128 <span id="L73" class="LineNr">73 </span>{
129 <span id="L74" class="LineNr">74 </span> std::vector<std::<span class="Type">complex</span><Floattype> > result(vec1.size());
130 <span id="L75" class="LineNr">75 </span> <span class="Statement">for</span> (Inttype i=<span class="Constant">0</span>; i<(Inttype)vec1.size(); ++i)
131 <span id="L76" class="LineNr">76 </span> result[i] = vec1[i] * vec2[i];
132 <span id="L77" class="LineNr">77 </span> <span class="Statement">return</span> result;
133 <span id="L78" class="LineNr">78 </span>}
134 <span id="L79" class="LineNr">79 </span>
135 <span id="L80" class="LineNr">80 </span><span class="Comment">// Normalize vector</span>
136 <span id="L81" class="LineNr">81 </span>Floattype cnorm(<span class="Type">const</span> std::vector<std::<span class="Type">complex</span><Floattype> > vec)
137 <span id="L82" class="LineNr">82 </span>{
138 <span id="L83" class="LineNr">83 </span> Floattype res = <span class="Constant">0.0f</span>;
139 <span id="L84" class="LineNr">84 </span> <span class="Statement">for</span> (Inttype i=<span class="Constant">0</span>; i<(Inttype)vec.size(); ++i)
140 <span id="L85" class="LineNr">85 </span> res += norm(vec[i])*norm(vec[i]);
141 <span id="L86" class="LineNr">86 </span> <span class="Statement">return</span> sqrt(res);
142 <span id="L87" class="LineNr">87 </span>}
143 <span id="L88" class="LineNr">88 </span>
144 <span id="L89" class="LineNr">89 </span><span class="PreProc">#endif</span>
145 </pre>
146 </body>
147 </html>
148 <!-- vim: set foldmethod=manual : -->