vector_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 : -->