Nrowmax.h 1.71 KB
Newer Older
Francois Gygi committed
1 2
////////////////////////////////////////////////////////////////////////////////
//
Francois Gygi committed
3 4 5 6
// Copyright (c) 2008 The Regents of the University of California
//
// This file is part of Qbox
//
Francois Gygi committed
7 8
// Qbox is distributed under the terms of the GNU General Public License
// as published by the Free Software Foundation, either version 2 of
Francois Gygi committed
9 10 11 12 13 14
// 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/>.
//
////////////////////////////////////////////////////////////////////////////////
//
Francois Gygi committed
15 16 17
// Nrowmax.h
//
////////////////////////////////////////////////////////////////////////////////
Francois Gygi committed
18
// $Id: Nrowmax.h,v 1.5 2008-09-08 15:56:18 fgygi Exp $
Francois Gygi committed
19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35

#ifndef NROWMAX_H
#define NROWMAX_H

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

#include "Sample.h"

class Nrowmax : public Var
{
  Sample *s;

  public:

36
  const char *name ( void ) const { return "nrowmax"; };
Francois Gygi committed
37 38 39 40 41 42 43 44 45

  int set ( int argc, char **argv )
  {
    if ( argc != 2 )
    {
      if ( ui->onpe0() )
      cout << " nrowmax takes only one value" << endl;
      return 1;
    }
46

Francois Gygi committed
47 48 49 50 51 52 53 54 55
    int v = atoi(argv[1]);
    if ( v <= 0 )
    {
      if ( ui->onpe0() )
        cout << " nrowmax must be positive" << endl;
      return 1;
    }

    s->wf.set_nrowmax(v);
56 57 58 59 60 61
    s->wf.update_occ(0.0);
    if ( s->wfv != 0 )
    {
      s->wfv->set_nrowmax(v);
      s->wfv->clear();
    }
62

Francois Gygi committed
63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78
    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 << setw(10) << s->wf.nrowmax();
     return st.str();
  }

  Nrowmax(Sample *sample) : s(sample) {};
};
#endif