1 //////////////////////////////////////////////////////////////////////////////// 2 // 3 // Copyright (c) 2008 The Regents of the University of California 4 // 5 // This file is part of Qbox 6 // 7 // Qbox is distributed under the terms of the GNU General Public License 8 // as published by the Free Software Foundation, either version 2 of 9 // the License, or (at your option) any later version. 10 // See the file COPYING in the root directory of this distribution 11 // or <http://www.gnu.org/licenses/>. 12 // 13 //////////////////////////////////////////////////////////////////////////////// 14 // 15 // Emass.h 16 // 17 //////////////////////////////////////////////////////////////////////////////// 18 19 #ifndef EMASS_H 20 #define EMASS_H 21 22 #include<iostream> 23 #include<iomanip> 24 #include<sstream> 25 #include<stdlib.h> 26 27 #include "Sample.h" 28 29 class Emass : public Var 30 { 31 Sample *s; 32 33 public: 34 name(void)35 const char *name ( void ) const { return "emass"; }; 36 set(int argc,char ** argv)37 int set ( int argc, char **argv ) 38 { 39 if ( argc != 2 ) 40 { 41 if ( ui->onpe0() ) 42 cout << " emass takes only one value" << endl; 43 return 1; 44 } 45 46 double v = atof(argv[1]); 47 if ( v < 0.0 ) 48 { 49 if ( ui->onpe0() ) 50 cout << " emass must be non-negative" << endl; 51 return 1; 52 } 53 54 s->ctrl.emass = v; 55 return 0; 56 } 57 print(void)58 string print (void) const 59 { 60 ostringstream st; 61 st.setf(ios::left,ios::adjustfield); 62 st << setw(10) << name() << " = "; 63 st.setf(ios::right,ios::adjustfield); 64 st << setw(10) << s->ctrl.emass; 65 return st.str(); 66 } 67 Emass(Sample * sample)68 Emass(Sample *sample) : s(sample) { s->ctrl.emass = 0.0; } 69 }; 70 #endif 71