예제 #1
0
파일: nmspecf2py.py 프로젝트: bjfar/pysusy
class NMspec:
    def __init__(self,infile,specout,decayout,omegaout,tunout,workdir, \
                    template=None,options={},silenceobjs=None,model='generalNMSSM'):
        """args:
        need to finish list
        model - model to use for computing spectrum. Currently valid options are
        'generalNMSSM'->(nmhdecay called) and 'mSUGRA'->(nmspec called)
        """
        # Either the file to use as input, or the name to given the input file
        # which will be created from the file 'template'.
        # (Need to use abspath since we have to change directory to run nmspec)
        self.infile = os.path.abspath(infile)
        # Names to give output files
        self.spectr = os.path.abspath(specout)
        self.decay = os.path.abspath(decayout)
        self.omega = os.path.abspath(omegaout)
        self.tuning = os.path.abspath(tunout)
        self.SLHAspectr = SpectrumFile(name=self.spectr, directory='')    #SpectrumFile has extra blueprints defined to deal with multi-index items
        self.SLHAomega = OmegaFile(name=self.omega, directory='')    #OmegaFile has extra blueprints defined to deal with the Channels block
	self.SLHAtuning = SpectrumFile(name=self.tuning, directory='')

        if template != None:
            self.template = template
            print "nmspec template SLHA input file: ", self.template
            self.SLHAtemplate = SLHAFile(name=self.template, directory='')       #File to modify to create infile
            self.SLHAtemplate.readdata()                                #Read data from the template file
        self.options = options
        
        # Add to self.options the appropriate switch for the selected model
        # note: I actually don't think this does anything. I think this switch
        # is just how the NMSSMTools run script decides which of nmspec,
        # nmhdecay, or nmgmsb to call. But let's set it anyway just in case.
        
        # Set which NMSSMTools code to run
        if model=='generalNMSSM':
            print 'Computing NMSSM spectrum using nmhdecay'
            self.runfunc = nmhdecay.spec
        elif model=='mSUGRA':
            print 'Computing NMSSM spectrum using nmspec'
            self.runfunc = nmspec.spec
        elif model=='GMSB':
            raise Exception("Sorry, model GMSB is not currently implemented! \
please set model='generalNMSSM' (for nmhdecay) or model='mSUGRA' (for nmspec)")
        else:
            raise Exception("model specification string does not match an \
implemented model. Please set model='generalNMSSM' (for nmhdecay) or \
model='mSUGRA' (for nmspec)")
        
        # Check that the config file specifying the location of nmspec exists.
        # It is not great that this is required, but I didn't have time or 
        # patience to make a fancy enough nmspec extension module that could
        # deal with this issue itself.
        cfg = ConfigParser.RawConfigParser()
        cfgfile = workdir+'/nmspecf2py.cfg'
        print 'Reading config file for nmspec wrapper: {0}'.format(cfgfile)
        try:
            cfg.readfp(open(cfgfile))
        except IOError as err:
            raise IOError('{0} - Cannot locate nmspecf2py config file: please \
create a file in nmspecf2py module work directory ("wrappers" if using pysusy3)\
 called "nmspecf2py.cfg", which has the contents:\n\
\n\
[settings]\n\
nmspecdir = <full path to directory containing nmspec>\n\
\n\
If using NMSSMTools_3.2.3 this path is "<parent>/NMSSMTools_3.2.3/main".\n\
Sorry that this is necessary; my nmspec extension module is not advanced enough\
 to figure this out for itself, and nmspec needs to access some files in the \
nmspec directories.'.format(err))
        self.nmspecdir = cfg.get('settings','nmspecdir')
        # END __init__
        
    def silence(self):
        """Kill the stdout of the current process. Lets us silence the
        output of the wrapped fortran code, which 
        sys.stdout = open('/dev/null'), for example, does not"""
        os.dup2(self.null_fds[0], 1)                                    # put /dev/null fds on 1 and 2
        os.dup2(self.null_fds[1], 2)
    
    def unsilence(self):
        """undoes the action of silence"""
        os.dup2(self.save[0], 1)                                        # restore file descriptors so I can print the results
        os.dup2(self.save[1], 2)
        # close the temporary fds
        #os.close(null_fds[0]) #don't think I want to do this, may as well keep using the same ones every loop
        #os.close(null_fds[1])
        
    def run(self):
        """Execute process
        For now just doing simple thing. See ISAJET wrapper if program isolation
        required (i.e. fork/multiprocessing)
        """
        # nmspec is a bit stupid and doesn't work correctly if cwd is not the
        # directory the program is installed into (since it tried to access 
        # files containing info about experimental constraints)
        # So need to change directory and back here (WILL CAUSE MAYHEM IF NMSPEC
        # CRASHES, KEEP AN EYE ON THIS)
        with chdir(self.nmspecdir):
                self.runfunc(self.infile, self.spectr, self.decay, self.omega, self.tuning)
    
    def checkoutput(self):
        """Check if errors occurred during running"""
        # Check spectrum output file for error signals
        self.SLHAspectr.readdata()
        try:
            block = self.SLHAspectr.block('SPINFO')
            item = block[4]
            errormsg = item.value
            return 1, errormsg   # Yes errors occurred
        except KeyError as err:
            if err.message==4: pass        # No errors occurred
            else: raise
       	
        if self.options['usemicromegas'] != 0:
            # Check darkmatter output file for error signals 
            self.SLHAomega.readdata()
            try:
                block = self.SLHAomega.block('RDINFO')
                item = block[4]
                errormsg = item.value
                #print 'testing', errormsg
                return 2, errormsg   # Yes errors occurred (inc. "neutralino not LSP")
            except KeyError as err:
                if err.message==4: pass        # No errors occurred
                else: raise    
        
        # Read in the tuning file; no error checks to perform. 
        self.SLHAtuning.readdata()
   
        # temp! Many points passing the checks which seem messed up, need to see what is wrong with them...
        #print "#====================================================="   
        #print "  'GOOD' POINT FOUND, SEARCHING FOR EXTRA PROBLEMS... "
        #print "#====================================================="   
        #print "SPINFO block:"
        #self.SLHAspectr.printblock('SPINFO')
        #print "RDINFO block:"
        #self.SLHAomega.printblock('RDINFO')
        #print "------------------------------------------------------"
        #print "parameter values:"
        #print "M2    : {0}".format(self.SLHAspectr.block('EXTPAR')[2]).rstrip('\n')
        #print "lambda: {0}".format(self.SLHAspectr.block('EXTPAR')[61]).rstrip('\n')
        #print "kappa : {0}".format(self.SLHAspectr.block('EXTPAR')[62]).rstrip('\n')
        #print "TanB  : {0}".format(self.SLHAspectr.block('MINPAR')[3]).rstrip('\n')
        #print "------------------------------------------------------"
        #print "tuning file values:"
        #try:
        #   self.SLHAtuning.block('TUNJACOB')[10]
        #   print "logJuds_a : {0}".format(self.SLHAtuning.block('TUNJACOB')[10]).rstrip('\n')
        #   print "logJuds_b : {0}".format(self.SLHAtuning.block('TUNJACOB')[11]).rstrip('\n')
        #   print "logJuds_c : {0}".format(self.SLHAtuning.block('TUNJACOB')[12]).rstrip('\n')
        #   print "logAJUDS  : {0}".format(self.SLHAtuning.block('TUNJACOB')[21]).rstrip('\n')
        #except TypeError:
        #   print "No TUNJACOB block!"
        #print "#====================================================="
        ## flush buffer to make sure this prints in the right place
        #sys.stdout.flush()  
 
        #Made it to the end of the checks! Therefore no errors occurred.
        return 0, ''
        
    def getspecobs(self,skiperrors=True):
        """Gather results of computations"""
        # self.SLHAspectr.readdata() # Now must run 'checkoutput' first
        return self.SLHAspectr.getobs(specdict)                            #extract observables from SLHA file using translation dictionary

    def gettunobs(self,skiperrors=True):
        """Gather results of computations"""
        #self.SLHAtuning.readdata()  # Doesn't get done in error checks, so do it here
        return self.SLHAtuning.getobs(tundict)                            #extract observables from SLHA file using translation dictionary

    def getomegaobs(self,skiperrors=True):
        """Gather results of computations"""
        if self.options['usemicromegas'] != 0:
            # extract observables from SLHA file using translation dictionary
            obsdict = self.SLHAomega.getobs(DMdict) 
        else:
            # Nothing was computed in this case
            obsdict = {}
        return obsdict 
    
    # Naming dictionary for observables read from decay file     
    entries = []
    higgses = [1,2,3]
    for h in higgses:
        # tuples are ("crushed" comment name, decaydict key)
        entries += [
                    ('#BR(H_%i->gluongluon)'%h, 'BR(H%i->gg)'%h),
                    ('#BR(H_%i->muonmuon)'%h,   'BR(H%i->mumu)'%h),
                    ('#BR(H_%i->tautau)'%h,     'BR(H%i->tautau)'%h),
                    ('#BR(H_%i->ssbar)'%h,      'BR(H%i->ssbar)'%h),
                    ('#BR(H_%i->ccbar)'%h,      'BR(H%i->ccbar)'%h),
                    ('#BR(H_%i->bbbar)'%h,      'BR(H%i->bbbar)'%h),
                    ('#BR(H_%i->ttbar)'%h,      'BR(H%i->ttbar)'%h),
                    ('#BR(H_%i->W+W-)'%h,       'BR(H%i->W+W-)'%h),
                    ('#BR(H_%i->ZZ)'%h,         'BR(H%i->ZZ)'%h),
                    ('#BR(H_%i->gammagamma)'%h, 'BR(H%i->aa)'%h),
                    ('#BR(H_%i->Zgamma)'%h,     'BR(H%i->Za)'%h),    
                   ]      
    decayentries = dict(entries)        
    
    def getdecayobs(self,skiperrors=True):
        """Reads the 'decay' output file of NMSSMTools
        This one has a weird (but SLHA compliant) format, so we cannot use the 
        standard pysusy block format reader. A custom one is used below.
        Only extracts Higgs decay information at this time."""
        decaydict = {}
        with open(self.decay,'r') as f:
            for line in f:
                words = line.split()
                # Check if we have reached a new section yet
                if words[:2]==['DECAY','25']:
                    decaydict['H1-width'] = float(words[2]); higgs=1; continue
                elif words[:2]==['DECAY','35']:
                    decaydict['H2-width'] = float(words[2]); higgs=2; continue
                elif words[:2]==['DECAY','45']:
                    decaydict['H3-width'] = float(words[2]); higgs=3; continue
                # Branching ratios (read based on "crushed" comment)
                crushedcomment = ''.join(words[4:])
                if crushedcomment in self.decayentries.keys():
                    decaydict[self.decayentries[crushedcomment]] \
                            = float(words[0])
        # Finished reading file!                                  
        return decaydict              
        
    def writeinput(self,paramvector):
        """Write the input SLHA file.
        Parameter names used in 'paramvector' dictionary must match
        'SLHAdict'
        paramvector - dictionary containing parameter names and values
        options - dictionary containing option names and values
        """
        if self.template == None:
            raise ValueError('Error in ISAJET wrapper: Attempted to \
create input SLHA file without specifying a template!')
        else:
            for param,val in self.options.items()+paramvector.items():
                try:
                    blockname,indices = specdict[param]                 #get the block and index for this parameter
                except KeyError, err:
                    msg = '{0} : Parameter name supplied in paramvector \
(or options) does not match any entry of the SLHA mapping dictionary (param = {1}, \
paramvector = {2}'.format(err,param,paramvector)
                    raise KeyError(msg)
                if len(indices)==1: 
                    self.SLHAtemplate.block(blockname)[indices[0]].value = val   #set the new parameter value in the template SLHA file object
                #elif len(indices)==2:  WONT WORK! SEE self.getobs
                #    self.SLHAtemplate.block(blockname)[indices[0]][indices[1]].value = val
                else:
                    raise KeyError('No entry found in template nmspec SLHA \
input file for specified parameter! (param={0}, blockname={1}, indices={2})'.format(param,blockname,indices))
            self.SLHAtemplate.copyto(self.infile)                       #save the modified template values into 'infile' in SLHA format
예제 #2
0
파일: isajetf2py.py 프로젝트: bjfar/pysusy
class Isajet:
    def __init__(self,infile,outfile,template=None,silenceobjs=None):
        self.infile = infile                                            #Either the file to use as input, or the name to given the input file which will be created from the file 'template'.
        self.outfile = outfile
        self.SLHAout = SpectrumFile(name=self.outfile, directory='')    #SpectrumFile has extra blueprints defined to deal with multi-index items
        if template != None:
            self.template = template
            self.SLHAtemplate = SLHAFile(name=self.template, directory='')       #File to modify to create infile
            self.SLHAtemplate.readdata()                                #Read data from the template file
        #save info about stdout for use by silence functions
        #NOW GET THESE EXTERNALLY SO THEY CAN BE SHARED BY ALL PROGRAMS
        ##self.null_fds = [os.open(os.devnull, os.O_RDWR) for x in xrange(2)] # open 2 fds
        ##self.save = os.dup(1), os.dup(2)                                # save the current file descriptors to a tuple
        #self.null_fds, self.save = silenceobjs
        
    def silence(self):
        """Kill the stdout of the current process. Lets us silence the
        output of the wrapped fortran code, which 
        sys.stdout = open('/dev/null'), for example, does not"""
        os.dup2(self.null_fds[0], 1)                                    # put /dev/null fds on 1 and 2
        os.dup2(self.null_fds[1], 2)
    
    def unsilence(self):
        """undoes the action of silence"""
        os.dup2(self.save[0], 1)                                        # restore file descriptors so I can print the results
        os.dup2(self.save[1], 2)
        # close the temporary fds
        #os.close(null_fds[0]) #don't think I want to do this, may as well keep using the same ones every loop
        #os.close(null_fds[1])
        
    def run(self):
        """Execute process
        Used to use multiprocessing module to spawn the new process, but
        it does weird things like 'import random', so now we'll just do
        the fork ourselves.
        """
        sys.stdout.flush()                                              #os.fork() clones the io buffer for these files, so make sure they get dumped to disk before the fork or the output will get duplicated. 
        sys.stderr.flush()
        childpid = os.fork()                                            #spawn child process and get its pid (or 0 if this is the child process)
        if childpid == 0:
            #self.silence()                                              # Don't want default output clogging up our log files so dump it all to dev/null
            with Silence(): isajet.isasugraslha(self.infile, self.outfile)              #run isajet in the child process, in case a STOP statement is encountered in the fortran code
            os._exit(0)                                                 #exit the child process with error code 0 (even if isajet fails we'll still do this coz we can tell how things went from the output file)
        else:
            os.waitpid(childpid, 0)
        #self.unsilence()    #do the unsilencing out here, since the child process can be killed by isajet
        #print 'hello'
        """The fork seems to be slower than using Process, somehow? Check this.
        update: apparently the average time comes out the same so I guess it is fine
        sys.stderr.write('RUNNING ISAJET')
        p = Process(target=isajet.isasugraslha, args=(self.infile, self.outfile))
        p.start()
        p.join()
        sys.stderr.write('ISAJET PROCESS REJOINED')
        """
        
    def getobs(self,skiperrors=True):
        """Gather results of computations"""
        self.SLHAout.readdata()
        return self.SLHAout.getobs(SLHAdict)                            #extract observables from SLHA file using translation dictionary
        
        """
        self.SLHAout.readdata()                                         #read SLHA file (NOTE: THIS BY FAR DOMINATES THE CPU TIME FOR getobs. OPTIMISE THIS IF REQUIRED)
        self.obs = {}                                                   #reset output dictionary
        for key,(blockname,indices) in SLHAdict.items():                #cycle through SLHA translation dictionary
            item = self.SLHAout.block(blockname)
            try:
                for i in indices:
                   item = item[i]                                       #dig through nested lists of items 
            except TypeError:                                      
                if item: item = item[indices]                           #if no more item lists are found, assume we have only one index
                else: return None
            except AttributeError:
                print "Error extracting value from file {0}, block {1}, index {2}. Blueprint may \
    be missing, perhaps because file object is created from a class lacking this blueprint, for \
    example from 'SLHAFile' rather than 'SpectrumFile'.".format(self.SLHAout.name,blockname,indices)
                raise
            self.obs[key] = item.value
            
            try:
                if len(indices)==1: 
                    self.obs[key] = self.SLHAout.block(blockname)[indices[0]].value  #extract observables from SLHA file object
                elif len(indices)==2:
                    print self.SLHAout.block(blockname)
                    print dir(self.SLHAout.block(blockname))
                    print self.SLHAout.block(blockname)[1]
                    item = self.SLHAout.block(blockname)
                    try:
                        for i in indices:
                            item = item[i]
                    except TypeError:
                        if item: item = item[index]
                    self.obs[key] = self.SLHAout.block(blockname,indices)
                    self.obs[key] = self.SLHAout.block(blockname)[indices[0],indices[1]].value
                else:
                    raise KeyError('Invalid indices found in isajet SLHA \
input file! (key={0}, blockname={1}, indices={2})'.format(key,blockname,indices))
            except (KeyError, TypeError):
                print 'Invalid indices found in isajet SLHA \
input file! (key={0}, blockname={1}, indices={2})'.format(key,blockname,indices)
                raise
        return self.obs
        """
        
    def writeinput(self,paramvector):
        """Write the input SLHA file.
        Parameter names used in 'paramvector' dictionary must match
        'SLHAdict'
        """
        if self.template == None:
            raise ValueError('Error in ISAJET wrapper: Attempted to \
create input SLHA file without specifying a template!')
        else:
            for param,val in paramvector.items():
                try:
                    blockname,indices = SLHAdict[param]                 #get the block and index for this parameter
                except KeyError, err:
                    msg = '{0} : Parameter name supplied in paramvector \
does not match any entry of the SLHA mapping dictionary (param = {1}, \
paramvector = {2}'.format(err,param,paramvector)
                    raise KeyError(msg)
                if len(indices)==1: 
                    self.SLHAtemplate.block(blockname)[indices[0]].value = val   #set the new parameter value in the template SLHA file object
                #elif len(indices)==2:  WONT WORK! SEE self.getobs
                #    self.SLHAtemplate.block(blockname)[indices[0]][indices[1]].value = val
                else:
                    raise KeyError('No entry found in template isajet SLHA \
input file for specified parameter! (param={0}, blockname={1}, indices={2})'.format(param,blockname,indices))
            self.SLHAtemplate.copyto(self.infile)                       #save the modified template values into 'infile' in SLHA format