PBatch.h

Go to the documentation of this file.

// Author: I. Froehlich
// Written: 14.02.2008
//
 
#ifndef _PBATCH_H_
#define _PBATCH_H_
 
#include "TInterpreter.h"
#include "TMethodCall.h"
#include "TMethodArg.h"
 
#include "PFormula.h"
#include "PValues.h"
#include "PParticle.h"
 
#include "TROOT.h"
#include "TTree.h"
#include "TH1.h"
#include "TH2.h"
#include "TH3.h"
#include "TGraph.h"
#include "TGraph2D.h"
 
#include <iostream>
#include "PDefines.h"
 
using namespace std;
 
#define MAX_COMMAND_POINTER    1000
#define MAX_COMMAND_TMETHODS   100
#define MAX_COMMAND_OPTIONS    10
#define MAX_STACK_GOSUB        20
#define MAX_READLINE_ARGS      20
 
 
#define IS_OBJECT 1
#define IS_DOUBLE 2
 
#define kFOREACH    2
#define kGOTO       4
#define kFOREACHEND 8
#define kUPDATE     16
#define kEOF        32
#define kPUSH       64
#define kPUSHBRANCH 128
#define kELSE       256
 
#define COMMAND_MASS  '1'
#define COMMAND_MASS2 '2'
#define COMMAND_GETBEAM   '3'
#define COMMAND_GETTARGET '4'
#define COMMAND_IF    '?'
#define COMMAND_ELSE  '%'
#define COMMAND_PLUS  '+'
#define COMMAND_MINUS '-'
#define COMMAND_DIV   '/'
#define COMMAND_MULT  '*'
#define COMMAND_IS    '='
#define COMMAND_EQUAL '~'
 
 
#define COMMAND_BOOST     'b'
#define COMMAND_COS       'c'
#define COMMAND_EVAL      'e'
#define COMMAND_FABS      'f'
#define COMMAND_INTERNAL  'i'
#define COMMAND_PRINT     'p'
#define COMMAND_ROT       'r'
#define COMMAND_THETA     't'
#define COMMAND_GETRANDOM 'v'
#define COMMAND_GETRANDOMX 'x'
#define COMMAND_GETRANDOMY 'y'
 
 
#define COMMAND_ANGLE    'A'
#define COMMAND_BRANCH   'B'
#define COMMAND_P3E      'E'
#define COMMAND_GOTO     'G'
#define COMMAND_PUSH     'H'
#define COMMAND_LABEL    'L'
#define COMMAND_PFORMULA 'P'
#define COMMAND_P3M      'Q'
#define COMMAND_RETURN   'R'
#define COMMAND_GOSUB    'S'
#define COMMAND_PVALUE   'V'
#define COMMAND_EXIT     'X'
#define COMMAND_ECHO     'Y'
#define COMMAND_READLINE 'Z'
#define COMMAND_FORMORE  '>'
#define COMMAND_FOREACH  '<'
 
 
 
#define METHOD_RETURN_DOUBLE 1
#define METHOD_RETURN_INT    2
#define METHOD_RETURN_VOID   4
#define METHOD_RETURN_PPARTICLE   8
 
#define STREAM_MAX_POS       "npar"
#define STREAM_DEFAULT_POS   "cpos"
 
class PBatch;
PBatch& fBatch();
PBatch* makeGlobalBatch();
 
 
class PBatch : public TObject {
 
 private:
 
    Bool_t CheckAndSplit(char * arg, char delim, int *key1, int *key2);
 
    Bool_t GetArguments(const char *a, /* const char *b, */
            char *name, char **function, char **arg1, char **arg2);
    Int_t  CheckObjectType(Int_t key);
    Int_t  GetKey(char *name, int fl, int makeflag);
    Int_t  GetDelimPosition(char *arg, char delim, Int_t *yes=NULL);
 
    Int_t  GetMethodHandle(char *name, Int_t flag=0);
    Int_t  CrackMethodArgs(char *name);
    Int_t  arg1, arg2, arg3, arg4;
 
    Int_t command_pointer, method_pointer, last_command_pointer;
    TMethodCall *methods[MAX_COMMAND_TMETHODS];
    char *method_name[MAX_COMMAND_TMETHODS];
    Int_t methods_flags[MAX_COMMAND_TMETHODS],
    methods_arg_flags[4][MAX_COMMAND_TMETHODS];
 
    char lst_command[MAX_COMMAND_POINTER];
    Int_t lst_command_int[MAX_COMMAND_POINTER];
    Int_t flag_command_int[MAX_COMMAND_POINTER];  //=0: PParticle, =1: PUtilsRengine
    Int_t lst_key_a[MAX_COMMAND_POINTER];
    Int_t error_flag[MAX_COMMAND_POINTER];
    Int_t lst_key[MAX_COMMAND_OPTIONS][MAX_COMMAND_POINTER];
    Int_t  lst_options_counter[MAX_COMMAND_POINTER];
    PFormula *lst_form[MAX_COMMAND_POINTER];
    //    Int_t lst_key2[MAX_COMMAND_POINTER];
    //    Int_t lst_key3[MAX_COMMAND_POINTER];
    //    Int_t lst_key4[MAX_COMMAND_POINTER];
 
    char *echo_string[MAX_COMMAND_POINTER];
    char *readline_format_string[MAX_COMMAND_POINTER];
    char *readline_string[MAX_COMMAND_POINTER];
    Double_t **readline_args[MAX_COMMAND_POINTER];
    int readline_num_args[MAX_COMMAND_POINTER];
    char *varlist;   //Allowed command for new variables
 
    Int_t batch_particle_param, batch_value_param, pid_param, num_command_param, num_batch_param, num_bulk_param,
    stream_default_pos_param, stream_max_pos_param, batch_update_param, batch_models_param;
    Int_t batch_histogram_param;
    Int_t num_batch,num_bulk, locnum_batch, locnum_bulk, locnum_command, locnum_old_command;  //Position in the projector & reaction loop
    Int_t locnum_branch; //current branch number
    Int_t else_position;
    Int_t current_position;
 
    static Int_t stack_num_batch[MAX_STACK_GOSUB], stack_num_bulk[MAX_STACK_GOSUB],
    stack_num_command[MAX_STACK_GOSUB], stack_num_pos;
 
    void  AddSpacePlaceholder(char *command);
    void  RemoveSpacePlaceholder(char *command);
    Int_t EvalPFormula(char *command);
 
 
    PValues pdummy;
 
    TH1       *fHisto1;
    TH2       *fHisto2;
    TH3       *fHisto3;
    TGraph    *fGraph;
    TGraph2D  *fGraph2D;
    Int_t      is_readonly;
    TH1D      **slicesx, **slicesy;
 
    FILE      *file, *tmp_file;
    PParticle *current_particle; //For push command
 
    Double_t *x, *y, *z;  //Axis values for _eval
    Int_t eval_err_dumped;
 
    Int_t status;
 
    TTree *tree;         //Pointer to storage tree
    Int_t *size_branches;
    Int_t *key_branches;
 
 
 public:
 
    //constructor
    PBatch();
 
    void Print(const Option_t *delme=NULL) const ;
 
    Bool_t AddCommand(const char *command); //adds a command line to batch
    Bool_t AddCommand(char command, int key_a, int key1, int key2, int key3=-1, int key4=-1, int key5=-1);
 
    void SetPosition(Int_t my_num_batch, Int_t my_num_bulk) {
    num_batch = my_num_batch;
    num_bulk  = my_num_bulk;
    };
    Int_t GetNewBatch()   {return locnum_batch;};
    Int_t GetNewBulk()    {return locnum_bulk;};
    Int_t GetNewCommand() {return locnum_command;};
    Int_t GetOldCommand() {return locnum_old_command;};
    Int_t GetBranch()     {return locnum_branch;};
    Int_t GetElsePosition() {return else_position;};
    Int_t GetCurrentPosition() {return current_position;};
 
    using TObject::Execute;
    Int_t Execute(Int_t command_pos=0, Int_t retval=kFALSE);
    Int_t ExecuteLastLine(void) {
    Bool_t retval = Execute(last_command_pointer);
    last_command_pointer = command_pointer;
    return retval;
    }
    Bool_t Execute(const char *command) {
    if (AddCommand(command)) {
        return (Bool_t)ExecuteLastLine();
    }
    last_command_pointer = command_pointer;
    Error("Execute", "Command not executed");
    return kFALSE;
    }
 
    void SetToolObject(TH1 *Histo1) {
    fHisto1 = Histo1;
    }
    void SetToolObject(TH2 *Histo2) {
    fHisto2 = Histo2;
    }
    void SetToolObject(TH3 *Histo3) {
    fHisto3 = Histo3;
    }
    void SetToolObject(TGraph *Graph) {
    fGraph = Graph;
    }
    void SetToolObject(TGraph2D *Graph2D) {
    fGraph2D = Graph2D;
    }
    void SetToolObject(FILE *f) {
    file = f;
    }
    void SetToolObjectTmp(FILE *f) {
    tmp_file = f;
    }
    Int_t IsReadonly() {
    return is_readonly;
    }
 
    PParticle *GetCurrentParticle(void) {
    return current_particle;
    }
 
    Int_t Status(void) {
    //workaround to check for a "[+]"
    //=1 for such a case
    //=0 in all other cases
    return status;
    }
 
    void SetVarList(char *my_x) {
    //Allowed commands for new variables
    //Format must be "a;b;c;" with trailing ;'s
    //If NULL don't care!
    varlist = my_x;
    };
 
    void ReplaceAll(TString *op, const char *oldstring, const char *newstring);
 
    void SetTree(TTree *my_tree) {tree = my_tree;};
    TTree *GetTree(void)         {return tree;};
 
    void SetSizeBranches(Int_t *my_size_branches) {size_branches = my_size_branches;};
    void SetKeysBranches(Int_t *my_key_branches)  {key_branches = my_key_branches;};
 
    ClassDef(PBatch, 0)  //Batch commands
};
 
#endif