from import_classes import *
appfamily=AppFamily('fome-gen', 'X_D0RELEASE', 'X_FOME_NAME')
filename = 'X_FILENAME'
t = SAMMCFile(filename,Events(X_FIRST_EVT, X_LAST_EVT, X_NEVTS),"fome",appfamily,"X_START_TIME","X_END_TIME",FILE_SIZE_KB,
{'Global':
  {'ProducedByName':'X_PRODUCER',
   'OriginName':'X_ORIGIN',
   'Phase':'X_MC_PHASE',
   'FacilityName':'X_FACILITY',
   'ProducedForName':'X_REQUESTER',
   'RunType':'Monte Carlo',
   'GroupName':'X_GROUP_NAME',
   'Stream':'notstreamed',
   'Description':'X_DESCRIPTION',
   'Stream': 'notstreamed',
   'RunjobVersion': 'X_RUNJOB_VERSION',
   'RequestID': 'X_REQUESTID',
   'WorkRequestID': 'X_WORKREQUESTID'
  },
'Fome' :
  {'collisionenergy': X_COLLISIONENERGY,
   'comments': 'X_COMMENTS',
   'decay': 'X_DECAY',
   'dr-gam-lep': X_DR-GAM-LEP,
   'dr-gam-lf': X_DR-GAM-LF,
   'dr-hf-hf': X_DR-HF-HF, 
   'dr-hf-lf': X_DR-HF-LF, 
   'dr-lep-hf': X_DR-LEP-HF, 
   'dr-lep-lep': X_DR-LEP-LEP, 
   'dr-lf-lep': DR-LF-LEP,
   'dr-lf-lf': X_DR-LF-LF, 
   'd-xsect-lo': X_D-XSECT-LO, 
   'd-xsect-nlo': X_D-XSECT-NLO,   
   'ewopt': X_EWOPT, 
   'fact-scheme': 'X_FACT-SCHEME',
   'fscal-opt': X_FSCAL-OPT, 
   'fscal-val': X_FSCAL-VAL, 
   'generator': 'X_GENERATOR',
   'generatorvsn': 'X_GENERATORVSN',
   'gl-scal-fact': X_GL-SCAL-FACT,
   'gupi-wgt': X_GUPI-WGT, 
   'in-ex-cl': 'X_IN-EX-CL', 
   'itmx1': X_ITMX1, 
   'itmx2': X_ITMX2, 
   'jet-algo': 'X_JET-ALGO', 
   'lambda_qcd4': X_LAMBDA_QCD4,
   'lambda_qcd5': X_LAMBDA_QCD5,
   'mass-max-34': X_MASS-MAX-34, 
   'mass-max-56': X_MASS-MAX-56, 
   'mass-min-34': X_MASS-MIN-34, 
   'mass-min-56': X_MASS-MIN-56, 
   'match-algo': 'X_MATCH-ALGO',
   'matching_dr': X_MATCHING_DR,
   'matching_opt': X_MATCHING_OPT,
   'matching_pt': X_MATCHING_PT,
   'mb': X_MB, 
   'mc': X_MC, 
   'met': X_MET, 
   'mha': X_MHA, 
   'mhh': X_MHH, 
   'mhl': X_MHL, 
   'mhp': X_MHP,
   'mod-couplings': X_MOD-COUPLINGS, 
   'mod-coupl-val': 'X_MOD-COUPL-VAL', 
   'mt': X_MT,
   'mx-wgt-f': X_MX-WGT-F, 
   'ncal1': X_NCAL1, 
   'ncal2': X_NCAL2, 
   'nevts': X_NEVTS, 
   'npartons': X_NPARTONS, 
   'part-kt-max': X_PART-KT-MAX, 
   'part-kt-min': X_PART-KT-MIN, 
   'pdfcode': 'X_PDFCODE', 
   'pdfsource': 'X_PDFSOURCE', 
   'po': 'X_PO', 
   'process': 'X_PROCESS', 
   'ptfr-gam-cone': X_PTFR-GAM-CONE, 
   'pt-min-gam1': X_PT-MIN-GAM1, 
   'pt-min-gam2': X_PT-MIN-GAM2, 
   'pt_min_hf': X_PT_MIN_HF, 
   'pt-min-lep1': X_PT-MIN-LEP1, 
   'pt-min-lep2': X_PT-MIN-LEP2,	        
   'pt_min_lf': X_PT_MIN_LF, 
   'ranseed1': X_RANSEED0,
   'ranseed1': '[X_RANSEED1_0, X_RANSEED1_1, X_RANSEED1_2,...]', 
   'ranseed2': '[X_RANSEED2_0, X_RANSEED2_1, X_RANSEED2_2,...]', 
   'ranseed3': '[X_RANSEED3_0, X_RANSEED3_1, X_RANSEED3_2,...]', 
   'ranseed4': '[X_RANSEED4_0, X_RANSEED4_1, X_RANSEED4_2,...]', 
   'runnumber': X_RUNNUMBER,
   'r-gam-cone': 'X_R-GAM-CONE', 
   'r-lep-cone': X_R-LEP-CONE, 
   'root-s': X_ROOT-S, 
   'rscal-opt': X_RSCAL-OPT, 
   'rscal-val': X_RSCAL-VAL, 
   'usemaxopt': 'X_USEMAXOPT',
   'xsect-lo': X_XSECT-LO, 
   'xsect-nlo': X_XSECT-NLO, 
   'y-max-gam1': X_Y-MAX-GAM1, 
   'y-max-gam2': X_Y-MAX-GAM2, 
   'y_max_hf': X_Y_MAX_HF, 
   'y-max-lep1': X_Y-MAX-LEP1, 
   'y-max-lep2': X_Y-MAX-LEP2, 
   'y_max_lf': X_Y_MAX_LF, 
   'y-min-gam1': X_Y-MIN-GAM1, 
   'y-min-gam2': X_Y-MIN-GAM2, 
   'y_min_hf': X_Y_MIN_HF, 
   'y-min-lep1': X_Y-MIN-LEP1, 
   'y-min-lep2': X_Y-MIN-LEP2, 
   'y_min_lf': X_Y_MIN_LF, 
   'z-w': X_Z-W
} 

})

if __name__ == '__main__':
    import re,sys
    
    if len(sys.argv) != 2: sys.exit(0)
    
    ValidStore = re.compile('store',re.IGNORECASE)
    ValidDecl  = re.compile('declare',re.IGNORECASE)
    script = sys.argv[0]
    arg    = sys.argv[1]
    mode   = None
    
    if ValidDecl.match(arg): mode = 'declare'
    if ValidStore.match(arg): mode = 'store'  
    if mode == None: sys.exit(0)

    sam = None
    try:
        import SamUserApi
        sam = SamUserApi.SamUserApi()
    except:
        print "Unable to import and Initialise SamUserApi"
        print sys.exc_info()[0]
        print sys.exc_info()[1]
        print "Unable to do store/declare for this file"
        sys.exit()

    if sam == None: sys.exit(0)

    if mode == 'store':
        print "Storing File: %s"%script
        try:
            result = sam.store(descrip=script,source='.')
            print result
            sys.exit(0)
        except:
            print "Error Storing File"
            print sys.exc_info()[0]
            print sys.exc_info()[1]
            sys.exit(1)
            
    if mode == 'declare':
        print "Declaring File: %s"%script
        try:
            result = sam.declare(args=script)
            print result
            sys.exit(0)
        except:
            print "Error Declaring File"
            print sys.exc_info()[0]
            print sys.exc_info()[1]
            sys.exit(1)