lankun/sjy01-preprocessing
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

xlsx

Browse Source
This commit is contained in:
nuknal
2024-05-21 18:05:10 +08:00
parent b60095e5b9
commit faf285775f
4 changed files with 7 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
cmd/extract.go
View File

@@ -28,6 +28,7 @@ var extractCmd = &cobra.Command{
for i, d := range dats { for i, d := range dats {
e.SeprateAuxAndImgData(d, i) e.SeprateAuxAndImgData(d, i)
} }
e.Cleanup()
} }
} else { } else {
p := &extract.Params{ p := &extract.Params{
@@ -43,6 +44,7 @@ var extractCmd = &cobra.Command{
for i, d := range dats { for i, d := range dats {
e.SeprateAuxAndImgData(d, i) e.SeprateAuxAndImgData(d, i)
} }
e.Cleanup()
} }
}, },
} }
@@ -58,7 +60,7 @@ func init() {
func params() []*extract.Params { func params() []*extract.Params {
var params []*extract.Params var params []*extract.Params
datas := []string{"051622", "051712", "051721", datas := []string{"051622", "051712", "051721",
"051813", "051821", "051823", "051814", "051821", "051823",
"051921", "051922", "052022"} "051921", "051922", "052022"}
for _, d := range datas { for _, d := range datas {
params = append(params, &extract.Params{ params = append(params, &extract.Params{

3
extract/aux.go
View File

@@ -34,6 +34,9 @@ func (e Extractor) ExtractAux(auxfile, xlsxfile string) error {
row := 2 row := 2
col := 1 col := 1
for i := 0; i < len(data); i += 24 + 128 + 512 { for i := 0; i < len(data); i += 24 + 128 + 512 {
if i+24+128+512 > len(data) {
break
}
var head AuxFrameHead var head AuxFrameHead
head.Decode(data[i : i+24]) head.Decode(data[i : i+24])
l0, _ := head.SaveXlsx(f, col, row) l0, _ := head.SaveXlsx(f, col, row)

2
extract/process.go
View File

@@ -12,7 +12,7 @@ type Extractor struct {
func NewExtractor(params *Params) *Extractor { func NewExtractor(params *Params) *Extractor {
os.MkdirAll(params.OutputPath, 0755) os.MkdirAll(params.OutputPath, 0755)
os.MkdirAll(params.TempPath, 0755) os.MkdirAll(params.TempPath, 0755)
return &Extractor{params: params} return &Extractor{params: params, Clean: true}
} }
func (e *Extractor) Cleanup() error { func (e *Extractor) Cleanup() error {

5
extract/trans_frame.go
View File

@@ -59,9 +59,6 @@ func (p *Extractor) ExtractOriginalImageData(aosDataFile string) ([]string, erro
snRange := map[int][]uint32{} snRange := map[int][]uint32{}
datSet := map[int][]byte{} datSet := map[int][]byte{}
fsn, _ := os.Create("demo/temp/tf_sn.txt")
defer fsn.Close()
var i int var i int
for i < len(aosData) { for i < len(aosData) {
if i+4 > len(aosData) { if i+4 > len(aosData) {
@@ -79,8 +76,6 @@ func (p *Extractor) ExtractOriginalImageData(aosDataFile string) ([]string, erro
tf.Decode(aosData[i : i+TransImageFrameLength]) tf.Decode(aosData[i : i+TransImageFrameLength])
fileno := int(tf.FileNo) fileno := int(tf.FileNo)
snRange[fileno] = append(snRange[fileno], tf.SNo) snRange[fileno] = append(snRange[fileno], tf.SNo)
fsn.WriteString(fmt.Sprintf("%d %d %d\n", i, tf.SNo, fileno))
if tf.SecretFlag == 0xAA { if tf.SecretFlag == 0xAA {
log.Info("secret frame is not supported") log.Info("secret frame is not supported")
break break