Initial commit of multiasset branch

src/opensesim/World.java

@@ -0,0 +1,145 @@
+/*
+ * Copyright (c) 2018, 7u83 <7u83@mail.ru>
+ * All rights reserved.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions are met:
+ *
+ * * Redistributions of source code must retain the above copyright notice, this
+ *   list of conditions and the following disclaimer.
+ * * Redistributions in binary form must reproduce the above copyright notice,
+ *   this list of conditions and the following disclaimer in the documentation
+ *   and/or other materials provided with the distribution.
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
+ * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+ * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
+ * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
+ * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
+ * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
+ * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
+ * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
+ * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
+ * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
+ * POSSIBILITY OF SUCH DAMAGE.
+ */
+package opensesim;
+
+import java.util.ArrayList;
+import java.util.Collection;
+import java.util.Collections;
+import java.util.HashMap;
+import java.util.HashSet;
+import opensesim.sesim.AssetPair;
+import opensesim.sesim.interfaces.Configurable;
+import opensesim.util.IDGenerator;
+import opensesim.util.IDGenerator.Id;
+import opensesim.util.SeSimException;
+import org.json.JSONArray;
+import org.json.JSONObject;
+
+/**
+ *
+ * @author 7u83 <7u83@mail.ru>
+ */
+public class World implements Configurable {
+
+    HashMap<Id, AbstractAsset> assetsById = new HashMap<>();
+    HashMap<String, AbstractAsset> assetsBySymbol = new HashMap<>();
+    IDGenerator assetIdGenerator = new IDGenerator();
+
+    HashSet<AssetPair> assetPairs = new HashSet<>();
+
+    ArrayList<Exchange> exchanges = new ArrayList<>();
+
+    /**
+     * Create a World object.
+     */
+    public World() {
+
+    }
+
+    public Collection<AbstractAsset> getAssetCollection() {
+        return Collections.unmodifiableCollection(assetsById.values());
+    }
+
+    public Collection<AssetPair> getAssetPairsCollection() {
+        return Collections.unmodifiableCollection(assetPairs);
+    }
+
+    public Collection<Exchange> getExchangeCollection() {
+        return Collections.unmodifiableCollection(exchanges);
+    }
+
+    public void add(AssetPair pair) {
+        assetPairs.add(pair);
+    }
+
+    public Exchange addExchange(String symbol) throws SeSimException {
+        for (Exchange ex : getExchangeCollection()) {
+            if (ex.getSymbol().equals(symbol)) {
+                throw (new SeSimException("Exchange already defined."));
+            }
+        }
+
+        Exchange ex = new Exchange(this, symbol);
+        exchanges.add(ex);
+        return ex;
+    }
+
+    static final String JSON_ASSET = "asset";
+    static final String JSON_EXCHANGES = "exchanges";
+
+    @Override
+    public JSONObject getConfig() {
+        JSONObject cfg = new JSONObject();
+
+        // save assets
+        JSONArray arr;
+        arr = new JSONArray();
+        for (AbstractAsset a : getAssetCollection()) {
+            arr.put(a.getConfig());
+        }
+        cfg.put(JSON_ASSET, arr);
+
+        // save exchanges
+        arr = new JSONArray();
+        for (Exchange ex : getExchangeCollection()) {
+            arr.put(ex.getConfig());
+        }
+        cfg.put(JSON_EXCHANGES, arr);
+
+        return cfg;
+    }
+
+    @Override
+    public void putConfig(JSONObject cfg) {
+        JSONArray arr = cfg.optJSONArray(JSON_ASSET);
+        if (arr == null) {
+            arr = new JSONArray();
+        }
+        for (Object o : arr) {
+            JSONObject acfg = (JSONObject) o;
+            AbstractAsset.create(this, acfg);
+        }
+    }
+
+    public AbstractAsset createAsset(Class cls, String symbol) throws Exception {
+        return AbstractAsset.create(this, cls, symbol);
+    }
+
+    /*
+        static public JSONArray toJson() {
+
+        JSONArray all = new JSONArray();
+        for (Map.Entry<Id, Asset> entry : assetsById.entrySet()) {
+            Id key = entry.getKey();
+            Asset value = entry.getValue();
+            all.put(value.getConfig());
+
+        }
+
+        return all;
+    }
+     */
+}