~polto/elphelogm/trunk

if((i = fwrite(p->og.header, 1, p->og.header_len, p->fd)) != p->og.header_len || p->og.body_len && (i = fwrite(p->og.body, 1, p->og.body_len, p->fd)) != p->og.body_len) {

136

//j=errno;

137

//fprintf(stderr,"\n%d %d %d\n",i,p->og.header_len,p->og.body_len);

138

//errno=j;

139

perror("\nfwrite");

140

exit(3);

141

}

142

}

143

}

144

if(p->imode && granulepos)

145

fflush(p->fd);

146

}

147

148

void *dump(param_t * p, int last)

149

{

150

int len = 0;

151

uint8_t *tempbuf;

152

uint32_t ts;

153

stream_header sh;

154

vorbis_comment *comment;

155

time_t t = time(0);

156

157

tempbuf = p->buffer;

158

if(p->start_time == 0)

159

p->start_time = t;

160

161

if(p->ftmpl && (!p->fname || !p->fd)) {

162

char *newname = (char *) malloc(10240);

163

if(!newname) {

164

perror("malloc");

165

exit(2);

166

}

167

strftime(newname, 10240, p->ftmpl, localtime(&t));

168

if(p->fd)

169

fclose(p->fd);

170

p->fd = fopen(newname, "w");

171

if(!p->fd) {

172

perror(newname);

173

exit(1);

174

}

175

msg(1, "\nSaving to file:%s\n", newname);

176

header_produced = 0;

177

p->cur_frames = 0;

178

p->start_time = t;

179

if(p->fname)

180

free(p->fname);

181

p->fname = newname;

182

}

183

184

if(header_produced && (p->width != peek_size(&(p->rtp_queue), 0) || p->height != peek_size(&(p->rtp_queue), 1)))

185

header_produced = 2;

186

187

if(header_produced != 1) {

188

if(!header_produced) {

189

ogg_stream_init(&(p->os), p->serialno);

190

p->packetno = 0;

191

if(!rawmode)

192

p->last_granulepos = 0;

193

}

194

header_produced = 1;

195

if(p->fps == 0.0 && p->rtp_queue.qframes > 1) {

196

p->fps = (double) VIDEO_TIMESCALE / (p->rtp_queue.head_f->next->timestamp - p->rtp_queue.head_f->timestamp);

197

}

198

// Write out the first page of stream

199

// because header must come together before any

200

// non-header pages.

201

*((unsigned char *) tempbuf) = PACKET_TYPE_HEADER;

202

memset(&sh, 0, sizeof(stream_header));

203

strcpy(sh.streamtype, "video");

204

memcpy(sh.subtype, p->codec, 4);

205

put_uint32(&sh.size, sizeof(sh));

206

put_uint64(&sh.time_unit, p->fps > 0.0 ? (ogg_int64_t) (10000000.0 / p->fps)

207

: (ogg_int64_t) 10000000 * VTSD / VIDEO_TIMESCALE);

208

put_uint64(&sh.samples_per_unit, 1);

209

put_uint32(&sh.default_len, 1);

210

put_uint32(&sh.buffersize, p->max_frame_size);

211

put_uint16(&sh.bits_per_sample, p->bpp);

212

put_uint32(&sh.sh.video.width, p->width = peek_size(&(p->rtp_queue), 0));

213

put_uint32(&sh.sh.video.height, p->height = peek_size(&(p->rtp_queue), 1));

214

memcpy(&tempbuf[1], &sh, sizeof(stream_header));

215

216

// put it into Ogg stream

217

write_ogg_packet(p, 1 + get_uint32(&sh.size), 0, 0);

218

219

// make comments headers

220

comment = generate_vorbis_comment();

221

len = comments_to_buffer(comment, (char *)tempbuf, p->max_frame_size + 1);

222

if(len < 0) {

223

msg(0, "error: buffer is too small for %d bytes " "(can hold %ld bytes).\n", len, p->max_frame_size + 1);

224

return 0;

225

}

226

if(comment != NULL) {

227

vorbis_comment_clear(comment);

228

free(comment);

229

}

230

231

write_ogg_packet(p, len, 0, 0);

232

233

// we are ready dumping stream data

234

} // if header_produced

235

236

len = get_frame(&(p->rtp_queue), &tempbuf[1], (int32_t *)&ts); // get jpeg picture from RTP buffer

237

238

if(len == 0)

239

return 0; // no frame

240

if(notfirst && (signed) ts - (signed) prevts < 0) {

241

msg(2, "\nskip old:%u (was:%u)\n", ts, prevts);

242

return 0;

243

}

244

if(snum > 0) {

245

msg(2, "\nskip:%u\n", ts);

246

snum--;

247

return 0;

248

}

249

msg(2, "\nout:%u\n", ts);

250

snum = p->snum;

251

if(notfirst && p->delta_ts != ts - prevts) {

252

//header_produced=2;

253

p->delta_ts = ts - prevts;

254

} else

255

notfirst = 1;

256

prevts = ts;

257

258

*((unsigned char *) tempbuf) = PACKET_IS_SYNCPOINT;

259

if(p->ftmpl && (p->max_size && ftell(p->fd) + len + 45 >= p->max_size || p->max_frames && p->cur_frames + 1 >= p->max_frames || p->max_time && t - p->start_time >= p->max_time))

260

last = 1;

261

262

write_ogg_packet(p, len + 1, p->last_granulepos += p->imode && p->rtp_queue.nb_frames > 0 ? 1 : (p->fps > 0.0 ? 1 + (p->delta_ts * p->fps - 1) / VIDEO_TIMESCALE : p->delta_ts / VTSD), last);

263

264

p->cur_frames++;

265

frames++;

266

267

// fprintf(stderr,"frame: %07lld fps: %.5f filesize:%d \r",frames,

268

// (double)VIDEO_TIMESCALE/p->delta_ts, ftell(p->fd));

269

270

if(last && p->fd) {

271

fclose(p->fd);

272

p->fd = 0;

273

if(p->fname) {

274

char *newname = (char *)malloc(10260);

275

int ren = 0, i, j;

276

if(!newname) {

277

perror("error");

278

exit(1);

279

}

280

for(i = j = 0; j < 10259 && p->fname[i];) {

281

if(p->fname[i] == '%' && p->fname[i + 1] == '$') {

282

ren = 1;

283

newname[j++] = p->fname[i++];

284

newname[j++] = '%';

285

i++;

286

} else if(p->fname[i] != '$')

287

newname[j++] = p->fname[i++];

288

else {

289

ren = 1;

290

switch (p->fname[++i]) {

291

default:

292

newname[j++] = '%';

293

continue;

294

case '$':

295

newname[j++] = '$';

296

i++;

297

continue;

298

case 'n':

299

i++;

300

j += snprintf(newname + j, 10260 - j, "%d", p->cur_frames);

301

continue;

302

case 't':

303

i++;

304

j += snprintf(newname + j, 10260 - j, "%014llu", p->last_granulepos - 1);

305

continue;

306

}

307

}

308

}

309

if(ren) {

310

char *newname2 = (char *)malloc(10260);

311

newname[j] = 0;

312

if(!newname2) {

313

perror("error");

314

exit(1);

315

}

316

strftime(newname2, 10260, newname, localtime(&t));

317

if(rename(p->fname, newname2) < 0) {

318

perror("file rename error");

319

exit(1);

320

}

321

msg(1, "\nFile renamed:%s\n", newname2);

322

free(newname2);

323

}

324

free(newname);

325

}

326

}

327

328

return NULL;

329

}