summaryrefslogtreecommitdiff
path: root/src/main/java/com/c2kernel/process/AbstractMain.java
blob: b594c1a3c9340f08de1a43dfca80be7f810194d2 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
/**************************************************************************
 * AbstractMain
 *
 * $Revision: 1.67 $
 * $Date: 2004/10/25 15:27:35 $
 *
 * Copyright (C) 2001 CERN - European Organization for Nuclear Research
 * All rights reserved.
 **************************************************************************/

package com.c2kernel.process;

import java.io.File;
import java.io.FileNotFoundException;
import java.io.FileOutputStream;
import java.io.PrintStream;
import java.util.Enumeration;
import java.util.Properties;

import com.c2kernel.utils.FileStringUtility;
import com.c2kernel.utils.Logger;

/**************************************************************************
 *
 * @author $Author: abranson $ $Date: 2004/10/25 15:27:35 $
 * @version $Revision: 1.67 $
 **************************************************************************/
abstract public class AbstractMain
{
    public static boolean runningAsWrapper = false;


   /**************************************************************************
    * reading and setting input paramaters
    **************************************************************************/
    public static Properties readC2KArgs( String[] args ) throws Exception {
        Properties c2kProps;
        Properties argProps = new Properties();
        int logLevel = 0;
        PrintStream logStream = System.out;

        int i = 0;
        while( i < args.length ) {
        	if (args[i].startsWith("-") && args[i].length()>1) {
        		String key = args[i].substring(1);
        		if (argProps.containsKey(key))
        			throw new Exception("Argument "+args[i]+" given twice");
        		String value = "";
        		if (!args[i+1].startsWith("-"))
        			value = args[++i];
        		argProps.put(key, value);
        		i++;
        	}
        	else
        		throw new Exception("Bad argument: "+args[i]);

        }
        
        if (argProps.containsKey("logFile"))
			try {
				logStream = new PrintStream(new FileOutputStream(argProps.getProperty("logFile"), true));
			} catch (FileNotFoundException e) {
				e.printStackTrace();
				throw new Exception("Logfile "+argProps.getProperty("logFile")+" cannot be created");
			}
        
        // Set up log stream
        if (argProps.containsKey("logLevel"))
        	logLevel = Integer.parseInt(argProps.getProperty("logLevel"));
        Logger.addLogStream(logStream, logLevel);

        // Dump params if log high enough
        
        if (Logger.doLog(3)) for (Enumeration<?> e = argProps.propertyNames(); e.hasMoreElements();) {
        	String next = (String)e.nextElement();
			System.out.println("AbstractMain: Param "+next+": "+argProps.getProperty(next));
		}

        String configPath = argProps.getProperty("config");
        if (configPath == null)
        	throw new Exception("Config file not specified");
        
        if (!new File(configPath).exists())
        	throw new Exception("Config file "+configPath+" not found");
        else
        	Logger.msg(0, "Config file: "+configPath);

        // Load config & connect files into c2kprops
        c2kProps = FileStringUtility.loadConfigFile(argProps.getProperty("config") );
        c2kProps.putAll(argProps); // args overlap config

        String connectFile = c2kProps.getProperty("connect");
        if (connectFile == null)
        	throw new Exception("Connect file not specified");
        
        if (!new File(connectFile).exists())
        	throw new Exception("Connect file "+connectFile+" not found");
        else
        	Logger.msg(0, "Connect file: "+connectFile);

       	FileStringUtility.appendConfigFile( c2kProps, connectFile);

       	if (!c2kProps.containsKey("LocalCentre")) {
       		String connectFileName = new File(connectFile).getName();
       		String centreId = connectFileName.substring(0, connectFileName.lastIndexOf(".clc"));
       		c2kProps.setProperty("LocalCentre", centreId);
       	}

        c2kProps.putAll(argProps); // args override connect file too
        Logger.msg(7, "AbstractMain::standardSetUp() - readC2KArgs() DONE.");

        return c2kProps;
    }

}