5bf93292 |
1 | //$Id$ |
2 | |
3 | // Author: Anders Vestbo <mailto:vestbo$fi.uib.no> |
4 | //*-- Copyright © ASV |
5 | |
6 | #include <stdio.h> |
7 | #include <stream.h> |
8 | #include <stdlib.h> |
9 | |
10 | #include "AliL3Compress.h" |
11 | #include "AliL3TrackArray.h" |
12 | #include "AliL3ModelTrack.h" |
13 | #include "bitio.h" |
14 | |
15 | ClassImp(AliL3Compress) |
16 | |
17 | AliL3Compress::AliL3Compress() |
18 | { |
19 | |
20 | } |
21 | |
22 | AliL3Compress::~AliL3Compress() |
23 | { |
24 | |
25 | } |
26 | |
27 | void AliL3Compress::Write2File(AliL3TrackArray *tracks) |
28 | { |
29 | FILE *file = fopen("data.raw","w"); |
30 | Short_t ntracks = tracks->GetNTracks(); |
31 | cout<<"Writing "<<ntracks<<" tracks to file"<<endl; |
32 | //Write the number of tracks at the beginning: |
33 | //fwrite(&ntracks,sizeof(Short_t),1,file); |
34 | |
35 | for(Int_t i=0; i<ntracks; i++) |
36 | { |
37 | AliL3ModelTrack *track = (AliL3ModelTrack*)tracks->GetCheckedTrack(i); |
38 | if(!track) continue; |
39 | AliL3TrackModel *model = track->GetModel(); |
40 | cout<<"Writing "<<model->fFirstPointX<<endl; |
41 | if(fwrite(model,sizeof(AliL3TrackModel),1,file)!=1) break; |
42 | for(Int_t j=0; j<model->fNClusters; j++) |
43 | { |
44 | AliL3ClusterModel *cl = track->GetClusterModel(j); |
45 | fwrite(cl,sizeof(AliL3ClusterModel),1,file); |
46 | } |
47 | } |
48 | fclose(file); |
49 | } |
50 | |
51 | void AliL3Compress::ReadFile() |
52 | { |
53 | FILE *file = fopen("data.raw","r"); |
54 | |
55 | AliL3TrackArray *tracks = new AliL3TrackArray("AliL3ModelTrack"); |
56 | Int_t ntracks=0; |
57 | |
58 | while(!feof(file)) |
59 | { |
60 | AliL3ModelTrack *track = (AliL3ModelTrack*)tracks->NextTrack(); |
61 | track->Init(0,0); |
62 | AliL3TrackModel *model = track->GetModel(); |
63 | AliL3ClusterModel *clusters = track->GetClusters(); |
64 | if(fread(model,sizeof(AliL3TrackModel),1,file)!=1) break; |
65 | cout<<"Read model "<<model->fFirstPointX<<endl; |
66 | if(fread(clusters,(model->fNClusters)*sizeof(AliL3ClusterModel),1,file)!=1) break; |
67 | ntracks++; |
68 | } |
69 | |
70 | delete tracks; |
71 | cout<<"Read "<<ntracks<<" tracks from file"<<endl; |
72 | fclose(file); |
73 | } |
74 | |
75 | void AliL3Compress::CompressFile() |
76 | { |
77 | |
78 | BIT_FILE *output = OpenOutputBitFile("test.raw"); |
79 | FILE *input = fopen("data.raw","r"); |
80 | |
81 | AliL3TrackModel track; |
82 | AliL3ClusterModel cluster; |
83 | Int_t temp; |
84 | |
85 | while(!feof(input)) |
86 | { |
87 | if(fread(&track,sizeof(AliL3TrackModel),1,input)!=1) break; |
88 | cout<<"Writing "<<sizeof(AliL3TrackModel)<<endl; |
89 | fwrite(&track,sizeof(AliL3TrackModel),1,output->file); |
90 | Int_t bitcount=0; |
91 | for(Int_t i=0; i<track.fNClusters; i++) |
92 | { |
93 | if(fread(&cluster,sizeof(AliL3ClusterModel),1,input)!=1) break; |
94 | Int_t flag = cluster.fPresent; |
95 | OutputBit(output,flag); |
96 | bitcount++; |
97 | if(!flag) continue; |
98 | temp = (Int_t)cluster.fDTime; |
99 | if(temp<0) |
100 | OutputBit(output,0); |
101 | else |
102 | OutputBit(output,1); |
103 | OutputBits(output,abs(temp),8); |
104 | temp = (Int_t)cluster.fDPad; |
105 | if(temp<0) |
106 | OutputBit(output,0); |
107 | else |
108 | OutputBit(output,1); |
109 | OutputBits(output,abs(temp),8); |
110 | bitcount+=8; |
111 | temp = (Int_t)cluster.fDCharge; |
112 | OutputBits(output,temp,10); |
113 | bitcount+=10; |
114 | |
115 | //Short_t temp=(Short_t)cluster.fDTime; |
116 | cout<<"flag "<<(int)flag<<" dtime "<<(int)cluster.fDTime<<" dpad "<<(int)cluster.fDPad<<" charge "<<cluster.fDCharge<<endl; |
117 | } |
118 | |
119 | } |
120 | |
121 | fclose(input); |
122 | CloseOutputBitFile(output); |
123 | } |
124 | |
125 | void AliL3Compress::ExpandFile() |
126 | { |
127 | BIT_FILE *input = OpenInputBitFile("test.raw"); |
128 | |
129 | AliL3TrackModel track; |
130 | AliL3ClusterModel cluster; |
131 | |
132 | fread(&track,sizeof(AliL3TrackModel),1,input->file); |
133 | for(Int_t i=0; i<track.fNClusters; i++) |
134 | { |
135 | Int_t temp,sign; |
136 | temp = InputBit(input); |
137 | if(!temp) break; |
138 | sign=InputBit(input); |
139 | temp = InputBits(input,8); |
140 | if(!sign) |
141 | temp*=-1; |
142 | cout<<"Dtime "<<temp; |
143 | sign=InputBit(input); |
144 | temp = InputBits(input,8); |
145 | if(!sign) |
146 | temp*=-1; |
147 | cout<<" DPad "<<temp; |
148 | temp=InputBits(input,10); |
149 | cout<<" Charge "<<temp<<endl; |
150 | } |
151 | CloseInputBitFile(input); |
152 | } |