Efield.h 1.72 KB
Newer Older
Francois Gygi committed
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35
////////////////////////////////////////////////////////////////////////////////
//
// Copyright (c) 2014 The Regents of the University of California
//
// This file is part of Qbox
//
// Qbox is distributed under the terms of the GNU General Public License
// as published by the Free Software Foundation, either version 2 of
// the License, or (at your option) any later version.
// See the file COPYING in the root directory of this distribution
// or <http://www.gnu.org/licenses/>.
//
////////////////////////////////////////////////////////////////////////////////
//
// Efield.h
//
////////////////////////////////////////////////////////////////////////////////

#ifndef EFIELD_H
#define EFIELD_H

#include<iostream>
#include<iomanip>
#include<sstream>
#include<stdlib.h>
#include"D3vector.h"

#include "Sample.h"

class Efield : public Var
{
  Sample *s;

  public:

36
  const char *name ( void ) const { return "e_field"; };
Francois Gygi committed
37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77

  int set ( int argc, char **argv )
  {
    if ( argc != 4 )
    {
      if ( ui->onpe0() )
      cout << " e_field takes 3 values" << endl;
      return 1;
    }

    double v0 = atof(argv[1]);
    double v1 = atof(argv[2]);
    double v2 = atof(argv[3]);

    s->ctrl.e_field[0] = v0;
    s->ctrl.e_field[1] = v1;
    s->ctrl.e_field[2] = v2;

    return 0;
  }

  string print (void) const
  {
     ostringstream st;
     st.setf(ios::left,ios::adjustfield);
     st << setw(10) << name() << " = ";
     st.setf(ios::right,ios::adjustfield);
     st << s->ctrl.e_field[0] << " "
        << s->ctrl.e_field[1] << " "
        << s->ctrl.e_field[2] << " ";
     return st.str();
  }

  Efield(Sample *sample) : s(sample)
  {
    s->ctrl.e_field[0] = 0.0;
    s->ctrl.e_field[1] = 0.0;
    s->ctrl.e_field[2] = 0.0;
  }
};
#endif