Herwig.java

Go to the documentation of this file.

package cedar.jetweb.generator;

import java.util.HashMap;
import java.util.Map;
import java.util.Map.Entry;
import java.util.Vector;

import java.io.BufferedWriter;
import java.io.IOException;

import cedar.jetweb.JetWebException;

import cedar.jetweb.model.MCProcessType;
//import cedar.jetweb.model.Collision;
import cedar.jetweb.model.Model;
import cedar.jetweb.db.DBGeneratorManager;
import cedar.jetweb.job.Cut;
import cedar.jetweb.job.CutCollection;

public class Herwig extends Generator {

    private HashMap<Integer, String> processNameMap_ = null;
    private HashMap<String, Vector<Integer>> processSelectorMap_ = 
    new HashMap<String, Vector<Integer>>();
    private boolean processNamesInit_ = true;

    private HashMap<String, String> allowedCuts_ = 
    new HashMap<String, String>();

    private boolean allowedCutsInit_ = true;

    protected Herwig(String version) {
    generatorName = "herwig";
    generatorVersion = version;

    }

    public String getProcessSelector(MCProcessType proc) {

    StringBuffer sel = new StringBuffer("");
    if(processNamesInit_) initProcessNames();

    Vector<Integer> PIds = proc.getCompatibleProcesses();

    Integer iprocNumber = 1;
    boolean found = false;
    for(Integer Id: PIds){

        MCProcessType subProc = new MCProcessType(Id);
        
        Vector<Integer> iprocs = 
        processSelectorMap_.get(subProc.getMCProcessName());
        
        if(iprocs!=null){
        found = true;
        for(Integer iproc: iprocs){
            sel.append("IPROC "+iprocNumber+" = "+iproc+"\n");
            ++iprocNumber;
        }
        sel.append("C\n");
        }
    }
    
    if(found) return sel.toString();

    if (proc.getMCProcessName().equals(MCProcessType.HIGHET)){
        if (proc.isPhotoproduction()){      
        sel.append("IPROC 1 = 5000\n");
        sel.append("IPROC 2 = 1500\n");
        return sel.toString();
        } else if (proc.isPP()){        
        sel.append("IPROC 1 = 1500\n");
        return sel.toString();
        } else if (proc.isGammaGamma()){
        sel.append("IPROC 1 = 6000\n");
        sel.append("IPROC 2 = 5000\n");
        sel.append("IPROC 3 = 1500\n");
        return sel.toString();
        } else {
        return "C Unknown process type";
        }
    } else if (proc.getMCProcessName().equals(MCProcessType.CHARM)){
        if (proc.isPhotoproduction()){      
        sel.append("IPROC 1 = 5004\n");
        sel.append("IPROC 2 = 1504\n");
        return sel.toString();
        } else if (proc.isPP()){        
        sel.append("IPROC 1 = 1504\n");
        return sel.toString();
        } else if (proc.isGammaGamma()){
        sel.append("IPROC 1 = 6004\n");
        sel.append("IPROC 2 = 5004\n");
        sel.append("IPROC 3 = 1504\n");
        return sel.toString();
        } else {
        return "C Unknown process type";
        }
    } else if (proc.getMCProcessName().equals(MCProcessType.BEAUTY)){
        if (proc.isPhotoproduction()){      
        sel.append("IPROC 1 = 5005\n");
        sel.append("IPROC 2 = 1505\n");
        return sel.toString();
        } else if (proc.isPP()){        
        sel.append("IPROC 1 = 1505\n");
        return sel.toString();
        } else if (proc.isGammaGamma()){
        sel.append("IPROC 1 = 6005\n");
        sel.append("IPROC 2 = 5005\n");
        sel.append("IPROC 3 = 1505\n");
        return sel.toString();
        } else {
        return "C Unknown process type";
        }
    } else {
        return "C Unknown process type";
    }
    }

    //public String getMCProcessName(HashMap<String,HashMap<Integer,Integer>> parameters) 
    public String getMCProcessName(HashMap<String, Object> parameters)
    throws JetWebException{

    Integer iproc = (Integer) parameters.get("IPROC");

    //  HashMap<Integer,Integer> iprocArray = parameters.get("IPROC");

    //Integer iproc = iprocArray.get(new Integer(1));
    if(iproc==null){
        throw new JetWebException
        ("IPROC not present in intParameters",
         "cedar.jetweb.generator.herwig: getMCProcessType()");
    }

    if(processNamesInit_) initProcessNames();

    String processName = processNameMap_.get(iproc);
    if(processName==null){
        throw new JetWebException
        ("Unknown IPROC number = "+iproc,
         "cedar.jetweb.generator.herwig: getMCProcessType()");
    }

    return processName;
    }


    public boolean processTrumps(Generator gen){
    Integer iproc= intParameters.get("IPROC");
    if(iproc==1500){
        iproc = gen.getIntParameters().get("IPROC");
        if(iproc==5000) return true;
    }
    return false;
    }

    public boolean checkParameters(){
    return true;
    }

    protected String getParticleName(String pName){
    return pName;
    }

    public void writeSteering(BufferedWriter ofile, Model model, 
                  MCProcessType proc) 
    throws JetWebException {

    try {
        // Common, except PDFs
        writeCommonFFKey(ofile,proc);
        // Turn on the appropriate HZTool routines
        writePapersFFKey(ofile,proc);
        // PDFs
        writePDFFFKey(ofile,model,proc);
        // All the other parameters
        writeGenFFKey(ofile,proc);

    } catch (IOException io) {
        throw new JetWebException(io,"Herwig.writeSteering failed in write to "+ofile);
    }

    }

    private void initAllowedCuts(){
    if(!allowedCutsInit_) return;
    allowedCutsInit_ = false;
    allowedCuts_.put("PTMin", "PTMIN");
    }

    //initialises the mapping between iproc and process name strings
    private void initProcessNames(){
    if(!processNamesInit_) return;
    processNamesInit_ = false;

    processSelectorMap_ = new HashMap<String, Vector<Integer>>();

    Vector<Integer> iprocs = new Vector<Integer>();

    iprocs.add(1500);
    
    processSelectorMap_.put(MCProcessType.PARTONPARTON, iprocs);

    iprocs = new Vector<Integer>();
    iprocs.add(1504);
    processSelectorMap_.put(MCProcessType.CHARMPARTONPARTON, iprocs);

    iprocs = new Vector<Integer>();
    iprocs.add(1505);
    processSelectorMap_.put(MCProcessType.BEAUTYPARTONPARTON, iprocs);

    iprocs = new Vector<Integer>();

    iprocs.add(5000);
    
    processSelectorMap_.put(MCProcessType.PHOTONPARTON, iprocs);

    iprocs = new Vector<Integer>();
    iprocs.add(5004);
    processSelectorMap_.put(MCProcessType.CHARMPHOTONPARTON, iprocs);

    iprocs = new Vector<Integer>();
    iprocs.add(5005);
    processSelectorMap_.put(MCProcessType.BEAUTYPHOTONPARTON, iprocs);

    iprocs = new Vector<Integer>();

    iprocs.add(6000);

    processSelectorMap_.put(MCProcessType.PHOTONPHOTON, iprocs);

    iprocs = new Vector<Integer>();
    iprocs.add(6004);
    processSelectorMap_.put(MCProcessType.CHARMPHOTONPHOTON, iprocs);
    
    iprocs = new Vector<Integer>();
    iprocs.add(6005);
    processSelectorMap_.put(MCProcessType.BEAUTYPHOTONPHOTON, iprocs);

    iprocs = new Vector<Integer>();

    iprocs.add(8000);

    processSelectorMap_.put(MCProcessType.MINBIAS, iprocs);

    processNameMap_ = new HashMap<Integer, String>();
    
    
    for(Integer sue=0; sue!=20000; sue+=10000){

        for(Integer iproc=100; iproc!=1146; ++iproc){
        processNameMap_.put(iproc+sue, MCProcessType.ANNIHILATION);
        }

        processNameMap_.put(1500+sue, MCProcessType.PARTONPARTON);
        processNameMap_.put(1504+sue, MCProcessType.CHARMPARTONPARTON);
        processNameMap_.put(1505+sue, MCProcessType.BEAUTYPARTONPARTON);
        processNameMap_.put(5000+sue, MCProcessType.PHOTONPARTON);
        processNameMap_.put(5004+sue, MCProcessType.CHARMPHOTONPARTON);
        processNameMap_.put(5005+sue, MCProcessType.BEAUTYPHOTONPARTON);
        processNameMap_.put(8000+sue, MCProcessType.MINBIAS);
        
        for(Integer iproc=9000; iproc!=9131; ++iproc){
        processNameMap_.put(iproc+sue, MCProcessType.DIS);

        if(iproc.toString().charAt(3)=='4') 
            processNameMap_.put(iproc+sue, MCProcessType.CHARMDIS);
        if(iproc.toString().charAt(3)=='5') 
            processNameMap_.put(iproc+sue, MCProcessType.BEAUTYDIS);

        if(iproc==9120)
            processNameMap_.put(iproc+sue, MCProcessType.CHARMDIS);
        if(iproc==9121)
            processNameMap_.put(iproc+sue, MCProcessType.BEAUTYDIS);
        
        
        if(iproc==9107) iproc+=3;
        if(iproc==9122) iproc+=8;
        }
    }
    
    return;
    }

    public boolean inModel(String pName){
    if (pName.equalsIgnoreCase("IPROC")) return false;
    if (pName.equalsIgnoreCase("PTMIN")) return false;
    if (pName.equalsIgnoreCase("JIMMY") || 
        pName.equalsIgnoreCase("PTJIM") ||
        pName.equalsIgnoreCase("PHRAD") ||
        pName.equalsIgnoreCase("PRRAD")){
        Integer iproc = intParameters.get("IPROC");
        if(iproc!=null && iproc==5000){
        return false;
        }
    }
    return true;
    }
    public boolean inProcessType(String pName){
    if (pName.equalsIgnoreCase("IPROC")) return true;
    return false;
    }

    public boolean inCuts(String pName){
    if (pName.equalsIgnoreCase("PTMIN")) return true;
    return false;
    }

    public Herwig addCut(Cut cut){

    if(allowedCutsInit_) initAllowedCuts();

    if(cuts_==null) cuts_ = new CutCollection();

    String cutName = allowedCuts_.get(cut.getName());
    
    //if cut not known by Herwig then return
    if(cutName==null) return this;

    // if existing cut in the collection supersedes this one 
    // then don't do anything
    if(!cuts_.addCut(cut)) return this;
    
    realParameters.put(cutName, cut.getValue());

    return this;
    }

    public CutCollection getCutCollection(){

    if(!dirtyCuts_) return cuts_;
    dirtyCuts_ = false;
    cuts_ = new CutCollection();

    if(allowedCutsInit_) initAllowedCuts();

    for(Map.Entry entry: allowedCuts_.entrySet()){
        String param = (String)entry.getValue();
        String name = (String)entry.getKey();
        Double value = realParameters.get(param);


        Cut tmp = new Cut(name, value);
        cuts_.addCut(tmp);
    }

    return cuts_;
    }

    public Double getPthat(){
    
    return  realParameters.get("PTMIN"); 
    
    }
    
}


/*


        // Choose ptmin. This should depend upon
        // process & generator, not depending on the model.
        // i.e. it is the kinematic cut (CKIN(3), PTMIN etc) not a
        // model parameter.
        if (line.startsWith("PTMIN")) {
            line = "PTMIN "+" to be done!";
        }

        // Choose ptjim.
        if (line.startsWith("PTJIM")) {
            line = "PTJIM "+model.getPtjim();
        }

        // Choose proton radius
        if (line.startsWith("PRRAD")) {
            line = "PRRAD "+model.getProtonRadius();
        }
        // Choose proton radius
        if (line.startsWith("PHRAD")) {
            line = "PHRAD "+model.getPhotonRadius();
        }
        // Choose underlying event model
        if (line.startsWith("JIMMY")) {
            line = "JIMMY "+model.getJimmy();
        }
        // Choose proton & photon intrinsic kt
        if (line.startsWith("PTRMS ")) {
            line = "PTRMS "+model.getPtrms();
        }
        // Choose Qspac
        if (line.startsWith("QSPAC")) {
            line = "QSPAC "+model.getQspac();
        }
        // Choose CLMAX
        if (line.startsWith("CLMAX")) {
            line = "CLMAX "+model.getClmax();
        }
        // Choose QCDLAM
        if (line.startsWith("QCDLAM")) {
            line = "QCDLAM "+model.getQcdlam();
        }
        // Choose VQCUT
        if (line.startsWith("VQCUT")) {
            line = "VQCUT "+model.getVqcut();
        }
        // Choose PRSOF
        if (line.startsWith("PRSOF")) {
            line = "PRSOF "+model.getPrsof();
        }
        // Choose IOPREM
        if (line.startsWith("IOPREM")) {
            line = "IOPREM "+model.getIoprem();
        }
        // Choose BTCLM
        if (line.startsWith("BTCLM")) {
            line = "BTCLM "+model.getBtclm();
        }
        // Choose PHAD
        if (line.startsWith("PHAD")) {
            line = "PHAD "+model.getPhad();
        }
        // Choose Psplt1
        if (line.startsWith("PSPLT1")) {
            line = "PSPLT1 "+model.getPsplt1();
        }
        // Choose Psplt2
        if (line.startsWith("PSPLT2")) {
            line = "PSPLT2 "+model.getPsplt2();
        }

        // Choose the process for HERWIG
        // This is a hack. 
        // It needs to be done more generally using a proper Generator class.
        if (line.startsWith("IPROC")){
            if (proc.getMCProcessName().equals(MCProcessType.HIGHET)){
            // Note that SUE is still on. To turn it off,
            // PRSOF has to be set to 0.
            line = "IPROC 1500";
            }
        }

        newCards.write(line);
        newCards.newLine();

        //  Now the detailed parameter list for PYTHIA.
        if (line.startsWith("c Detailed Pythia parameters")){

            Map mstp = new TreeMap(model.getMstp());
            Map parp = new TreeMap(model.getParp());

            Iterator iparam = mstp.entrySet().iterator();
            while (iparam.hasNext()){
            Entry mstpEntry = (Entry)iparam.next();

            Integer mstpValue = (Integer)mstpEntry.getValue();
            Integer key = (Integer)mstpEntry.getKey();
            if (mstpValue!=null) {
                newCards.write("MSTP "+key+"="+mstpValue);
                newCards.newLine();
            }
            }
 
            iparam = parp.entrySet().iterator();
            while (iparam.hasNext()){
            Entry parpEntry = (Entry)iparam.next();

            Double parpValue = (Double)parpEntry.getValue();
            Integer key = (Integer)parpEntry.getKey();
            if (parpValue!=null) {
                newCards.write("PARP "+key+"="
                       +JetWebConfig.getFormat().format(parpValue.floatValue()));
                newCards.newLine();
            }
            } 

            Map mstj = new TreeMap(model.getMstj());
            Map parj = new TreeMap(model.getParj());

            iparam = mstj.entrySet().iterator();
            while (iparam.hasNext()){
            Entry mstjEntry = (Entry)iparam.next();

            Integer mstjValue = (Integer)mstjEntry.getValue();
            Integer key = (Integer)mstjEntry.getKey();
            if (mstjValue!=null) {
                newCards.write("MSTJ "+key+"="+mstjValue);
                newCards.newLine();
            }
            }
 
            iparam = parj.entrySet().iterator();
            while (iparam.hasNext()){
            Entry parjEntry = (Entry)iparam.next();

            Double parjValue = (Double)parjEntry.getValue();
            Integer key = (Integer)parjEntry.getKey();
            if (parjValue!=null) {
                newCards.write("PARJ "+key+"="
                       +JetWebConfig.getFormat().format(parjValue.floatValue()));
                newCards.newLine();
            }
            } 
        }

        if (line.startsWith("c List of HZTOOL routines")){
            //newCards.write(line);
            //newCards.newLine();
            HashSet paperSwitches = new HashSet(PaperBank.getNPapers());

            Enumeration plotList = PaperBank.getAllPlots();
            while (plotList.hasMoreElements()){
            RealPlot plot = (RealPlot)plotList.nextElement();
            if (plot.getCollisionId()==collision.getId()){
                paperSwitches.add(new Integer(plot.getPaperId()));
            }
            }
            for (Iterator papersOn = paperSwitches.iterator();papersOn.hasNext();){
            newCards.write("ROUTINE "+(Integer)papersOn.next()+" = TRUE");
            newCards.newLine();
            }
            

        }
        }



        */

    /*
    // Special check on some HERWIG parameters to make sure
    // they are consistent.  Must all move to the herwig class
    // soon
    if (getGenerator().equalsIgnoreCase("herwig")){
        try {
        
        // If JIMMY is off, set its parameters to the default values.
        if ( getRunSeries().getJimmy().intValue()==0){
            if (getRunSeries().getMCProcessType().hasPhoton()){
            runSeries.setPhad(JetWebConfig.DefaultPhad);
            runSeries.setPhotonRadius(JetWebConfig.DefaultPhrad);
            }
            if (getRunSeries().getMCProcessType().hasProton()){
            runSeries.setProtonRadius(JetWebConfig.DefaultPrrad);
            }
            runSeries.setPtjim(JetWebConfig.DefaultPtjim);
        }
        
        } catch (Exception e){
        valid=false;
        System.out.println("Logfile WARNING: badly populated HERWIG params");
        }
        
    }
    */

Generated Wed Jan 17 09:14:27 GMT 2007