package com.smtaiserver.smtaiserver.gismap.theme;
|
|
import java.util.ArrayList;
|
import java.util.Date;
|
import java.util.HashMap;
|
import java.util.LinkedHashMap;
|
import java.util.List;
|
import java.util.Map;
|
import java.util.Map.Entry;
|
|
import org.apache.logging.log4j.LogManager;
|
import org.apache.logging.log4j.Logger;
|
|
import com.smtaiserver.smtaiserver.core.SMTAIServerApp;
|
import com.smtaiserver.smtaiserver.database.SMTDatabase;
|
import com.smtaiserver.smtaiserver.database.SMTDatabase.DBQueryNotify;
|
import com.smtaiserver.smtaiserver.database.SMTDatabase.DBRecord;
|
import com.smtaiserver.smtaiserver.gismap.theme.SMTMapThemeTableDef.SMTMapThemeColumn;
|
import com.smtaiserver.smtaiserver.javaai.ast.ASTDBMap;
|
import com.smtservlet.util.Json;
|
import com.smtservlet.util.SMTJsonWriter;
|
import com.smtservlet.util.SMTStatic;
|
|
public abstract class SMTMapThemeDef
|
{
|
public enum SMTMapThemeStyleID
|
{
|
PSTYLE,
|
PSIZE,
|
PCOLOR,
|
|
LSTYLE,
|
LSIZE,
|
LCOLOR,
|
|
TSTRING,
|
TSIZE,
|
TCOLOR
|
}
|
|
|
public static class SMTMapThemeLegendValue
|
{
|
public String _value;
|
public String _style;
|
|
public SMTMapThemeLegendValue(SMTMapThemeColumn columnDef, String styleName, Json jsonValue)
|
{
|
_value = columnDef.convValueToSQL(jsonValue.getJson("value").asString());
|
_style = SMTMapThemeDef.convStyleLegendStyle(styleName, jsonValue.getJson("style").asString());
|
}
|
}
|
|
public static class SMTMapThemeLegendStyle
|
{
|
public SMTMapThemeColumn _columnDef;
|
public String _default;
|
public String _title;
|
public List<SMTMapThemeLegendValue> _listLegendValue = new ArrayList<>();
|
|
public SMTMapThemeLegendStyle(String styleName, SMTMapThemeTableDef tableDef, Json jsonLegend) throws Exception
|
{
|
_columnDef = tableDef.getColumnDef(jsonLegend.getJson("column").asString());
|
_title = jsonLegend.safeGetStr("title", null);
|
_default = SMTMapThemeDef.convStyleLegendStyle(styleName, jsonLegend.getJson("default").asString());
|
|
|
for(Json jsonLegendValue : jsonLegend.getJson("legends").asJsonList())
|
{
|
_listLegendValue.add(new SMTMapThemeLegendValue(_columnDef, styleName, jsonLegendValue));
|
}
|
}
|
|
public void appendToSQL(StringBuilder sbSQL)
|
{
|
// 如果没有分级，则直接使用缺省值
|
if(_listLegendValue.size() == 0)
|
{
|
sbSQL.append(_default);
|
}
|
// 生成分级SQL
|
else
|
{
|
sbSQL.append("CASE\n");
|
for(SMTMapThemeLegendValue legendValue : _listLegendValue)
|
{
|
sbSQL.append("WHEN " + _columnDef._name + " " + _columnDef._op + " " + legendValue._value + " THEN " + legendValue._style + "\n");
|
}
|
sbSQL.append("ELSE " + _default + " END");
|
}
|
}
|
}
|
|
private static class SMTMapThemeStyleDef
|
{
|
public SMTMapThemeTableDef _tableDef;
|
public Map<String, SMTMapThemeLegendStyle> _mapStyle2Legend = new HashMap<>();
|
private static Logger _logger = LogManager.getLogger(SMTMapThemeStyleDef.class);
|
|
public SMTMapThemeStyleDef(Json jsonConfig) throws Exception
|
{
|
_tableDef = SMTAIServerApp.getApp().getMapThemeTableDef(jsonConfig.getJson("theme_table").asString());
|
|
for(Entry<String, Json> entry : jsonConfig.getJson("styles").asJsonMap().entrySet())
|
{
|
SMTMapThemeLegendStyle legendStyle = new SMTMapThemeLegendStyle(entry.getKey(), _tableDef, entry.getValue());
|
_mapStyle2Legend.put(entry.getKey(), legendStyle);
|
}
|
|
}
|
|
public void queryThemeStyles(ASTDBMap dbMap, Date curTime, SMTMapThemeStyleMap themeStyleMap) throws Exception
|
{
|
// 生成SQL语句
|
StringBuilder sbSQL = new StringBuilder();
|
sbSQL.append("SELECT OTYPE, ONAME\n");
|
for(Entry<String, SMTMapThemeLegendStyle> entry : _mapStyle2Legend.entrySet())
|
{
|
sbSQL.append(",");
|
|
SMTMapThemeLegendStyle legendStyle = entry.getValue();
|
legendStyle.appendToSQL(sbSQL);
|
sbSQL.append(" AS " + entry.getKey() + "\n");
|
}
|
|
sbSQL.append("FROM (\n");
|
|
Map<String, Object> mapArgValues = new HashMap<>();
|
mapArgValues.put("cur_time", curTime);
|
_tableDef.appendToSQL(mapArgValues, sbSQL);
|
sbSQL.append(") T");
|
|
// 执行SQL，并将结果导入themeStyleMap
|
SMTDatabase db = dbMap.getDatabase(_tableDef.getDSId());
|
_logger.info(sbSQL.toString());
|
db.querySQLNotify(sbSQL.toString(), null, new DBQueryNotify()
|
{
|
@Override
|
public boolean onNextRecord(DBRecord rec) throws Exception
|
{
|
themeStyleMap.appendStyle(rec);
|
return true;
|
}
|
});
|
}
|
|
}
|
|
public static class SMTMapThemeStyleMap
|
{
|
private static class SMTMapThemeStyleCache
|
{
|
public Object[] _styles;
|
public int _pos;
|
|
public SMTMapThemeStyleCache(Object[] styles, int pos)
|
{
|
_styles = styles;
|
_pos = pos;
|
}
|
}
|
|
private Map<String, Map<String, Object[]>> _mapOTYPE2ONAME2Style = new HashMap<>();
|
|
public void appendStyle(DBRecord rec) throws Exception
|
{
|
String OTYPE = rec.getString("OTYPE");
|
String ONAME = rec.getString("ONAME");
|
|
// 获取或创建OTYPE
|
Map<String, Object[]> mapONAME2Style = _mapOTYPE2ONAME2Style.get(OTYPE);
|
if(mapONAME2Style == null)
|
{
|
mapONAME2Style = new HashMap<>();
|
_mapOTYPE2ONAME2Style.put(OTYPE, mapONAME2Style);
|
}
|
|
// 获取或创建ONAME
|
Object[] themeStyle = mapONAME2Style.get(ONAME);
|
if(themeStyle == null)
|
{
|
themeStyle = new Object[SMTMapThemeDef._styleCount];
|
mapONAME2Style.put(ONAME, themeStyle);
|
}
|
|
// 合并style
|
for(SMTMapThemeStyleID styleId : SMTMapThemeStyleID.values())
|
{
|
String styleName = styleId.toString();
|
if("MAX".equals(styleName))
|
continue;
|
int pos = rec.getColIndex(styleName);
|
if(pos < 0)
|
continue;
|
|
Object value;
|
if(styleName.endsWith("COLOR") || styleName.equals("TSTRING"))
|
value = rec.getString(pos);
|
else
|
value = rec.getInteger(pos);
|
|
if(value == null)
|
continue;
|
|
themeStyle[styleId.ordinal()] = value;
|
}
|
}
|
|
public void styleToJson(SMTJsonWriter jsonWr)
|
{
|
Map<String, SMTMapThemeStyleCache> mapKey2Styles = new LinkedHashMap<>();
|
|
// 加入元素
|
for(Entry<String, Map<String, Object[]>> entryOTYPE : _mapOTYPE2ONAME2Style.entrySet())
|
{
|
StringBuilder sbStyleKey = new StringBuilder();
|
jsonWr.beginMap("O_" + entryOTYPE.getKey());
|
for(Entry<String, Object[]> entryONAME : entryOTYPE.getValue().entrySet())
|
{
|
// 计算当前元素的style的key
|
Object[] styles = entryONAME.getValue();
|
sbStyleKey.setLength(0);
|
for(SMTMapThemeStyleID styleId : SMTMapThemeStyleID.values())
|
{
|
if(styleId.toString().equals("TSTRING"))
|
continue;
|
|
Object value = styles[styleId.ordinal()];
|
sbStyleKey.append((value == null ? "null" : SMTStatic.toString(value)) + ",");
|
}
|
|
// 根据key找到对应的style，如果不存在，则创建
|
SMTMapThemeStyleCache styleCache = mapKey2Styles.get(sbStyleKey.toString());
|
if(styleCache == null)
|
{
|
styleCache = new SMTMapThemeStyleCache(styles, mapKey2Styles.size());
|
mapKey2Styles.put(sbStyleKey.toString(), styleCache);
|
}
|
|
// 将对象加入元素表
|
String ONAME = entryONAME.getKey();
|
jsonWr.beginMap(ONAME);
|
{
|
jsonWr.addKeyValue("style_id", styleCache._pos);
|
|
String TSTRING = (String)styles[SMTMapThemeStyleID.TSTRING.ordinal()];
|
if(!SMTStatic.isNullOrEmpty(TSTRING))
|
jsonWr.addKeyValue("TSTRING", TSTRING);
|
}
|
jsonWr.endMap();
|
}
|
jsonWr.endMap();
|
}
|
|
// 加入风格
|
jsonWr.beginArray("styles");
|
for(Entry<String, SMTMapThemeStyleCache> entry : mapKey2Styles.entrySet())
|
{
|
jsonWr.beginMap(null);
|
for(SMTMapThemeStyleID styleId : SMTMapThemeStyleID.values())
|
{
|
if(styleId.toString().equals("TSTRING"))
|
continue;
|
|
Object value = entry.getValue()._styles[styleId.ordinal()];
|
if(value == null)
|
continue;
|
|
jsonWr.addKeyValue(styleId.toString(), SMTStatic.toString(value));
|
}
|
jsonWr.endMap();
|
}
|
jsonWr.endArray();
|
}
|
}
|
|
protected String _id;
|
protected String _title;
|
protected String _group;
|
protected static int _styleCount = SMTMapThemeStyleID.values().length;
|
protected List<SMTMapThemeStyleDef> _listThemeStyleDef = new ArrayList<>();
|
|
public void initInstance(DBRecord rec) throws Exception
|
{
|
_id = rec.getString("theme_id");
|
_title = rec.getString("theme_title");
|
_group = rec.getString("theme_group");
|
Json jsonThemeList = Json.read(rec.getString("theme_legend"));
|
|
for(Json jsonTheme : jsonThemeList.getJson("themes").asJsonList())
|
{
|
SMTMapThemeStyleDef themeStyleDef = new SMTMapThemeStyleDef(jsonTheme);
|
_listThemeStyleDef.add(themeStyleDef);
|
}
|
}
|
|
public void legendToJson(SMTJsonWriter jsonWr)
|
{
|
jsonWr.beginArray("legends");
|
for(SMTMapThemeStyleDef themeStyleDef : _listThemeStyleDef)
|
{
|
for(Entry<String, SMTMapThemeLegendStyle> entry : themeStyleDef._mapStyle2Legend.entrySet())
|
{
|
SMTMapThemeLegendStyle legendStyle = entry.getValue();
|
if(SMTStatic.isNullOrEmpty(legendStyle._title))
|
continue;
|
|
jsonWr.beginMap(null);
|
{
|
jsonWr.addKeyValue("title", legendStyle._title);
|
jsonWr.addKeyValue("style", entry.getKey());
|
jsonWr.addKeyValue("default", legendStyle._default.replace("'", ""));
|
jsonWr.addKeyValue("operate", legendStyle._columnDef._op);
|
jsonWr.beginArray("legend");
|
for(SMTMapThemeLegendValue legendValue : legendStyle._listLegendValue)
|
{
|
jsonWr.beginMap(null);
|
{
|
jsonWr.addKeyValue("style", legendValue._style.replace("'", ""));
|
jsonWr.addKeyValue("value", legendValue._value.replace("'", "").replace("::timestamp", ""));
|
}
|
jsonWr.endMap();
|
}
|
jsonWr.endArray();
|
}
|
jsonWr.endMap();
|
}
|
}
|
jsonWr.endArray();
|
}
|
|
private static String convStyleLegendStyle(String styleName, String style)
|
{
|
if(styleName.endsWith("COLOR"))
|
return "'" + style + "'";
|
|
return style;
|
}
|
|
public String getId()
|
{
|
return _id;
|
}
|
|
public String getTitle()
|
{
|
return _title;
|
}
|
|
public String getGroup()
|
{
|
return _group;
|
}
|
|
public void queryThemeStyles(ASTDBMap dbMap, Date curTime, SMTMapThemeStyleMap themeStyleMap) throws Exception
|
{
|
for(SMTMapThemeStyleDef themeStyleDef : _listThemeStyleDef)
|
{
|
themeStyleDef.queryThemeStyles(dbMap, curTime, themeStyleMap);
|
}
|
|
}
|
}
|
