package com.iailab.netsdk.demo.frame.TargetRecognition;
|
|
import java.awt.BorderLayout;
|
import java.awt.Dimension;
|
import java.awt.FlowLayout;
|
import java.awt.GridLayout;
|
import java.awt.event.ActionEvent;
|
import java.awt.event.ActionListener;
|
import java.awt.event.WindowAdapter;
|
import java.awt.event.WindowEvent;
|
import java.util.ArrayList;
|
import java.util.HashMap;
|
import java.util.Vector;
|
|
import javax.swing.JButton;
|
import javax.swing.JComboBox;
|
import javax.swing.JDialog;
|
import javax.swing.JLabel;
|
import javax.swing.JOptionPane;
|
import javax.swing.JPanel;
|
import javax.swing.JScrollPane;
|
import javax.swing.JTable;
|
import javax.swing.JTextField;
|
import javax.swing.table.DefaultTableCellRenderer;
|
import javax.swing.table.DefaultTableModel;
|
|
import com.iailab.netsdk.common.*;
|
import com.iailab.netsdk.demo.module.LoginModule;
|
import com.iailab.netsdk.demo.module.TargetRecognitionModule;
|
import com.iailab.netsdk.lib.ToolKits;
|
import com.iailab.netsdk.lib.NetSDKLib.NET_FACERECONGNITION_GROUP_INFO;
|
|
public class DispositionOperateDialog extends JDialog {
|
/**
|
*
|
*/
|
private static final long serialVersionUID = 1L;
|
|
// 通道列表
|
private Vector<String> chnlist = new Vector<String>();
|
|
// 通道个数
|
private int nChn = LoginModule.m_stDeviceInfo.byChanNum;
|
|
// 通道列表, 用于布控
|
private ArrayList< Integer> arrayList = new ArrayList<Integer>();
|
|
// key：通道 value：相似度, 用于撤控
|
private HashMap<Integer, Integer> hashMap = new HashMap<Integer, Integer>();
|
|
String groupId = "";
|
String groupName = "";
|
|
public DispositionOperateDialog(String groupId, String groupName) {
|
setTitle(Res.string().getDisposition() + "/" + Res.string().getDelDisposition());
|
setLayout(new BorderLayout());
|
setModal(true);
|
pack();
|
setSize(450, 400);
|
setResizable(false);
|
setLocationRelativeTo(null);
|
setDefaultCloseOperation(DISPOSE_ON_CLOSE); // 释放窗体
|
|
for(int i = 1; i < nChn + 1; i++) {
|
chnlist.add(Res.string().getChannel() + " " + String.valueOf(i));
|
}
|
|
this.groupId = groupId;
|
this.groupName = groupName;
|
|
DispositionListPanel dispositionListPanel = new DispositionListPanel();
|
DispositionInfoPanel dispositionInfoPanel = new DispositionInfoPanel();
|
|
add(dispositionListPanel, BorderLayout.CENTER);
|
add(dispositionInfoPanel, BorderLayout.NORTH);
|
|
findChnAndSimilary();
|
|
addWindowListener(new WindowAdapter() {
|
public void windowClosing(WindowEvent e) {
|
dispose();
|
}
|
});
|
}
|
|
/*
|
* 布控显示列表
|
*/
|
private class DispositionListPanel extends JPanel {
|
private static final long serialVersionUID = 1L;
|
public DispositionListPanel() {
|
BorderEx.set(this, "", 2);
|
setLayout(new BorderLayout());
|
|
JPanel panel = new JPanel();
|
JPanel panel2 = new JPanel();
|
JPanel panel3 = new JPanel();
|
|
Dimension dimension = new Dimension();
|
dimension.width = 145;
|
panel.setPreferredSize(dimension);
|
panel.setLayout(new BorderLayout());
|
panel.add(panel2, BorderLayout.NORTH);
|
panel.add(panel3, BorderLayout.SOUTH);
|
|
addBtn = new JButton(Res.string().getAdd());
|
refreshBtn = new JButton(Res.string().getFresh());
|
panel2.setLayout(new GridLayout(2, 1));
|
panel2.add(addBtn);
|
panel2.add(refreshBtn);
|
|
dispositionBtn = new JButton(Res.string().getDisposition());
|
delDispositionBtn = new JButton(Res.string().getDelDisposition());
|
panel3.setLayout(new GridLayout(2, 1));
|
panel3.add(dispositionBtn);
|
panel3.add(delDispositionBtn);
|
|
data = new Object[512][2];
|
defaultTableModel = new DefaultTableModel(data, Res.string().getDispositionTable());
|
table = new JTable(defaultTableModel){ // 列表不可编辑
|
private static final long serialVersionUID = 1L;
|
@Override
|
public boolean isCellEditable(int row, int column) {
|
return false;
|
}
|
};
|
|
DefaultTableCellRenderer dCellRenderer = new DefaultTableCellRenderer();
|
dCellRenderer.setHorizontalAlignment(JLabel.CENTER);
|
table.setDefaultRenderer(Object.class, dCellRenderer);
|
|
add(new JScrollPane(table), BorderLayout.CENTER);
|
add(panel, BorderLayout.EAST);
|
|
// 添加通道和相似度到列表
|
addBtn.addActionListener(new ActionListener() {
|
@Override
|
public void actionPerformed(ActionEvent arg0) {
|
boolean isExit = false;
|
|
String chn = String.valueOf(chnComboBox.getSelectedIndex() + 1);
|
String similary = similaryTextField.getText();
|
|
if(similaryTextField.getText().equals("")
|
|| Integer.parseInt(similaryTextField.getText()) > 100) {
|
JOptionPane.showMessageDialog(null, Res.string().getSimilarityRange(), Res.string().getErrorMessage(), JOptionPane.ERROR_MESSAGE);
|
return;
|
}
|
|
// 如果存在了通道号，列表修改相应的相似度
|
for(int i = 0; i < nChn; i++) {
|
if(chn.equals(String.valueOf(defaultTableModel.getValueAt(i, 0)).trim())) {
|
defaultTableModel.setValueAt(similary, i, 1);
|
isExit = true;
|
break;
|
}
|
}
|
|
if(isExit) {
|
return;
|
}
|
|
// 如果不存在通道号，按顺序添加人列表
|
for(int i = 0; i < nChn; i++) {
|
if(String.valueOf(defaultTableModel.getValueAt(i, 0)).trim().equals("")
|
|| defaultTableModel.getValueAt(i, 0) == null) {
|
defaultTableModel.setValueAt(chn, i, 0);
|
defaultTableModel.setValueAt(similary, i, 1);
|
break;
|
}
|
}
|
}
|
});
|
|
// 刷新已布控的通道和相似度信息
|
refreshBtn.addActionListener(new ActionListener() {
|
@Override
|
public void actionPerformed(ActionEvent arg0) {
|
findChnAndSimilary();
|
}
|
});
|
|
// 布控
|
dispositionBtn.addActionListener(new ActionListener() {
|
@Override
|
public void actionPerformed(ActionEvent arg0) {
|
hashMap.clear();
|
|
// 获取列表里的数据
|
for(int i = 0; i < nChn; i++) {
|
// 判断通道号是否为空
|
if(!String.valueOf(defaultTableModel.getValueAt(i, 0)).trim().equals("")
|
&& defaultTableModel.getValueAt(i, 0) != null) {
|
|
// 判断相似度是否为空
|
if(!String.valueOf(defaultTableModel.getValueAt(i, 1)).trim().equals("")
|
&& defaultTableModel.getValueAt(i, 1) != null) {
|
hashMap.put(Integer.parseInt(String.valueOf(defaultTableModel.getValueAt(i, 0)).trim()),
|
Integer.parseInt(String.valueOf(defaultTableModel.getValueAt(i, 1)).trim()));
|
} else {
|
hashMap.put(Integer.parseInt(String.valueOf(defaultTableModel.getValueAt(i, 0)).trim()), 0);
|
}
|
}
|
}
|
System.out.println("size:" + hashMap.size());
|
if(hashMap.size() == 0) {
|
JOptionPane.showMessageDialog(null, Res.string().getAddDispositionInfo(), Res.string().getErrorMessage(), JOptionPane.ERROR_MESSAGE);
|
return;
|
}
|
|
if(!TargetRecognitionModule.putDisposition(goroupIdTextField.getText(), hashMap)) {
|
JOptionPane.showMessageDialog(null, Res.string().getFailed() + "," + ToolKits.getErrorCodeShow(), Res.string().getErrorMessage(), JOptionPane.ERROR_MESSAGE);
|
} else {
|
JOptionPane.showMessageDialog(null, Res.string().getSucceed(), Res.string().getPromptMessage(), JOptionPane.INFORMATION_MESSAGE);
|
}
|
|
// 刷新列表
|
findChnAndSimilary();
|
}
|
});
|
|
delDispositionBtn.addActionListener(new ActionListener() {
|
@Override
|
public void actionPerformed(ActionEvent arg0) {
|
arrayList.clear();
|
|
// 获取所有选中行数
|
int[] rows = null;
|
rows = table.getSelectedRows();
|
|
if(rows == null) {
|
JOptionPane.showMessageDialog(null, Res.string().getSelectDelDispositionInfo(), Res.string().getErrorMessage(), JOptionPane.ERROR_MESSAGE);
|
return;
|
}
|
|
// 获取所有选中，非空的通道号
|
for(int i = 0; i < rows.length; i++) {
|
if(!String.valueOf(defaultTableModel.getValueAt(rows[i], 0)).trim().equals("")
|
&& defaultTableModel.getValueAt(rows[i], 0) != null) {
|
arrayList.add(Integer.parseInt(String.valueOf(defaultTableModel.getValueAt(rows[i], 0)).trim()));
|
}
|
}
|
|
if(arrayList.size() == 0) {
|
JOptionPane.showMessageDialog(null, Res.string().getSelectDelDispositionInfo(), Res.string().getErrorMessage(), JOptionPane.ERROR_MESSAGE);
|
return;
|
}
|
|
if(!TargetRecognitionModule.delDisposition(goroupIdTextField.getText(), arrayList)) {
|
JOptionPane.showMessageDialog(null, Res.string().getFailed() + "," + ToolKits.getErrorCodeShow(), Res.string().getErrorMessage(), JOptionPane.ERROR_MESSAGE);
|
} else {
|
JOptionPane.showMessageDialog(null, Res.string().getSucceed(), Res.string().getPromptMessage(), JOptionPane.INFORMATION_MESSAGE);
|
}
|
|
// 刷新列表
|
findChnAndSimilary();
|
}
|
});
|
}
|
}
|
|
/*
|
* 布控信息
|
*/
|
private class DispositionInfoPanel extends JPanel {
|
/**
|
*
|
*/
|
private static final long serialVersionUID = 1L;
|
public DispositionInfoPanel() {
|
BorderEx.set(this, "", 2);
|
setLayout(new FlowLayout());
|
|
Dimension dimension = new Dimension();
|
dimension.height = 80;
|
setPreferredSize(dimension);
|
|
JLabel goroupIdLabel = new JLabel(Res.string().getFaceGroupId(), JLabel.CENTER);
|
JLabel goroupNameLabel = new JLabel(Res.string().getFaceGroupName(), JLabel.CENTER);
|
JLabel chnLabel = new JLabel(Res.string().getChannel(), JLabel.CENTER);
|
JLabel similaryLabel = new JLabel(Res.string().getSimilarity(), JLabel.CENTER);
|
|
goroupIdTextField = new JTextField();
|
goroupNameTextField = new JTextField();
|
chnComboBox = new JComboBox(chnlist);
|
similaryTextField = new JTextField();
|
|
Dimension dimension1 = new Dimension();
|
dimension1.width = 80;
|
dimension1.height = 20;
|
goroupIdLabel.setPreferredSize(dimension1);
|
goroupNameLabel.setPreferredSize(dimension1);
|
chnLabel.setPreferredSize(dimension1);
|
similaryLabel.setPreferredSize(dimension1);
|
goroupIdTextField.setPreferredSize(new Dimension(120, 20));
|
goroupNameTextField.setPreferredSize(new Dimension(120, 20));
|
chnComboBox.setPreferredSize(new Dimension(120, 20));
|
similaryTextField.setPreferredSize(new Dimension(120, 20));
|
|
add(goroupIdLabel);
|
add(goroupIdTextField);
|
add(goroupNameLabel);
|
add(goroupNameTextField);
|
add(chnLabel);
|
add(chnComboBox);
|
add(similaryLabel);
|
add(similaryTextField);
|
|
ToolKits.limitTextFieldLength(similaryTextField, 3);
|
|
goroupIdTextField.setEditable(false);
|
goroupNameTextField.setEditable(false);
|
|
goroupIdTextField.setText(groupId);
|
goroupNameTextField.setText(groupName);
|
}
|
}
|
|
// 查找人脸库的布控通道以及对应的相似度
|
private void findChnAndSimilary() {
|
// 清空列表
|
defaultTableModel.setRowCount(0);
|
defaultTableModel.setRowCount(512);
|
|
// 查询布控信息
|
NET_FACERECONGNITION_GROUP_INFO[] groupInfos = TargetRecognitionModule.findGroupInfo(goroupIdTextField.getText());
|
|
if(groupInfos == null) {
|
return;
|
}
|
|
for(int i = 0; i < groupInfos[0].nRetChnCount; i++) {
|
defaultTableModel.setValueAt(String.valueOf(groupInfos[0].nChannel[i] + 1), i, 0);
|
defaultTableModel.setValueAt(String.valueOf(groupInfos[0].nSimilarity[i]), i, 1);
|
}
|
}
|
|
private Object[][] data;
|
private DefaultTableModel defaultTableModel;
|
private JTable table;
|
|
private JButton addBtn;
|
private JButton refreshBtn;
|
private JButton dispositionBtn;
|
private JButton delDispositionBtn;
|
private JTextField goroupIdTextField;
|
private JTextField goroupNameTextField;
|
private JComboBox chnComboBox;
|
private JTextField similaryTextField;
|
}
|
