libgender/shader.C

#include <cassert>

#include "shader.h"
#include "shader_vars.h"

#include <sstream>
#include <iomanip>

namespace shader {

  using namespace std;

  refcounted::~refcounted ()
  {
    if (refcnt)
      abort ();
  }

  void refcounted::refcnt_destroy () const
  {
    if (!--refcnt)
      delete this; // quite a bit of code...
  }

  const char str_float  [] = "float";
  const char str_vec2   [] = "vec2";
  const char str_vec3   [] = "vec3";
  const char str_vec4   [] = "vec4";
  const char str_mat2   [] = "mat2";
  const char str_mat3   [] = "mat3";
  const char str_mat4   [] = "mat4";

  const char str_sampler_1d             [] = "sampler1D";
  const char str_sampler_1d_shadow      [] = "sampler1DShadow";
  const char str_sampler_2d             [] = "sampler2D";
  const char str_sampler_2d_shadow      [] = "sampler2DShadow";
  const char str_sampler_2d_rect        [] = "sampler2DRect";
  const char str_sampler_2d_rect_shadow [] = "sampler2DRectShadow";
  const char str_sampler_3d             [] = "sampler3D";
  const char str_sampler_3d_rect        [] = "sampler3DRect";
  const char str_sampler_cube           [] = "samplerCube";

  unsigned int var_i::next_id = 0;

  var_i::var_i (const char *domainstr, const char *typestr)
  : domainstr (domainstr), typestr (typestr)
  {
  }

  var_i::~var_i ()
  {
  }

#if 0
  stream_i::stream_i (const char *strtype)
  : var_i (strtype)
  {
    sprintf (name, "V%d", ++next_id);
  }
#endif

  temporary_i::temporary_i (const char *strtype)
  : var_i (0, strtype)
  {
    sprintf (name, "T%d", ++next_id);
  }

  varying_i::varying_i (const char *strtype)
  : var_i ("varying", strtype)
  {
    sprintf (name, "V%d", ++next_id);
  }

  uniform_i::uniform_i (const char *strtype)
  : var_i ("uniform", strtype)
  {
    sprintf (name, "U%d", ++next_id);
  }

  ////////////////////////////////////////////////////////////////////////////

  void var_i::operator ()() const
  {
    compile::cur->code << name;

    if (compile::cur->first (this))
      {
        if (domainstr)
          compile::cur->global << domainstr << ' ' << typestr << ' ' << name << ";\n";
        else
          compile::cur->local << "  " << typestr << ' ' << name << ";\n";
      }
  }

  ////////////////////////////////////////////////////////////////////////////

  int texture_units::unit_count = 8;
  int texture_units::units[8] = { 7, 6, 5, 4, 3, 2, 1, 0 };

  shader_object_i *cur = 0;

  shader_object_i::shader_object_i (GLenum type)
  : type (type)
  {
    id = glCreateShaderObjectARB (type);
    assert (id);
  }

  shader_object_i::~shader_object_i ()
  {
    glDeleteObjectARB (id);
  }

  static string linify (const string &s)
  {
    ostringstream o;

    int b = 0, e;
    int l = 1;
    do {
      o << setw (3) << l << ": ";
      e = s.find ('\n', b);
      if (e == string::npos)
        e = s.size ();

      o << s.substr (b, e - b + 1);
      b = e + 1;
      l++;
    } while (b < s.size ());

    return o.str ();
  }

  void shader_object_i::start ()
  {
    compile::cur = new compile::shader_builder;
  }

  void shader_object_i::stop ()
  {
    compile::shader_builder &b = *compile::cur;

    ostringstream os;

    os << b.global.str ()
       << "\nvoid main (void)\n"
       << "{\n"
       << b.local.str ()
       << "\n"
       << b.code.str ()
       << "}\n";

    delete cur;
    cur = 0;

    string src = os.str ();
    const char *sptr = src.data ();
    const int slen = src.size ();

    printf ("%s\n", linify (src).c_str ());
    glShaderSourceARB (id, 1, &sptr, &slen);
    glCompileShaderARB (id);

    GLint compiled;
    glGetObjectParameterivARB (id, GL_OBJECT_COMPILE_STATUS_ARB, &compiled);

    if (!compiled)
      {
        char infolog[8192];
        glGetInfoLogARB (id, 8192, NULL, infolog);
        printf ("%s\n", linify (src).c_str ());
        printf ("%s\n", infolog);
        abort ();
      }
  }

  ////////////////////////////////////////////////////////////////////////////
  
  GLint uniform_i::location ()
  {
    assert (program_object_i::cur);

    GLint &rid = program_object_i::cur->uloc[this];

    if (!rid)
      rid = glGetUniformLocationARB (program_object_i::cur->id, name);

    return rid;
  }

  program_object_i *program_object_i::cur; // currently bound program

  program_object_i::program_object_i ()
  {
    id = glCreateProgramObjectARB ();
    assert (id);

    glAttachObjectARB (id, vsh->id);
    glAttachObjectARB (id, fsh->id);
  }

  program_object_i::~program_object_i ()
  {
    glDeleteProgramsARB (1, &id);
  }

  void program_object_i::link ()
  {
    glLinkProgramARB (id);
        
    GLint linked;
    glGetObjectParameterivARB (id, GL_OBJECT_LINK_STATUS_ARB, &linked);

    if (!linked)
      {
        char infolog[8192];
        glGetInfoLogARB (id, 8192, NULL, infolog);
        printf ("LINK-INFOLOG<%s>\n", infolog);
        abort ();
      }

    uloc.clear ();
  }

  void program_object_i::enable ()
  {
    glUseProgramObjectARB (id);
    //TODO: set samplers here
    cur = this;
  }

  void program_object_i::disable ()
  {
    //TODO: clear samplers here
    glUseProgramObjectARB (0);
    cur = 0;
  }

  const sl_expr< sl_string<60> > sl_convert< ::vec2 >::convert (const ::vec2 &v)
  {
    sl_string<60> s;
    sprintf (s.str, "vec2 (%g, %g)", v.x, v.y);
    return s;
  }

  const sl_expr< sl_string<80> > sl_convert< ::vec3 >::convert (const ::vec3 &v)
  {
    sl_string<80> s;
    sprintf (s.str, "vec3 (%g, %g, %g)", v.x, v.y, v.z);
    return s;
  }

  const sl_expr< sl_string<100> > sl_convert< ::vec4 >::convert (const ::vec4 &v)
  {
    sl_string<100> s;
    sprintf (s.str, "vec4 (%g, %g, %g, %g)", v.x, v.y, v.z, v.w);
    return s;
  }

  namespace compile {

    shader_builder *cur = 0;

    bool shader_builder::first (const void *p)
    {
      if (seen.find (p) == seen.end ())
        {
          seen.insert (p);
          return true;
        }

      return false;
    }
  }

  void debdebdebdebug ()//D
  {
    return;

    using namespace compile;

    vertex_shader vsh;

    vsh->start ();

    temp_4f lightpos;
    temp_4f wpos;

    lightpos = vec4 (0, 10, 0, 1);
    wpos = model_view_matrix * vin.vertex;
    vout.position = vin.vertex * model_view_matrix_inverse;
    vout.tex_coord[0] = vin.tex_coord[0];
    vout.tex_coord[1] = normalize (lightpos - wpos);
    vout.tex_coord[2] = normalize (wpos);
    //vout.tex_coord[3] = normalize (xyz (model_view_matrix_inverse_transpose) * vin.normal);
    //vout.tex_coord[4] = normalize (xyz (projection_matrix_inverse_transpose) - wpos);

    vsh->stop ();

    fragment_shader fsh;

    fsh->start ();

    xyz (fout.frag_color) = noise3 (x (fin.frag_coord) * y (fin.frag_coord));

    temp_1f spec_expon;
    spec_expon = 200;

    fsh->stop ();

    //abort ();
  }

}

Revision:	1.27
Committed:	Wed Nov 3 01:12:34 2004 UTC (19 years, 7 months ago) by root
Content type:	text/plain
Branch:	MAIN
Changes since 1.26:	+10 -10 lines
Log Message:	* empty log message *
#	Content
1	#include <cassert>
2
3	#include "shader.h"
4	#include "shader_vars.h"
5
6	#include <sstream>
7	#include <iomanip>
8
9	namespace shader {
10
11	using namespace std;
12
13	refcounted::~refcounted ()
14	{
15	if (refcnt)
16	abort ();
17	}
18
19	void refcounted::refcnt_destroy () const
20	{
21	if (!--refcnt)
22	delete this; // quite a bit of code...
23	}
24
25	const char str_float [] = "float";
26	const char str_vec2 [] = "vec2";
27	const char str_vec3 [] = "vec3";
28	const char str_vec4 [] = "vec4";
29	const char str_mat2 [] = "mat2";
30	const char str_mat3 [] = "mat3";
31	const char str_mat4 [] = "mat4";
32
33	const char str_sampler_1d [] = "sampler1D";
34	const char str_sampler_1d_shadow [] = "sampler1DShadow";
35	const char str_sampler_2d [] = "sampler2D";
36	const char str_sampler_2d_shadow [] = "sampler2DShadow";
37	const char str_sampler_2d_rect [] = "sampler2DRect";
38	const char str_sampler_2d_rect_shadow [] = "sampler2DRectShadow";
39	const char str_sampler_3d [] = "sampler3D";
40	const char str_sampler_3d_rect [] = "sampler3DRect";
41	const char str_sampler_cube [] = "samplerCube";
42
43	unsigned int var_i::next_id = 0;
44
45	var_i::var_i (const char domainstr, const char typestr)
46	: domainstr (domainstr), typestr (typestr)
47	{
48	}
49
50	var_i::~var_i ()
51	{
52	}
53
54	#if 0
55	stream_i::stream_i (const char *strtype)
56	: var_i (strtype)
57	{
58	sprintf (name, "V%d", ++next_id);
59	}
60	#endif
61
62	temporary_i::temporary_i (const char *strtype)
63	: var_i (0, strtype)
64	{
65	sprintf (name, "T%d", ++next_id);
66	}
67
68	varying_i::varying_i (const char *strtype)
69	: var_i ("varying", strtype)
70	{
71	sprintf (name, "V%d", ++next_id);
72	}
73
74	uniform_i::uniform_i (const char *strtype)
75	: var_i ("uniform", strtype)
76	{
77	sprintf (name, "U%d", ++next_id);
78	}
79
80	////////////////////////////////////////////////////////////////////////////
81
82	void var_i::operator ()() const
83	{
84	compile::cur->code << name;
85
86	if (compile::cur->first (this))
87	{
88	if (domainstr)
89	compile::cur->global << domainstr << ' ' << typestr << ' ' << name << ";\n";
90	else
91	compile::cur->local << " " << typestr << ' ' << name << ";\n";
92	}
93	}
94
95	////////////////////////////////////////////////////////////////////////////
96
97	int texture_units::unit_count = 8;
98	int texture_units::units[8] = { 7, 6, 5, 4, 3, 2, 1, 0 };
99
100	shader_object_i *cur = 0;
101
102	shader_object_i::shader_object_i (GLenum type)
103	: type (type)
104	{
105	id = glCreateShaderObjectARB (type);
106	assert (id);
107	}
108
109	shader_object_i::~shader_object_i ()
110	{
111	glDeleteObjectARB (id);
112	}
113
114	static string linify (const string &s)
115	{
116	ostringstream o;
117
118	int b = 0, e;
119	int l = 1;
120	do {
121	o << setw (3) << l << ": ";
122	e = s.find ('\n', b);
123	if (e == string::npos)
124	e = s.size ();
125
126	o << s.substr (b, e - b + 1);
127	b = e + 1;
128	l++;
129	} while (b < s.size ());
130
131	return o.str ();
132	}
133
134	void shader_object_i::start ()
135	{
136	compile::cur = new compile::shader_builder;
137	}
138
139	void shader_object_i::stop ()
140	{
141	compile::shader_builder &b = *compile::cur;
142
143	ostringstream os;
144
145	os << b.global.str ()
146	<< "\nvoid main (void)\n"
147	<< "{\n"
148	<< b.local.str ()
149	<< "\n"
150	<< b.code.str ()
151	<< "}\n";
152
153	delete cur;
154	cur = 0;
155
156	string src = os.str ();
157	const char *sptr = src.data ();
158	const int slen = src.size ();
159
160	printf ("%s\n", linify (src).c_str ());
161	glShaderSourceARB (id, 1, &sptr, &slen);
162	glCompileShaderARB (id);
163
164	GLint compiled;
165	glGetObjectParameterivARB (id, GL_OBJECT_COMPILE_STATUS_ARB, &compiled);
166
167	if (!compiled)
168	{
169	char infolog[8192];
170	glGetInfoLogARB (id, 8192, NULL, infolog);
171	printf ("%s\n", linify (src).c_str ());
172	printf ("%s\n", infolog);
173	abort ();
174	}
175	}
176
177	////////////////////////////////////////////////////////////////////////////
178
179	GLint uniform_i::location ()
180	{
181	assert (program_object_i::cur);
182
183	GLint &rid = program_object_i::cur->uloc[this];
184
185	if (!rid)
186	rid = glGetUniformLocationARB (program_object_i::cur->id, name);
187
188	return rid;
189	}
190
191	program_object_i *program_object_i::cur; // currently bound program
192
193	program_object_i::program_object_i ()
194	{
195	id = glCreateProgramObjectARB ();
196	assert (id);
197
198	glAttachObjectARB (id, vsh->id);
199	glAttachObjectARB (id, fsh->id);
200	}
201
202	program_object_i::~program_object_i ()
203	{
204	glDeleteProgramsARB (1, &id);
205	}
206
207	void program_object_i::link ()
208	{
209	glLinkProgramARB (id);
210
211	GLint linked;
212	glGetObjectParameterivARB (id, GL_OBJECT_LINK_STATUS_ARB, &linked);
213
214	if (!linked)
215	{
216	char infolog[8192];
217	glGetInfoLogARB (id, 8192, NULL, infolog);
218	printf ("LINK-INFOLOG<%s>\n", infolog);
219	abort ();
220	}
221
222	uloc.clear ();
223	}
224
225	void program_object_i::enable ()
226	{
227	glUseProgramObjectARB (id);
228	//TODO: set samplers here
229	cur = this;
230	}
231
232	void program_object_i::disable ()
233	{
234	//TODO: clear samplers here
235	glUseProgramObjectARB (0);
236	cur = 0;
237	}
238
239	const sl_expr< sl_string<60> > sl_convert< ::vec2 >::convert (const ::vec2 &v)
240	{
241	sl_string<60> s;
242	sprintf (s.str, "vec2 (%g, %g)", v.x, v.y);
243	return s;
244	}
245
246	const sl_expr< sl_string<80> > sl_convert< ::vec3 >::convert (const ::vec3 &v)
247	{
248	sl_string<80> s;
249	sprintf (s.str, "vec3 (%g, %g, %g)", v.x, v.y, v.z);
250	return s;
251	}
252
253	const sl_expr< sl_string<100> > sl_convert< ::vec4 >::convert (const ::vec4 &v)
254	{
255	sl_string<100> s;
256	sprintf (s.str, "vec4 (%g, %g, %g, %g)", v.x, v.y, v.z, v.w);
257	return s;
258	}
259
260	namespace compile {
261
262	shader_builder *cur = 0;
263
264	bool shader_builder::first (const void *p)
265	{
266	if (seen.find (p) == seen.end ())
267	{
268	seen.insert (p);
269	return true;
270	}
271
272	return false;
273	}
274	}
275
276	void debdebdebdebug ()//D
277	{
278	return;
279
280	using namespace compile;
281
282	vertex_shader vsh;
283
284	vsh->start ();
285
286	temp_4f lightpos;
287	temp_4f wpos;
288
289	lightpos = vec4 (0, 10, 0, 1);
290	wpos = model_view_matrix * vin.vertex;
291	vout.position = vin.vertex * model_view_matrix_inverse;
292	vout.tex_coord[0] = vin.tex_coord[0];
293	vout.tex_coord[1] = normalize (lightpos - wpos);
294	vout.tex_coord[2] = normalize (wpos);
295	//vout.tex_coord[3] = normalize (xyz (model_view_matrix_inverse_transpose) * vin.normal);
296	//vout.tex_coord[4] = normalize (xyz (projection_matrix_inverse_transpose) - wpos);
297
298	vsh->stop ();
299
300	fragment_shader fsh;
301
302	fsh->start ();
303
304	xyz (fout.frag_color) = noise3 (x (fin.frag_coord) * y (fin.frag_coord));
305
306	temp_1f spec_expon;
307	spec_expon = 200;
308
309	fsh->stop ();
310
311	//abort ();
312	}
313
314	}
315