package com.iailab.netsdk.demo.frame.TargetRecognition;
|
|
import java.awt.BorderLayout;
|
import java.awt.Dimension;
|
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.io.UnsupportedEncodingException;
|
|
import javax.swing.JButton;
|
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.ListSelectionModel;
|
import javax.swing.table.DefaultTableCellRenderer;
|
import javax.swing.table.DefaultTableModel;
|
|
import com.iailab.netsdk.common.BorderEx;
|
import com.iailab.netsdk.common.Res;
|
import com.iailab.netsdk.demo.module.TargetRecognitionModule;
|
import com.iailab.netsdk.lib.NetSDKLib.*;
|
import com.iailab.netsdk.lib.ToolKits;
|
|
public class GroupOperateDialog extends JDialog{
|
|
/**
|
*
|
*/
|
private static final long serialVersionUID = 1L;
|
|
// 人脸库名称
|
private String inputGroupName = "";
|
|
// 布控界面
|
public DispositionOperateDialog dispositionOperateDialog = null;
|
|
// 人员操作界面
|
public PersonOperateDialog personOperateDialog = null;
|
|
public GroupOperateDialog() {
|
setTitle(Res.string().getGroupOperate());
|
setLayout(new BorderLayout());
|
setModal(true);
|
pack();
|
setSize(650, 360);
|
setResizable(false);
|
setLocationRelativeTo(null);
|
setDefaultCloseOperation(DISPOSE_ON_CLOSE); // 释放窗体
|
|
GroupListPanel GroupPanel = new GroupListPanel();
|
GroupOperatePanel GroupOperatePanel = new GroupOperatePanel();
|
|
add(GroupPanel, BorderLayout.CENTER);
|
add(GroupOperatePanel, BorderLayout.EAST);
|
|
findGroupInfo();
|
|
addWindowListener(new WindowAdapter() {
|
public void windowClosing(WindowEvent e) {
|
dispose();
|
}
|
});
|
}
|
|
/*
|
* 人脸库显示列表
|
*/
|
private class GroupListPanel extends JPanel {
|
private static final long serialVersionUID = 1L;
|
public GroupListPanel() {
|
BorderEx.set(this, "", 2);
|
setLayout(new BorderLayout());
|
|
data = new Object[20][3];
|
defaultTableModel = new DefaultTableModel(data, Res.string().getGroupTable());
|
table = new JTable(defaultTableModel) { // 列表不可编辑
|
private static final long serialVersionUID = 1L;
|
@Override
|
public boolean isCellEditable(int row, int column) {
|
return false;
|
}
|
};
|
|
table.getColumnModel().getColumn(0).setPreferredWidth(80);
|
table.getColumnModel().getColumn(1).setPreferredWidth(280);
|
table.getColumnModel().getColumn(2).setPreferredWidth(100);
|
|
table.setSelectionMode(ListSelectionModel.SINGLE_SELECTION); // 只能选中一行
|
|
DefaultTableCellRenderer dCellRenderer = new DefaultTableCellRenderer();
|
dCellRenderer.setHorizontalAlignment(JLabel.CENTER);
|
table.setDefaultRenderer(Object.class, dCellRenderer);
|
|
add(new JScrollPane(table), BorderLayout.CENTER);
|
|
}
|
}
|
|
/*
|
* 人脸库操作
|
*/
|
private class GroupOperatePanel extends JPanel {
|
/**
|
*
|
*/
|
private static final long serialVersionUID = 1L;
|
public GroupOperatePanel() {
|
BorderEx.set(this, "", 2);
|
setLayout(new BorderLayout());
|
|
Dimension dimension = new Dimension();
|
dimension.width = 230;
|
setPreferredSize(dimension);
|
|
JPanel GroupPanel = new JPanel();
|
JPanel panel = new JPanel();
|
|
add(GroupPanel, BorderLayout.CENTER);
|
add(panel, BorderLayout.SOUTH);
|
|
/*JButton searchByPicBtn = new JButton(Res.string().getSearchByPic());*/
|
JButton personOperateBtn = new JButton(Res.string().getPersonOperate());
|
panel.setPreferredSize(new Dimension(230, 45));
|
panel.setLayout(new GridLayout(2, 1));
|
/*panel.add(searchByPicBtn);*/
|
panel.add(personOperateBtn);
|
|
/*
|
* 人脸库增删改， 布控、撤控
|
*/
|
JButton refreshBtn = new JButton(Res.string().getFresh());
|
JButton addGroupBtn = new JButton(Res.string().getAddGroup());
|
JButton modifyGroupBtn = new JButton(Res.string().getModifyGroup());
|
JButton deleteGroupBtn = new JButton(Res.string().getDelGroup());
|
JButton dispositionBtn = new JButton(Res.string().getDisposition() + "/" + Res.string().getDelDisposition());
|
|
GroupPanel.setLayout(new GridLayout(12, 1));
|
GroupPanel.add(refreshBtn);
|
GroupPanel.add(addGroupBtn);
|
GroupPanel.add(modifyGroupBtn);
|
GroupPanel.add(deleteGroupBtn);
|
GroupPanel.add(dispositionBtn);
|
|
// 刷新人脸库列表
|
refreshBtn.addActionListener(new ActionListener() {
|
@Override
|
public void actionPerformed(ActionEvent arg0) {
|
findGroupInfo();
|
}
|
});
|
|
// 添加人脸库
|
addGroupBtn.addActionListener(new ActionListener() {
|
|
@Override
|
public void actionPerformed(ActionEvent arg0) {
|
inputGroupName = JOptionPane.showInputDialog(GroupOperateDialog.this,
|
Res.string().getInputGroupName(), "");
|
|
if(inputGroupName == null) { // 取消或者关闭按钮
|
return;
|
}
|
|
if(TargetRecognitionModule.addGroup(inputGroupName)) {
|
JOptionPane.showMessageDialog(null, Res.string().getSucceed(), Res.string().getPromptMessage(), JOptionPane.INFORMATION_MESSAGE);
|
} else {
|
JOptionPane.showMessageDialog(null, ToolKits.getErrorCodeShow(), Res.string().getErrorMessage(), JOptionPane.ERROR_MESSAGE);
|
}
|
|
// 更新人脸库列表
|
findGroupInfo();
|
}
|
});
|
|
// 修改人脸库
|
modifyGroupBtn.addActionListener(new ActionListener() {
|
|
@Override
|
public void actionPerformed(ActionEvent arg0) {
|
int row = -1;
|
row = table.getSelectedRow(); //获得所选的单行
|
|
if(row < 0) {
|
JOptionPane.showMessageDialog(null, Res.string().getSelectGroup(), Res.string().getErrorMessage(), JOptionPane.ERROR_MESSAGE);
|
return;
|
}
|
|
if(defaultTableModel.getValueAt(row, 0) == null || String.valueOf(defaultTableModel.getValueAt(row, 0)).trim().equals("")) {
|
JOptionPane.showMessageDialog(null, Res.string().getSelectGroup(), Res.string().getErrorMessage(), JOptionPane.ERROR_MESSAGE);
|
return;
|
}
|
|
inputGroupName = JOptionPane.showInputDialog(GroupOperateDialog.this,
|
Res.string().getInputGroupName(), String.valueOf(defaultTableModel.getValueAt(row, 1)).trim());
|
|
if(inputGroupName == null) { // 取消或者关闭按钮
|
return;
|
}
|
|
if(TargetRecognitionModule.modifyGroup(inputGroupName, String.valueOf(defaultTableModel.getValueAt(row, 0)).trim())) {
|
JOptionPane.showMessageDialog(null, Res.string().getSucceed(), Res.string().getPromptMessage(), JOptionPane.INFORMATION_MESSAGE);
|
} else {
|
JOptionPane.showMessageDialog(null, ToolKits.getErrorCodeShow(), Res.string().getErrorMessage(), JOptionPane.ERROR_MESSAGE);
|
}
|
|
// 更新人脸库列表
|
findGroupInfo();
|
}
|
});
|
|
// 删除人脸库
|
deleteGroupBtn.addActionListener(new ActionListener() {
|
|
@Override
|
public void actionPerformed(ActionEvent arg0) {
|
int row = -1;
|
row = table.getSelectedRow(); //获得所选的单行
|
|
if(row < 0) {
|
JOptionPane.showMessageDialog(null, Res.string().getSelectGroup(), Res.string().getErrorMessage(), JOptionPane.ERROR_MESSAGE);
|
return;
|
}
|
|
if(defaultTableModel.getValueAt(row, 0) == null || String.valueOf(defaultTableModel.getValueAt(row, 0)).trim().equals("")) {
|
JOptionPane.showMessageDialog(null, Res.string().getSelectGroup(), Res.string().getErrorMessage(), JOptionPane.ERROR_MESSAGE);
|
return;
|
}
|
|
if(!TargetRecognitionModule.deleteGroup(String.valueOf(defaultTableModel.getValueAt(row, 0)).trim())) {
|
JOptionPane.showMessageDialog(null, ToolKits.getErrorCodeShow(), Res.string().getErrorMessage(), JOptionPane.ERROR_MESSAGE);
|
} else {
|
JOptionPane.showMessageDialog(null, Res.string().getSucceed(), Res.string().getPromptMessage(), JOptionPane.INFORMATION_MESSAGE);
|
}
|
|
// 更新人脸库列表
|
findGroupInfo();
|
}
|
});
|
|
// 布控/撤控
|
dispositionBtn.addActionListener(new ActionListener() {
|
|
@Override
|
public void actionPerformed(ActionEvent arg0) {
|
int row = -1;
|
row = table.getSelectedRow(); //获得所选的单行
|
|
if(row < 0) {
|
JOptionPane.showMessageDialog(null, Res.string().getSelectGroup(), Res.string().getErrorMessage(), JOptionPane.ERROR_MESSAGE);
|
return;
|
}
|
|
if(defaultTableModel.getValueAt(row, 0) == null || String.valueOf(defaultTableModel.getValueAt(row, 0)).trim().equals("")) {
|
JOptionPane.showMessageDialog(null, Res.string().getSelectGroup(), Res.string().getErrorMessage(), JOptionPane.ERROR_MESSAGE);
|
return;
|
}
|
|
dispositionOperateDialog = new DispositionOperateDialog(String.valueOf(defaultTableModel.getValueAt(row, 0)).trim(),
|
String.valueOf(defaultTableModel.getValueAt(row, 1)).trim());
|
dispositionOperateDialog.setVisible(true);
|
|
}
|
});
|
|
/*// 以图搜图
|
searchByPicBtn.addActionListener(new ActionListener() {
|
@Override
|
public void actionPerformed(ActionEvent arg0) {
|
SearchByPicDialog dialog = new SearchByPicDialog();
|
dialog.setVisible(true);
|
}
|
});*/
|
|
// 人员操作
|
personOperateBtn.addActionListener(new ActionListener() {
|
|
@Override
|
public void actionPerformed(ActionEvent arg0) {
|
int row = -1;
|
row = table.getSelectedRow(); //获得所选的单行
|
|
if(row < 0) {
|
JOptionPane.showMessageDialog(null, Res.string().getSelectGroup(), Res.string().getErrorMessage(), JOptionPane.ERROR_MESSAGE);
|
return;
|
}
|
|
if(defaultTableModel.getValueAt(row, 0) == null || String.valueOf(defaultTableModel.getValueAt(row, 0)).trim().equals("")) {
|
JOptionPane.showMessageDialog(null, Res.string().getSelectGroup(), Res.string().getErrorMessage(), JOptionPane.ERROR_MESSAGE);
|
return;
|
}
|
|
personOperateDialog = new PersonOperateDialog(String.valueOf(defaultTableModel.getValueAt(row, 0)).trim(),
|
String.valueOf(defaultTableModel.getValueAt(row, 1)).trim());
|
personOperateDialog.setVisible(true);
|
}
|
});
|
}
|
}
|
|
/*
|
* 查找所有人脸库
|
*/
|
private void findGroupInfo() {
|
// 清空列表
|
for(int i = 0; i < 20; i++) {
|
for(int j = 0; j < 3; j++) {
|
defaultTableModel.setValueAt("", i, j);
|
}
|
}
|
|
// 查询人脸库
|
NET_FACERECONGNITION_GROUP_INFO[] groupInfoArr = TargetRecognitionModule.findGroupInfo("");
|
if(groupInfoArr != null) {
|
for(int i = 0; i < groupInfoArr.length; i++) {
|
defaultTableModel.setValueAt(new String(groupInfoArr[i].szGroupId).trim(), i, 0);
|
try {
|
defaultTableModel.setValueAt(new String(groupInfoArr[i].szGroupName, "GBK").trim(), i, 1);
|
} catch (UnsupportedEncodingException e1) {
|
e1.printStackTrace();
|
}
|
defaultTableModel.setValueAt(String.valueOf(groupInfoArr[i].nGroupSize).trim(), i, 2);
|
}
|
}
|
}
|
|
private Object[][] data;
|
private DefaultTableModel defaultTableModel;
|
private JTable table;
|
}
|
